🆕 页面新增迎新开关实现

app/friend.go

@@ -62,6 +62,29 @@ func ChangeEnableGroupRankStatus(ctx *gin.Context) {
 	ctx.String(http.StatusOK, "操作成功")
 }
 
+// ChangeEnableWelcomeStatus
+// @description: 修改是否开启迎新
+// @param ctx
+func ChangeEnableWelcomeStatus(ctx *gin.Context) {
+	var p changeStatusParam
+	if err := ctx.ShouldBindJSON(&p); err != nil {
+		ctx.String(http.StatusBadRequest, "参数错误")
+		return
+	}
+	log.Printf("待修改的群Id：%s", p.WxId)
+
+	err := client.MySQL.Model(&entity.Friend{}).
+		Where("wxid = ?", p.WxId).
+		Update("`enable_welcome`", gorm.Expr(" !`enable_welcome`")).Error
+	if err != nil {
+		log.Printf("修改开启迎新失败：%s", err)
+		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
+		return
+	}
+
+	ctx.String(http.StatusOK, "操作成功")
+}
+
 // ChangeSkipGroupRankStatus
 // @description: 修改是否跳过水群排行榜
 // @param ctx

entity/friend.go

@@ -14,6 +14,7 @@ type Friend struct {
 	PinyinAll      string `json:"pinyinAll"`                                                // 昵称全拼
 	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"` // 是否使用聊天排行
+	EnableWelcome  bool   `json:"enableWelcome" gorm:"type:tinyint(1) default 0 not null"`  // 是否启用迎新
 	IsOk           bool   `json:"isOk" gorm:"type:tinyint(1) default 0 not null"`           // 是否正常
 }
 

handler/at_message.go

@@ -4,9 +4,10 @@ import (
 	"context"
 	"fmt"
 	"github.com/sashabaranov/go-openai"
+	"go-wechat/client"
 	"go-wechat/config"
 	"go-wechat/entity"
-	"go-wechat/service"
+	"go-wechat/model"
 	"go-wechat/utils"
 	"log"
 	"regexp"
@@ -16,26 +17,15 @@ import (
 // handleAtMessage
 // @description: 处理At机器人的消息
 // @param m
-func handleAtMessage(m entity.Message) {
+func handleAtMessage(m model.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 {
+	var count int64
+	client.MySQL.Model(&entity.Friend{}).Where("enable_ai IS TRUE").Where("wxid = ?", m.FromUser).Count(&count)
+	if count < 1 {
 		return
 	}
 
@@ -63,9 +53,9 @@ func handleAtMessage(m entity.Message) {
 	})
 
 	// 配置模型
-	model := openai.GPT3Dot5Turbo0613
+	chatModel := openai.GPT3Dot5Turbo0613
 	if config.Conf.Ai.Model != "" {
-		model = config.Conf.Ai.Model
+		chatModel = config.Conf.Ai.Model
 	}
 
 	// 默认使用AI回复
@@ -77,7 +67,7 @@ func handleAtMessage(m entity.Message) {
 	resp, err := client.CreateChatCompletion(
 		context.Background(),
 		openai.ChatCompletionRequest{
-			Model:    model,
+			Model:    chatModel,
 			Messages: messages,
 		},
 	)

handler/parse.go

@@ -6,6 +6,7 @@ import (
 	"go-wechat/model"
 	"go-wechat/service"
 	"go-wechat/types"
+	"go-wechat/utils"
 	"log"
 	"net"
 	"strings"
@@ -23,43 +24,47 @@ func Parse(remoteAddr net.Addr, msg []byte) {
 		return
 	}
 	// 提取出群成员信息
-	groupUser := ""
-	msgStr := m.Content
+	//groupUser := ""
+	//msgStr := m.Content
 	if strings.Contains(m.FromUser, "@") {
-		switch m.Type {
-		case types.MsgTypeRecalled:
-			// 消息撤回
-		case types.MsgTypeSys:
-			// 系统消息
-			go handleSysMessage(m)
-		default:
-			// 默认消息处理
-			groupUser = strings.Split(m.Content, "\n")[0]
-			groupUser = strings.ReplaceAll(groupUser, ":", "")
-
-			// 文字消息单独提出来处理一下
-			msgStr = strings.Join(strings.Split(m.Content, "\n")[1:], "\n")
+		// 群消息，处理一下消息和发信人
+		groupUser := strings.Split(m.Content, "\n")[0]
+		groupUser = strings.ReplaceAll(groupUser, ":", "")
+		// 如果两个id一致，说明是系统发的
+		if m.FromUser != groupUser {
+			m.GroupUser = groupUser
 		}
+		// 用户的操作单独提出来处理一下
+		m.Content = 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)
+	log.Printf("%s\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", remoteAddr, m.FromUser, m.GroupUser, m.Type, m.Content)
+
+	// 异步处理消息
+	go func() {
+		if m.IsNewUserJoin() {
+			// 欢迎新成员
+			go handleNewUserJoin(m)
+		} else if m.IsAt() {
+			// @机器人的消息
+			go handleAtMessage(m)
+		} else if !strings.Contains(m.FromUser, "@") && m.Type == types.MsgTypeText {
+			// 私聊消息处理
+			utils.SendMessage(m.FromUser, "", "暂未开启私聊AI", 0)
+		}
+	}()
 
 	// 转换为结构体之后入库
 	var ent entity.Message
 	ent.MsgId = m.MsgId
 	ent.CreateTime = m.CreateTime
 	ent.CreateAt = time.Unix(int64(m.CreateTime), 0)
-	ent.Content = msgStr
+	ent.Content = m.Content
 	ent.FromUser = m.FromUser
-	ent.GroupUser = groupUser
+	ent.GroupUser = m.GroupUser
 	ent.ToUser = m.ToUser
 	ent.Type = m.Type
 	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

@@ -1,19 +1,25 @@
 package handler
 
 import (
+	"go-wechat/client"
+	"go-wechat/entity"
 	"go-wechat/model"
 	"go-wechat/utils"
-	"strings"
 )
 
-// handleSysMessage
-// @description: 系统消息处理
+// handleNewUserJoin
+// @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)
+func handleNewUserJoin(m model.Message) {
+	// 判断是否开启迎新
+	var count int64
+	client.MySQL.Model(&entity.Friend{}).Where("enable_welcome IS TRUE").Where("wxid = ?", m.FromUser).Count(&count)
+	if count < 1 {
+		return
 	}
+
+	// 发一张图乐呵乐呵
+
+	// 自己欢迎自己图片地址  D:\Share\emoticon\welcome-yourself.gif
+	utils.SendImage(m.FromUser, "D:\\Share\\emoticon\\welcome-yourself.gif", 0)
 }

model/message.go

@@ -1,18 +1,87 @@
 package model
 
-import "go-wechat/types"
+import (
+	"encoding/xml"
+	"go-wechat/types"
+	"strings"
+)
 
 // Message
 // @description: 消息
 type Message struct {
-	MsgId              int64             `json:"msgId" gorm:"primarykey"`
+	MsgId              int64             `json:"msgId"`
 	CreateTime         int               `json:"createTime"`
 	Content            string            `json:"content"`
-	DisplayFullContent string            `json:"displayFullContent" gorm:"-"`
+	DisplayFullContent string            `json:"displayFullContent"`
 	FromUser           string            `json:"fromUser"`
+	GroupUser          string            `json:"-"`
 	MsgSequence        int               `json:"msgSequence"`
 	Pid                int               `json:"pid"`
 	Signature          string            `json:"signature"`
 	ToUser             string            `json:"toUser"`
 	Type               types.MessageType `json:"type"`
 }
+
+// systemMsgDataXml
+// @description: 微信系统消息的xml结构
+type systemMsgDataXml struct {
+	SysMsg sysMsg `xml:"sysmsg"`
+	Type   string `xml:"type,attr"`
+}
+
+// sysMsg
+// @description: 消息主体
+type sysMsg struct{}
+
+// IsPat
+// @description: 是否是拍一拍消息
+// @receiver m
+// @return bool
+func (m Message) IsPat() bool {
+	// 解析xml
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+
+	return m.Type == types.MsgTypeRecalled && d.Type == "pat"
+}
+
+// IsRevokeMsg
+// @description: 是否是撤回消息
+// @receiver m
+// @return bool
+func (m Message) IsRevokeMsg() bool {
+	// 解析xml
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+
+	return m.Type == types.MsgTypeRecalled && d.Type == "revokemsg"
+}
+
+// IsNewUserJoin
+// @description: 是否是新人入群
+// @receiver m
+// @return bool
+func (m Message) IsNewUserJoin() bool {
+	sysFlag := m.Type == types.MsgTypeSys && strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
+	if sysFlag {
+		return true
+	}
+	// 解析另一种情况
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+	return m.Type == types.MsgTypeSys && d.Type == "delchatroommember"
+}
+
+// IsAt
+// @description: 是否是At机器人的消息
+// @receiver m
+// @return bool
+func (m Message) IsAt() bool {
+	return strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你")
+}

router/router.go

@@ -22,6 +22,7 @@ func Init(g *gin.Engine) {
 	// 接口
 	api := g.Group("/api")
 	api.PUT("/ai/status", app.ChangeEnableAiStatus)               // 修改是否开启AI状态
+	api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus)     // 修改是否开启迎新状态
 	api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态
 	api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus)     // 修改是否跳过水群排行榜状态
 	api.GET("/group/users", app.GetGroupUsers)                    // 获取群成员列表

tasks/friends/friends.go

@@ -65,6 +65,7 @@ func Sync() {
 				Pinyin:        friend.Pinyin,
 				PinyinAll:     friend.PinyinAll,
 				Wxid:          friend.Wxid,
+				IsOk:          true,
 			}).Error
 			if err != nil {
 				log.Printf("新增好友失败: %s", err.Error())

types/message.go

@@ -4,6 +4,7 @@ import "fmt"
 
 type MessageType int
 
+// 微信定义的消息类型
 const (
 	MsgTypeText           MessageType = 1     // 文本消息
 	MsgTypeImage          MessageType = 3     // 图片消息

views/index.html

@@ -90,6 +90,7 @@
                         <th>是否在通讯录</th>
                         <th>是否启用AI</th>
                         <th>是否启用水群排行榜</th>
+                        <th>是否启用迎新</th>
                         <th>操作</th>
                     </tr>
                     </thead>
@@ -134,6 +135,14 @@
                                 <div class="swap-off">❌已禁用</div>
                             </label>
                         </td>
+                        <td>
+                            <label class="swap swap-flip {{ checkSwap .EnableWelcome }}">
+                                <input type="checkbox" onclick="changeWelcomeEnableStatus({{.Wxid}})"/>
+
+                                <div class="swap-on">✔️已启用</div>
+                                <div class="swap-off">❌已禁用</div>
+                            </label>
+                        </td>
                         <td>
                             <button class="btn btn-link" onclick="getGroupUsers({{.Wxid}}, {{.Nickname}})">查看群成员</button>
                         </td>

views/static/js/index.js

@@ -35,6 +35,22 @@ function changeGroupRankEnableStatus(wxId) {
     })
 }
 
+// 修改欢迎语开启状态
+function changeWelcomeEnableStatus(wxId) {
+    axios({
+        method: 'put',
+        url: '/api/welcome/status',
+        data: {
+            wxId: wxId
+        }
+    }).then(function (response) {
+        console.log(`返回结果: ${JSON.stringify(response)}`);
+    }).catch(function (error) {
+        console.log(`错误信息: ${error}`);
+        alert("修改失败")
+    })
+}
+
 // 修改群成员是否参与排行榜状态
 function changeUserGroupRankSkipStatus(groupId, userId) {
     console.log("修改水群排行榜开启状态: ", groupId, userId)

vo/friend.go

@@ -14,6 +14,7 @@ type FriendItem struct {
 	Wxid           string         // 微信原始Id
 	EnableAi       bool           // 是否使用AI
 	EnableChatRank bool           // 是否使用聊天排行
+	EnableWelcome  bool           // 是否使用迎新
 	IsOk           bool           // 是否还在通讯库(群聊是要还在群里也算)
 	LastActiveTime types.DateTime // 最后活跃时间
 }