🔥 移除测试代码

client/mysql.go

@@ -1,6 +1,7 @@
 package client
 
 import (
+	"go-wechat/config"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 	"log"
@@ -10,11 +11,9 @@ import (
 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,
+		DSN:                     config.Conf.MySQL.GetDSN(),
 		DontSupportRenameIndex:  true, // 重命名索引时采用删除并新建的方式
 		DontSupportRenameColumn: true, // 用 `change` 重命名列
 	}

config.yaml

@@ -1,13 +1,41 @@
+# 微信HOOK配置
+wechat:
+  # 微信HOOK接口地址
+  host: 10.0.0.73:19088
+  # 是否在启动的时候自动设置hook服务的回调
+  autoSetCallback: false
+  # 回调IP，如果是Docker运行，本参数必填，如果Docker修改了映射，格式为 ip:port
+  callback: 10.0.0.51
+
+# 数据库
+mysql:
+  host: 10.0.0.31
+  port: 3307
+  user: wechat
+  password: wechat123
+  db: wechat
+
 task:
   enable: true
   syncFriends:
       enable: true
-      cron: '0 * * * *'
+      cron: '*/5 * * * *' # 五分钟一次
   waterGroup:
-      enable: true
-      cron: '30 9 * * *'
-      groups:
-        - '18958257758@chatroom'
-        - '49448748645@chatroom'
-      blacklist:
-        - 'wxid_778868788691exit2'
+      enable: false
+      cron:
+        yesterday: '30 9 * * *' # 每天9:30
+        week: '30 9 * * 1' # 每周一9:30
+        month: '30 9 1 * *' # 每月1号9:30
+
+# AI回复
+ai:
+  # 是否启用
+  enable: false
+  # 模型，不填默认gpt-3.5-turbo-0613
+  model: gpt-3.5-turbo-0613
+  # OpenAI Api key
+  apiKey: sk-xxxx
+  # 接口代理域名，不填默认ChatGPT官方地址
+  baseUrl: https://sxxx
+  # 人设
+  personality: 你的名字叫张三，你是一个百科机器人，你的爱好是看电影，你的性格是开朗的，你的专长是讲故事，你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣，也不会讨论任何与政治相关的话题，你甚至可以拒绝回答这一类话题。

config/ai.go

@@ -0,0 +1,11 @@
+package config
+
+// ai
+// @description: AI配置
+type ai struct {
+	Enable      bool   `json:"enable" yaml:"enable"`           // 是否启用AI
+	Model       string `json:"model" yaml:"model"`             // 模型
+	ApiKey      string `json:"apiKey" yaml:"apiKey"`           // API Key
+	BaseUrl     string `json:"baseUrl" yaml:"baseUrl"`         // API地址
+	Personality string `json:"personality" yaml:"personality"` // 人设
+}

config/config.go

@@ -6,20 +6,7 @@ var Conf Config
 // @description: 配置
 type Config struct {
 	Task   task   `json:"task" yaml:"task"`     // 定时任务配置
-}
-
-// task
-// @description: 定时任务
-type task struct {
-	Enable      bool `json:"enable" yaml:"enable"` // 是否启用
-	SyncFriends struct {
-		Enable bool   `json:"enable" yaml:"enable"` // 是否启用
-		Cron   string `json:"cron" yaml:"cron"`     // 定时任务表达式
-	} `json:"syncFriends" yaml:"syncFriends"` // 同步好友
-	WaterGroup struct {
-		Enable    bool     `json:"enable" yaml:"enable"`       // 是否启用
-		Cron      string   `json:"cron" yaml:"cron"`           // 定时任务表达式
-		Groups    []string `json:"groups" yaml:"groups"`       // 启用的群Id
-		Blacklist []string `json:"blacklist" yaml:"blacklist"` // 黑名单
-	} `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
+	MySQL  mysql  `json:"mysql" yaml:"mysql"`   // MySQL 配置
+	Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
+	Ai     ai     `json:"ai" yaml:"ai"`         // AI配置
 }

config/mysql.go

@@ -0,0 +1,24 @@
+package config
+
+import (
+	"fmt"
+)
+
+// mysql
+// @description: MySQL配置
+type mysql struct {
+	Host     string `mapstructure:"host" yaml:"host"`         // 主机
+	Port     int    `mapstructure:"port" yaml:"port"`         // 端口
+	User     string `mapstructure:"user" yaml:"user"`         // 用户名
+	Password string `mapstructure:"password" yaml:"password"` // 密码
+	Db       string `mapstructure:"db" yaml:"db"`             // 数据库名称
+}
+
+// GetDSN
+// @description: 返回 MySQL 连接字符串
+// @receiver c
+// @return string
+func (c mysql) GetDSN() string {
+	return fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		c.User, c.Password, c.Host, c.Port, c.Db)
+}

config/task.go

@@ -0,0 +1,31 @@
+package config
+
+// task
+// @description: 定时任务
+type task struct {
+	Enable      bool        `json:"enable" yaml:"enable"`           // 是否启用
+	SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友
+	WaterGroup  waterGroup  `json:"waterGroup" yaml:"waterGroup"`   // 水群排行榜
+}
+
+// syncFriends
+// @description: 同步好友
+type syncFriends struct {
+	Enable bool   `json:"enable" yaml:"enable"` // 是否启用
+	Cron   string `json:"cron" yaml:"cron"`     // 定时任务表达式
+}
+
+// waterGroup
+// @description: 水群排行榜
+type waterGroup struct {
+	Enable bool           `json:"enable" yaml:"enable"` // 是否启用
+	Cron   waterGroupCron `json:"cron" yaml:"cron"`     // 定时任务表达式
+}
+
+// waterGroupCron
+// @description: 水群排行榜定时任务
+type waterGroupCron struct {
+	Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
+	Week      string `json:"week" yaml:"week"`           // 周排行榜
+	Month     string `json:"month" yaml:"month"`         // 月排行榜
+}

config/wechat.go

@@ -0,0 +1,33 @@
+package config
+
+import "strings"
+
+// wxHelper
+// @description: 微信助手
+type wechat struct {
+	Host            string `json:"host" yaml:"host"`                       // 接口地址
+	AutoSetCallback bool   `json:"autoSetCallback" yaml:"autoSetCallback"` // 是否自动设置回调地址
+	Callback        string `json:"callback" yaml:"callback"`               // 回调地址
+}
+
+// Check
+// @description: 检查配置是否可用
+// @receiver w
+// @return bool
+func (w wechat) Check() bool {
+	if w.Host == "" {
+		return false
+	}
+	if w.AutoSetCallback && w.Callback == "" {
+		return false
+	}
+	return true
+}
+
+func (w wechat) GetURL(uri string) string {
+	host := w.Host
+	if !strings.HasPrefix(w.Host, "http://") {
+		host = "http://" + w.Host
+	}
+	return host + uri
+}

docker-compose.yaml

@@ -0,0 +1,48 @@
+version: '3.9'
+
+services:
+  wechat:
+    image: lxh01/wxhelper-docker:3.9.5.81
+    container_name: gw-wechat
+    restart: unless-stopped
+    environment:
+      - WINEDEBUG=fixme-all
+    volumes:
+      - ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
+    ports:
+      - "8080:8080"
+      - "19088:19088"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
+      interval: 60s
+      timeout: 10s
+      retries: 5
+
+
+  mysql:
+    image: mysql:8
+    container_name: gw-db
+    restart: unless-stopped
+    depends_on:
+      wechat:
+        condition: service_healthy
+    environment:
+      - MYSQL_ROOT_PASSWORD=wechat
+      - MYSQL_USER=wechat
+      - MYSQL_PASSWORD=wechat
+      - MYSQL_DATABASE=wechat
+    volumes:
+      - ./data/db:/var/lib/mysql
+
+
+  wxhelper:
+    image: gitee.ltd/lxh/go-wxhelper:latest
+    container_name: gw-service
+    restart: unless-stopped
+    depends_on:
+      - mysql
+    volumes:
+      # 配置文件请参阅项目根目录的config.yaml文件
+      - ./config/config.yaml:/app/config.yaml
+    ports:
+      - "19099:19099"

entity/friend.go

@@ -10,6 +10,8 @@ type Friend struct {
 	Pinyin         string `json:"pinyin"`                                                   // 昵称拼音大写首字母
 	PinyinAll      string `json:"pinyinAll"`                                                // 昵称全拼
 	Wxid           string `json:"wxid"`                                                     // 微信原始Id
+	EnableAi       bool   `json:"enableAI" gorm:"type:tinyint(1) default 0 not null"`       // 是否使用AI
+	EnableChatRank bool   `json:"enableChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否使用聊天排行
 }
 
 func (Friend) TableName() string {
@@ -24,8 +26,9 @@ type GroupUser struct {
 	HeadImage    string     `json:"headImage"`                                              // 头像
 	Nickname     string     `json:"nickname"`                                               // 昵称
 	Wxid         string     `json:"wxid"`                                                   // 微信Id
-	IsMember  bool      `json:"isMember" gorm:"type:tinyint(1)"` // 是否群成员
-	LeaveTime time.Time `json:"leaveTime"`                       // 离开时间
+	IsMember     bool       `json:"isMember" gorm:"type:tinyint(1) default 0 not null"`     // 是否群成员
+	LeaveTime    *time.Time `json:"leaveTime"`                                              // 离开时间
+	SkipChatRank bool       `json:"skipChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否跳过聊天排行
 }
 
 func (GroupUser) TableName() string {

go.mod

@@ -3,9 +3,11 @@ module go-wechat
 go 1.21
 
 require (
+	github.com/duke-git/lancet/v2 v2.2.7
 	github.com/fsnotify/fsnotify v1.6.0
 	github.com/go-co-op/gocron v1.34.1
 	github.com/go-resty/resty/v2 v2.8.0
+	github.com/sashabaranov/go-openai v1.17.5
 	github.com/spf13/viper v1.17.0
 	gorm.io/driver/mysql v1.5.1
 	gorm.io/gorm v1.25.4

go.sum

@@ -51,6 +51,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/duke-git/lancet/v2 v2.2.7 h1:u9zr6HR+MDUvZEtTlAFtSTIgZfEFsN7cKi27n5weZsw=
+github.com/duke-git/lancet/v2 v2.2.7/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
 github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
 github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
 github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=
@@ -175,6 +177,8 @@ github.com/sagikazarmark/locafero v0.3.0 h1:zT7VEGWC2DTflmccN/5T1etyKvxSxpHsjb9c
 github.com/sagikazarmark/locafero v0.3.0/go.mod h1:w+v7UsPNFwzF1cHuOajOOzoq4U7v/ig1mpRjqV+Bu1U=
 github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
 github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
+github.com/sashabaranov/go-openai v1.17.5 h1:ItBzlrrfTtkFWOFlgfOhk3y/xRBC4PJol4gdbiK7hgg=
+github.com/sashabaranov/go-openai v1.17.5/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
 github.com/spf13/afero v1.10.0 h1:EaGW2JJh15aKOejeuJ+wpFSHnbd7GE6Wvp3TsNhb6LY=

handler/at_message.go

@@ -0,0 +1,93 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"github.com/sashabaranov/go-openai"
+	"go-wechat/config"
+	"go-wechat/entity"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"regexp"
+	"strings"
+)
+
+// handleAtMessage
+// @description: 处理At机器人的消息
+// @param m
+func handleAtMessage(m entity.Message) {
+	if !config.Conf.Ai.Enable {
+		return
+	}
+
+	// 取出所有启用了AI的好友或群组
+	us, err := service.GetAllEnableAI()
+	if err != nil {
+		utils.SendMessage(m.FromUser, m.GroupUser, "#系统异常\n"+err.Error(), 0)
+		return
+	}
+	// 判断是否启用，如果没有启用，直接返回
+	var canUse bool
+	for _, u := range us {
+		if u.Wxid == m.FromUser {
+			canUse = true
+			break
+		}
+	}
+	if !canUse {
+		return
+	}
+
+	// 预处理一下发送的消息，用正则去掉@机器人的内容
+	re := regexp.MustCompile(`@([^ ]+)`)
+	matches := re.FindStringSubmatch(m.Content)
+	if len(matches) > 0 {
+		// 过滤掉第一个匹配到的
+		m.Content = strings.Replace(m.Content, matches[0], "", 1)
+	}
+
+	// 组装消息体
+	messages := make([]openai.ChatCompletionMessage, 0)
+	if config.Conf.Ai.Personality != "" {
+		// 填充人设
+		messages = append(messages, openai.ChatCompletionMessage{
+			Role:    openai.ChatMessageRoleSystem,
+			Content: config.Conf.Ai.Personality,
+		})
+	}
+	// 填充用户消息
+	messages = append(messages, openai.ChatCompletionMessage{
+		Role:    openai.ChatMessageRoleUser,
+		Content: m.Content,
+	})
+
+	// 配置模型
+	model := openai.GPT3Dot5Turbo0613
+	if config.Conf.Ai.Model != "" {
+		model = config.Conf.Ai.Model
+	}
+
+	// 默认使用AI回复
+	conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
+	if config.Conf.Ai.BaseUrl != "" {
+		conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
+	}
+	client := openai.NewClientWithConfig(conf)
+	resp, err := client.CreateChatCompletion(
+		context.Background(),
+		openai.ChatCompletionRequest{
+			Model:    model,
+			Messages: messages,
+		},
+	)
+
+	if err != nil {
+		log.Printf("OpenAI聊天发起失败: %v", err.Error())
+		utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
+		return
+	}
+
+	// 发送消息
+	utils.SendMessage(m.FromUser, m.GroupUser, "\n"+resp.Choices[0].Message.Content, 0)
+}

handler/parse.go

@@ -26,8 +26,14 @@ func Parse(remoteAddr net.Addr, msg []byte) {
 	groupUser := ""
 	msgStr := m.Content
 	if strings.Contains(m.FromUser, "@") {
-		// 系统消息不单独处理
-		if m.Type != types.MsgTypeRecalled && m.Type != types.MsgTypeSys {
+		switch m.Type {
+		case types.MsgTypeRecalled:
+			// 消息撤回
+		case types.MsgTypeSys:
+			// 系统消息
+			go handleSysMessage(m)
+		default:
+			// 默认消息处理
 			groupUser = strings.Split(m.Content, "\n")[0]
 			groupUser = strings.ReplaceAll(groupUser, ":", "")
 
@@ -50,5 +56,10 @@ func Parse(remoteAddr net.Addr, msg []byte) {
 	ent.DisplayFullContent = m.DisplayFullContent
 	ent.Raw = string(msg)
 
+	// 处理At机器人的消息
+	if strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你") {
+		go handleAtMessage(ent)
+	}
+
 	go service.SaveMessage(ent)
 }

handler/sys_message.go

@@ -0,0 +1,19 @@
+package handler
+
+import (
+	"go-wechat/model"
+	"go-wechat/utils"
+	"strings"
+)
+
+// handleSysMessage
+// @description: 系统消息处理
+// @param m
+func handleSysMessage(m model.Message) {
+	// 有人进群
+	if strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊") {
+		// 发一张图乐呵乐呵
+		// 自己欢迎自己图片地址  D:\Share\emoticon\welcome-yourself.gif
+		utils.SendImage(m.FromUser, "D:\\Share\\emoticon\\welcome-yourself.gif", 0)
+	}
+}

initialization/mysql.go

@@ -1,8 +1,10 @@
-package config
+package initialization
 
 import (
 	"github.com/fsnotify/fsnotify"
 	"github.com/spf13/viper"
+	"go-wechat/client"
+	"go-wechat/config"
 	"log"
 )
 
@@ -21,23 +23,29 @@ func InitConfig() {
 		log.Panicf("读取配置文件失败: %v", err)
 	}
 	// 绑定配置文件
-	if err := vp.Unmarshal(&Conf); err != nil {
+	if err := vp.Unmarshal(&config.Conf); err != nil {
 		log.Panicf("配置文件解析失败: %v", err)
 	}
-	log.Printf("配置文件解析完成: %+v", Conf)
+	log.Printf("配置文件解析完成: %+v", config.Conf)
+	if !config.Conf.Wechat.Check() {
+		log.Panicf("微信HOOK配置缺失")
+	}
 	// 初始化数据库连接
-	//db.Init()
+	client.InitMySQLClient()
 	//redis.Init()
 
 	// 下面的代码是配置变动之后自动刷新的
 	vp.WatchConfig()
 	vp.OnConfigChange(func(e fsnotify.Event) {
 		// 绑定配置文件
-		if err := vp.Unmarshal(&Conf); err != nil {
+		if err := vp.Unmarshal(&config.Conf); err != nil {
 			log.Printf("配置文件更新失败: %v", err)
 		} else {
+			if !config.Conf.Wechat.Check() {
+				log.Panicf("微信HOOK配置缺失")
+			}
 			// 初始化数据库连接
-			//db.Init()
+			client.InitMySQLClient()
 			//redis.Init()
 		}
 	})

main.go

@@ -2,18 +2,19 @@ package main
 
 import (
 	"bytes"
-	"go-wechat/client"
 	"go-wechat/config"
 	"go-wechat/handler"
+	"go-wechat/initialization"
 	"go-wechat/tasks"
+	"go-wechat/utils"
 	"io"
 	"log"
 	"net"
+	"time"
 )
 
 func init() {
-	config.InitConfig()
-	client.InitMySQLClient()
+	initialization.InitConfig()
 	tasks.InitTasks()
 }
 
@@ -37,6 +38,13 @@ func process(conn net.Conn) {
 }
 
 func main() {
+	// 如果启用了自动配置回调，就设置一下
+	if config.Conf.Wechat.AutoSetCallback {
+		utils.ClearCallback()
+		time.Sleep(500 * time.Millisecond) // 休眠五百毫秒再设置
+		utils.SetCallback(config.Conf.Wechat.Callback)
+	}
+
 	// 建立 tcp 服务
 	listen, err := net.Listen("tcp", "0.0.0.0:19099")
 	if err != nil {

readme.md

@@ -1,6 +1,101 @@
-
-
-### 手动打包镜像
+## 食用方式
+0. 新建一个文件夹
 ```shell
-docker build --push -t repo.lxh.io/lxh/go-wxhelper:latest .
+mkdir wechat-hook # 名字随便写
+cd wechat-hook
+```
+1. 创建配置文件`config.yaml`
+```shell
+mkdir config # 先创建一个文件夹保存配置文件，文件名不要变
+vim config.yaml # 编辑配置文件，内容如下，最新配置请参考项目里的config.yaml文件
+```
+```yaml
+# 微信HOOK配置
+wechat:
+  # 微信HOOK接口地址
+  host: wechat:19088
+  # 是否在启动的时候自动设置hook服务的回调
+  autoSetCallback: true
+  # 回调IP，如果是Docker运行，本参数必填，如果Docker修改了映射，格式为 ip:port，如果使用项目提供的docker-compsoe.yaml文件启动，可以不写
+  callback: 
+
+# 数据库
+mysql:
+  host: mysql
+  port: 3306
+  user: wechat
+  password: wechat
+  db: wechat
+
+task:
+  enable: false
+  syncFriends:
+      enable: true
+      cron: '0 * * * *'
+  waterGroup:
+      enable: true
+      cron:
+        yesterday: '30 9 * * *' # 每天9:30
+        week: '30 9 * * 1' # 每周一9:30
+        month: '30 9 1 * *' # 每月1号9:30
+```
+
+2. 创建`docker-compose.yaml`文件
+```yaml
+version: '3.9'
+
+services:
+  wechat:
+    image: lxh01/wxhelper-docker:3.9.5.81
+    container_name: gw-wechat
+    restart: unless-stopped
+    environment:
+      - WINEDEBUG=fixme-all
+    volumes:
+      - ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
+    ports:
+      - "8080:8080"
+      - "19088:19088"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
+      interval: 60s
+      timeout: 10s
+      retries: 5
+
+
+  mysql:
+    image: mysql:8
+    container_name: gw-db
+    restart: unless-stopped
+    depends_on:
+      wechat:
+        condition: service_healthy
+    environment:
+      - MYSQL_ROOT_PASSWORD=wechat
+      - MYSQL_USER=wechat
+      - MYSQL_PASSWORD=wechat
+      - MYSQL_DATABASE=wechat
+    volumes:
+      - ./data/db:/var/lib/mysql
+
+
+  wxhelper:
+    image: gitee.ltd/lxh/go-wxhelper:latest
+    container_name: gw-service
+    restart: unless-stopped
+    depends_on:
+      - mysql
+    volumes:
+      # 配置文件请参阅项目根目录的config.yaml文件
+      - ./config/config.yaml:/app/config.yaml
+    ports:
+      - "19099:19099"
+
+```
+
+3. 启动
+```shell
+# 以下命令选个能用的就行
+docker-compose up -d # 老版本
+docker compose up -d # 新版本
 ```

service/friend.go

@@ -0,0 +1,23 @@
+package service
+
+import (
+	"go-wechat/client"
+	"go-wechat/entity"
+)
+
+// GetAllEnableAI
+// @description: 取出所有启用了AI的好友或群组
+// @return []entity.Friend
+func GetAllEnableAI() (records []entity.Friend, err error) {
+	err = client.MySQL.Where("enable_ai = ?", 1).Find(&records).Error
+	return
+}
+
+// GetAllEnableChatRank
+// @description: 取出所有启用了聊天排行榜的群组
+// @return records
+// @return err
+func GetAllEnableChatRank() (records []entity.Friend, err error) {
+	err = client.MySQL.Where("enable_chat_rank = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
+	return
+}

tasks/friends/friends.go

@@ -1,9 +1,10 @@
-package tasks
+package friends
 
 import (
 	"encoding/json"
 	"github.com/go-resty/resty/v2"
 	"go-wechat/client"
+	"go-wechat/config"
 	"go-wechat/constant"
 	"go-wechat/entity"
 	"go-wechat/model"
@@ -19,15 +20,15 @@ import (
 // http客户端
 var hc = resty.New()
 
-// syncFriends
+// Sync
 // @description: 同步好友列表
-func syncFriends() {
+func Sync() {
 	var base model.Response[[]model.FriendItem]
 
 	resp, err := hc.R().
 		SetHeader("Content-Type", "application/json;chartset=utf-8").
 		SetResult(&base).
-		Post("http://10.0.0.73:19088/api/getContactList")
+		Post(config.Conf.Wechat.GetURL("/api/getContactList"))
 	if err != nil {
 		log.Printf("获取好友列表失败: %s", err.Error())
 		return
@@ -105,7 +106,7 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
 		SetHeader("Content-Type", "application/json;chartset=utf-8").
 		SetBody(string(pbs)).
 		SetResult(&baseResp).
-		Post("http://10.0.0.73:19088/api/getMemberFromChatRoom")
+		Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
 	if err != nil {
 		log.Printf("获取群成员信息失败: %s", err.Error())
 		return
@@ -189,7 +190,7 @@ func getContactProfile(wxid string) (ent model.ContactProfile, err error) {
 		SetHeader("Content-Type", "application/json;chartset=utf-8").
 		SetBody(string(pbs)).
 		SetResult(&baseResp).
-		Post("http://10.0.0.73:19088/api/getContactProfile")
+		Post(config.Conf.Wechat.GetURL("/api/getContactProfile"))
 	if err != nil {
 		log.Printf("获取成员详情失败: %s", err.Error())
 		return

tasks/tasks.go

@@ -3,6 +3,8 @@ package tasks
 import (
 	"github.com/go-co-op/gocron"
 	"go-wechat/config"
+	"go-wechat/tasks/friends"
+	"go-wechat/tasks/watergroup"
 	"log"
 	"time"
 )
@@ -20,14 +22,22 @@ func InitTasks() {
 
 	// 水群排行
 	if config.Conf.Task.WaterGroup.Enable {
-		log.Printf("水群排行任务已启用，执行表达式: %s", config.Conf.Task.WaterGroup.Cron)
-		_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron).Do(yesterday)
+		log.Printf("水群排行任务已启用，执行表达式: %+v", config.Conf.Task.WaterGroup.Cron)
+		if config.Conf.Task.WaterGroup.Cron.Yesterday != "" {
+			_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Yesterday).Do(watergroup.Yesterday)
+		}
+		if config.Conf.Task.WaterGroup.Cron.Week != "" {
+			_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Week).Do(watergroup.Week)
+		}
+		if config.Conf.Task.WaterGroup.Cron.Month != "" {
+			_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Month).Do(watergroup.Month)
+		}
 	}
 
 	// 更新好友列表
 	if config.Conf.Task.SyncFriends.Enable {
 		log.Printf("更新好友列表任务已启用，执行表达式: %s", config.Conf.Task.SyncFriends.Cron)
-		_, _ = s.Cron(config.Conf.Task.SyncFriends.Cron).Do(syncFriends)
+		_, _ = s.Cron(config.Conf.Task.SyncFriends.Cron).Do(friends.Sync)
 	}
 
 	// 开启定时任务

tasks/water_group.go

@@ -1,87 +0,0 @@
-package tasks
-
-import (
-	"fmt"
-	"go-wechat/client"
-	"go-wechat/config"
-	"go-wechat/entity"
-	"go-wechat/utils"
-	"log"
-	"strings"
-)
-
-// 水群排行榜
-
-// yesterday
-// @description: 昨日排行榜
-func yesterday() {
-	for _, id := range config.Conf.Task.WaterGroup.Groups {
-		dealYesterday(id)
-	}
-}
-
-// dealYesterday
-// @description: 处理请求
-// @param gid
-func dealYesterday(gid string) {
-	notifyMsgs := []string{"#昨日水群排行榜"}
-
-	// 获取昨日消息总数
-	var yesterdayMsgCount int64
-	err := client.MySQL.Model(&entity.Message{}).
-		Where("from_user = ?", gid).
-		Where("DATEDIFF(create_at,NOW()) = -1").
-		Count(&yesterdayMsgCount).Error
-	if err != nil {
-		log.Printf("获取昨日消息总数失败, 错误信息: %v", err)
-		return
-	}
-	log.Printf("昨日消息总数: %d", yesterdayMsgCount)
-	if yesterdayMsgCount == 0 {
-		return
-	}
-
-	notifyMsgs = append(notifyMsgs, " ")
-	notifyMsgs = append(notifyMsgs, fmt.Sprintf("昨日消息总数: %d", yesterdayMsgCount))
-
-	// 返回数据
-	type record struct {
-		GroupUser string
-		Nickname  string
-		Count     int64
-	}
-
-	var records []record
-	tx := client.MySQL.Table("t_message AS tm").
-		Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id").
-		Select("tm.group_user", "tgu.nickname", "count( 1 ) AS `count`").
-		Where("tm.from_user = ?", gid).
-		Where("tm.type < 10000").
-		Where("DATEDIFF(tm.create_at,NOW()) = -1").
-		Group("tm.group_user, tgu.nickname").Order("`count` DESC").
-		Limit(10)
-
-	// 黑名单
-	blacklist := config.Conf.Task.WaterGroup.Blacklist
-	// 如果有黑名单，过滤掉
-	if len(blacklist) > 0 {
-		tx.Where("tm.group_user NOT IN (?)", blacklist)
-	}
-
-	err = tx.Find(&records).Error
-
-	if err != nil {
-		log.Printf("获取昨日消息失败, 错误信息: %v", err)
-		return
-	}
-	notifyMsgs = append(notifyMsgs, " ")
-	for i, r := range records {
-		log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
-		notifyMsgs = append(notifyMsgs, fmt.Sprintf("#%d: %s -> %d条", i+1, r.Nickname, r.Count))
-	}
-
-	notifyMsgs = append(notifyMsgs, " \n请未上榜的群友多多反思。")
-
-	log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
-	go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
-}

tasks/watergroup/month.go

@@ -0,0 +1,85 @@
+package watergroup
+
+import (
+	"fmt"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"strings"
+	"time"
+)
+
+// Month
+// @description: 月排行榜
+func Month() {
+	groups, err := service.GetAllEnableChatRank()
+	if err != nil {
+		log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
+		return
+	}
+
+	for _, group := range groups {
+		// 消息统计
+		dealMonth(group.Wxid)
+		// 获取上个月月份
+		yd := time.Now().Local().AddDate(0, 0, -1).Format("200601")
+		// 发送词云
+		fileName := fmt.Sprintf("%s_%s.png", yd, group.Wxid)
+		utils.SendImage(group.Wxid, "D:\\Share\\wordcloud\\"+fileName, 0)
+	}
+}
+
+// dealMonth
+// @description: 处理请求
+// @param gid
+func dealMonth(gid string) {
+	monthStr := time.Now().Local().AddDate(0, 0, -1).Format("2006年01月")
+	notifyMsgs := []string{fmt.Sprintf("#%s水群排行榜", monthStr)}
+
+	// 获取上月消息总数
+	records, err := getRankData(gid, "month")
+	if err != nil {
+		log.Printf("获取上月消息排行失败, 错误信息: %v", err)
+		return
+	}
+	log.Printf("上月消息总数: %+v", records)
+	// 莫得消息，直接返回
+	if len(records) == 0 {
+		log.Printf("上月群[%s]无对话记录", gid)
+		return
+	}
+	// 计算消息总数
+	var msgCount int64
+	for _, v := range records {
+		msgCount += v.Count
+	}
+	// 组装消息总数推送信息
+	notifyMsgs = append(notifyMsgs, " ")
+	notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ %s本群 %d 位朋友共产生 %d 条发言", monthStr, len(records), msgCount))
+	notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
+
+	notifyMsgs = append(notifyMsgs, " ")
+	for i, r := range records {
+		// 只取前十条
+		if i >= 10 {
+			break
+		}
+
+		log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
+		badge := "🏆"
+		switch i {
+		case 0:
+			badge = "🥇"
+		case 1:
+			badge = "🥈"
+		case 2:
+			badge = "🥉"
+		}
+		notifyMsgs = append(notifyMsgs, fmt.Sprintf("%s %s -> %d条", badge, r.Nickname, r.Count))
+	}
+
+	notifyMsgs = append(notifyMsgs, fmt.Sprintf(" \n🎉感谢以上群友%s对群活跃做出的卓越贡献，也请未上榜的群友多多反思。", monthStr))
+
+	log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
+	go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
+}

tasks/watergroup/utils.go

@@ -0,0 +1,41 @@
+package watergroup
+
+import "go-wechat/client"
+
+// rankUser
+// @description: 排行榜用户
+type rankUser struct {
+	GroupUser string // 微信Id
+	Nickname  string // 昵称
+	Count     int64  // 消息数
+}
+
+// getRankData
+// @description: 获取消息排行榜
+// @param groupId string 群Id
+// @param d string 模式(yesterday | week | month)
+// @return rank
+// @return err
+func getRankData(groupId, date string) (rank []rankUser, err error) {
+	tx := client.MySQL.Table("t_message AS tm").
+		Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id AND tgu.skip_chat_rank = 0").
+		Select("tm.group_user", "tgu.nickname", "count( 1 ) AS `count`").
+		Where("tm.from_user = ?", groupId).
+		Where("tm.type < 10000").
+		Group("tm.group_user, tgu.nickname").
+		Order("`count` DESC")
+
+	// 根据参数获取不同日期的数据
+	switch date {
+	case "yesterday":
+		tx.Where("DATEDIFF(tm.create_at,NOW()) = -1")
+	case "week":
+		tx.Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1")
+	case "month":
+		tx.Where("PERIOD_DIFF(date_format(now(), '%Y%m'), date_format(create_at, '%Y%m')) = 1")
+	}
+
+	// 查询指定时间段全部数据
+	err = tx.Find(&rank).Error
+	return
+}

tasks/watergroup/week.go

@@ -0,0 +1,83 @@
+package watergroup
+
+import (
+	"fmt"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"strings"
+	"time"
+)
+
+// Week
+// @description: 周排行榜
+func Week() {
+	groups, err := service.GetAllEnableChatRank()
+	if err != nil {
+		log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
+		return
+	}
+
+	for _, group := range groups {
+		// 消息统计
+		dealWeek(group.Wxid)
+		// 获取上周周数
+		year, weekNo := time.Now().Local().AddDate(0, 0, -1).ISOWeek()
+		// 发送词云
+		fileName := fmt.Sprintf("%d%d_%s.png", year, weekNo, group.Wxid)
+		utils.SendImage(group.Wxid, "D:\\Share\\wordcloud\\"+fileName, 0)
+	}
+}
+
+// dealWeek
+// @description: 处理请求
+// @param gid
+func dealWeek(gid string) {
+	notifyMsgs := []string{"#上周水群排行榜"}
+
+	// 获取上周消息总数
+	records, err := getRankData(gid, "week")
+	if err != nil {
+		log.Printf("获取上周消息排行失败, 错误信息: %v", err)
+		return
+	}
+	log.Printf("上周消息总数: %+v", records)
+	// 莫得消息，直接返回
+	if len(records) == 0 {
+		log.Printf("上周群[%s]无对话记录", gid)
+		return
+	}
+	// 计算消息总数
+	var msgCount int64
+	for _, v := range records {
+		msgCount += v.Count
+	}
+	// 组装消息总数推送信息
+	notifyMsgs = append(notifyMsgs, " ")
+	notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ 上周本群 %d 位朋友共产生 %d 条发言", len(records), msgCount))
+	notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
+
+	notifyMsgs = append(notifyMsgs, " ")
+	for i, r := range records {
+		// 只取前十条
+		if i >= 10 {
+			break
+		}
+		log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
+		badge := "🏆"
+		switch i {
+		case 0:
+			badge = "🥇"
+		case 1:
+			badge = "🥈"
+		case 2:
+			badge = "🥉"
+		}
+		notifyMsgs = append(notifyMsgs, fmt.Sprintf("%s %s -> %d条", badge, r.Nickname, r.Count))
+	}
+
+	notifyMsgs = append(notifyMsgs, " \n🎉感谢以上群友上周对群活跃做出的卓越贡献，也请未上榜的群友多多反思。")
+
+	log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
+	go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
+}

tasks/watergroup/yesterday.go

@@ -0,0 +1,85 @@
+package watergroup
+
+import (
+	"fmt"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"strings"
+	"time"
+)
+
+// 水群排行榜
+
+// Yesterday
+// @description: 昨日排行榜
+func Yesterday() {
+	groups, err := service.GetAllEnableChatRank()
+	if err != nil {
+		log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
+		return
+	}
+
+	for _, group := range groups {
+		// 消息统计
+		dealYesterday(group.Wxid)
+		// 获取昨日日期
+		yd := time.Now().Local().AddDate(0, 0, -1).Format("20060102")
+		// 发送词云
+		fileName := fmt.Sprintf("%s_%s.png", yd, group.Wxid)
+		utils.SendImage(group.Wxid, "D:\\Share\\wordcloud\\"+fileName, 0)
+	}
+}
+
+// dealYesterday
+// @description: 处理请求
+// @param gid
+func dealYesterday(gid string) {
+	notifyMsgs := []string{"#昨日水群排行榜"}
+
+	// 获取昨日消息总数
+	records, err := getRankData(gid, "yesterday")
+	if err != nil {
+		log.Printf("获取昨日消息排行失败, 错误信息: %v", err)
+		return
+	}
+	log.Printf("昨日消息总数: %+v", records)
+	// 莫得消息，直接返回
+	if len(records) == 0 {
+		log.Printf("昨日群[%s]无对话记录", gid)
+		return
+	}
+	// 计算消息总数
+	var msgCount int64
+	for _, v := range records {
+		msgCount += v.Count
+	}
+	// 组装消息总数推送信息
+	notifyMsgs = append(notifyMsgs, " ")
+	notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ 昨日本群 %d 位朋友共产生 %d 条发言", len(records), msgCount))
+	notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
+
+	notifyMsgs = append(notifyMsgs, " ")
+	for i, r := range records {
+		// 只取前十条
+		if i >= 10 {
+			break
+		}
+		log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
+		badge := "🏆"
+		switch i {
+		case 0:
+			badge = "🥇"
+		case 1:
+			badge = "🥈"
+		case 2:
+			badge = "🥉"
+		}
+		notifyMsgs = append(notifyMsgs, fmt.Sprintf("%s %s -> %d条", badge, r.Nickname, r.Count))
+	}
+
+	notifyMsgs = append(notifyMsgs, " \n🎉感谢以上群友昨日对群活跃做出的卓越贡献，也请未上榜的群友多多反思。")
+
+	log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
+	go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
+}

utils/callback.go

@@ -0,0 +1,62 @@
+package utils
+
+import (
+	"encoding/json"
+	"github.com/duke-git/lancet/v2/netutil"
+	"github.com/go-resty/resty/v2"
+	"go-wechat/config"
+	"log"
+	"net"
+	"strconv"
+	"strings"
+)
+
+// ClearCallback
+// @description: 清理微信HOOK回调
+func ClearCallback() {
+	res := resty.New()
+	resp, err := res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		Post(config.Conf.Wechat.GetURL("/api/unhookSyncMsg"))
+	if err != nil {
+		log.Panicf("清理微信HOOK回调失败: %s", err.Error())
+	}
+	log.Printf("清理微信HOOK回调结果: %s", resp.String())
+}
+
+// SetCallback
+// @description: 设置微信HOOK回调
+// @param host
+func SetCallback(userHost string) {
+	// 获取本机IP地址
+	host := net.ParseIP(netutil.GetInternalIp()).String()
+	port := 19099
+	if userHost != "" {
+		uh := strings.Split(strings.TrimSpace(userHost), ":")
+		host = uh[0]
+		if len(uh) == 2 {
+			port, _ = strconv.Atoi(uh[1])
+		}
+	}
+
+	// 组装参数
+	param := map[string]any{
+		"port":       port, // socket端口
+		"ip":         host, // socketIP
+		"url":        "",   // http接口地址
+		"timeout":    3000, // 超时毫秒数
+		"enableHttp": 0,    // 是否使用http接口
+	}
+	pbs, _ := json.Marshal(param)
+	log.Printf("设置微信HOOK回调参数: %s", string(pbs))
+
+	res := resty.New()
+	resp, err := res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		Post(config.Conf.Wechat.GetURL("/api/hookSyncMsg"))
+	if err != nil {
+		log.Panicf("设置微信HOOK回调失败: %s", err.Error())
+	}
+	log.Printf("设置微信HOOK回调结果: %s", resp.String())
+}

utils/send.go

@@ -3,6 +3,7 @@ package utils
 import (
 	"encoding/json"
 	"github.com/go-resty/resty/v2"
+	"go-wechat/config"
 	"log"
 	"time"
 )
@@ -17,18 +18,30 @@ func SendMessage(toId, atId, msg string, retryCount int) {
 		log.Printf("重试五次失败，停止发送")
 		return
 	}
+
 	// 组装参数
 	param := map[string]any{
 		"wxid": toId, // 群或好友Id
 		"msg":  msg,  // 消息
 	}
+
+	// 接口地址
+	apiUrl := config.Conf.Wechat.GetURL("/api/sendTextMsg")
+	if atId != "" {
+		apiUrl = config.Conf.Wechat.GetURL("/api/sendAtText")
+		param = map[string]any{
+			"chatRoomId": toId,
+			"wxids":      atId,
+			"msg":        msg, // 消息
+		}
+	}
 	pbs, _ := json.Marshal(param)
 
 	res := resty.New()
 	resp, err := res.R().
 		SetHeader("Content-Type", "application/json;chartset=utf-8").
 		SetBody(string(pbs)).
-		Post("http://10.0.0.73:19088/api/sendTextMsg")
+		Post(apiUrl)
 	if err != nil {
 		log.Printf("发送文本消息失败: %s", err.Error())
 		// 休眠五秒后重新发送
@@ -37,3 +50,35 @@ func SendMessage(toId, atId, msg string, retryCount int) {
 	}
 	log.Printf("发送文本消息结果: %s", resp.String())
 }
+
+// SendImage
+// @description: 发送图片
+// @param toId string 群或者好友Id
+// @param imgPath string 图片路径
+// @param retryCount int 重试次数
+func SendImage(toId, imgPath string, retryCount int) {
+	if retryCount > 5 {
+		log.Printf("重试五次失败，停止发送")
+		return
+	}
+
+	// 组装参数
+	param := map[string]any{
+		"wxid":      toId,    // 群或好友Id
+		"imagePath": imgPath, // 图片地址
+	}
+	pbs, _ := json.Marshal(param)
+
+	res := resty.New()
+	resp, err := res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		Post(config.Conf.Wechat.GetURL("/api/sendImagesMsg"))
+	if err != nil {
+		log.Printf("发送图片消息失败: %s", err.Error())
+		// 休眠五秒后重新发送
+		time.Sleep(5 * time.Second)
+		SendImage(toId, imgPath, retryCount+1)
+	}
+	log.Printf("发送图片消息结果: %s", resp.String())
+}