add go

2023-04-29 10:00:48 +08:00
parent bc57b9a54d
commit 9b99c62d00
9 changed files with 3087 additions and 0 deletions
@@ -0,0 +1,342 @@
+package wcf
+
+import (
+	logs "github.com/danbai225/go-logs"
+	"go.nanomsg.org/mangos/v3"
+	"go.nanomsg.org/mangos/v3/protocol"
+	"go.nanomsg.org/mangos/v3/protocol/pair1"
+	_ "go.nanomsg.org/mangos/v3/transport/all"
+	"google.golang.org/protobuf/proto"
+	"strconv"
+	"strings"
+)
+
+type Client struct {
+	add     string
+	socket  protocol.Socket
+	RecvTxt bool
+}
+
+func (c *Client) conn() error {
+	socket, err := pair1.NewSocket()
+	if err != nil {
+		return err
+	}
+	err = socket.Dial(c.add)
+	if err != nil {
+		return err
+	}
+	c.socket = socket
+	return err
+}
+func (c *Client) send(data []byte) error {
+	return c.socket.Send(data)
+}
+func (c *Client) Recv() (*Response, error) {
+	msg := &Response{}
+	recv, err := c.socket.Recv()
+	if err != nil {
+		return msg, err
+	}
+	err = proto.Unmarshal(recv, msg)
+	return msg, err
+}
+func (c *Client) IsLogin() bool {
+	err := c.send(genFunReq(Functions_FUNC_IS_LOGIN).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	if recv.GetStatus() == 1 {
+		return true
+	}
+	return false
+}
+func (c *Client) GetSelfWXID() string {
+	err := c.send(genFunReq(Functions_FUNC_GET_SELF_WXID).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStr()
+}
+func (c *Client) GetMsgTypes() map[int32]string {
+	err := c.send(genFunReq(Functions_FUNC_GET_MSG_TYPES).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetTypes().GetTypes()
+}
+func (c *Client) GetContacts() []*RpcContact {
+	err := c.send(genFunReq(Functions_FUNC_GET_CONTACTS).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetContacts().GetContacts()
+}
+func (c *Client) GetDBNames() []string {
+	err := c.send(genFunReq(Functions_FUNC_GET_DB_NAMES).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetDbs().Names
+}
+func (c *Client) GetDBTables(tab string) []*DbTable {
+	req := genFunReq(Functions_FUNC_GET_DB_TABLES)
+	str := &Request_Str{Str: tab}
+	req.Msg = str
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetTables().GetTables()
+}
+func (c *Client) ExecDBQuery(db, sql string) []*DbRow {
+	req := genFunReq(Functions_FUNC_EXEC_DB_QUERY)
+	q := Request_Query{
+		Query: &DbQuery{
+			Db:  db,
+			Sql: sql,
+		},
+	}
+	req.Msg = &q
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetRows().GetRows()
+}
+func (c *Client) AcceptFriend() int32 {
+	err := c.send(genFunReq(Functions_FUNC_ACCEPT_FRIEND).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+func (c *Client) AddChatRoomMembers(roomId string, wxIds []string) int32 {
+	req := genFunReq(Functions_FUNC_ADD_ROOM_MEMBERS)
+	q := Request_M{
+		M: &AddMembers{Roomid: roomId,
+			Wxids: strings.Join(wxIds, ",")},
+	}
+	req.Msg = &q
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+func (c *Client) GetUserInfo() *UserInfo {
+	err := c.send(genFunReq(Functions_FUNC_GET_USER_INFO).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetUi()
+}
+
+/*
+SendTxt
+@param msg:      消息内容（如果是 @ 消息则需要有跟 @ 的人数量相同的 @）
+@param receiver: 消息接收人，私聊为 wxid（wxid_xxxxxxxxxxxxxx），群聊为roomid（xxxxxxxxxx@chatroom）
+@param ates:    群聊时要 @ 的人（私聊时为空字符串），多个用逗号分隔。@所有人 用notify@all（必须是群主或者管理员才有权限）
+*/
+func (c *Client) SendTxt(msg string, receiver string, ates []string) int32 {
+	req := genFunReq(Functions_FUNC_SEND_TXT)
+	req.Msg = &Request_Txt{
+		Txt: &TextMsg{
+			Msg:      msg,
+			Receiver: receiver,
+			Aters:    strings.Join(ates, ","),
+		},
+	}
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+
+/*
+SendIMG
+path 绝对路径
+*/
+func (c *Client) SendIMG(path string, receiver string) int32 {
+	req := genFunReq(Functions_FUNC_SEND_IMG)
+	req.Msg = &Request_File{
+		File: &PathMsg{
+			Path:     path,
+			Receiver: receiver,
+		},
+	}
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+
+/*
+SendFile
+path 绝对路径
+*/
+func (c *Client) SendFile(path string, receiver string) int32 {
+	req := genFunReq(Functions_FUNC_SEND_FILE)
+	req.Msg = &Request_File{
+		File: &PathMsg{
+			Path:     path,
+			Receiver: receiver,
+		},
+	}
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+func (c *Client) SendXml(path, content, receiver string, Type int32) int32 {
+	req := genFunReq(Functions_FUNC_SEND_XML)
+	req.Msg = &Request_Xml{
+		Xml: &XmlMsg{
+			Receiver: receiver,
+			Content:  content,
+			Path:     path,
+			Type:     Type,
+		},
+	}
+	err := c.send(req.build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	return recv.GetStatus()
+}
+func (c *Client) EnableRecvTxt() int32 {
+	err := c.send(genFunReq(Functions_FUNC_ENABLE_RECV_TXT).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	c.RecvTxt = true
+	return recv.GetStatus()
+}
+func (c *Client) DisableRecvTxt() int32 {
+	err := c.send(genFunReq(Functions_FUNC_DISABLE_RECV_TXT).build())
+	if err != nil {
+		logs.Err(err)
+	}
+	recv, err := c.Recv()
+	if err != nil {
+		logs.Err(err)
+	}
+	c.RecvTxt = false
+	return recv.GetStatus()
+}
+func (c *Client) OnMSG(f func(msg *WxMsg)) error {
+	socket, err := pair1.NewSocket()
+	if err != nil {
+		return err
+	}
+	socket.SetOption(mangos.OptionRecvDeadline, 2000)
+	socket.SetOption(mangos.OptionSendDeadline, 2000)
+	err = socket.Dial(addPort(c.add))
+
+	if err != nil {
+		return err
+	}
+	for c.RecvTxt {
+		msg := &Response{}
+		recv, err := socket.Recv()
+		if err != nil {
+			return err
+		}
+		logs.Info(len(recv))
+		_ = proto.Unmarshal(recv, msg)
+		go f(msg.GetWxmsg())
+	}
+	return err
+}
+func NewWCF(add string) (*Client, error) {
+	if add == "" {
+		add = "tcp://127.0.0.1:10086"
+	}
+	client := &Client{add: add}
+	err := client.conn()
+	return client, err
+}
+
+type cmdMSG struct {
+	*Request
+}
+
+func (c *cmdMSG) build() []byte {
+	marshal, _ := proto.Marshal(c)
+	return marshal
+}
+func genFunReq(fun Functions) *cmdMSG {
+	return &cmdMSG{
+		&Request{Func: fun,
+			Msg: nil},
+	}
+}
+func addPort(add string) string {
+	parts := strings.Split(add, ":")
+	port, _ := strconv.Atoi(parts[2])
+	newPort := port + 1
+	return parts[0] + ":" + parts[1] + ":" + strconv.Itoa(newPort)
+}
@@ -0,0 +1,152 @@
+package wcf
+
+import (
+	logs "github.com/danbai225/go-logs"
+	"go.nanomsg.org/mangos/v3/protocol/pair1"
+	_ "go.nanomsg.org/mangos/v3/transport/all"
+	"google.golang.org/protobuf/proto"
+	"testing"
+)
+
+func TestConn(t *testing.T) {
+	socket, err := pair1.NewSocket()
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	err = socket.Dial("tcp://127.0.0.1:10086")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	response := Response{
+		Func: Functions_FUNC_IS_LOGIN,
+		Msg:  nil,
+	}
+	marshal, err := proto.Marshal(&response)
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	socket.Send(marshal)
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	recv, err := socket.Recv()
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	msg := Response{}
+	err = proto.Unmarshal(recv, &msg)
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(msg.GetStatus())
+
+}
+func TestIsLogin(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.IsLogin())
+}
+func TestGetSelfWXID(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.GetSelfWXID())
+}
+func TestGetMsgTypes(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.GetMsgTypes())
+}
+func TestGetContacts(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	for _, contact := range wcf.GetContacts() {
+		logs.Info(contact.Remark, contact.Wxid)
+	}
+	logs.Info(wcf.GetContacts())
+}
+func TestGetDBNames(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.GetDBNames())
+}
+func TestGetDBTables(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.GetDBTables("ChatMsg.db"))
+}
+func TestExecDBQuery(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.ExecDBQuery("ChatMsg.db", "SELECT * FROM Name2ID_v1"))
+}
+func TestAcceptFriend(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.AcceptFriend())
+}
+func TestGetUserInfo(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.GetUserInfo())
+}
+func TestSendTxT(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.SendTxt(" Hello @ 淡白", "45415088466@chatroom", []string{"wxid_xxxxxxxx"}))
+}
+func TestSendIMG(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.SendIMG("C:\\Users\\danbai\\Desktop\\work\\code\\WeChatFerry-go\\图片1.png", "wxid_qvo0irhbw9fk22"))
+}
+func TestOnMSG(t *testing.T) {
+	wcf, err := NewWCF("")
+	if err != nil {
+		logs.Err(err)
+		return
+	}
+	logs.Info(wcf.EnableRecvTxt())
+	defer wcf.DisableRecvTxt()
+	wcf.OnMSG(func(msg *WxMsg) {
+		logs.Info(msg.GetContent())
+	})
+}