Merge pull request 'hotfix' (#94) from hotfix into main

Reviewed-on: lxh/go-wxhelper#94

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, "操作成功")
+}

config.yaml

@@ -25,6 +25,8 @@ system:
     news: true
     # 热榜
     hotTop: true
+    # 每天免费 AI 次数限制
+    aiFreeLimit: 50
 
 # 微信HOOK配置
 wechat:

config/system.go

@@ -22,4 +22,5 @@ type defaultRule struct {
 	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免费次数
 }

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,8 @@ import (
 	"time"
 )
 
+var notifyMap = make(map[string]bool)
+
 // AI
 // @description: AI消息
 // @param m
@@ -37,6 +39,23 @@ func AI(m *plugin.MessageContext) {
 	if !friendInfo.EnableAi {
 		return
 	}
+	if friendInfo.AiUsedToday > 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 +151,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 +181,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

router/router.go

@@ -26,6 +26,7 @@ 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.PUT("/welcome/status", app.ChangeEnableWelcomeStatus)     // 修改是否开启迎新状态
 	api.PUT("/command/status", app.ChangeEnableCommandStatus)     // 修改是否开启指令状态

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{}).
+		Update("`ai_used_today`", 0).Error
+	if err != nil {
+		log.Printf("清空AI今日使用次数失败, 错误信息: %v", err)
+	}
+}

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

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/send.go

@@ -2,6 +2,7 @@ package utils
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/go-resty/resty/v2"
 	"go-wechat/common/current"
@@ -15,9 +16,10 @@ 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 retryCount > 5 {
 		log.Printf("重试五次失败，停止发送")
+		err = errors.New("重试五次失败，停止发送")
 		return
 	}
 
@@ -48,9 +50,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

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

@@ -73,8 +73,7 @@
                         <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
+                          <option value="" {{ if eq .Prompt "" }}selected{{ end }}>默认</option>
-                          "" }}selected{{ end }}>默认</option>
 
                           {{$usePrompt := .Prompt}}
                           {{ range $.assistant }}
@@ -89,6 +88,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,13 @@
             {{ end }}
           </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">
+            {{ .info.AiUsedToday }}次(限制{{ .info.AiFreeLimit }}次)
+          </dd>
+        </div>
 
         {{$isGroup := checkIsGroup .info.Wxid}}
         {{ if eq $isGroup true }}

views/static/js/index.js

@@ -291,3 +291,29 @@ 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();
+  })
+}