Merge pull request '🐛 Fix a bug.' (#105) from hotfix into main

Reviewed-on: #105

app/other.go

@@ -0,0 +1,56 @@
+package app
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"go-wechat/model/vo"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"net/http"
+)
+
+// SendAiSummary
+// @description: 发送AI摘要
+// @param ctx
+func SendAiSummary(ctx *gin.Context) {
+	// 获取群Id
+	groupId := ctx.Query("id")
+	if groupId == "" {
+		ctx.String(http.StatusForbidden, "群Id不能为空")
+		return
+	}
+
+	// 取出群名称
+	groupInfo, err := service.GetFriendInfoById(groupId)
+	if err != nil {
+		ctx.String(http.StatusInternalServerError, "获取群信息失败")
+		return
+	}
+
+	// 获取对话记录
+	var records []vo.TextMessageItem
+	if records, err = service.GetTextMessagesById(groupId); err != nil {
+		log.Printf("获取群[%s]对话记录失败, 错误信息: %v", groupId, err)
+		ctx.String(http.StatusInternalServerError, "获取群对话记录失败")
+		return
+	}
+	if len(records) < 10 {
+		ctx.String(http.StatusForbidden, "群对话记录不足10条，建议自己看")
+		return
+	}
+
+	// 组装对话记录为字符串
+	var replyMsg string
+	replyMsg, err = utils.GetAiSummary(groupInfo.Nickname, records)
+	if err != nil {
+		log.Printf("群聊记录总结失败: %v", err.Error())
+		ctx.String(http.StatusInternalServerError, "群聊消息总结失败，错误信息: "+err.Error())
+		return
+	}
+
+	replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了，让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg)
+	log.Printf("群[%s]对话记录总结成功，总结内容: %s", groupInfo.Nickname, replyMsg)
+	_ = utils.SendMessage(groupId, "", replyMsg, 0)
+	ctx.String(http.StatusOK, "操作完成")
+}

config.yaml

@@ -31,7 +31,7 @@ system:
 # 微信HOOK配置
 wechat:
   # 微信HOOK接口地址
-  host: 10.0.0.79:19088
+  host: 10.0.0.79:19098
   # 微信容器映射出来的vnc页面地址，没有就不填
   #  vncUrl: http://192.168.1.175:19087/vnc_lite.html
   # 是否在启动的时候自动设置hook服务的回调
@@ -61,9 +61,9 @@ task:
   hotTop:
     enable: true
     cron: '0 */1 * * *' # 每小时一次
-  syncFriends:
+  syncFriends: # 逻辑已修改，每天大半夜同步一次即可(主要是用以修正数据)，平时会根据收到的消息针对性同步
     enable: false
-    cron: '*/5 * * * *' # 五分钟一次
+    cron: '0 4 * * *' # 每天4:00
   groupSummary:
     enable: false
     cron: '30 0 * * *' # 每天0:30
@@ -81,7 +81,7 @@ mq:
   enable: false
   # RabbitMQ配置
   rabbitmq:
-    host: 10.0.0.247
+    host: 10.0.0.31
     port: 5672
     user: wechat
     password: wechat123
@@ -102,6 +102,7 @@ ai:
   # 人设
   personality: 你的名字叫张三，你是一个百科机器人，你的爱好是看电影，你的性格是开朗的，你的专长是讲故事，你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣，也不会讨论任何与政治相关的话题，你甚至可以拒绝回答这一类话题。
   models:
+    - name: 默认
     - name: ChatGPT-4
       model: gpt-4
       canManager: false
@@ -139,3 +140,7 @@ resource:
   wordcloud:
     type: image
     path: D:\Share\wordcloud\%s
+  # 邀请入群消息回复
+  invitation-join-group:
+    type: text
+    path: "您的邀请消息已收到啦，正在通知我的主人来同意请求。在我加群之后将会进行初始化操作，直到收到我主动发送的消息就是初始化完成咯，在那之前请耐心等待喔~"

initialization/plugin.go

@@ -6,7 +6,6 @@ import (
 	plugin "go-wechat/plugin"
 	"go-wechat/plugin/plugins"
 	"go-wechat/service"
-	"go-wechat/types"
 )
 
 // Plugin
@@ -31,7 +30,7 @@ func Plugin() {
 	}, plugins.NotifyInvitationJoinGroup)
 	// 被移除群聊通知到配置用户
 	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		return m.Type == types.MsgTypeSys
+		return m.IsRemoveFromChatroom()
 	}, plugins.NotifyRemoveFromChatroom)
 	// 响应好友添加成功消息
 	dispatcher.RegisterHandler(func(m *dto.Message) bool {

main.go

@@ -12,6 +12,8 @@ import (
 	"html/template"
 	"log"
 	"net/http"
+	"os"
+	"strconv"
 	"strings"
 	"time"
 )
@@ -22,6 +24,13 @@ func init() {
 	initialization.Plugin()              // 注册插件
 	tasks.InitTasks()                    // 初始化定时任务
 	mq.Init()                            // 初始化MQ
+
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
+		log.Printf("已设置环境变量DONT_SEND，不发送消息")
+	}
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SAVE")); flag {
+		log.Printf("已设置环境变量DONT_SAVE，不保存消息")
+	}
 }
 
 func main() {

model/dto/message.go

@@ -250,5 +250,31 @@ func (m Message) IsJoinToGroup() (flag bool) {
 	if m.Type != types.MsgTypeSys {
 		return
 	}
-	return strings.Contains(m.Content, "\"邀请你加入了群聊，群聊参与人还有：")
+	flag = strings.Contains(m.Content, "\"邀请你加入了群聊，群聊参与人还有：")
+	if flag {
+		return
+	}
+	return strings.Contains(m.Content, "\"邀请你和\"") && strings.Contains(m.Content, "\"加入了群聊")
+}
+
+// IsRemoveFromChatroom
+// @description: 是否是被移出群聊消息
+// @receiver m
+// @return flag
+func (m Message) IsRemoveFromChatroom() (flag bool) {
+	if m.Type != types.MsgTypeSys {
+		return
+	}
+	return strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊")
+}
+
+// IsChangeGroupName
+// @description: 是否是修改群名称消息
+// @receiver m
+// @return flag
+func (m Message) IsChangeGroupName() (flag bool) {
+	if m.Type != types.MsgTypeSys {
+		return
+	}
+	return strings.HasPrefix(m.Content, "\"修改群名为“")
 }

plugin/plugins/ai.go

@@ -43,7 +43,7 @@ func AI(m *plugin.MessageContext) {
 	if !friendInfo.EnableAi {
 		return
 	}
-	if friendInfo.AiUsedToday > 0 && friendInfo.AiUsedToday >= friendInfo.AiFreeLimit {
+	if friendInfo.AiFreeLimit > 0 && friendInfo.AiUsedToday >= friendInfo.AiFreeLimit {
 		if notifyMap[m.FromUser] {
 			return
 		}

plugin/plugins/notify2configuser.go

@@ -2,28 +2,41 @@ package plugins
 
 import (
 	"fmt"
+	"go-wechat/client"
 	"go-wechat/config"
+	"go-wechat/model/entity"
 	"go-wechat/plugin"
 	"go-wechat/service"
 	"go-wechat/utils"
-	"strings"
 )
 
 // NotifyInvitationJoinGroup
 // @description: 通知邀请入群消息到配置用户
 // @param m
 func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
-	// 先回复一条固定句子
-	utils.SendMessage(m.FromUser, m.GroupUser, "您的邀请消息已收到啦，正在通知我的主人来同意请求。在我加群之后将会进行初始化操作，直到收到我主动发送的消息就是初始化完成咯，在那之前请耐心等待喔~", 0)
+	// 先回复一条固定消息
+	if conf, e := config.Conf.Resource["invitation-join-group"]; e {
+		// 发送一条新消息
+		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)
+		}
+	}
 
-	// 如果是邀请进群，推送到配置的用户
-	if flag, dec := m.IsInvitationJoinGroup(); flag {
-		for _, user := range config.Conf.System.NewFriendNotify.ToUser {
-			if user != "" {
-				// 发送一条新消息
-				dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec)
-				utils.SendMessage(user, "", dec, 0)
-			}
+	// 推送到配置的用户
+	_, dec := m.IsInvitationJoinGroup()
+	for _, user := range config.Conf.System.NewFriendNotify.ToUser {
+		if user != "" {
+			// 发送一条新消息
+			dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec)
+			_ = utils.SendMessage(user, "", dec, 0)
 		}
 	}
 }
@@ -32,24 +45,24 @@ func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
 // @description: 被移除群聊通知到配置用户
 // @param m
 func NotifyRemoveFromChatroom(m *plugin.MessageContext) {
-	// 如果是被移出群聊，推送到配置的用户
-	if strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊") {
-		// 调用一下退出群聊接口，防止被移出后还能从好友列表接口拉到相关信息
-		utils.QuitChatroom(m.FromUser, 0)
+	// 调用一下退出群聊接口，防止被移出后还能从好友列表接口拉到相关信息
+	utils.QuitChatroom(m.FromUser, 0)
 
-		// 取出群名称
-		groupInfo, err := service.GetFriendInfoById(m.FromUser)
-		if err != nil {
-			return
-		}
-		// 组装消息
-		msg := fmt.Sprintf("#移除群聊提醒\n\n群Id: %s\n群名称: %s\n事件: %s", m.FromUser, groupInfo.Nickname, m.Content)
+	// 取出群名称
+	groupInfo, err := service.GetFriendInfoById(m.FromUser)
+	if err != nil {
+		return
+	}
+	// 组装消息
+	msg := fmt.Sprintf("#移除群聊提醒\n\n群Id: %s\n群名称: %s\n事件: %s", m.FromUser, groupInfo.Nickname, m.Content)
 
-		for _, user := range config.Conf.System.NewFriendNotify.ToUser {
-			if user != "" {
-				// 发送一条新消息
-				utils.SendMessage(user, "", msg, 0)
-			}
+	for _, user := range config.Conf.System.NewFriendNotify.ToUser {
+		if user != "" {
+			// 发送一条新消息
+			_ = utils.SendMessage(user, "", msg, 0)
 		}
 	}
+
+	// 修改is_ok状态
+	client.MySQL.Model(&entity.Friend{}).Where("wxid = ?", m.FromUser).Update("is_ok", false)
 }

plugin/plugins/systemmessgae.go

@@ -2,7 +2,9 @@ package plugins
 
 import (
 	"go-wechat/plugin"
+	"go-wechat/service"
 	"go-wechat/utils"
+	"time"
 )
 
 // ReplyNewFriend
@@ -13,6 +15,17 @@ func ReplyNewFriend(m *plugin.MessageContext) {
 		_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化，请稍等几分钟，初始化完成之后我将主动告知您。", 0)
 	}
 	if m.IsOldFriendBack() {
-		_ = utils.SendMessage(m.FromUser, m.GroupUser, "嘿，我的朋友，你为何要离我而去？又为何去而复返？", 0)
+		_ = utils.SendMessage(m.FromUser, "", "嘿，我的朋友，你为何要离我而去？又为何去而复返？", 0)
 	}
+
+	go func() {
+		// 等待5秒
+		time.Sleep(5 * time.Second)
+		// 刷新好友列表
+		service.SyncFriend()
+		// 如果是加入群，刷新群成员列表
+		if m.IsJoinToGroup() {
+			service.SyncGroupMembers(m.FromUser)
+		}
+	}()
 }

plugin/plugins/welconenew.go

@@ -5,6 +5,7 @@ import (
 	"go-wechat/config"
 	"go-wechat/model/entity"
 	"go-wechat/plugin"
+	"go-wechat/service"
 	"go-wechat/utils"
 )
 
@@ -28,7 +29,7 @@ func WelcomeNew(m *plugin.MessageContext) {
 	switch conf.Type {
 	case "text":
 		// 文字类型
-		utils.SendMessage(m.FromUser, "", conf.Path, 0)
+		_ = utils.SendMessage(m.FromUser, "", conf.Path, 0)
 	case "image":
 		// 图片类型
 		utils.SendImage(m.FromUser, conf.Path, 0)
@@ -36,4 +37,7 @@ func WelcomeNew(m *plugin.MessageContext) {
 		// 表情类型
 		utils.SendEmotion(m.FromUser, conf.Path, 0)
 	}
+
+	// 刷新群成员列表
+	go service.SyncGroupMembers(m.FromUser)
 }

router/router.go

@@ -28,6 +28,7 @@ func Init(g *gin.Engine) {
 	api.POST("/ai/model", app.ChangeUseAiModel)                   // 修改使用的AI模型
 	api.POST("/ai/free", app.ChangeAiFreeLimit)                   // 修改AI免费次数
 	api.POST("/ai/assistant", app.ChangeUseAiAssistant)           // 修改使用的AI助手
+	api.POST("/ai/summary", app.SendAiSummary)                    // 手动发送AI聊天总结
 	api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus)     // 修改是否开启迎新状态
 	api.PUT("/command/status", app.ChangeEnableCommandStatus)     // 修改是否开启指令状态
 	api.PUT("/news/status", app.ChangeEnableNewsStatus)           // 修改是否开启早报状态

service/friendsync.go

@@ -0,0 +1,257 @@
+package service
+
+import (
+	"go-wechat/client"
+	"go-wechat/common/types"
+	"go-wechat/config"
+	"go-wechat/model/entity"
+	"go-wechat/utils"
+	"log"
+	"strings"
+	"time"
+)
+
+// SyncFriend
+// @description: 同步好友列表
+func SyncFriend() {
+	// 获取好友列表
+	friends, err := utils.GetFriendList()
+	if err != nil {
+		log.Printf("获取好友列表失败: %s", err.Error())
+		return
+	}
+	// 当前获取到的成员Id，用于后续设置is_ok状态
+	nowIds := make([]string, 0)
+	// 取出已存在的成员信息
+	var oldData []entity.Friend
+	err = client.MySQL.Find(&oldData).Error
+	if err != nil {
+		log.Printf("查询好友列表失败: %s", err.Error())
+		return
+	}
+	// 将历史数据整理成map
+	oldMap := make(map[string]bool)
+	for _, item := range oldData {
+		oldMap[item.Wxid] = item.IsOk
+	}
+
+	// 新增的成员，用于通知给指定的人
+	var notifyMap = make(map[string]string)
+
+	// 开启事务
+	tx := client.MySQL.Begin()
+	defer tx.Commit()
+
+	// 循环获取到的好友列表
+	for _, item := range friends {
+		// 填充当前存在的账号
+		nowIds = append(nowIds, item.Wxid)
+
+		// 判断是否已经存在
+		if _, ok := oldMap[item.Wxid]; ok {
+			// 已存在，修改
+			pm := map[string]any{
+				"nickname":       item.Nickname,
+				"custom_account": item.CustomAccount,
+				"pinyin":         item.Pinyin,
+				"pinyin_all":     item.PinyinAll,
+				"is_ok":          true,
+			}
+			err = tx.Model(&entity.Friend{}).Where("wxid = ?", item.Wxid).Updates(pm).Error
+			if err != nil {
+				log.Printf("修改好友失败: %s", err.Error())
+				continue
+			}
+
+			// 如果已存在但是是已退出的群，也通知一下
+			if !oldMap[item.Wxid] {
+				notifyMap[item.Wxid] = item.Nickname + " #秽土转生"
+
+				// 通知一下，初始化完成
+				if conf, e := config.Conf.Resource["introduce"]; e {
+					// 发送一条新消息
+					switch conf.Type {
+					case "text":
+						// 文字类型
+						_ = utils.SendMessage(item.Wxid, "", conf.Path, 0)
+					case "image":
+						// 图片类型
+						utils.SendImage(item.Wxid, conf.Path, 0)
+					case "emotion":
+						// 表情类型
+						utils.SendEmotion(item.Wxid, conf.Path, 0)
+					}
+				}
+				// 发送配置网页
+				if config.Conf.System.Domain != "" {
+					title := "欢迎使用微信机器人(切勿转发)"
+					desc := "点我可以配置功能喔，提示非微信官方网页，点击继续访问即可"
+					url := utils.GetManagerUrl(item.Wxid)
+					utils.SendPublicMsg(item.Wxid, title, desc, url, 0)
+				}
+			}
+		} else {
+			// 新增
+			err = tx.Create(&entity.Friend{
+				CustomAccount:  item.CustomAccount,
+				Nickname:       item.Nickname,
+				Pinyin:         item.Pinyin,
+				PinyinAll:      item.PinyinAll,
+				Wxid:           item.Wxid,
+				IsOk:           true,
+				EnableAi:       config.Conf.System.DefaultRule.Ai,
+				EnableChatRank: config.Conf.System.DefaultRule.ChatRank,
+				EnableSummary:  config.Conf.System.DefaultRule.Summary,
+				EnableWelcome:  config.Conf.System.DefaultRule.Welcome,
+				EnableNews:     config.Conf.System.DefaultRule.News,
+				EnableHotTop:   config.Conf.System.DefaultRule.HotTop,
+				AiFreeLimit:    config.Conf.System.DefaultRule.AiFreeLimit,
+				ClearMember:    0,
+				LastActive:     types.DateTime(time.Now().Local()),
+			}).Error
+			if err != nil {
+				log.Printf("新增好友失败: %s", err.Error())
+				continue
+			}
+			notifyMap[item.Wxid] = item.Nickname
+			if conf, e := config.Conf.Resource["introduce"]; e {
+				// 发送一条新消息
+				switch conf.Type {
+				case "text":
+					// 文字类型
+					_ = utils.SendMessage(item.Wxid, "", conf.Path, 0)
+				case "image":
+					// 图片类型
+					utils.SendImage(item.Wxid, conf.Path, 0)
+				case "emotion":
+					// 表情类型
+					utils.SendEmotion(item.Wxid, conf.Path, 0)
+				}
+			}
+			// 发送配置网页
+			if config.Conf.System.Domain != "" {
+				title := "欢迎使用微信机器人(切勿转发)"
+				desc := "点我可以配置功能喔，提示非微信官方网页，点击继续访问即可"
+				url := utils.GetManagerUrl(item.Wxid)
+				utils.SendPublicMsg(item.Wxid, title, desc, url, 0)
+			}
+		}
+	}
+
+	// 通知有新成员
+	if len(notifyMap) > 0 && config.Conf.System.NewFriendNotify.Enable {
+		// 组装成一句话
+		msg := []string{"#新好友通知\n"}
+		for wxId, nickname := range notifyMap {
+			msg = append(msg, "微信Id: "+wxId+"\n昵称: "+nickname)
+		}
+		for _, user := range config.Conf.System.NewFriendNotify.ToUser {
+			if user != "" {
+				// 发送一条新消息
+				_ = utils.SendMessage(user, "", strings.Join(msg, "\n-------\n"), 0)
+			}
+		}
+	}
+
+	// 清理不在列表中的好友
+	clearPm := map[string]any{
+		"is_ok": false,
+	}
+	err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Updates(clearPm).Error
+	if err != nil {
+		log.Printf("清理好友失败: %s", err.Error())
+	}
+
+	log.Println("同步好友列表完成")
+}
+
+// SyncGroupMembers
+// @description: 同步群成员
+// @param wxId
+func SyncGroupMembers(wxId string) {
+	membersIds, admin, err := utils.GetGroupMembers(wxId)
+	if err != nil {
+		return
+	}
+
+	// 修改不在数组的群成员状态为不在
+	pm := map[string]any{
+		"is_member":  false,
+		"leave_time": time.Now().Local(),
+	}
+	err = client.MySQL.Model(&entity.GroupUser{}).
+		Where("group_id = ?", wxId).
+		Where("is_member IS TRUE").
+		Where("wxid NOT IN (?)", membersIds).
+		Updates(pm).Error
+	if err != nil {
+		log.Printf("修改群成员状态失败: %s", err.Error())
+		return
+	}
+
+	// 取出当前数据库存在的成员信息
+	var oldData []entity.GroupUser
+	err = client.MySQL.Model(&entity.GroupUser{}).
+		Where("group_id = ?", wxId).
+		Find(&oldData).Error
+	if err != nil {
+		log.Printf("查询群成员失败: %s", err.Error())
+		return
+	}
+	// 将历史数据整理成map
+	oldMap := make(map[string]bool)
+	for _, item := range oldData {
+		oldMap[item.Wxid] = item.IsMember
+	}
+
+	// 循环获取到的群成员列表
+	for _, wxid := range membersIds {
+		// 如果历史数据中存在，且是成员，跳过
+		if isMember, ok := oldMap[wxid]; ok && isMember {
+			continue
+		}
+		// 获取成员信息
+		cp, e := utils.GetContactProfile(wxid)
+		if e != nil {
+			log.Printf("获取成员信息失败: %s", e.Error())
+			continue
+		}
+		if cp.Wxid != "" {
+			if _, ok := oldMap[wxid]; ok {
+				// 历史数据中存在，修改
+				// 修改
+				gupm := map[string]any{
+					"account":    cp.Account,
+					"head_image": cp.HeadImage,
+					"nickname":   cp.Nickname,
+					"is_member":  true,
+					"is_admin":   cp.Wxid == admin,
+					"leave_time": nil,
+				}
+				err = client.MySQL.Model(&entity.GroupUser{}).Where("group_id = ?", wxId).Where("wxid = ?", wxid).Updates(gupm).Error
+				if err != nil {
+					log.Printf("修改群成员失败: %s", err.Error())
+					continue
+				}
+			} else {
+				// 新增的
+				// 新增
+				err = client.MySQL.Create(&entity.GroupUser{
+					GroupId:    wxId,
+					Account:    cp.Account,
+					HeadImage:  cp.HeadImage,
+					Nickname:   cp.Nickname,
+					Wxid:       cp.Wxid,
+					IsMember:   true,
+					IsAdmin:    cp.Wxid == admin,
+					JoinTime:   time.Now().Local(),
+					LastActive: time.Now().Local(),
+				}).Error
+				if err != nil {
+					log.Printf("新增群成员失败: %s", err.Error())
+					continue
+				}
+			}
+		}
+	}
+}

tasks/summary/summary.go

@@ -1,15 +1,11 @@
 package summary
 
 import (
-	"context"
 	"fmt"
-	"github.com/sashabaranov/go-openai"
-	"go-wechat/config"
 	"go-wechat/model/vo"
 	"go-wechat/service"
 	"go-wechat/utils"
 	"log"
-	"strings"
 	"time"
 )
 
@@ -37,63 +33,21 @@ func AiSummary() {
 			continue
 		}
 		// 组装对话记录为字符串
-		var content []string
-		for _, record := range records {
-			content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。")))
-		}
-
-		msgTmp := `请帮我总结一下一下的群聊内容的梗概，生成的梗概需要尽可能详细，需要带上一些聊天关键信息，并且带上群友名字。
-注意，他们可能是多个话题，请仔细甄别。
-每一行代表一个人的发言，每一行的的格式为： {"{nickname}": "{content}"}--end--
-
-群名称: %s
-聊天记录如下:
-%s
-`
-
-		msg := fmt.Sprintf(msgTmp, group.Nickname, strings.Join(content, "\n"))
-
-		// AI总结
-		messages := []openai.ChatCompletionMessage{
-			{
-				Role:    openai.ChatMessageRoleUser,
-				Content: msg,
-			},
-		}
-
-		// 默认使用AI回复
-		conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
-		if config.Conf.Ai.BaseUrl != "" {
-			conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
-		}
-		ai := openai.NewClientWithConfig(conf)
-		var resp openai.ChatCompletionResponse
-		resp, err = ai.CreateChatCompletion(
-			context.Background(),
-			openai.ChatCompletionRequest{
-				Model:    config.Conf.Ai.SummaryModel,
-				Messages: messages,
-			},
-		)
-
+		var replyMsg string
+		replyMsg, err = utils.GetAiSummary(group.Nickname, records)
 		if err != nil {
 			log.Printf("群聊记录总结失败: %v", err.Error())
-			utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败，错误信息: "+err.Error(), 0)
+			_ = utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败，错误信息: "+err.Error(), 0)
 			continue
 		}
 
-		// 返回消息为空
-		if resp.Choices[0].Message.Content == "" {
-			utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败，AI返回结果为空", 0)
-			continue
-		}
-		replyMsg := fmt.Sprintf("#昨日消息总结\n又是一天过去了，让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", resp.Choices[0].Message.Content)
+		replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了，让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg)
 		//log.Printf("群[%s]对话记录总结成功，总结内容: %s", group.Wxid, replyMsg)
-		utils.SendMessage(group.Wxid, "", replyMsg, 0)
+		_ = utils.SendMessage(group.Wxid, "", replyMsg, 0)
 
-		// 判断耗时是否达到15秒，不足就等待
-		if used := time.Now().Sub(start); used < 15*time.Second {
-			time.Sleep(15*time.Second - used)
+		// 判断耗时是否达到一分钟，不足就等待
+		if used := time.Now().Sub(start); used < time.Minute {
+			time.Sleep(time.Minute - used)
 		}
 	}
 }

utils/friend.go

@@ -0,0 +1,105 @@
+package utils
+
+import (
+	"encoding/json"
+	"github.com/go-resty/resty/v2"
+	"go-wechat/common/constant"
+	"go-wechat/config"
+	"go-wechat/model/dto"
+	"log"
+	"slices"
+	"strings"
+)
+
+// http客户端
+var hc = resty.New()
+
+// GetFriendList
+// @description: 获取好友列表
+// @return friends
+// @return err
+func GetFriendList() (friends []dto.FriendItem, err error) {
+	var base dto.Response[[]dto.FriendItem]
+	resp, err := hc.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetResult(&base).
+		Post(config.Conf.Wechat.GetURL("/api/getContactList"))
+	if err != nil {
+		log.Printf("获取好友列表失败: %s", err.Error())
+		return
+	}
+	log.Printf("获取好友列表结果: %s", resp.String())
+
+	// 循环获取到的好友列表
+	for _, item := range base.Data {
+		// 跳过特殊账号
+		// 跳过公众号和企业微信好友
+		if strings.Contains(item.Wxid, "gh_") || strings.Contains(item.Wxid, "@openim") {
+			continue
+		}
+		// 特殊Id跳过
+		if slices.Contains(constant.SpecialId, item.Wxid) {
+			continue
+		}
+		// 添加到待返回列表
+		friends = append(friends, item)
+	}
+
+	return
+}
+
+// GetGroupMembers
+// @description: 获取指定群成员
+// @param wxId 群Id
+// @return ids 群成员id数组
+// @return members 群成员信息
+// @return err 错误信息
+func GetGroupMembers(wxId string) (ids []string, admin string, err error) {
+	var base dto.Response[dto.GroupUser]
+	// 组装参数
+	param := map[string]any{
+		"chatRoomId": wxId, // 群Id
+	}
+	pbs, _ := json.Marshal(param)
+
+	_, err = hc.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		SetResult(&base).
+		Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
+	if err != nil {
+		log.Printf("获取群成员信息失败: %s", err.Error())
+		return
+	}
+	// 昵称Id
+	ids = strings.Split(base.Data.Members, "^G")
+	admin = base.Data.Admin
+	return
+}
+
+// GetContactProfile
+// @description: 获取联系人信息
+// @param wxId 微信Id
+// @return info 联系人信息
+// @return err 错误信息
+func GetContactProfile(wxId string) (info dto.ContactProfile, err error) {
+	var baseResp dto.Response[dto.ContactProfile]
+
+	// 组装参数
+	param := map[string]any{
+		"wxid": wxId, // 群Id
+	}
+	pbs, _ := json.Marshal(param)
+
+	_, err = hc.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		SetResult(&baseResp).
+		Post(config.Conf.Wechat.GetURL("/api/getContactProfile"))
+	if err != nil {
+		log.Printf("获取成员详情失败: %s", err.Error())
+		return
+	}
+	info = baseResp.Data
+	return
+}

utils/send.go

@@ -8,6 +8,8 @@ import (
 	"go-wechat/common/current"
 	"go-wechat/config"
 	"log"
+	"os"
+	"strconv"
 	"time"
 )
 
@@ -17,6 +19,10 @@ import (
 // @param atId
 // @param msg
 func SendMessage(toId, atId, msg string, retryCount int) (err error) {
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
+		return
+	}
+
 	if retryCount > 5 {
 		log.Printf("重试五次失败，停止发送")
 		err = errors.New("重试五次失败，停止发送")
@@ -62,6 +68,10 @@ func SendMessage(toId, atId, msg string, retryCount int) (err error) {
 // @param imgPath string 图片路径
 // @param retryCount int 重试次数
 func SendImage(toId, imgPath string, retryCount int) {
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
+		return
+	}
+
 	if retryCount > 5 {
 		log.Printf("重试五次失败，停止发送")
 		return
@@ -94,6 +104,10 @@ func SendImage(toId, imgPath string, retryCount int) {
 // @param emotionHash string 表情包hash(md5值)
 // @param retryCount int 重试次数
 func SendEmotion(toId, emotionHash string, retryCount int) {
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
+		return
+	}
+
 	if retryCount > 5 {
 		log.Printf("重试五次失败，停止发送")
 		return
@@ -200,6 +214,10 @@ func QuitChatroom(chatRoomId string, retryCount int) {
 // @param url string 链接
 // @param retryCount int 重试次数
 func SendPublicMsg(wxId, title, digest, url string, retryCount int) {
+	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
+		return
+	}
+
 	if retryCount > 5 {
 		log.Printf("重试五次失败，停止发送")
 		return

utils/summary.go

@@ -0,0 +1,108 @@
+package utils
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/sashabaranov/go-openai"
+	"go-wechat/config"
+	"go-wechat/model/vo"
+	"log"
+	"strings"
+)
+
+// GetAiSummary
+// @description: AI总结群聊记录
+// @param groupName
+// @param records
+// @return replyMsg
+// @return err
+func GetAiSummary(groupName string, records []vo.TextMessageItem) (replyMsg string, err error) {
+	// 组装对话记录为字符串
+	var content []string
+	for _, record := range records {
+		content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。")))
+	}
+
+	prompt := `# Role: 数据分析师和信息整理专家
+        # Background: 用户需要从群聊记录中提取有价值的信息，包括技术话题讨论、工具分享、网站或资源链接。
+        # Profile: 你是一位专业的数据分析师，擅长从大量文本中提取关键信息，并能够高效地整理和总结内容。
+        # Skills: 数据分析、信息提取、内容总结、网络资源评估。
+        # Goals: 设计一个流程，帮助用户识别群聊中的技术话题、共享的工具和资源，并高效地总结每个话题的要点。
+        # Constrains: 该流程需要简洁明了，易于用户理解和操作，同时确保不遗漏任何重要信息。
+        # OutputFormat: 结构化的文本总结，包含关键讨论点、结论等。
+        # Workflow:
+        ## 1. 快速浏览群聊记录
+         - 通过快速浏览聊天记录，初步识别出主要话题。
+        ## 2. 识别和分类主要话题
+         - 识别群聊中的主要话题，分类讨论内容。
+         - 每个话题独立提取，避免混淆。
+		 - 话题识别尽可能准确，确保不遗漏重要信息。
+        ## 3. 编写总结概括
+         - 对每个话题进行总结，包含以下内容：
+           - 关键讨论点：主要的讨论内容和观点。
+           - 参与成员：参与讨论的成员的昵称。
+           - 结论：得出的结论或共识，内容要尽可能详细。
+        ## 4. 组织信息
+         - 以逻辑清晰、易于理解的方式组织信息。
+         - 使用结构化文本，使总结内容易于阅读。
+        ## 5. 生成最终总结
+         - 根据上述步骤，生成每个话题的详细总结。
+         - 确保每个总结完整且有条理。
+         - 返回数据格式为结构化文本，方便用户查阅。
+        ## 6. 群聊记录消息格式
+        格式: {"{nickname}": "{content}"}--end--
+        # Examples:
+        - 话题：技术讨论 - AI在医疗领域的应用
+          - 关键讨论点：讨论了AI在医疗诊断中的应用前景和现有挑战。
+          - 成员：张三、李四、王五。
+          - 结论：AI技术有望提高医疗诊断的准确性和效率，但数据隐私和模型解释性仍需解决。
+
+        - 话题：美食讨论 - 杭州到底有没有美食
+          - 关键讨论点：讨论了杭州的美食有哪些、分布地区、评价如何。
+          - 成员：张三、李四、王五。
+          - 结论：杭州好像没有美食。
+`
+
+	msg := fmt.Sprintf("群名称: %s\n聊天记录如下:\n%s", groupName, strings.Join(content, "\n"))
+
+	// AI总结
+	messages := []openai.ChatCompletionMessage{
+		{
+			Role:    openai.ChatMessageRoleSystem,
+			Content: prompt,
+		},
+		{
+			Role:    openai.ChatMessageRoleUser,
+			Content: msg,
+		},
+	}
+
+	// 默认使用AI回复
+	conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
+	if config.Conf.Ai.BaseUrl != "" {
+		conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
+	}
+	ai := openai.NewClientWithConfig(conf)
+	var resp openai.ChatCompletionResponse
+	resp, err = ai.CreateChatCompletion(
+		context.Background(),
+		openai.ChatCompletionRequest{
+			Model:    config.Conf.Ai.SummaryModel,
+			Messages: messages,
+		},
+	)
+
+	if err != nil {
+		log.Printf("群聊记录总结失败: %v", err.Error())
+		return
+	}
+
+	// 返回消息为空
+	if resp.Choices[0].Message.Content == "" {
+		err = errors.New("AI返回结果为空")
+		return
+	}
+	replyMsg = resp.Choices[0].Message.Content
+	return
+}

utils/url.go

@@ -16,6 +16,6 @@ func GetManagerUrl(wxId string) (url string) {
 	// base64一下
 	encodeString := base64.StdEncoding.EncodeToString([]byte(url))
 	// 拼接新链接（这个是一个已备案的域名）
-	url = "https://redirect.wjg95.cn/?s=" + encodeString
+	url = "https://redirect.gitee.ltd/?s=" + encodeString
 	return
 }

views/components.html

@@ -57,6 +57,9 @@
 
 <!-- 消息总结 -->
 {{define "summary"}}
+{{ if eq .EnableSummary true }}
+<button type="button" class="btn-link float-end text-blue-600" onclick="sendAiSummary({{.Wxid}})">手动发送总结</button>
+{{ end }}
 <button type="button"
         class="{{ if eq .EnableSummary true }}bg-green-600{{ else }}bg-gray-200{{ end }} relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out focus:outline-none focus:ring-2 focus:ring-green-600 focus:ring-offset-2"
         role="switch" aria-checked="false" onclick="changeSummaryEnableStatus({{.Wxid}})">

views/group.html

@@ -56,6 +56,7 @@
                   </div>
                   {{ if eq .EnableAi true }}
                   <div class="float-end">
+                    <!-- 模型 -->
                     <div>
                       <label>
                         <select class="block w-full rounded-md border-0 py-1.5 pl-3 pr-10 text-gray-900 ring-1 ring-inset ring-gray-300 focus:ring-2 focus:ring-green-600 sm:text-sm sm:leading-6" onchange="aiModelChange(event, {{.Wxid}})">
@@ -68,6 +69,7 @@
                         </select>
                       </label>
                     </div>
+                    <!-- AI角色 -->
                     <div class="float-end mt-1">
                       <label>
                         <select

views/static/js/index.js

@@ -317,3 +317,30 @@ function changeAiFreeLimit(wxid, limitNumber) {
     window.location.reload();
   })
 }
+
+// 发送AI群聊总结
+function sendAiSummary(wxid) {
+  if (wxid === '') {
+    alert("wxid不能为空")
+    return
+  }
+  if (!confirm("确定发送AI群聊总结吗?")) {
+    return
+  }
+  // 请求接口
+  axios({
+    method: 'post',
+    url: '/api/ai/summary',
+    params: {
+      id: wxid,
+    }
+  }).then(function (response) {
+    console.log(`返回结果: ${JSON.stringify(response)}`);
+    alert(`${response.data}`)
+  }).catch(function (error) {
+    console.log(`错误信息: ${error}`);
+    alert(`${response.data}`)
+  }).finally(function () {
+    window.location.reload();
+  })
+}