🎨 优化迎新逻辑，引入资源概念

common/current/robot.go

@@ -0,0 +1,19 @@
+package current
+
+import "go-wechat/model"
+
+var robotInfo model.RobotUserInfo
+
+// SetRobotInfo
+// @description: 设置机器人信息
+// @param info
+func SetRobotInfo(info model.RobotUserInfo) {
+	robotInfo = info
+}
+
+// GetRobotInfo
+// @description: 获取机器人信息
+// @return model.RobotUserInfo
+func GetRobotInfo() model.RobotUserInfo {
+	return robotInfo
+}

config.yaml

@@ -43,8 +43,9 @@ ai:
   # 人设
   personality: 你的名字叫张三，你是一个百科机器人，你的爱好是看电影，你的性格是开朗的，你的专长是讲故事，你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣，也不会讨论任何与政治相关的话题，你甚至可以拒绝回答这一类话题。
 
-# 资源配置（这玩意儿是机器人那个机器上的smb文件夹）
+# 资源配置
-# 暂时没用 - 2023-12-6
 resource:
-  localPath: /files
+  # 欢迎新成员表情包
-  remotePath: D:\Share\
+  welcomeNew:
+    type: emotion
+    path: 58e4150be2bba8f7b71974b10391f9e9

config/config.go

@@ -5,8 +5,9 @@ var Conf Config
 // Config
 // @description: 配置
 type Config struct {
-	Task   task   `json:"task" yaml:"task"`     // 定时任务配置
+	Task     task                    `json:"task" yaml:"task"`         // 定时任务配置
-	MySQL  mysql  `json:"mysql" yaml:"mysql"`   // MySQL 配置
+	MySQL    mysql                   `json:"mysql" yaml:"mysql"`       // MySQL 配置
-	Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
+	Wechat   wechat                  `json:"wechat" yaml:"wechat"`     // 微信助手
-	Ai     ai     `json:"ai" yaml:"ai"`         // AI配置
+	Ai       ai                      `json:"ai" yaml:"ai"`             // AI配置
+	Resource map[string]resourceItem `json:"resource" yaml:"resource"` // 资源配置
 }

config/resource.go

@@ -0,0 +1,8 @@
+package config
+
+// resourceItem
+// @description: 资源项
+type resourceItem struct {
+	Type string `json:"type" yaml:"type"` // 类型
+	Path string `json:"path" yaml:"path"` // 路径
+}

handler/parse.go

@@ -42,6 +42,7 @@ func Parse(remoteAddr net.Addr, msg []byte) {
 	// 异步处理消息
 	go func() {
 		if m.IsNewUserJoin() {
+			log.Printf("%s -> 开始迎新 -> %s", m.FromUser, m.Content)
 			// 欢迎新成员
 			go handleNewUserJoin(m)
 		} else if m.IsAt() {

handler/sys_message.go

@@ -2,9 +2,11 @@ package handler
 
 import (
 	"go-wechat/client"
+	"go-wechat/config"
 	"go-wechat/entity"
 	"go-wechat/model"
 	"go-wechat/utils"
+	"log"
 )
 
 // handleNewUserJoin
@@ -13,13 +15,33 @@ import (
 func handleNewUserJoin(m model.Message) {
 	// 判断是否开启迎新
 	var count int64
-	_ = client.MySQL.Model(&entity.Friend{}).Where("enable_welcome IS TRUE").Where("wxid = ?", m.FromUser).Count(&count).Error
+	err := client.MySQL.Model(&entity.Friend{}).
+		Where("enable_welcome IS TRUE").
+		Where("wxid = ?", m.FromUser).
+		Count(&count).Error
+	if err != nil {
+		log.Printf("查询是否开启迎新失败: %s", err.Error())
+		return
+	}
 	if count < 1 {
 		return
 	}
 
-	// 发一张图乐呵乐呵
+	// 读取欢迎新成员配置
-
+	conf, ok := config.Conf.Resource["welcomeNew"]
-	// 自己欢迎自己图片地址  D:\Share\emoticon\welcome-yourself.gif
+	if !ok {
-	utils.SendImage(m.FromUser, "D:\\Share\\emoticon\\welcome-yourself.gif", 0)
+		// 未配置，跳过
+		return
+	}
+	switch conf.Type {
+	case "text":
+		// 文字类型
+		utils.SendMessage(m.FromUser, "", conf.Path, 0)
+	case "image":
+		// 图片类型
+		utils.SendImage(m.FromUser, conf.Path, 0)
+	case "emotion":
+		// 表情类型
+		utils.SendEmotion(m.FromUser, conf.Path, 0)
+	}
 }

initialization/wechat.go

@@ -0,0 +1,31 @@
+package initialization
+
+import (
+	"github.com/go-resty/resty/v2"
+	"go-wechat/common/current"
+	"go-wechat/config"
+	"go-wechat/model"
+	"log"
+)
+
+// InitWechatRobotInfo
+// @description: 初始化微信机器人信息
+func InitWechatRobotInfo() {
+	// 获取数据
+	var base model.Response[model.RobotUserInfo]
+	_, err := resty.New().R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetResult(&base).
+		Post(config.Conf.Wechat.GetURL("/api/userInfo"))
+	if err != nil {
+		log.Printf("获取机器人信息失败: %s", err.Error())
+		return
+	}
+
+	log.Printf("机器人Id: %s", base.Data.WxId)
+	log.Printf("机器人微信号: %s", base.Data.Account)
+	log.Printf("机器人名称: %s", base.Data.Name)
+
+	// 设置为单例
+	current.SetRobotInfo(base.Data)
+}

main.go

@@ -16,8 +16,9 @@ import (
 )
 
 func init() {
-	initialization.InitConfig()
+	initialization.InitConfig()          // 初始化配置
-	tasks.InitTasks()
+	initialization.InitWechatRobotInfo() // 初始化机器人信息
+	tasks.InitTasks()                    // 初始化定时任务
 }
 
 func main() {

model/userinfo.go

@@ -0,0 +1,18 @@
+package model
+
+// RobotUserInfo
+// @description: 机器人用户信息
+type RobotUserInfo struct {
+	WxId            string `json:"wxid"`            // 微信Id
+	Account         string `json:"account"`         // 微信号
+	Name            string `json:"name"`            // 昵称
+	HeadImage       string `json:"headImage"`       // 头像
+	Mobile          string `json:"mobile"`          // 手机
+	Signature       string `json:"signature"`       // 个人签名
+	Country         string `json:"country"`         // 国家
+	Province        string `json:"province"`        // 省
+	City            string `json:"city"`            // 城市
+	CurrentDataPath string `json:"currentDataPath"` // 当前数据目录,登录的账号目录
+	DataSavePath    string `json:"dataSavePath"`    // 微信保存目录
+	DbKey           string `json:"dbKey"`           // 数据库的SQLCipher的加密key，可以使用该key配合decrypt.py解密数据库
+}

utils/send.go

@@ -2,7 +2,9 @@ package utils
 
 import (
 	"encoding/json"
+	"fmt"
 	"github.com/go-resty/resty/v2"
+	"go-wechat/common/current"
 	"go-wechat/config"
 	"log"
 	"time"
@@ -82,3 +84,40 @@ func SendImage(toId, imgPath string, retryCount int) {
 	}
 	log.Printf("发送图片消息结果: %s", resp.String())
 }
+
+// SendEmotion
+// @description: 发送自定义表情包
+// @param toId string 群或者好友Id
+// @param emotionHash string 表情包hash(md5值)
+// @param retryCount int 重试次数
+func SendEmotion(toId, emotionHash string, retryCount int) {
+	if retryCount > 5 {
+		log.Printf("重试五次失败，停止发送")
+		return
+	}
+
+	// 组装表情包本地地址
+	// 规则：机器人数据目录\FileStorage\CustomEmotion\表情包hash前两位\表情包hash
+	emotionPath := fmt.Sprintf("%sFileStorage\\CustomEmotion\\%s\\%s",
+		current.GetRobotInfo().CurrentDataPath, emotionHash[:2], emotionHash)
+
+	// 组装参数
+	param := map[string]any{
+		"wxid":     toId,        // 群或好友Id
+		"filePath": emotionPath, // 图片地址
+	}
+	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, emotionHash, retryCount+1)
+	}
+	log.Printf("发送表情包消息结果: %s", resp.String())
+}