lxh/go-wxhelper
4
4
Fork 5
Code Issues 3 Pull Requests Actions Packages Projects 2 Releases 114 Wiki Activity

🎉 first commit

Browse Source
This commit is contained in:
李寻欢 2023-09-21 17:33:59 +08:00
commit 744fb7b1d0
18 changed files with 648 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
.gitignore vendored Normal file
View File

@ -0,0 +1,9 @@
.idea
vendor
logs
*.exe
*.pprof
cache
log
dist
*.log

18
Dockerfile Normal file
View File

@ -0,0 +1,18 @@
FROM golang:alpine as builder
WORKDIR /builder
COPY . .
ENV GO111MODULE=on
ENV GOPROXY=https://goproxy.cn,direct
RUN go mod download && go build -o app
RUN ls -lh && chmod +x ./app
FROM repo.lxh.io/alpine:3.16.0 as runner
LABEL org.opencontainers.image.authors="lxh@cxh.cn"
EXPOSE 19099
WORKDIR /app
COPY --from=builder /builder/app ./app
CMD ./app

28
client/mysql.go Normal file
View File

@ -0,0 +1,28 @@
package client
import (
"gorm.io/driver/mysql"
"gorm.io/gorm"
"log"
)
// MySQL客户端
var MySQL *gorm.DB
func InitMySQLClient() {
dsn := "wechat:wechat123@tcp(10.0.0.31:3307)/wechat?charset=utf8mb4&parseTime=True&loc=Local"
// 创建连接对象
mysqlConfig := mysql.Config{
DSN: dsn,
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式
DontSupportRenameColumn: true, // 用 `change` 重命名列
}
conn, err := gorm.Open(mysql.New(mysqlConfig))
if err != nil {
log.Panicf("初始化MySQL连接失败, 错误信息: %v", err)
} else {
log.Println("MySQL连接成功")
}
MySQL = conn
}

6
constant/wxid.go Normal file
View File

@ -0,0 +1,6 @@
package constant
var SpecialId = []string{"filehelper", "newsapp", "fmessage", "weibo", "qqmail", "tmessage", "qmessage", "qqsync",
"floatbottle", "lbsapp", "shakeapp", "medianote", "qqfriend", "readerapp", "blogapp", "facebookapp", "masssendapp",
"meishiapp", "feedsapp", "voip", "blogappweixin", "weixin", "brandsessionholder", "weixinreminder", "officialaccounts",
"notification_messages", "wxitil", "userexperience_alarm", "notification_messages", "exmail_tool"}

20
entity/friend.go Normal file
View File

@ -0,0 +1,20 @@
package entity
// 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"` // 微信原始Id
}
func (Friend) TableName() string {
return "t_friend"
}

25
entity/message.go Normal file
View File

@ -0,0 +1,25 @@
package entity
import (
"go-wechat/types"
"time"
)
// Message
// @description: 消息数据库结构体
type Message struct {
MsgId int64 `gorm:"primaryKey"` // 消息Id
CreateTime int // 发送时间戳
CreateAt time.Time // 发送时间
Type types.MessageType // 消息类型
Content string // 内容
DisplayFullContent string // 显示的完整内容
FromUser string // 发送者
GroupUser string // 群成员
ToUser string // 接收者
Raw string // 原始通知字符串
}
func (Message) TableName() string {
return "t_message"
}

20
go.mod Normal file
View File

@ -0,0 +1,20 @@
module go-wechat
go 1.21
require (
github.com/go-co-op/gocron v1.34.1
github.com/go-resty/resty/v2 v2.8.0
gorm.io/driver/mysql v1.5.1
gorm.io/gorm v1.25.4
)
require (
github.com/go-sql-driver/mysql v1.7.0 // indirect
github.com/google/uuid v1.3.1 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
github.com/robfig/cron/v3 v3.0.1 // indirect
go.uber.org/atomic v1.9.0 // indirect
golang.org/x/net v0.15.0 // indirect
)

91
go.sum Normal file
View File

@ -0,0 +1,91 @@
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/go-co-op/gocron v1.34.1 h1:g4Y1ePFin2z4Rbb5gTHNYfj36moY4kU9s0ZmGy/ZddQ=
github.com/go-co-op/gocron v1.34.1/go.mod h1:NLi+bkm4rRSy1F8U7iacZOz0xPseMoIOnvabGoSe/no=
github.com/go-resty/resty/v2 v2.8.0 h1:J29d0JFWwSWrDCysnOK/YjsPMLQTx0TvgJEHVGvf2L8=
github.com/go-resty/resty/v2 v2.8.0/go.mod h1:UCui0cMHekLrSntoMyofdSTaPpinlRHFtPpizuyDW2w=
github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc=
github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
github.com/google/uuid v1.3.1 h1:KjJaJ9iWZ3jOFZIf1Lqf4laDRCasjl0BCmnEGxkdLb4=
github.com/google/uuid v1.3.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
github.com/rogpeppe/go-internal v1.8.1/go.mod h1:JeRgkft04UBgHMgCIwADu4Pn6Mtm5d4nPKWu0nJ5d+o=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
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.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc=
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.15.0 h1:ugBLEUaxABaB5AJqW9enI0ACdci2RUd4eP51NTBvuJ8=
golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
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.12.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.12.0/go.mod h1:owVbMEjm3cBLCHdkQu9b1opXd4ETQWc3BhuQGKgXgvU=
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/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=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gorm.io/driver/mysql v1.5.1 h1:WUEH5VF9obL/lTtzjmML/5e6VfFR/788coz2uaVCAZw=
gorm.io/driver/mysql v1.5.1/go.mod h1:Jo3Xu7mMhCyj8dlrb3WoCaRd1FhsVh+yMXb1jUInf5o=
gorm.io/gorm v1.25.1/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
gorm.io/gorm v1.25.4 h1:iyNd8fNAe8W9dvtlgeRI5zSVZPsq3OpcTu37cYcpCmw=
gorm.io/gorm v1.25.4/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=

54
handler/parse.go Normal file
View File

@ -0,0 +1,54 @@
package handler
import (
"encoding/json"
"go-wechat/entity"
"go-wechat/model"
"go-wechat/service"
"go-wechat/types"
"log"
"net"
"strings"
"time"
)
// Parse
// @description: 解析消息
// @param msg
func Parse(remoteAddr net.Addr, msg []byte) {
var m model.Message
if err := json.Unmarshal(msg, &m); err != nil {
log.Printf("[%s]消息解析失败： %v", remoteAddr, err)
log.Printf("[%s]消息内容： %d -> %v", remoteAddr, len(msg), string(msg))
return
}
// 提取出群成员信息
groupUser := ""
msgStr := m.Content
if strings.Contains(m.FromUser, "@") {
// 系统消息不单独处理
if m.Type != types.MsgTypeRecalled && m.Type != types.MsgTypeSys {
groupUser = strings.Split(m.Content, "\n")[0]
groupUser = strings.ReplaceAll(groupUser, ":", "")
// 文字消息单独提出来处理一下
msgStr = strings.Join(strings.Split(m.Content, "\n")[1:], "\n")
}
}
log.Printf("%s\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", remoteAddr, m.FromUser, groupUser, m.Type, msgStr)
// 转换为结构体之后入库
var ent entity.Message
ent.MsgId = m.MsgId
ent.CreateTime = m.CreateTime
ent.CreateAt = time.Unix(int64(m.CreateTime), 0)
ent.Content = msgStr
ent.FromUser = m.FromUser
ent.GroupUser = groupUser
ent.ToUser = m.ToUser
ent.Type = m.Type
ent.DisplayFullContent = m.DisplayFullContent
ent.Raw = string(msg)
go service.SaveMessage(ent)
}

55
main.go Normal file
View File

@ -0,0 +1,55 @@
package main
import (
"bytes"
"go-wechat/client"
"go-wechat/handler"
"go-wechat/tasks"
"io"
"log"
"net"
)
func init() {
client.InitMySQLClient()
tasks.InitTasks()
}
func process(conn net.Conn) {
// 处理完关闭连接
defer func() {
log.Printf("处理完成: -> %s", conn.RemoteAddr())
_ = conn.Close()
}()
var buf bytes.Buffer
if _, err := io.Copy(&buf, conn); err != nil {
log.Printf("[%s]返回数据失败，错误信息: %v", conn.RemoteAddr(), err)
}
log.Printf("[%s]数据长度: %d", conn.RemoteAddr(), buf.Len())
go handler.Parse(conn.RemoteAddr(), buf.Bytes())
// 将接受到的数据返回给客户端
if _, err := conn.Write([]byte("200 OK")); err != nil {
log.Printf("[%s]返回数据失败，错误信息: %v", conn.RemoteAddr(), err)
}
}
func main() {
// 建立 tcp 服务
listen, err := net.Listen("tcp", "0.0.0.0:19099")
if err != nil {
log.Printf("listen failed, err:%v", err)
return
}
for {
// 等待客户端建立连接
conn, err := listen.Accept()
if err != nil {
log.Printf("accept failed, err:%v", err)
continue
}
// 启动一个单独的 goroutine 去处理连接
go process(conn)
}
}

36
model/friend.go Normal file
View File

@ -0,0 +1,36 @@
package model
// FriendItem
// @description: 好友列表数据
type FriendItem 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"` // 微信原始Id
}
// GroupUser
// @description: 群成员返回结果
type GroupUser struct {
Admin string `json:"admin"` // 群主微信
AdminNickname string `json:"adminNickname"` // 群主昵称
ChatRoomId string `json:"chatRoomId"` // 群Id
MemberNickname string `json:"memberNickname"` // 成员昵称 `^G`切割
Members string `json:"members"` // 成员Id `^G`切割
}
// ContactProfile
// @description: 好友资料
type ContactProfile struct {
Account string `json:"account"` // 账号
HeadImage string `json:"headImage"` // 头像
Nickname string `json:"nickname"` // 昵称
V3 string `json:"v3"` // v3
Wxid string `json:"wxid"` // 微信Id
}

18
model/message.go Normal file
View File

@ -0,0 +1,18 @@
package model
import "go-wechat/types"
// Message
// @description: 消息
type Message struct {
MsgId int64 `json:"msgId" gorm:"primarykey"`
CreateTime int `json:"createTime"`
Content string `json:"content"`
DisplayFullContent string `json:"displayFullContent" gorm:"-"`
FromUser string `json:"fromUser"`
MsgSequence int `json:"msgSequence"`
Pid int `json:"pid"`
Signature string `json:"signature"`
ToUser string `json:"toUser"`
Type types.MessageType `json:"type"`
}

9
model/response.go Normal file
View File

@ -0,0 +1,9 @@
package model
// Response
// @description: 基础返回结构体
type Response[T any] struct {
Code int `json:"code"` // 状态码
Data T `json:"data"` // 数据
Msg string `json:"msg"` // 消息
}

28
service/message.go Normal file
View File

@ -0,0 +1,28 @@
package service
import (
"go-wechat/client"
"go-wechat/entity"
"log"
)
// SaveMessage
// @description: 消息入库
// @param msg
func SaveMessage(msg entity.Message) {
// 检查消息是否存在，存在就跳过
var count int64
err := client.MySQL.Model(&entity.Message{}).Where("msg_id = ?", msg.MsgId).Count(&count).Error
if err != nil {
log.Printf("检查消息是否存在失败, 错误信息: %v", err)
return
}
if count > 0 {
return
}
err = client.MySQL.Create(&msg).Error
if err != nil {
log.Printf("消息入库失败, 错误信息: %v", err)
}
log.Printf("消息入库成功，消息Id: %d", msg.MsgId)
}

111
tasks/friends.go Normal file
View File

@ -0,0 +1,111 @@
package tasks
import (
"encoding/json"
"github.com/go-resty/resty/v2"
"go-wechat/constant"
"go-wechat/entity"
"go-wechat/model"
"log"
"slices"
"strings"
)
// 同步群成员
// syncFriends
// @description: 同步好友列表
func syncFriends() {
var base model.Response[[]entity.Friend]
client := resty.New()
resp, err := client.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetResult(&base).
Post("http://10.0.0.73:19088/api/getContactList")
if err != nil {
log.Printf("获取好友列表失败: %s", err.Error())
return
}
log.Printf("获取好友列表结果: %s", resp.String())
for _, friend := range base.Data {
if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") {
continue
}
// 特殊Id跳过
if slices.Contains(constant.SpecialId, friend.Wxid) {
continue
}
log.Printf("昵称: %s -> 类型: %d -> 微信号: %s -> 微信原始Id: %s", friend.Nickname, friend.Type, friend.CustomAccount, friend.Wxid)
// 群成员，同步一下成员信息
if strings.Contains(friend.Wxid, "@chatroom") {
syncGroupUsers(friend.Wxid)
}
}
}
// syncGroupUsers
// @description: 同步群成员
// @param gid
func syncGroupUsers(gid string) {
var baseResp model.Response[model.GroupUser]
// 组装参数
param := map[string]any{
"chatRoomId": gid, // 群Id
}
pbs, _ := json.Marshal(param)
client := resty.New()
_, err := client.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&baseResp).
Post("http://10.0.0.73:19088/api/getMemberFromChatRoom")
if err != nil {
log.Printf("获取群成员信息失败: %s", err.Error())
return
}
// 昵称Id
wxIds := strings.Split(baseResp.Data.Members, "^G")
log.Printf(" 群成员数: %d", len(wxIds))
for _, wxid := range wxIds {
// 获取成员信息
cp, _ := getContactProfile(wxid)
if cp.Wxid != "" {
log.Printf(" 微信Id: %s -> 昵称: %s -> 微信号: %s", wxid, cp.Nickname, cp.Account)
}
}
}
// getContactProfile
// @description: 获取成员详情
// @param wxid
// @return ent
// @return err
func getContactProfile(wxid string) (ent model.ContactProfile, err error) {
var baseResp model.Response[model.ContactProfile]
// 组装参数
param := map[string]any{
"wxid": wxid, // 群Id
}
pbs, _ := json.Marshal(param)
client := resty.New()
_, err = client.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&baseResp).
Post("http://10.0.0.73:19088/api/getContactProfile")
if err != nil {
log.Printf("获取成员详情失败: %s", err.Error())
return
}
ent = baseResp.Data
return
}

24
tasks/tasks.go Normal file
View File

@ -0,0 +1,24 @@
package tasks
import (
"github.com/go-co-op/gocron"
"log"
"time"
)
// InitTasks
// @description: 初始化定时任务
func InitTasks() {
// 定时任务发送消息
s := gocron.NewScheduler(time.Local)
// 每天早上九点半发送前一天的水群排行
_, _ = s.Every(1).Day().At("09:30").Do(yesterday)
// 每小时更新一次好友列表
_, _ = s.Every(1).Minute().Do(syncFriends)
// 开启定时任务
s.StartAsync()
log.Println("定时任务初始化成功")
}

46
tasks/water_group.go Normal file
View File

@ -0,0 +1,46 @@
package tasks
import (
"go-wechat/client"
"go-wechat/entity"
"log"
)
// 水群排行榜
// yesterday
// @description: 昨日排行榜
func yesterday() {
// 获取昨日消息总数
var yesterdayMsgCount int64
err := client.MySQL.Model(&entity.Message{}).
Where("from_user = ?", "18958257758@chatroom").
Where("DATEDIFF(create_at,NOW()) = -1").
Count(&yesterdayMsgCount).Error
if err != nil {
log.Printf("获取昨日消息总数失败, 错误信息: %v", err)
return
}
log.Printf("昨日消息总数: %d", yesterdayMsgCount)
// 返回数据
type record struct {
GroupUser string
Count int64
}
var records []record
err = client.MySQL.Model(&entity.Message{}).
Select("group_user", "count( 1 ) AS `count`").
Where("from_user = ?", "18958257758@chatroom").
Where("DATEDIFF(create_at,NOW()) = -1").
Group("group_user").Order("`count` DESC").
Limit(5).Find(&records).Error
if err != nil {
log.Printf("获取昨日消息失败, 错误信息: %v", err)
return
}
for _, r := range records {
log.Printf("账号: %s -> %d", r.GroupUser, r.Count)
}
}

50
types/message.go Normal file
View File

@ -0,0 +1,50 @@
package types
import "fmt"
type MessageType int
const (
MsgTypeText MessageType = 1 // 文本消息
MsgTypeImage MessageType = 3 // 图片消息
MsgTypeVoice MessageType = 34 // 语音消息
MsgTypeVerify MessageType = 37 // 认证消息
MsgTypePossibleFriend MessageType = 40 // 好友推荐消息
MsgTypeShareCard MessageType = 42 // 名片消息
MsgTypeVideo MessageType = 43 // 视频消息
MsgTypeEmoticon MessageType = 47 // 表情消息
MsgTypeLocation MessageType = 48 // 地理位置消息
MsgTypeApp MessageType = 49 // APP消息
MsgTypeVoip MessageType = 50 // VOIP消息
MsgTypeVoipNotify MessageType = 52 // VOIP结束消息
MsgTypeVoipInvite MessageType = 53 // VOIP邀请
MsgTypeMicroVideo MessageType = 62 // 小视频消息
MsgTypeSys MessageType = 10000 // 系统消息
MsgTypeRecalled MessageType = 10002 // 消息撤回
)
var MessageTypeMap = map[MessageType]string{
MsgTypeText: "文本消息",
MsgTypeImage: "图片消息",
MsgTypeVoice: "语音消息",
MsgTypeVerify: "认证消息",
MsgTypePossibleFriend: "好友推荐消息",
MsgTypeShareCard: "名片消息",
MsgTypeVideo: "视频消息",
MsgTypeEmoticon: "表情消息",
MsgTypeLocation: "地理位置消息",
MsgTypeApp: "APP消息",
MsgTypeVoip: "VOIP消息",
MsgTypeVoipNotify: "VOIP结束消息",
MsgTypeVoipInvite: "VOIP邀请",
MsgTypeMicroVideo: "小视频消息",
MsgTypeSys: "系统消息",
MsgTypeRecalled: "消息撤回",
}
func (mt MessageType) String() string {
if msg, ok := MessageTypeMap[mt]; ok {
return msg
}
return fmt.Sprintf("未知消息类型(%d)", mt)
}