Compare commits
No commits in common. "main" and "v1.2.40" have entirely different histories.
@ -280,26 +280,3 @@ func AutoClearMembers(ctx *gin.Context) {
|
|||||||
|
|
||||||
ctx.String(http.StatusOK, "操作成功")
|
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, "操作成功")
|
|
||||||
}
|
|
||||||
|
56
app/other.go
56
app/other.go
@ -1,56 +0,0 @@
|
|||||||
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, "操作完成")
|
|
||||||
}
|
|
17
config.yaml
17
config.yaml
@ -4,7 +4,7 @@ system:
|
|||||||
alApiToken: xxx
|
alApiToken: xxx
|
||||||
# urlc.cn的Token,用来生成短链接
|
# urlc.cn的Token,用来生成短链接
|
||||||
urlcApiToken: xxx
|
urlcApiToken: xxx
|
||||||
# 系统访问域名,必须是包括 http[s]:// 的完整域名
|
# 系统访问域名
|
||||||
domain: https://wechat.abc.com
|
domain: https://wechat.abc.com
|
||||||
# 添加新好友或群之后通知给指定的人
|
# 添加新好友或群之后通知给指定的人
|
||||||
newFriendNotify:
|
newFriendNotify:
|
||||||
@ -25,13 +25,11 @@ system:
|
|||||||
news: true
|
news: true
|
||||||
# 热榜
|
# 热榜
|
||||||
hotTop: true
|
hotTop: true
|
||||||
# 每天免费 AI 次数限制
|
|
||||||
aiFreeLimit: 50
|
|
||||||
|
|
||||||
# 微信HOOK配置
|
# 微信HOOK配置
|
||||||
wechat:
|
wechat:
|
||||||
# 微信HOOK接口地址
|
# 微信HOOK接口地址
|
||||||
host: 10.0.0.79:19098
|
host: 10.0.0.79:19088
|
||||||
# 微信容器映射出来的vnc页面地址,没有就不填
|
# 微信容器映射出来的vnc页面地址,没有就不填
|
||||||
# vncUrl: http://192.168.1.175:19087/vnc_lite.html
|
# vncUrl: http://192.168.1.175:19087/vnc_lite.html
|
||||||
# 是否在启动的时候自动设置hook服务的回调
|
# 是否在启动的时候自动设置hook服务的回调
|
||||||
@ -61,9 +59,9 @@ task:
|
|||||||
hotTop:
|
hotTop:
|
||||||
enable: true
|
enable: true
|
||||||
cron: '0 */1 * * *' # 每小时一次
|
cron: '0 */1 * * *' # 每小时一次
|
||||||
syncFriends: # 逻辑已修改,每天大半夜同步一次即可(主要是用以修正数据),平时会根据收到的消息针对性同步
|
syncFriends:
|
||||||
enable: false
|
enable: false
|
||||||
cron: '0 4 * * *' # 每天4:00
|
cron: '*/5 * * * *' # 五分钟一次
|
||||||
groupSummary:
|
groupSummary:
|
||||||
enable: false
|
enable: false
|
||||||
cron: '30 0 * * *' # 每天0:30
|
cron: '30 0 * * *' # 每天0:30
|
||||||
@ -81,7 +79,7 @@ mq:
|
|||||||
enable: false
|
enable: false
|
||||||
# RabbitMQ配置
|
# RabbitMQ配置
|
||||||
rabbitmq:
|
rabbitmq:
|
||||||
host: 10.0.0.31
|
host: 10.0.0.247
|
||||||
port: 5672
|
port: 5672
|
||||||
user: wechat
|
user: wechat
|
||||||
password: wechat123
|
password: wechat123
|
||||||
@ -102,7 +100,6 @@ ai:
|
|||||||
# 人设
|
# 人设
|
||||||
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
|
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
|
||||||
models:
|
models:
|
||||||
- name: 默认
|
|
||||||
- name: ChatGPT-4
|
- name: ChatGPT-4
|
||||||
model: gpt-4
|
model: gpt-4
|
||||||
canManager: false
|
canManager: false
|
||||||
@ -140,7 +137,3 @@ resource:
|
|||||||
wordcloud:
|
wordcloud:
|
||||||
type: image
|
type: image
|
||||||
path: D:\Share\wordcloud\%s
|
path: D:\Share\wordcloud\%s
|
||||||
# 邀请入群消息回复
|
|
||||||
invitation-join-group:
|
|
||||||
type: text
|
|
||||||
path: "您的邀请消息已收到啦,正在通知我的主人来同意请求。在我加群之后将会进行初始化操作,直到收到我主动发送的消息就是初始化完成咯,在那之前请耐心等待喔~"
|
|
||||||
|
@ -16,11 +16,10 @@ type newFriendNotify struct {
|
|||||||
|
|
||||||
// 默认规则
|
// 默认规则
|
||||||
type defaultRule struct {
|
type defaultRule struct {
|
||||||
Ai bool `json:"ai" yaml:"ai"` // 是否启用AI
|
Ai bool `json:"ai" yaml:"ai"` // 是否启用AI
|
||||||
ChatRank bool `json:"chatRank" yaml:"chatRank"` // 是否启用聊天排行榜
|
ChatRank bool `json:"chatRank" yaml:"chatRank"` // 是否启用聊天排行榜
|
||||||
Summary bool `json:"summary" yaml:"summary"` // 是否启用聊天总结
|
Summary bool `json:"summary" yaml:"summary"` // 是否启用聊天总结
|
||||||
Welcome bool `json:"welcome" yaml:"welcome"` // 是否启用欢迎新成员
|
Welcome bool `json:"welcome" yaml:"welcome"` // 是否启用欢迎新成员
|
||||||
News bool `json:"news" yaml:"news"` // 是否启用每日早报
|
News bool `json:"news" yaml:"news"` // 是否启用每日早报
|
||||||
HotTop bool `json:"hotTop" yaml:"hotTop"` // 是否启用热门话题
|
HotTop bool `json:"hotTop" yaml:"hotTop"` // 是否启用热门话题
|
||||||
AiFreeLimit int `json:"aiFreeLimit" yaml:"aiFreeLimit"` // AI免费次数
|
|
||||||
}
|
}
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
plugin "go-wechat/plugin"
|
plugin "go-wechat/plugin"
|
||||||
"go-wechat/plugin/plugins"
|
"go-wechat/plugin/plugins"
|
||||||
"go-wechat/service"
|
"go-wechat/service"
|
||||||
|
"go-wechat/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Plugin
|
// Plugin
|
||||||
@ -26,15 +27,15 @@ func Plugin() {
|
|||||||
// 通知邀请入群消息到配置用户
|
// 通知邀请入群消息到配置用户
|
||||||
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
||||||
flag, _ := m.IsInvitationJoinGroup()
|
flag, _ := m.IsInvitationJoinGroup()
|
||||||
return flag && !m.IsGroup()
|
return flag
|
||||||
}, plugins.NotifyInvitationJoinGroup)
|
}, plugins.NotifyInvitationJoinGroup)
|
||||||
// 被移除群聊通知到配置用户
|
// 被移除群聊通知到配置用户
|
||||||
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
||||||
return m.IsRemoveFromChatroom()
|
return m.Type == types.MsgTypeSys
|
||||||
}, plugins.NotifyRemoveFromChatroom)
|
}, plugins.NotifyRemoveFromChatroom)
|
||||||
// 响应好友添加成功消息
|
// 响应好友添加成功消息
|
||||||
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
dispatcher.RegisterHandler(func(m *dto.Message) bool {
|
||||||
return m.IsNewFriendAdd() || m.IsJoinToGroup() || m.IsOldFriendBack()
|
return m.Type == types.MsgTypeSys
|
||||||
}, plugins.ReplyNewFriend)
|
}, plugins.ReplyNewFriend)
|
||||||
|
|
||||||
// 私聊指令消息
|
// 私聊指令消息
|
||||||
|
9
main.go
9
main.go
@ -12,8 +12,6 @@ import (
|
|||||||
"html/template"
|
"html/template"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@ -24,13 +22,6 @@ func init() {
|
|||||||
initialization.Plugin() // 注册插件
|
initialization.Plugin() // 注册插件
|
||||||
tasks.InitTasks() // 初始化定时任务
|
tasks.InitTasks() // 初始化定时任务
|
||||||
mq.Init() // 初始化MQ
|
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() {
|
func main() {
|
||||||
|
@ -219,62 +219,3 @@ func (m Message) IsInvitationJoinGroup() (flag bool, str string) {
|
|||||||
}
|
}
|
||||||
return
|
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, "\"修改群名为“")
|
|
||||||
}
|
|
||||||
|
@ -25,8 +25,6 @@ type Friend struct {
|
|||||||
ClearMember int `json:"clearMember"` // 清理成员配置(多少天未活跃的)
|
ClearMember int `json:"clearMember"` // 清理成员配置(多少天未活跃的)
|
||||||
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常
|
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常
|
||||||
UsedTokens int `json:"usedTokens"` // 已使用的AI Token数量
|
UsedTokens int `json:"usedTokens"` // 已使用的AI Token数量
|
||||||
AiFreeLimit int `json:"aiFreeLimit"` // AI免费次数
|
|
||||||
AiUsedToday int `json:"aiUsedToday"` // 今日已使用的AI次数
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Friend) TableName() string {
|
func (Friend) TableName() string {
|
||||||
|
@ -24,8 +24,6 @@ type FriendItem struct {
|
|||||||
EnableHotTop bool // 是否启用热搜
|
EnableHotTop bool // 是否启用热搜
|
||||||
ClearMember int // 清理成员配置(多少天未活跃的)
|
ClearMember int // 清理成员配置(多少天未活跃的)
|
||||||
IsOk bool // 是否还在通讯库(群聊是要还在群里也算)
|
IsOk bool // 是否还在通讯库(群聊是要还在群里也算)
|
||||||
AiFreeLimit int // AI免费次数
|
|
||||||
AiUsedToday int // 今日已使用的AI次数
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GroupUserItem
|
// GroupUserItem
|
||||||
|
@ -19,12 +19,6 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
// 已经通知过的群组或者好友map
|
|
||||||
var notifyMap = make(map[string]bool)
|
|
||||||
|
|
||||||
// 拉取最近消息条数
|
|
||||||
const fetchMessageCount = 20
|
|
||||||
|
|
||||||
// AI
|
// AI
|
||||||
// @description: AI消息
|
// @description: AI消息
|
||||||
// @param m
|
// @param m
|
||||||
@ -43,23 +37,6 @@ func AI(m *plugin.MessageContext) {
|
|||||||
if !friendInfo.EnableAi {
|
if !friendInfo.EnableAi {
|
||||||
return
|
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(`@([^ | ]+)`)
|
re := regexp.MustCompile(`@([^ | ]+)`)
|
||||||
@ -155,17 +132,17 @@ func AI(m *plugin.MessageContext) {
|
|||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("OpenAI聊天发起失败: %v", err.Error())
|
log.Printf("OpenAI聊天发起失败: %v", err.Error())
|
||||||
_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI聊天初始化失败,我已经通知我主人来修啦,请稍候一下下喔~", 0)
|
utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 返回消息为空
|
// 返回消息为空
|
||||||
if len(resp.Choices) == 0 || resp.Choices[0].Message.Content == "" {
|
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
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 异步更新一下已使用的AI tokens
|
// 异步更新一下已使用的AI次数
|
||||||
go service.UpdateUsedAiTokens(m.FromUser, resp.Usage.TotalTokens)
|
go service.UpdateUsedAiTokens(m.FromUser, resp.Usage.TotalTokens)
|
||||||
|
|
||||||
// 保存一下AI 返回的消息,消息 Id 使用传入 Id 的负数
|
// 保存一下AI 返回的消息,消息 Id 使用传入 Id 的负数
|
||||||
@ -185,7 +162,7 @@ func AI(m *plugin.MessageContext) {
|
|||||||
if m.GroupUser != "" {
|
if m.GroupUser != "" {
|
||||||
replyMsg = "\n" + resp.Choices[0].Message.Content
|
replyMsg = "\n" + resp.Choices[0].Message.Content
|
||||||
}
|
}
|
||||||
err = utils.SendMessage(m.FromUser, m.GroupUser, replyMsg, 0)
|
utils.SendMessage(m.FromUser, m.GroupUser, replyMsg, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
// getGroupUserMessages
|
// getGroupUserMessages
|
||||||
@ -202,7 +179,7 @@ func getGroupUserMessages(msgId int64, groupId, groupUserId string) (records []e
|
|||||||
Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
|
Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
|
||||||
Where(subQuery).
|
Where(subQuery).
|
||||||
Order("create_at desc").
|
Order("create_at desc").
|
||||||
Limit(fetchMessageCount).Find(&records)
|
Limit(4).Find(&records)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -218,6 +195,6 @@ func getUserPrivateMessages(userId string) (records []entity.Message) {
|
|||||||
Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
|
Where("create_at >= DATE_SUB(NOW(),INTERVAL 30 MINUTE)").
|
||||||
Where(subQuery).
|
Where(subQuery).
|
||||||
Order("create_at desc").
|
Order("create_at desc").
|
||||||
Limit(fetchMessageCount).Find(&records)
|
Limit(4).Find(&records)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
@ -2,41 +2,28 @@ package plugins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"go-wechat/client"
|
|
||||||
"go-wechat/config"
|
"go-wechat/config"
|
||||||
"go-wechat/model/entity"
|
|
||||||
"go-wechat/plugin"
|
"go-wechat/plugin"
|
||||||
"go-wechat/service"
|
"go-wechat/service"
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
// NotifyInvitationJoinGroup
|
// NotifyInvitationJoinGroup
|
||||||
// @description: 通知邀请入群消息到配置用户
|
// @description: 通知邀请入群消息到配置用户
|
||||||
// @param m
|
// @param m
|
||||||
func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
|
func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
|
||||||
// 先回复一条固定消息
|
// 先回复一条固定句子
|
||||||
if conf, e := config.Conf.Resource["invitation-join-group"]; e {
|
utils.SendMessage(m.FromUser, m.GroupUser, "您的邀请消息已收到啦,正在通知我的主人来同意请求。在我加群之后将会进行初始化操作,直到收到我主动发送的消息就是初始化完成咯,在那之前请耐心等待喔~", 0)
|
||||||
// 发送一条新消息
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 推送到配置的用户
|
// 如果是邀请进群,推送到配置的用户
|
||||||
_, dec := m.IsInvitationJoinGroup()
|
if flag, dec := m.IsInvitationJoinGroup(); flag {
|
||||||
for _, user := range config.Conf.System.NewFriendNotify.ToUser {
|
for _, user := range config.Conf.System.NewFriendNotify.ToUser {
|
||||||
if user != "" {
|
if user != "" {
|
||||||
// 发送一条新消息
|
// 发送一条新消息
|
||||||
dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec)
|
dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec)
|
||||||
_ = utils.SendMessage(user, "", dec, 0)
|
utils.SendMessage(user, "", dec, 0)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -45,24 +32,24 @@ func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
|
|||||||
// @description: 被移除群聊通知到配置用户
|
// @description: 被移除群聊通知到配置用户
|
||||||
// @param m
|
// @param m
|
||||||
func NotifyRemoveFromChatroom(m *plugin.MessageContext) {
|
func NotifyRemoveFromChatroom(m *plugin.MessageContext) {
|
||||||
// 调用一下退出群聊接口,防止被移出后还能从好友列表接口拉到相关信息
|
// 如果是被移出群聊,推送到配置的用户
|
||||||
utils.QuitChatroom(m.FromUser, 0)
|
if strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊") {
|
||||||
|
// 调用一下退出群聊接口,防止被移出后还能从好友列表接口拉到相关信息
|
||||||
|
utils.QuitChatroom(m.FromUser, 0)
|
||||||
|
|
||||||
// 取出群名称
|
// 取出群名称
|
||||||
groupInfo, err := service.GetFriendInfoById(m.FromUser)
|
groupInfo, err := service.GetFriendInfoById(m.FromUser)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 组装消息
|
// 组装消息
|
||||||
msg := fmt.Sprintf("#移除群聊提醒\n\n群Id: %s\n群名称: %s\n事件: %s", m.FromUser, groupInfo.Nickname, m.Content)
|
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 {
|
for _, user := range config.Conf.System.NewFriendNotify.ToUser {
|
||||||
if user != "" {
|
if user != "" {
|
||||||
// 发送一条新消息
|
// 发送一条新消息
|
||||||
_ = utils.SendMessage(user, "", msg, 0)
|
utils.SendMessage(user, "", msg, 0)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 修改is_ok状态
|
|
||||||
client.MySQL.Model(&entity.Friend{}).Where("wxid = ?", m.FromUser).Update("is_ok", false)
|
|
||||||
}
|
}
|
||||||
|
@ -2,30 +2,17 @@ package plugins
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"go-wechat/plugin"
|
"go-wechat/plugin"
|
||||||
"go-wechat/service"
|
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
"time"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
// ReplyNewFriend
|
// ReplyNewFriend
|
||||||
// @description: 响应好友添加成功消息
|
// @description: 响应好友添加成功消息
|
||||||
// @param m
|
// @param m
|
||||||
func ReplyNewFriend(m *plugin.MessageContext) {
|
func ReplyNewFriend(m *plugin.MessageContext) {
|
||||||
if m.IsNewFriendAdd() || m.IsJoinToGroup() {
|
isNewFriend := strings.HasPrefix(m.Content, "你已添加了") && strings.HasSuffix(m.Content, ",现在可以开始聊天了。")
|
||||||
_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化,请稍等几分钟,初始化完成之后我将主动告知您。", 0)
|
isNewChatroom := strings.Contains(m.Content, "\"邀请你加入了群聊,群聊参与人还有:")
|
||||||
|
if isNewFriend || isNewChatroom {
|
||||||
|
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)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
}
|
||||||
|
@ -5,7 +5,6 @@ import (
|
|||||||
"go-wechat/config"
|
"go-wechat/config"
|
||||||
"go-wechat/model/entity"
|
"go-wechat/model/entity"
|
||||||
"go-wechat/plugin"
|
"go-wechat/plugin"
|
||||||
"go-wechat/service"
|
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -29,7 +28,7 @@ func WelcomeNew(m *plugin.MessageContext) {
|
|||||||
switch conf.Type {
|
switch conf.Type {
|
||||||
case "text":
|
case "text":
|
||||||
// 文字类型
|
// 文字类型
|
||||||
_ = utils.SendMessage(m.FromUser, "", conf.Path, 0)
|
utils.SendMessage(m.FromUser, "", conf.Path, 0)
|
||||||
case "image":
|
case "image":
|
||||||
// 图片类型
|
// 图片类型
|
||||||
utils.SendImage(m.FromUser, conf.Path, 0)
|
utils.SendImage(m.FromUser, conf.Path, 0)
|
||||||
@ -37,7 +36,4 @@ func WelcomeNew(m *plugin.MessageContext) {
|
|||||||
// 表情类型
|
// 表情类型
|
||||||
utils.SendEmotion(m.FromUser, conf.Path, 0)
|
utils.SendEmotion(m.FromUser, conf.Path, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 刷新群成员列表
|
|
||||||
go service.SyncGroupMembers(m.FromUser)
|
|
||||||
}
|
}
|
||||||
|
@ -26,9 +26,7 @@ func Init(g *gin.Engine) {
|
|||||||
api := g.Group("/api")
|
api := g.Group("/api")
|
||||||
api.PUT("/ai/status", app.ChangeEnableAiStatus) // 修改是否开启AI状态
|
api.PUT("/ai/status", app.ChangeEnableAiStatus) // 修改是否开启AI状态
|
||||||
api.POST("/ai/model", app.ChangeUseAiModel) // 修改使用的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/assistant", app.ChangeUseAiAssistant) // 修改使用的AI助手
|
||||||
api.POST("/ai/summary", app.SendAiSummary) // 手动发送AI聊天总结
|
|
||||||
api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态
|
api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态
|
||||||
api.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态
|
api.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态
|
||||||
api.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报状态
|
api.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报状态
|
||||||
|
@ -152,25 +152,3 @@ func UpdateUsedAiTokens(wxId string, tokens int) {
|
|||||||
log.Printf("更新AI使用次数失败, 错误信息: %v", err)
|
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
@ -1,257 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -3,7 +3,6 @@ package cleargroupuser
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"go-wechat/client"
|
"go-wechat/client"
|
||||||
"go-wechat/common/current"
|
|
||||||
"go-wechat/model/entity"
|
"go-wechat/model/entity"
|
||||||
"go-wechat/service"
|
"go-wechat/service"
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
@ -16,7 +15,7 @@ import (
|
|||||||
func ClearGroupUser() {
|
func ClearGroupUser() {
|
||||||
groups, err := service.GetAllEnableClearGroup()
|
groups, err := service.GetAllEnableClearGroup()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("获取启用了末位淘汰的群组失败, 错误信息: %v", err)
|
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -57,9 +56,7 @@ func ClearGroupUser() {
|
|||||||
// @param days 需要清理的未活跃的天数
|
// @param days 需要清理的未活跃的天数
|
||||||
// @return members
|
// @return members
|
||||||
func getNeedDeleteMembers(groupId string, days int) (members []entity.GroupUser) {
|
func getNeedDeleteMembers(groupId string, days int) (members []entity.GroupUser) {
|
||||||
err := client.MySQL.Model(&entity.GroupUser{}).
|
err := client.MySQL.Model(&entity.GroupUser{}).Where("group_id = ?", groupId).
|
||||||
Where("group_id = ?", groupId).
|
|
||||||
Where("wxid != ?", current.GetRobotInfo().WxId).
|
|
||||||
Where("is_member IS TRUE").
|
Where("is_member IS TRUE").
|
||||||
Where("DATEDIFF( NOW(), last_active ) >= ?", days).
|
Where("DATEDIFF( NOW(), last_active ) >= ?", days).
|
||||||
Order("last_active DESC").
|
Order("last_active DESC").
|
||||||
|
@ -1,9 +0,0 @@
|
|||||||
package friends
|
|
||||||
|
|
||||||
import "go-wechat/service"
|
|
||||||
|
|
||||||
// ClearAiUsedToday
|
|
||||||
// @description: 清空AI日使用次数
|
|
||||||
func ClearAiUsedToday() {
|
|
||||||
service.ClearAiUsedToday()
|
|
||||||
}
|
|
@ -42,19 +42,6 @@ func Sync() {
|
|||||||
|
|
||||||
nowIds := []string{}
|
nowIds := []string{}
|
||||||
|
|
||||||
// 取出已存在的成员信息
|
|
||||||
var oldData []entity.Friend
|
|
||||||
err = tx.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 newItmes = make(map[string]string)
|
var newItmes = make(map[string]string)
|
||||||
|
|
||||||
@ -70,7 +57,12 @@ func Sync() {
|
|||||||
nowIds = append(nowIds, friend.Wxid)
|
nowIds = append(nowIds, friend.Wxid)
|
||||||
|
|
||||||
// 判断是否存在,不存在的话就新增,存在就修改一下名字
|
// 判断是否存在,不存在的话就新增,存在就修改一下名字
|
||||||
if _, e := oldMap[friend.Wxid]; !e {
|
var count int64
|
||||||
|
err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Count(&count).Error
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if count == 0 {
|
||||||
// 新增
|
// 新增
|
||||||
err = tx.Create(&entity.Friend{
|
err = tx.Create(&entity.Friend{
|
||||||
CustomAccount: friend.CustomAccount,
|
CustomAccount: friend.CustomAccount,
|
||||||
@ -85,7 +77,6 @@ func Sync() {
|
|||||||
EnableWelcome: config.Conf.System.DefaultRule.Welcome,
|
EnableWelcome: config.Conf.System.DefaultRule.Welcome,
|
||||||
EnableNews: config.Conf.System.DefaultRule.News,
|
EnableNews: config.Conf.System.DefaultRule.News,
|
||||||
EnableHotTop: config.Conf.System.DefaultRule.HotTop,
|
EnableHotTop: config.Conf.System.DefaultRule.HotTop,
|
||||||
AiFreeLimit: config.Conf.System.DefaultRule.AiFreeLimit,
|
|
||||||
ClearMember: 0,
|
ClearMember: 0,
|
||||||
LastActive: types.DateTime(time.Now().Local()),
|
LastActive: types.DateTime(time.Now().Local()),
|
||||||
}).Error
|
}).Error
|
||||||
@ -110,9 +101,9 @@ func Sync() {
|
|||||||
}
|
}
|
||||||
// 发送配置网页
|
// 发送配置网页
|
||||||
if config.Conf.System.Domain != "" {
|
if config.Conf.System.Domain != "" {
|
||||||
title := "欢迎使用微信机器人(切勿转发)"
|
title := "欢迎使用微信机器人"
|
||||||
desc := "点我可以配置功能喔,提示非微信官方网页,点击继续访问即可"
|
desc := "点我可以配置功能喔,提示非微信官方网页,点击继续访问即可"
|
||||||
url := utils.GetManagerUrl(friend.Wxid)
|
url := config.Conf.System.Domain + "/manager.html?id=" + friend.Wxid
|
||||||
utils.SendPublicMsg(friend.Wxid, title, desc, url, 0)
|
utils.SendPublicMsg(friend.Wxid, title, desc, url, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -129,33 +120,6 @@ func Sync() {
|
|||||||
log.Printf("修改好友失败: %s", err.Error())
|
log.Printf("修改好友失败: %s", err.Error())
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
// 如果已存在但是是已退出的群,也通知一下
|
|
||||||
if !oldMap[friend.Wxid] {
|
|
||||||
newItmes[friend.Wxid] = friend.Nickname
|
|
||||||
|
|
||||||
// 通知一下,初始化完成
|
|
||||||
if conf, ok := config.Conf.Resource["introduce"]; ok {
|
|
||||||
// 发送一条新消息
|
|
||||||
switch conf.Type {
|
|
||||||
case "text":
|
|
||||||
// 文字类型
|
|
||||||
utils.SendMessage(friend.Wxid, "", conf.Path, 0)
|
|
||||||
case "image":
|
|
||||||
// 图片类型
|
|
||||||
utils.SendImage(friend.Wxid, conf.Path, 0)
|
|
||||||
case "emotion":
|
|
||||||
// 表情类型
|
|
||||||
utils.SendEmotion(friend.Wxid, conf.Path, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// 发送配置网页
|
|
||||||
if config.Conf.System.Domain != "" {
|
|
||||||
title := "欢迎使用微信机器人(切勿转发)"
|
|
||||||
desc := "点我可以配置功能喔,提示非微信官方网页,点击继续访问即可"
|
|
||||||
url := utils.GetManagerUrl(friend.Wxid)
|
|
||||||
utils.SendPublicMsg(friend.Wxid, title, desc, url, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 群成员,同步一下成员信息
|
// 群成员,同步一下成员信息
|
||||||
|
@ -1,12 +1,15 @@
|
|||||||
package summary
|
package summary
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"github.com/sashabaranov/go-openai"
|
||||||
|
"go-wechat/config"
|
||||||
"go-wechat/model/vo"
|
"go-wechat/model/vo"
|
||||||
"go-wechat/service"
|
"go-wechat/service"
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
"log"
|
"log"
|
||||||
"time"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
// AiSummary
|
// AiSummary
|
||||||
@ -19,9 +22,6 @@ func AiSummary() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, group := range groups {
|
for _, group := range groups {
|
||||||
// 记录开始时间
|
|
||||||
var start = time.Now()
|
|
||||||
|
|
||||||
// 获取对话记录
|
// 获取对话记录
|
||||||
var records []vo.TextMessageItem
|
var records []vo.TextMessageItem
|
||||||
if records, err = service.GetTextMessagesById(group.Wxid); err != nil {
|
if records, err = service.GetTextMessagesById(group.Wxid); err != nil {
|
||||||
@ -33,21 +33,56 @@ func AiSummary() {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
// 组装对话记录为字符串
|
// 组装对话记录为字符串
|
||||||
var replyMsg string
|
var content []string
|
||||||
replyMsg, err = utils.GetAiSummary(group.Nickname, records)
|
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,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("群聊记录总结失败: %v", err.Error())
|
log.Printf("群聊记录总结失败: %v", err.Error())
|
||||||
_ = utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败,错误信息: "+err.Error(), 0)
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了,让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg)
|
// 返回消息为空
|
||||||
//log.Printf("群[%s]对话记录总结成功,总结内容: %s", group.Wxid, replyMsg)
|
if resp.Choices[0].Message.Content == "" {
|
||||||
_ = utils.SendMessage(group.Wxid, "", replyMsg, 0)
|
continue
|
||||||
|
|
||||||
// 判断耗时是否达到一分钟,不足就等待
|
|
||||||
if used := time.Now().Sub(start); used < time.Minute {
|
|
||||||
time.Sleep(time.Minute - used)
|
|
||||||
}
|
}
|
||||||
|
replyMsg := fmt.Sprintf("#昨日消息总结\n又是一天过去了,让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", resp.Choices[0].Message.Content)
|
||||||
|
//log.Printf("群[%s]对话记录总结成功,总结内容: %s", group.Wxid, replyMsg)
|
||||||
|
utils.SendMessage(group.Wxid, "", replyMsg, 0)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -65,8 +65,6 @@ func InitTasks() {
|
|||||||
|
|
||||||
// 每天0点检查一次处理清理群成员
|
// 每天0点检查一次处理清理群成员
|
||||||
_, _ = s.Cron("0 0 * * *").Do(cleargroupuser.ClearGroupUser)
|
_, _ = s.Cron("0 0 * * *").Do(cleargroupuser.ClearGroupUser)
|
||||||
// 每天0点清空AI日使用次数
|
|
||||||
_, _ = s.Cron("0 0 * * *").Do(friends.ClearAiUsedToday)
|
|
||||||
|
|
||||||
// 开启定时任务
|
// 开启定时任务
|
||||||
s.StartAsync()
|
s.StartAsync()
|
||||||
|
105
utils/friend.go
105
utils/friend.go
@ -1,105 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
@ -2,14 +2,11 @@ package utils
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-resty/resty/v2"
|
"github.com/go-resty/resty/v2"
|
||||||
"go-wechat/common/current"
|
"go-wechat/common/current"
|
||||||
"go-wechat/config"
|
"go-wechat/config"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -18,14 +15,9 @@ import (
|
|||||||
// @param toId
|
// @param toId
|
||||||
// @param atId
|
// @param atId
|
||||||
// @param msg
|
// @param msg
|
||||||
func SendMessage(toId, atId, msg string, retryCount int) (err error) {
|
func SendMessage(toId, atId, msg string, retryCount int) {
|
||||||
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if retryCount > 5 {
|
if retryCount > 5 {
|
||||||
log.Printf("重试五次失败,停止发送")
|
log.Printf("重试五次失败,停止发送")
|
||||||
err = errors.New("重试五次失败,停止发送")
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -56,10 +48,9 @@ func SendMessage(toId, atId, msg string, retryCount int) (err error) {
|
|||||||
log.Printf("发送文本消息失败: %s", err.Error())
|
log.Printf("发送文本消息失败: %s", err.Error())
|
||||||
// 休眠五秒后重新发送
|
// 休眠五秒后重新发送
|
||||||
time.Sleep(5 * time.Second)
|
time.Sleep(5 * time.Second)
|
||||||
return SendMessage(toId, atId, msg, retryCount+1)
|
SendMessage(toId, atId, msg, retryCount+1)
|
||||||
}
|
}
|
||||||
log.Printf("发送文本消息结果: %s", resp.String())
|
log.Printf("发送文本消息结果: %s", resp.String())
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendImage
|
// SendImage
|
||||||
@ -68,10 +59,6 @@ func SendMessage(toId, atId, msg string, retryCount int) (err error) {
|
|||||||
// @param imgPath string 图片路径
|
// @param imgPath string 图片路径
|
||||||
// @param retryCount int 重试次数
|
// @param retryCount int 重试次数
|
||||||
func SendImage(toId, imgPath string, retryCount int) {
|
func SendImage(toId, imgPath string, retryCount int) {
|
||||||
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if retryCount > 5 {
|
if retryCount > 5 {
|
||||||
log.Printf("重试五次失败,停止发送")
|
log.Printf("重试五次失败,停止发送")
|
||||||
return
|
return
|
||||||
@ -104,10 +91,6 @@ func SendImage(toId, imgPath string, retryCount int) {
|
|||||||
// @param emotionHash string 表情包hash(md5值)
|
// @param emotionHash string 表情包hash(md5值)
|
||||||
// @param retryCount int 重试次数
|
// @param retryCount int 重试次数
|
||||||
func SendEmotion(toId, emotionHash string, retryCount int) {
|
func SendEmotion(toId, emotionHash string, retryCount int) {
|
||||||
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if retryCount > 5 {
|
if retryCount > 5 {
|
||||||
log.Printf("重试五次失败,停止发送")
|
log.Printf("重试五次失败,停止发送")
|
||||||
return
|
return
|
||||||
@ -214,10 +197,6 @@ func QuitChatroom(chatRoomId string, retryCount int) {
|
|||||||
// @param url string 链接
|
// @param url string 链接
|
||||||
// @param retryCount int 重试次数
|
// @param retryCount int 重试次数
|
||||||
func SendPublicMsg(wxId, title, digest, url string, retryCount int) {
|
func SendPublicMsg(wxId, title, digest, url string, retryCount int) {
|
||||||
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if retryCount > 5 {
|
if retryCount > 5 {
|
||||||
log.Printf("重试五次失败,停止发送")
|
log.Printf("重试五次失败,停止发送")
|
||||||
return
|
return
|
||||||
|
108
utils/summary.go
108
utils/summary.go
@ -1,108 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
21
utils/url.go
21
utils/url.go
@ -1,21 +0,0 @@
|
|||||||
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
|
|
||||||
}
|
|
@ -57,9 +57,6 @@
|
|||||||
|
|
||||||
<!-- 消息总结 -->
|
<!-- 消息总结 -->
|
||||||
{{define "summary"}}
|
{{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"
|
<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"
|
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}})">
|
role="switch" aria-checked="false" onclick="changeSummaryEnableStatus({{.Wxid}})">
|
||||||
|
@ -94,26 +94,6 @@
|
|||||||
</label>
|
</label>
|
||||||
</dd>
|
</dd>
|
||||||
</div>
|
</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 }}
|
{{ end }}
|
||||||
|
|
||||||
<div class="flex justify-between gap-x-4 py-3">
|
<div class="flex justify-between gap-x-4 py-3">
|
||||||
|
@ -56,7 +56,6 @@
|
|||||||
</div>
|
</div>
|
||||||
{{ if eq .EnableAi true }}
|
{{ if eq .EnableAi true }}
|
||||||
<div class="float-end">
|
<div class="float-end">
|
||||||
<!-- 模型 -->
|
|
||||||
<div>
|
<div>
|
||||||
<label>
|
<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}})">
|
<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}})">
|
||||||
@ -69,13 +68,13 @@
|
|||||||
</select>
|
</select>
|
||||||
</label>
|
</label>
|
||||||
</div>
|
</div>
|
||||||
<!-- AI角色 -->
|
|
||||||
<div class="float-end mt-1">
|
<div class="float-end mt-1">
|
||||||
<label>
|
<label>
|
||||||
<select
|
<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"
|
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}})">
|
onchange="aiAssistantChange(event, {{.Wxid}})">
|
||||||
<option value="" {{ if eq .Prompt "" }}selected{{ end }}>默认</option>
|
<option value="" {{ if eq .Prompt
|
||||||
|
"" }}selected{{ end }}>默认</option>
|
||||||
|
|
||||||
{{$usePrompt := .Prompt}}
|
{{$usePrompt := .Prompt}}
|
||||||
{{ range $.assistant }}
|
{{ range $.assistant }}
|
||||||
@ -90,26 +89,6 @@
|
|||||||
{{ end }}
|
{{ end }}
|
||||||
</dd>
|
</dd>
|
||||||
</div>
|
</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">
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
<dt class="text-gray-500">水群排行榜</dt>
|
<dt class="text-gray-500">水群排行榜</dt>
|
||||||
|
@ -81,39 +81,19 @@
|
|||||||
{{ end }}
|
{{ end }}
|
||||||
</dd>
|
</dd>
|
||||||
</div>
|
</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}}
|
{{$isGroup := checkIsGroup .info.Wxid}}
|
||||||
{{ if eq $isGroup true }}
|
{{ if eq $isGroup true }}
|
||||||
<!-- 水群排行榜 -->
|
<!-- 水群排行榜 -->
|
||||||
<div class="flex justify-between gap-x-4 py-3 items-center">
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
<dt class="text-gray-500">
|
<dt class="text-gray-500">水群排行榜</dt>
|
||||||
水群排行榜
|
|
||||||
<br/>
|
|
||||||
<span class="text-red-300">* 每天早上七点半定时发送</span>
|
|
||||||
</dt>
|
|
||||||
<dd class="flex items-start gap-x-2">
|
<dd class="flex items-start gap-x-2">
|
||||||
{{ template "chatRank" .info }}
|
{{ template "chatRank" .info }}
|
||||||
</dd>
|
</dd>
|
||||||
</div>
|
</div>
|
||||||
<!-- 群聊总结 -->
|
<!-- 群聊总结 -->
|
||||||
<div class="flex justify-between gap-x-4 py-3 items-center">
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
<dt class="text-gray-500">
|
<dt class="text-gray-500">群聊总结</dt>
|
||||||
群聊总结
|
|
||||||
<br/>
|
|
||||||
<span class="text-red-300">* 每天0点30定时发送</span>
|
|
||||||
</dt>
|
|
||||||
<dd class="flex items-start gap-x-2">
|
<dd class="flex items-start gap-x-2">
|
||||||
{{ template "summary" .info }}
|
{{ template "summary" .info }}
|
||||||
</dd>
|
</dd>
|
||||||
@ -128,26 +108,18 @@
|
|||||||
{{ end }}
|
{{ end }}
|
||||||
<!-- 早报 -->
|
<!-- 早报 -->
|
||||||
<div class="flex justify-between gap-x-4 py-3 items-center">
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
<dt class="text-gray-500">
|
<dt class="text-gray-500">早报</dt>
|
||||||
早报
|
|
||||||
<br/>
|
|
||||||
<span class="text-red-300">* 每天早上八点定时发送</span>
|
|
||||||
</dt>
|
|
||||||
<dd class="flex items-start gap-x-2">
|
<dd class="flex items-start gap-x-2">
|
||||||
{{ template "news" .info }}
|
{{ template "news" .info }}
|
||||||
</dd>
|
</dd>
|
||||||
</div>
|
</div>
|
||||||
<!-- 热榜 -->
|
<!-- 热榜 -->
|
||||||
<!-- <div class="flex justify-between gap-x-4 py-3 items-center">-->
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
<!-- <dt class="text-gray-500">-->
|
<dt class="text-gray-500">热榜</dt>
|
||||||
<!-- 热榜-->
|
<dd class="flex items-start gap-x-2">
|
||||||
<!-- <br/>-->
|
{{ template "hotTop" .info }}
|
||||||
<!-- <span class="text-red-300">* 每个整点定时发送</span>-->
|
</dd>
|
||||||
<!-- </dt>-->
|
</div>
|
||||||
<!-- <dd class="flex items-start gap-x-2">-->
|
|
||||||
<!-- {{ template "hotTop" .info }}-->
|
|
||||||
<!-- </dd>-->
|
|
||||||
<!-- </div>-->
|
|
||||||
{{ if eq $isGroup true }}
|
{{ if eq $isGroup true }}
|
||||||
<!-- 自动清理不活跃成员 -->
|
<!-- 自动清理不活跃成员 -->
|
||||||
<div class="flex justify-between gap-x-4 py-3 items-center">
|
<div class="flex justify-between gap-x-4 py-3 items-center">
|
||||||
|
@ -291,56 +291,3 @@ function changeClearMember(wxid, oldVal, newVal) {
|
|||||||
window.location.reload();
|
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();
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user