🎨 优化传递给AI的消息，去掉艾特机器人那一段字符串

config.yaml

@@ -3,7 +3,7 @@ wechat:
   # 微信HOOK接口地址
   host: 10.0.0.73:19088
   # 是否在启动的时候自动设置hook服务的回调
-  autoSetCallback: true
+  autoSetCallback: false
   # 回调IP，如果是Docker运行，本参数必填，如果Docker修改了映射，格式为 ip:port
   callback: 10.0.0.51
 
@@ -16,12 +16,12 @@ mysql:
   db: wechat
 
 task:
-  enable: false
+  enable: true
   syncFriends:
       enable: true
-      cron: '0 * * * *'
+      cron: '*/5 * * * *' # 五分钟一次
   waterGroup:
-      enable: true
+      enable: false
       cron: '30 9 * * *'
       # 需要发送水群排行榜的群Id
       groups:
@@ -29,4 +29,13 @@ task:
         - '49448748645@chatroom'
       # 不计入统计范围的用户Id
       blacklist:
-        - 'wxid_7788687886912'
+        - 'wxid_7788687886912'
+
+# AI回复
+ai:
+  # 是否启用
+  enable: false
+  # OpenAI Api key
+  apiKey: sk-xxxx
+  # 接口代理域名，不填默认ChatGPT官方地址
+  baseUrl: https://sxxx

config/ai.go

@@ -0,0 +1,9 @@
+package config
+
+// ai
+// @description: AI配置
+type ai struct {
+	Enable  bool   `json:"enable" yaml:"enable"`   // 是否启用AI
+	ApiKey  string `json:"apiKey" yaml:"apiKey"`   // API Key
+	BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
+}

config/config.go

@@ -8,6 +8,7 @@ type Config struct {
 	Task   task   `json:"task" yaml:"task"`     // 定时任务配置
 	MySQL  mysql  `json:"mysql" yaml:"mysql"`   // MySQL 配置
 	Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
+	Ai     ai     `json:"ai" yaml:"ai"`         // AI配置
 }
 
 // task

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

@@ -19,13 +19,13 @@ func (Friend) TableName() string {
 // GroupUser
 // @description: 群成员
 type GroupUser struct {
-	GroupId   string    `json:"groupId"`                         // 群Id
+	GroupId   string     `json:"groupId"`                         // 群Id
-	Account   string    `json:"account"`                         // 账号
+	Account   string     `json:"account"`                         // 账号
-	HeadImage string    `json:"headImage"`                       // 头像
+	HeadImage string     `json:"headImage"`                       // 头像
-	Nickname  string    `json:"nickname"`                        // 昵称
+	Nickname  string     `json:"nickname"`                        // 昵称
-	Wxid      string    `json:"wxid"`                            // 微信Id
+	Wxid      string     `json:"wxid"`                            // 微信Id
-	IsMember  bool      `json:"isMember" gorm:"type:tinyint(1)"` // 是否群成员
+	IsMember  bool       `json:"isMember" gorm:"type:tinyint(1)"` // 是否群成员
-	LeaveTime time.Time `json:"leaveTime"`                       // 离开时间
+	LeaveTime *time.Time `json:"leaveTime"`                       // 离开时间
 }
 
 func (GroupUser) TableName() string {

go.mod

@@ -7,6 +7,7 @@ require (
 	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

@@ -177,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,58 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"github.com/sashabaranov/go-openai"
+	"go-wechat/config"
+	"go-wechat/entity"
+	"go-wechat/utils"
+	"log"
+	"regexp"
+	"strings"
+)
+
+// handleAtMessage
+// @description: 处理At机器人的消息
+// @param m
+func handleAtMessage(m entity.Message) {
+	if !config.Conf.Ai.Enable {
+		return
+	}
+
+	// 预处理一下发送的消息，用正则去掉@机器人的内容
+	re := regexp.MustCompile(`@([^ ]+)`)
+	matches := re.FindStringSubmatch(m.Content)
+	if len(matches) > 0 {
+		// 过滤掉第一个匹配到的
+		m.Content = strings.Replace(m.Content, matches[0], "", 1)
+	}
+
+	// 默认使用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: openai.GPT3Dot5Turbo0613,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Role:    openai.ChatMessageRoleUser,
+					Content: m.Content,
+				},
+			},
+		},
+	)
+
+	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, "@") {
-		// 系统消息不单独处理
+		switch m.Type {
-		if m.Type != types.MsgTypeRecalled && m.Type != types.MsgTypeSys {
+		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)
+	}
+}

readme.md

@@ -1,6 +1,104 @@
-
+## 食用方式
-
+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 # 编辑配置文件，内容如下
+```
+```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: '30 9 * * *'
+      # 需要发送水群排行榜的群Id
+      groups:
+        - '11111@chatroom' # 自行配置
+      # 不计入统计范围的用户Id
+      blacklist:
+        - 'wxid_xxxx' # 自行配置
+```
+
+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 # 新版本
 ```

utils/send.go

@@ -18,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(config.Conf.Wechat.GetURL("/api/sendTextMsg"))
+		Post(apiUrl)
 	if err != nil {
 		log.Printf("发送文本消息失败: %s", err.Error())
 		// 休眠五秒后重新发送
@@ -38,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())
+}