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

Reviewed-on: lxh/go-wxhelper#105

app/friend.go

@@ -280,3 +280,26 @@ func AutoClearMembers(ctx *gin.Context) {
 
 	ctx.String(http.StatusOK, "操作成功")
 }
+
+// ChangeAiFreeLimit
+// @description: 修改AI免费次数
+// @param ctx
+func ChangeAiFreeLimit(ctx *gin.Context) {
+	var p autoClearMembers
+	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("`ai_free_limit`", p.Days).Error
+	if err != nil {
+		log.Printf("修改AI免费次数失败：%s", err)
+		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
+		return
+	}
+
+	ctx.String(http.StatusOK, "操作成功")
+}

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

@@ -4,7 +4,7 @@ system:
   alApiToken: xxx
   # urlc.cn的Token，用来生成短链接
   urlcApiToken: xxx
-  # 系统访问域名
+  # 系统访问域名，必须是包括 http[s]:// 的完整域名
   domain: https://wechat.abc.com
   # 添加新好友或群之后通知给指定的人
   newFriendNotify:
@@ -25,11 +25,13 @@ system:
     news: true
     # 热榜
     hotTop: true
+    # 每天免费 AI 次数限制
+    aiFreeLimit: 50
 
 # 微信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服务的回调
@@ -59,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
@@ -79,7 +81,7 @@ mq:
   enable: false
   # RabbitMQ配置
   rabbitmq:
-    host: 10.0.0.247
+    host: 10.0.0.31
     port: 5672
     user: wechat
     password: wechat123
@@ -100,6 +102,7 @@ ai:
   # 人设
   personality: 你的名字叫张三，你是一个百科机器人，你的爱好是看电影，你的性格是开朗的，你的专长是讲故事，你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣，也不会讨论任何与政治相关的话题，你甚至可以拒绝回答这一类话题。
   models:
+    - name: 默认
     - name: ChatGPT-4
       model: gpt-4
       canManager: false
@@ -137,3 +140,7 @@ resource:
   wordcloud:
     type: image
     path: D:\Share\wordcloud\%s
+  # 邀请入群消息回复
+  invitation-join-group:
+    type: text
+    path: "您的邀请消息已收到啦，正在通知我的主人来同意请求。在我加群之后将会进行初始化操作，直到收到我主动发送的消息就是初始化完成咯，在那之前请耐心等待喔~"

config/system.go

@@ -16,10 +16,11 @@ type newFriendNotify struct {
 
 // 默认规则
 type defaultRule struct {
-	Ai       bool `json:"ai" yaml:"ai"`             // 是否启用AI
-	ChatRank bool `json:"chatRank" yaml:"chatRank"` // 是否启用聊天排行榜
-	Summary  bool `json:"summary" yaml:"summary"`   // 是否启用聊天总结
-	Welcome  bool `json:"welcome" yaml:"welcome"`   // 是否启用欢迎新成员
-	News     bool `json:"news" yaml:"news"`         // 是否启用每日早报
-	HotTop   bool `json:"hotTop" yaml:"hotTop"`     // 是否启用热门话题
+	Ai          bool `json:"ai" yaml:"ai"`                   // 是否启用AI
+	ChatRank    bool `json:"chatRank" yaml:"chatRank"`       // 是否启用聊天排行榜
+	Summary     bool `json:"summary" yaml:"summary"`         // 是否启用聊天总结
+	Welcome     bool `json:"welcome" yaml:"welcome"`         // 是否启用欢迎新成员
+	News        bool `json:"news" yaml:"news"`               // 是否启用每日早报
+	HotTop      bool `json:"hotTop" yaml:"hotTop"`           // 是否启用热门话题
+	AiFreeLimit int  `json:"aiFreeLimit" yaml:"aiFreeLimit"` // AI免费次数
 }

initialization/plugin.go

@@ -6,7 +6,6 @@ import (
 	plugin "go-wechat/plugin"
 	"go-wechat/plugin/plugins"
 	"go-wechat/service"
-	"go-wechat/types"
 )
 
 // Plugin
@@ -31,11 +30,11 @@ 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 {
-		return m.Type == types.MsgTypeSys
+		return m.IsNewFriendAdd() || m.IsJoinToGroup() || m.IsOldFriendBack()
 	}, plugins.ReplyNewFriend)
 
 	// 私聊指令消息

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

@@ -219,3 +219,62 @@ func (m Message) IsInvitationJoinGroup() (flag bool, str string) {
 	}
 	return
 }
+
+// IsNewFriendAdd
+// @description: 是否是新好友添加消息
+// @receiver m
+// @return flag
+func (m Message) IsNewFriendAdd() (flag bool) {
+	if m.Type != types.MsgTypeSys {
+		return
+	}
+	return strings.HasPrefix(m.Content, "你已添加了") && strings.HasSuffix(m.Content, "，现在可以开始聊天了。")
+}
+
+// IsOldFriendBack
+// @description: 是否是老好友回归消息
+// @receiver m
+// @return flag
+func (m Message) IsOldFriendBack() (flag bool) {
+	if m.Type != types.MsgTypeSys {
+		return
+	}
+	return m.Content == "以上是打招呼的内容"
+}
+
+// IsJoinToGroup
+// @description: 是否是加入群聊消息
+// @receiver m
+// @return flag
+func (m Message) IsJoinToGroup() (flag bool) {
+	if m.Type != types.MsgTypeSys {
+		return
+	}
+	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, "\"修改群名为“")
+}

model/entity/friend.go

@@ -25,6 +25,8 @@ type Friend struct {
 	ClearMember    int            `json:"clearMember"`                                              // 清理成员配置（多少天未活跃的）
 	IsOk           bool           `json:"isOk" gorm:"type:tinyint(1) default 0 not null"`           // 是否正常
 	UsedTokens     int            `json:"usedTokens"`                                               // 已使用的AI Token数量
+	AiFreeLimit    int            `json:"aiFreeLimit"`                                              // AI免费次数
+	AiUsedToday    int            `json:"aiUsedToday"`                                              // 今日已使用的AI次数
 }
 
 func (Friend) TableName() string {

model/vo/friend.go

@@ -24,6 +24,8 @@ type FriendItem struct {
 	EnableHotTop   bool           // 是否启用热搜
 	ClearMember    int            // 清理成员配置（多少天未活跃的）
 	IsOk           bool           // 是否还在通讯库(群聊是要还在群里也算)
+	AiFreeLimit    int            // AI免费次数
+	AiUsedToday    int            // 今日已使用的AI次数
 }
 
 // GroupUserItem

plugin/plugins/ai.go

@@ -19,6 +19,12 @@ import (
 	"time"
 )
 
+// 已经通知过的群组或者好友map
+var notifyMap = make(map[string]bool)
+
+// 拉取最近消息条数
+const fetchMessageCount = 20
+
 // AI
 // @description: AI消息
 // @param m
@@ -37,6 +43,23 @@ func AI(m *plugin.MessageContext) {
 	if !friendInfo.EnableAi {
 		return
 	}
+	if friendInfo.AiFreeLimit > 0 && friendInfo.AiUsedToday >= friendInfo.AiFreeLimit {
+		if notifyMap[m.FromUser] {
+			return
+		}
+		_ = utils.SendMessage(m.FromUser, "", fmt.Sprintf("本群今天的免费次数已经用完啦，明天再来找我聊天吧~\n每天限制%d次，0点自动重置", friendInfo.AiFreeLimit), 0)
+		notifyMap[m.FromUser] = true
+		return
+	} else {
+		notifyMap[m.FromUser] = false
+	}
+
+	var err error
+	defer func() {
+		if err == nil {
+			service.UpdateAiUsedToday(m.FromUser)
+		}
+	}()
 
 	// 预处理一下发送的消息，用正则去掉@机器人的内容
 	re := regexp.MustCompile(`@([^ | ]+)`)
@@ -132,17 +155,17 @@ func AI(m *plugin.MessageContext) {
 
 	if err != nil {
 		log.Printf("OpenAI聊天发起失败: %v", err.Error())
-		utils.SendMessage(m.FromUser, m.GroupUser, "AI聊天初始化失败，我已经通知我主人来修啦，请稍候一下下喔～", 0)
+		_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI聊天初始化失败，我已经通知我主人来修啦，请稍候一下下喔～", 0)
 		return
 	}
 
 	// 返回消息为空
 	if len(resp.Choices) == 0 || resp.Choices[0].Message.Content == "" {
-		utils.SendMessage(m.FromUser, m.GroupUser, "AI似乎抽风了，没有告诉我你需要的回答~", 0)
+		_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI似乎抽风了，没有告诉我你需要的回答~", 0)
 		return
 	}
 
-	// 异步更新一下已使用的AI次数
+	// 异步更新一下已使用的AI tokens
 	go service.UpdateUsedAiTokens(m.FromUser, resp.Usage.TotalTokens)
 
 	// 保存一下AI 返回的消息，消息 Id 使用传入 Id 的负数
@@ -162,7 +185,7 @@ func AI(m *plugin.MessageContext) {
 	if m.GroupUser != "" {
 		replyMsg = "\n" + resp.Choices[0].Message.Content
 	}
-	utils.SendMessage(m.FromUser, m.GroupUser, replyMsg, 0)
+	err = utils.SendMessage(m.FromUser, m.GroupUser, replyMsg, 0)
 }
 
 // getGroupUserMessages
@@ -179,7 +202,7 @@ func getGroupUserMessages(msgId int64, groupId, groupUserId string) (records []e
 		Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
 		Where(subQuery).
 		Order("create_at desc").
-		Limit(4).Find(&records)
+		Limit(fetchMessageCount).Find(&records)
 	return
 }
 
@@ -195,6 +218,6 @@ func getUserPrivateMessages(userId string) (records []entity.Message) {
 		Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
 		Where(subQuery).
 		Order("create_at desc").
-		Limit(4).Find(&records)
+		Limit(fetchMessageCount).Find(&records)
 	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,17 +2,30 @@ package plugins
 
 import (
 	"go-wechat/plugin"
+	"go-wechat/service"
 	"go-wechat/utils"
-	"strings"
+	"time"
 )
 
 // ReplyNewFriend
 // @description: 响应好友添加成功消息
 // @param m
 func ReplyNewFriend(m *plugin.MessageContext) {
-	isNewFriend := strings.HasPrefix(m.Content, "你已添加了") && strings.HasSuffix(m.Content, "，现在可以开始聊天了。")
-	isNewChatroom := strings.Contains(m.Content, "\"邀请你加入了群聊，群聊参与人还有：")
-	if isNewFriend || isNewChatroom {
-		utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化，请稍等几分钟，初始化完成之后我将主动告知您。", 0)
+	if m.IsNewFriendAdd() || m.IsJoinToGroup() {
+		_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化，请稍等几分钟，初始化完成之后我将主动告知您。", 0)
 	}
+	if m.IsOldFriendBack() {
+		_ = 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

@@ -26,7 +26,9 @@ func Init(g *gin.Engine) {
 	api := g.Group("/api")
 	api.PUT("/ai/status", app.ChangeEnableAiStatus)               // 修改是否开启AI状态
 	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/friend.go

@@ -152,3 +152,25 @@ func UpdateUsedAiTokens(wxId string, tokens int) {
 		log.Printf("更新AI使用次数失败, 错误信息: %v", err)
 	}
 }
+
+// UpdateAiUsedToday
+// @description: 更新AI今日使用次数
+// @param wxId
+func UpdateAiUsedToday(wxId string) {
+	err := client.MySQL.Model(&entity.Friend{}).
+		Where("wxid = ?", wxId).
+		Update("`ai_used_today`", gorm.Expr(" `ai_used_today` + 1")).Error
+	if err != nil {
+		log.Printf("更新AI今日使用次数失败, 错误信息: %v", err)
+	}
+}
+
+// ClearAiUsedToday
+// @description: 清空AI今日使用次数
+func ClearAiUsedToday() {
+	err := client.MySQL.Model(&entity.Friend{}).Where("is_ok = 1").
+		Update("`ai_used_today`", 0).Error
+	if err != nil {
+		log.Printf("清空AI今日使用次数失败, 错误信息: %v", err)
+	}
+}

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/friends/ai.go

@@ -0,0 +1,9 @@
+package friends
+
+import "go-wechat/service"
+
+// ClearAiUsedToday
+// @description: 清空AI日使用次数
+func ClearAiUsedToday() {
+	service.ClearAiUsedToday()
+}

tasks/friends/friends.go

@@ -85,6 +85,7 @@ func Sync() {
 				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
@@ -109,9 +110,9 @@ func Sync() {
 			}
 			// 发送配置网页
 			if config.Conf.System.Domain != "" {
-				title := "欢迎使用微信机器人"
+				title := "欢迎使用微信机器人(切勿转发)"
 				desc := "点我可以配置功能喔，提示非微信官方网页，点击继续访问即可"
-				url := config.Conf.System.Domain + "/manager.html?id=" + friend.Wxid
+				url := utils.GetManagerUrl(friend.Wxid)
 				utils.SendPublicMsg(friend.Wxid, title, desc, url, 0)
 			}
 
@@ -149,9 +150,9 @@ func Sync() {
 				}
 				// 发送配置网页
 				if config.Conf.System.Domain != "" {
-					title := "欢迎使用微信机器人"
+					title := "欢迎使用微信机器人(切勿转发)"
 					desc := "点我可以配置功能喔，提示非微信官方网页，点击继续访问即可"
-					url := config.Conf.System.Domain + "/manager.html?id=" + friend.Wxid
+					url := utils.GetManagerUrl(friend.Wxid)
 					utils.SendPublicMsg(friend.Wxid, title, desc, url, 0)
 				}
 			}

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)
 		}
 	}
 }

tasks/tasks.go

@@ -65,6 +65,8 @@ func InitTasks() {
 
 	// 每天0点检查一次处理清理群成员
 	_, _ = s.Cron("0 0 * * *").Do(cleargroupuser.ClearGroupUser)
+	// 每天0点清空AI日使用次数
+	_, _ = s.Cron("0 0 * * *").Do(friends.ClearAiUsedToday)
 
 	// 开启定时任务
 	s.StartAsync()

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

@@ -2,11 +2,14 @@ package utils
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/go-resty/resty/v2"
 	"go-wechat/common/current"
 	"go-wechat/config"
 	"log"
+	"os"
+	"strconv"
 	"time"
 )
 
@@ -15,9 +18,14 @@ import (
 // @param toId
 // @param atId
 // @param msg
-func SendMessage(toId, atId, msg string, retryCount int) {
+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("重试五次失败，停止发送")
 		return
 	}
 
@@ -48,9 +56,10 @@ func SendMessage(toId, atId, msg string, retryCount int) {
 		log.Printf("发送文本消息失败: %s", err.Error())
 		// 休眠五秒后重新发送
 		time.Sleep(5 * time.Second)
-		SendMessage(toId, atId, msg, retryCount+1)
+		return SendMessage(toId, atId, msg, retryCount+1)
 	}
 	log.Printf("发送文本消息结果: %s", resp.String())
+	return
 }
 
 // SendImage
@@ -59,6 +68,10 @@ func SendMessage(toId, atId, msg string, retryCount int) {
 // @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
@@ -91,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
@@ -197,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

@@ -0,0 +1,21 @@
+package utils
+
+import (
+	"encoding/base64"
+	"fmt"
+	"go-wechat/config"
+)
+
+// GetManagerUrl
+// @description: 获取管理页面链接
+// @param wxId
+// @return newUrl
+func GetManagerUrl(wxId string) (url string) {
+	// 生成管理页面链接
+	url = fmt.Sprintf("%s/manager.html?id=%s", config.Conf.System.Domain, wxId)
+	// base64一下
+	encodeString := base64.StdEncoding.EncodeToString([]byte(url))
+	// 拼接新链接（这个是一个已备案的域名）
+	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/friend.html

@@ -94,6 +94,26 @@
                   </label>
                 </dd>
               </div>
+
+              <!-- 每日 AI 免费次数限制 -->
+              <div class="flex justify-between gap-x-4 py-3 items-center">
+                <dt class="text-gray-500">
+                  每日 AI 免费次数
+                </dt>
+                <dd class="flex items-start gap-x-2 items-center">
+                  <div class="relative rounded-md">
+                    <label>
+                      <input type="number" id="auto-ai-{{ .Wxid }}" min="0" class="block w-1/2 float-end rounded-md border-0 py-1.5 text-gray-900 shadow-sm ring-1 ring-inset ring-gray-300 placeholder:text-gray-400 focus:ring-2 focus:ring-inset focus:ring-indigo-600 sm:text-sm sm:leading-6" placeholder="每日 AI 免费次数限制"
+                             value="{{.AiFreeLimit}}"
+                             onblur="changeAiFreeLimit({{.Wxid}}, this.value)"
+                      >
+                    </label>
+                    <div class="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-3">
+                      <span class="text-gray-500 sm:text-sm" id="price-currency">次</span>
+                    </div>
+                  </div>
+                </dd>
+              </div>
               {{ end }}
 
               <div class="flex justify-between gap-x-4 py-3">

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,13 +69,13 @@
                         </select>
                       </label>
                     </div>
+                    <!-- AI角色 -->
                     <div class="float-end mt-1">
                       <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="aiAssistantChange(event, {{.Wxid}})">
-                          <option value="" {{ if eq .Prompt
-                          "" }}selected{{ end }}>默认</option>
+                          <option value="" {{ if eq .Prompt "" }}selected{{ end }}>默认</option>
 
                           {{$usePrompt := .Prompt}}
                           {{ range $.assistant }}
@@ -89,6 +90,26 @@
                   {{ end }}
                 </dd>
               </div>
+              <!-- 每日 AI 免费次数限制 -->
+              <div class="flex justify-between gap-x-4 py-3 items-center">
+                <dt class="text-gray-500">
+                  每日 AI 免费次数限制
+                  <span class="text-red-300">(0表示不限)</span>
+                </dt>
+                <dd class="flex items-start gap-x-2 items-center">
+                  <div class="relative rounded-md">
+                    <label>
+                      <input type="number" id="auto-ai-{{ .Wxid }}" min="0" class="block w-1/2 float-end rounded-md border-0 py-1.5 text-gray-900 shadow-sm ring-1 ring-inset ring-gray-300 placeholder:text-gray-400 focus:ring-2 focus:ring-inset focus:ring-indigo-600 sm:text-sm sm:leading-6" placeholder="每日 AI 免费次数限制"
+                             value="{{.AiFreeLimit}}"
+                             onblur="changeAiFreeLimit({{.Wxid}}, this.value)"
+                      >
+                    </label>
+                    <div class="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-3">
+                      <span class="text-gray-500 sm:text-sm" id="price-currency">次</span>
+                    </div>
+                  </div>
+                </dd>
+              </div>
               <!-- 水群排行榜 -->
               <div class="flex justify-between gap-x-4 py-3 items-center">
                 <dt class="text-gray-500">水群排行榜</dt>

views/manager-one.html

@@ -81,6 +81,18 @@
             {{ end }}
           </dd>
         </div>
+        <!-- 今日 AI 对话已使用次数 -->
+        <div class="flex justify-between gap-x-4 py-3 items-center">
+          <dt class="text-gray-500">
+            今日 AI 对话已使用次数
+            <br/>
+            <span class="text-red-300">* 每天0点重置</span>
+            <p class="text-gray-400">需要扩容请添加微信: _Elixir</p>
+          </dt>
+          <dd class="flex items-start gap-x-2">
+            {{ .info.AiUsedToday }}次{{ if lt 1 .info.AiFreeLimit }}(限制{{ .info.AiFreeLimit }}次){{ end }}
+          </dd>
+        </div>
 
         {{$isGroup := checkIsGroup .info.Wxid}}
         {{ if eq $isGroup true }}

views/static/js/index.js

@@ -291,3 +291,56 @@ function changeClearMember(wxid, oldVal, newVal) {
     window.location.reload();
   })
 }
+
+// 修改每日免费 AI 对话次数
+function changeAiFreeLimit(wxid, limitNumber) {
+  limitNumber = Number(limitNumber)
+
+  if (limitNumber < 0) {
+    alert('值不能小于0')
+  }
+  // 请求接口
+  axios({
+    method: 'post',
+    url: '/api/ai/free',
+    data: {
+      wxid: wxid,
+      days: Number(limitNumber)
+    }
+  }).then(function (response) {
+    console.log(`返回结果: ${JSON.stringify(response)}`);
+    alert(`${response.data}`)
+  }).catch(function (error) {
+    console.log(`错误信息: ${error}`);
+    alert("修改失败")
+  }).finally(function () {
+    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();
+  })
+}