From 14d407eff1759697e69da39fff156613aab2358f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= <lxh@cxh.cn>
Date: Mon, 4 Dec 2023 14:17:52 +0800
Subject: [PATCH] =?UTF-8?q?:new:=20=E6=96=B0=E5=A2=9E=E8=BF=8E=E6=96=B0?=
 =?UTF-8?q?=E5=BC=80=E5=85=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 entity/friend.go       |  1 +
 handler/sys_message.go | 24 +++++++++++++++---------
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/entity/friend.go b/entity/friend.go
index 4a5cbda..8f8ebd9 100644
--- a/entity/friend.go
+++ b/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"`           // 是否正常
 }
 
diff --git a/handler/sys_message.go b/handler/sys_message.go
index 88e9b42..212e9c8 100644
--- a/handler/sys_message.go
+++ b/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)
 }