This commit is contained in:
coward 2024-01-05 11:54:25 +08:00
commit 9680512f0a
8 changed files with 382 additions and 0 deletions

24
.gitignore vendored Normal file
View File

@ -0,0 +1,24 @@
### Go template
# If you prefer the allow list template instead of the deny list, see community template:
# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
#
# Binaries for programs and plugins
*.exe
*.exe~
*.dll
*.so
*.dylib
# Test binary, built with `go test -c`
*.test
# Output of the go coverage tool, specifically when used with LiteIDE
*.out
# Dependency directories (remove the comment below to include it)
# vendor/
.idea
# Go workspace file
go.work

10
README.md Normal file
View File

@ -0,0 +1,10 @@
# wx_helper_api_sdk
## install
```go
go get gitea.mrx.ltd/go-pkg/wx_helper_sdk
```
## use
```go
没啥好用的自己看client_test.go文件大同小异有些没写到测试用例里边自己搞定
```

194
client.go Normal file
View File

@ -0,0 +1,194 @@
package wx_helper_sdk
import (
"encoding/json"
"fmt"
"github.com/go-resty/resty/v2"
"log"
"strings"
)
type WxHelper struct {
Host string
Port uint
tmpResult CommonResponse
client *resty.Client
}
func NewWxHelper(host string, port uint) WxHelper {
hc := resty.New()
return WxHelper{
host,
port,
CommonResponse{},
hc,
}
}
// GetAddr
// @description: 获取请求地址
// @receiver w
// @param uri
// @return string
func (w *WxHelper) GetAddr(uri string) string {
if !strings.HasPrefix(w.Host, "http") && !strings.HasPrefix(w.Host, "https://") {
w.Host = fmt.Sprintf("http://%s", w.Host)
}
return fmt.Sprintf("%s:%d%s", w.Host, w.Port, uri)
}
// SetHook
// @description: 设置回调
// @receiver w
// @param req
// @return err
func (w *WxHelper) SetHook(param *SetHook) (err error) {
req := w.client.R()
req.Header.Set("Content-Type", "application/json")
req.SetBody(param)
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/hookSyncMsg"))
if err != nil {
return
}
if w.tmpResult.Code != 0 {
return fmt.Errorf("设置回调失败: %v", w.tmpResult.Msg)
}
return
}
// UnHook
// @description: 取消回调设置
// @receiver w
// @return err
func (w *WxHelper) UnHook() (err error) {
req := w.client.R()
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/unhookSyncMsg"))
if err != nil {
return
}
if w.tmpResult.Code != 0 {
return fmt.Errorf("取消回调设置失败: %v", w.tmpResult.Msg)
}
return
}
// CheckUserIsLogin
// @description: 校验用户是否已经登陆
// @receiver *WxHelper
// @return bool
func (w *WxHelper) CheckUserIsLogin() bool {
req := w.client.R()
_, err := req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/checkLogin"))
if err != nil {
log.Fatalf("获取用户登陆状态失败: %v", err.Error())
return false
}
if w.tmpResult.Code != 1 {
return false
}
return true
}
// GetCurrentLoginUserInfo
// @description: 获取当前登陆用户信息
// @receiver w
// @return response
// @return err
func (w *WxHelper) GetCurrentLoginUserInfo() (response UserInfo, err error) {
req := w.client.R()
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/userInfo"))
if err != nil {
log.Fatalf("获取用户登陆状态失败: %v", err.Error())
return
}
if w.tmpResult.Code != 1 {
return
}
data, _ := json.Marshal(w.tmpResult.Data)
_ = json.Unmarshal(data, &response)
return
}
// GetContactList
// @description: 获取好友列表
// @receiver w
// @param isShowTool 是否展示一些工具聊天,什么文件助手呀,订阅号等等没用的东西
// @return friends
// @return err
func (w *WxHelper) GetContactList(isShowTool bool) (friends []Friend, err error) {
req := w.client.R()
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/getContactList"))
if err != nil {
log.Fatalf("获取好友列表失败: %v", err.Error())
return
}
if w.tmpResult.Code != 1 {
return
}
data, _ := json.Marshal(w.tmpResult.Data)
_ = json.Unmarshal(data, &friends)
// 如果不需要展示那些工具号,直接剔除
if !isShowTool {
var onlyFriends []Friend
for _, v := range friends {
// 固定开头
if strings.HasPrefix(v.Wxid, "wxid") {
onlyFriends = append(onlyFriends, v)
}
}
friends = onlyFriends
}
return
}
// SendAtText
// @description: 群内圈人文本消息发送
// @receiver w
// @param param
// @return err
func (w *WxHelper) SendAtText(param *SendAtText) (err error) {
req := w.client.R()
req.Header.Set("Content-Type", "application/json")
req.SetBody(param)
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/sendAtText"))
if err != nil {
return fmt.Errorf("发送群内圈人消息失败: %v", err.Error())
}
if w.tmpResult.Code <= 0 {
return fmt.Errorf("发送群内圈人消息失败: %v", w.tmpResult.Msg)
}
return
}
// SendText
// @description: 发送文本信息(私聊)
// @receiver w
// @param param
// @return err
func (w *WxHelper) SendText(param *SendText) (err error) {
req := w.client.R()
req.Header.Set("Content-Type", "application/json")
req.SetBody(param)
_, err = req.SetResult(&w.tmpResult).Post(w.GetAddr("/api/sendAtText"))
if err != nil {
return fmt.Errorf("发送私聊文本信息失败: %v", err.Error())
}
if w.tmpResult.Code == 0 {
return fmt.Errorf("发送私聊文本信息失败: %v", w.tmpResult.Msg)
}
return
}

36
client_test.go Normal file
View File

@ -0,0 +1,36 @@
package wx_helper_sdk
import (
"fmt"
"testing"
)
var wxh WxHelper
func init() {
wxh = NewWxHelper("", 0)
}
func TestClientCheckUserIsLogin(t *testing.T) {
ok := wxh.CheckUserIsLogin()
fmt.Println(ok)
}
func TestClientGetCurrentLoginUser(t *testing.T) {
info, err := wxh.GetCurrentLoginUserInfo()
if err != nil {
t.Logf("failed: %v", err.Error())
return
}
fmt.Printf("%+v\n", info)
}
func TestClientGetContactList(t *testing.T) {
list, err := wxh.GetContactList(false)
if err != nil {
return
}
fmt.Printf("%+v", list)
}

8
go.mod Normal file
View File

@ -0,0 +1,8 @@
module gitea.mrx.ltd/go-pkg/wx_helper_sdk
go 1.21
require (
github.com/go-resty/resty/v2 v2.11.0 // indirect
golang.org/x/net v0.17.0 // indirect
)

43
go.sum Normal file
View File

@ -0,0 +1,43 @@
github.com/go-resty/resty/v2 v2.11.0 h1:i7jMfNOJYMp69lq7qozJP+bjgzfAzeOhuGlyDrqxT/8=
github.com/go-resty/resty/v2 v2.11.0/go.mod h1:iiP/OpA0CkcL3IGt1O0+/SIItFUbkkyw5BGXiVdTu+A=
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
golang.org/x/term v0.13.0/go.mod h1:LTmsnFJwVN6bCy1rVCoS+qHT1HhALEFxKncY3WNNh4U=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
golang.org/x/time v0.3.0/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=

26
request.go Normal file
View File

@ -0,0 +1,26 @@
package wx_helper_sdk
// SetHook
// @description: 设置回调
type SetHook struct {
Port string `json:"port"`
Ip string `json:"ip"`
Url string `json:"url"`
Timeout string `json:"timeout"`
EnableHttp string `json:"enableHttp"`
}
// SendAtText
// @description: 发送群里圈人文本信息
type SendAtText struct {
Wxids string `json:"wxids"`
ChatRoomId string `json:"chatRoomId"`
Msg string `json:"msg"`
}
// SendText
// @description: 发送文本信息(私聊)
type SendText struct {
Wxid string `json:"wxid"`
Msg string `json:"msg"`
}

41
response.go Normal file
View File

@ -0,0 +1,41 @@
package wx_helper_sdk
// CommonResponse
// @description: 公共返回
type CommonResponse struct {
Code int `json:"code"`
Msg string `json:"msg"`
Data any `json:"data"`
}
// UserInfo
// @description: 当前用户信息
type UserInfo struct {
Account string `json:"account"`
City string `json:"city"`
Country string `json:"country"`
CurrentDataPath string `json:"currentDataPath"`
DataSavePath string `json:"dataSavePath"`
DbKey string `json:"dbKey"`
HeadImage string `json:"headImage"`
Mobile string `json:"mobile"`
Name string `json:"name"`
Province string `json:"province"`
Signature string `json:"signature"`
Wxid string `json:"wxid"`
}
// Friend
// @description: 朋友信息
type Friend struct {
CustomAccount string `json:"customAccount"`
EncryptName string `json:"encryptName"`
Nickname string `json:"nickname"`
Pinyin string `json:"pinyin"`
PinyinAll string `json:"pinyinAll"`
Reserved1 int `json:"reserved1"`
Reserved2 int `json:"reserved2"`
Type int `json:"type"`
VerifyFlag int `json:"verifyFlag"`
Wxid string `json:"wxid"`
}