Compare commits

...

118 Commits
v1.0.30 ... v2

Author SHA1 Message Date
李寻欢
55d219b364 🎨 代码完善 2024-07-04 17:16:37 +08:00
李寻欢
a4234532cd 🎨 代码完善 2024-07-04 14:58:12 +08:00
李寻欢
3275a64852 🎨 代码完善 2024-07-04 14:57:31 +08:00
李寻欢
21505702e6 🆕 新增前端脚手架 2024-07-04 14:49:46 +08:00
李寻欢
45db832dad 🆕 新增前端脚手架 2024-07-04 14:46:12 +08:00
李寻欢
b806ade655 🔥 改为前后端分离形式 2024-07-04 14:10:46 +08:00
李寻欢
f39f46bfbf 🎨 页面美化 2024-06-19 14:21:07 +08:00
李寻欢
5819ac3c04 🎨 页面美化 2024-06-19 08:41:58 +08:00
李寻欢
c4964a9e21 🐛 Fix a bug. 2024-06-18 17:30:05 +08:00
李寻欢
c0dcffce9d 🎨 群组管理页面优化 2024-06-18 17:14:46 +08:00
李寻欢
b14dbe0d1d 🆕 新增AI角色管理(WIP) 2024-06-18 17:02:37 +08:00
李寻欢
b024600ef0 🆕 修改群组页面为卡片 2024-06-18 16:23:09 +08:00
李寻欢
bc2893fad1 🐛 修复自动删除群成员失败的BUG 2024-06-18 14:43:15 +08:00
李寻欢
a098da39ee 🆕 修改好友列表设置AI角色为数据库配置 2024-06-17 17:30:50 +08:00
李寻欢
42339e3c51 🎨 页面优化 2024-06-17 08:58:01 +08:00
李寻欢
20aeeefb3c Merge remote-tracking branch 'origin/hotfix' into hotfix 2024-06-14 08:36:27 +08:00
李寻欢
08ffd4de6c 🎨 修改一个错别字 2024-06-14 08:36:21 +08:00
李寻欢
2238e23c8d 🎨 优化好友列表同步处理逻辑 2024-06-13 07:02:00 +08:00
李寻欢
0f506e5afc 🎨 优化好友列表同步处理逻辑 2024-06-12 21:47:35 +08:00
李寻欢
44c45d11f2 Merge remote-tracking branch 'origin/hotfix' into hotfix 2024-06-06 14:18:23 +08:00
李寻欢
50e91680bb 🎨 逻辑优化 2024-06-06 14:18:18 +08:00
李寻欢
b2598f2406 🎨 消息总结新增传入群名称和过滤排行榜及前一天的总结消息 2024-05-31 07:27:17 +08:00
李寻欢
6818b10f4a 🎨 优化消息格式 2024-05-29 14:12:26 +08:00
李寻欢
727b06e143 🎨 优化消息格式 2024-05-16 08:32:44 +08:00
李寻欢
c0c3864a8e 🐛 修复清理不活跃成员功能的BUG 2024-05-16 00:12:57 +08:00
李寻欢
8c2ab9376c 🐛 修复清理不活跃成员功能的BUG 2024-05-16 00:12:47 +08:00
李寻欢
9c7e93660d 🆕 新增清理不活跃成员功能 2024-05-15 17:19:32 +08:00
李寻欢
4cc50718e2 Merge branch 'main' into hotfix 2024-05-15 11:15:38 +08:00
李寻欢
534f7a0ec8 🆕 新增每日早报功能 2024-05-15 11:15:15 +08:00
李寻欢
16838ff80f 🐛 Fix a bug. 2024-05-14 15:57:31 +08:00
李寻欢
ce37a269e1 Merge pull request '🆕 新增默认功能权限配置' (#46) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/46
2024-05-14 12:09:26 +08:00
李寻欢
aa78f3940c 🆕 新增默认功能权限配置 2024-05-14 12:08:30 +08:00
李寻欢
1c55900291 Merge pull request '🎨 优化发给新成员的消息' (#45) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/45
2024-05-14 11:51:31 +08:00
李寻欢
386e0dd6d3 🎨 优化发给新成员的消息 2024-05-14 11:50:27 +08:00
李寻欢
9bec70319b Merge pull request '🆕 添加群成员显示(WIP)' (#44) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/44
2024-05-09 09:45:38 +08:00
李寻欢
bd33759b80 🆕 添加群成员显示(WIP) 2024-05-09 09:44:57 +08:00
李寻欢
45109b02ef Merge pull request '🎨 优化消息总结查询SQL,支持更多消息类型' (#43) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/43
2024-04-29 10:13:34 +08:00
李寻欢
e8523c1e3e 🎨 优化消息总结查询SQL,支持更多消息类型 2024-04-29 10:12:50 +08:00
李寻欢
765e0ead7f Merge pull request 'hotfix' (#42) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/42
2024-04-24 17:24:26 +08:00
李寻欢
e7117d227b 🎨 好友列表页显示效果优化 2024-04-24 17:23:59 +08:00
李寻欢
448ac83514 🎨 首页显示效果优化 2024-04-24 15:49:17 +08:00
李寻欢
5024fb4b05 Merge pull request '🔥 重构管理页面,长得更好看一些了' (#41) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/41
2024-04-23 10:45:38 +08:00
李寻欢
1fca021b38 🔥 重构管理页面,长得更好看一些了 2024-04-23 10:42:08 +08:00
李寻欢
70fc7add78 Merge pull request '🎨 优化AI对话角色提示词' (#40) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/40
2024-04-19 09:55:27 +08:00
李寻欢
ab5a8092f7 🎨 优化AI对话角色提示词 2024-04-19 09:54:50 +08:00
李寻欢
21a6f75ed3 Merge pull request '🎨 优化AI对话角色设置,支持为群或者好友单独设置角色(需手动修改数据库)' (#39) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/39
2024-04-16 17:19:31 +08:00
李寻欢
daa36f308b 🎨 优化AI对话角色设置,支持为群或者好友单独设置角色(需手动修改数据库) 2024-04-16 17:19:08 +08:00
李寻欢
e78f6a6d6e Merge pull request '🎨 优化群聊总结返回消息内容' (#38) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/38
2024-04-15 16:22:43 +08:00
李寻欢
bb92c67069 🎨 优化群聊总结返回消息内容 2024-04-15 16:22:17 +08:00
李寻欢
542a2d5ef6 Merge pull request '🎨 优化消息总结提示词' (#37) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/37
2024-04-15 15:10:32 +08:00
李寻欢
c13771be22 🎨 优化消息总结提示词 2024-04-15 15:09:40 +08:00
李寻欢
982ddef0da Merge pull request '🎨 总结群聊时,不足100条记录的不总结' (#36) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/36
2024-04-12 11:40:51 +08:00
李寻欢
99447d332d 🎨 总结群聊时,不足100条记录的不总结 2024-04-12 11:40:40 +08:00
李寻欢
3478f4f9e2 Merge pull request 'hotfix' (#35) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/35
2024-04-12 11:38:10 +08:00
李寻欢
b3ed0fcc6f 🆕 新增群聊对话记录总结 2024-04-12 11:37:21 +08:00
李寻欢
3da8b327d0 chore(go-wxhelper): add AI command functionality 🤖
Add AI command functionality to handle AI commands in the WeChat plugin. Includes options to enable or disable AI features.
2024-04-12 10:48:46 +08:00
李寻欢
0262fdd9f7 Merge pull request 'hotfix' (#34) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/34
2024-04-09 14:43:20 +08:00
李寻欢
f775f1d67d Merge remote-tracking branch 'origin/hotfix' into hotfix 2024-04-09 14:42:59 +08:00
李寻欢
43f9d590b7 🎨 脚本优化 2024-04-09 14:42:47 +08:00
李寻欢
22c7d3683b Merge pull request 'hotfix' (#33) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/33
2024-03-22 23:07:45 +08:00
李寻欢
d589a9615e Merge remote-tracking branch 'origin/hotfix' into hotfix 2024-03-22 23:07:00 +08:00
李寻欢
13205e445f 🎨 新增 MQ 开关 2024-03-22 23:06:54 +08:00
李寻欢
a414a98a51 Merge pull request ' 群消息统计新增前十名消息数占比' (#31) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/31
2024-03-13 10:33:15 +08:00
李寻欢
d21089ab69 群消息统计新增前十名消息数占比 2024-03-13 10:32:20 +08:00
李寻欢
a274f085f8 Merge pull request 'hotfix' (#30) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/30
2024-03-13 10:03:10 +08:00
李寻欢
8ddc40cc01 群消息统计新增活跃用户人均消息条数和中位数 2024-03-13 10:02:40 +08:00
李寻欢
c0c810d02e 群消息统计新增活跃用户人均消息条数和中位数 2024-03-13 10:01:52 +08:00
李寻欢
905dab5ab8 Merge pull request '🐛 Fix a bug.' (#29) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/29
2024-03-12 08:37:32 +08:00
李寻欢
ddb0db0b6a 🐛 Fix a bug. 2024-03-12 08:37:05 +08:00
李寻欢
8bac050a02 Merge pull request 'hotfix' (#28) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/28
2024-03-11 14:56:45 +08:00
李寻欢
a548af9de2 🐛 修复活跃度不发送的BUG 2024-03-11 14:55:10 +08:00
李寻欢
64c3c9c78b 🎨 逻辑优化 2024-03-10 07:36:10 +08:00
李寻欢
0f7cf5515d Merge pull request '🐛 Fix a bug.' (#27) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/27
2024-03-08 07:45:51 +08:00
李寻欢
a2d84dea08 🐛 Fix a bug. 2024-03-08 07:45:05 +08:00
李寻欢
d3f8a59390 Merge pull request 'hotfix' (#26) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/26
2024-03-07 10:04:47 +08:00
李寻欢
165fefdb48 🐛 修复扫码加入群聊的成员无法识别的BUG 2024-03-07 10:04:03 +08:00
李寻欢
22474efc57 🎨 逻辑完善 2024-03-07 09:49:37 +08:00
李寻欢
797821e2ed 🎨 逻辑完善 2024-03-07 09:48:52 +08:00
李寻欢
79cbeb6ea5 Merge pull request '🆕 水群排行榜新增群活跃度' (#25) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/25
2024-03-07 09:43:27 +08:00
李寻欢
49705a03a8 🆕 水群排行榜新增群活跃度 2024-03-07 09:42:59 +08:00
李寻欢
40e97608d3 Merge pull request '📝 完善文档' (#24) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/24
2024-02-26 15:30:56 +08:00
李寻欢
fe7bc02c08 📝 完善文档 2024-02-26 15:30:40 +08:00
李寻欢
bb2c4e919e Merge pull request 'hotfix' (#23) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/23
2024-02-26 15:27:05 +08:00
李寻欢
0ea189fb82 📝 完善文档 2024-02-26 15:26:37 +08:00
李寻欢
778db8e349 📝 完善文档 2024-02-26 15:24:42 +08:00
李寻欢
4b7ad4126c Merge pull request '🆕 新增MQ消息通道' (#22) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/22
2024-02-19 14:17:59 +08:00
李寻欢
1413181bd4 🆕 新增MQ消息通道 2024-02-19 14:17:26 +08:00
李寻欢
a11ba51246 Merge pull request '🆕 控制页面新增设置AI模型' (#21) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/21
2024-01-31 12:03:01 +08:00
李寻欢
ea4262adf0 🆕 控制页面新增设置AI模型 2024-01-31 12:02:33 +08:00
李寻欢
a9f6c9ff0d Merge pull request '🐛 Fix a bug.' (#20) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/20
2024-01-26 12:00:53 +08:00
李寻欢
244bff5714 🐛 Fix a bug. 2024-01-26 12:00:29 +08:00
李寻欢
b46271a111 Merge pull request '🐛 Fix a bug.' (#19) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/19
2024-01-25 16:33:59 +08:00
李寻欢
5d11cc7c8a 🐛 Fix a bug. 2024-01-25 16:33:45 +08:00
李寻欢
c6da056f98 Merge pull request '🎨 逻辑优化,那个网页,大大地提速' (#18) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/18
2024-01-25 16:27:51 +08:00
李寻欢
61e03a6a7b 🎨 逻辑优化,那个网页,大大地提速 2024-01-25 16:27:12 +08:00
李寻欢
8ff18c3626 Merge pull request '🎨 逻辑优化' (#17) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/17
2024-01-25 11:46:30 +08:00
李寻欢
6acb3b79ec 🎨 逻辑优化 2024-01-25 11:46:03 +08:00
李寻欢
f895d045a7 Merge pull request '🐛 修复AI开关未生效的BUG' (#16) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/16
2024-01-23 16:23:52 +08:00
李寻欢
eb2e292e8e 🐛 修复AI开关未生效的BUG 2024-01-23 16:23:07 +08:00
李寻欢
8f378e7fc9 Merge pull request '🔧 新增数据库配置模型' (#15) from hotfix into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/15
2024-01-22 08:32:00 +08:00
李寻欢
497e17ec62 🔧 新增数据库配置模型 2024-01-22 08:31:38 +08:00
李寻欢
0a421dbb9d Merge pull request 'dev' (#13) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/13
2024-01-15 11:48:29 +08:00
李寻欢
0e38036a68 🐛 修复AI插件会响应@所有人消息事件 2024-01-15 11:48:03 +08:00
李寻欢
7ff54e4d95 ⬆️ 更新项目依赖 2024-01-13 15:30:55 +08:00
李寻欢
be15d42d93 Merge pull request 'dev' (#12) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/12
2024-01-12 22:54:50 +08:00
李寻欢
6a879cbef6 🆕 群成员新增是否群主 2024-01-12 22:54:27 +08:00
李寻欢
01051ff606 🆕 网页新增控制是否打开指令功能 2024-01-12 22:20:14 +08:00
李寻欢
450857dee8 Merge pull request '🐛 Fix a bug. #7' (#11) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/11
2024-01-12 21:51:25 +08:00
李寻欢
77c5a96c76 🐛 Fix a bug. #7 2024-01-12 21:50:56 +08:00
李寻欢
0c2010c348 Merge pull request '🐛 修复AI插件会响应@所有人消息事件 #7' (#9) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/9
2024-01-09 11:12:52 +08:00
李寻欢
2918924107 🐛 修复AI插件会响应@所有人消息事件 2024-01-09 11:11:09 +08:00
李寻欢
f796e7b3a8 Merge pull request 'dev' (#8) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/8
2024-01-09 10:49:21 +08:00
李寻欢
83efe20ddf Merge remote-tracking branch 'origin/dev' into dev 2024-01-09 10:48:30 +08:00
李寻欢
f0bb46b9ab 🆕 新增显示vnc页面 2024-01-09 10:48:21 +08:00
李寻欢
68ae670429 🐛 Fix a bug. 2024-01-01 09:38:55 +08:00
李寻欢
1d14b036ed 🐛 Fix a bug. 2024-01-01 09:28:11 +08:00
李寻欢
f5cc7953f4 Merge pull request '🆕 新增水群年度排行榜' (#6) from dev into main
Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/6
2024-01-01 00:59:54 +08:00
李寻欢
0db8e6bf95 🆕 新增水群年度排行榜 2024-01-01 00:57:52 +08:00
124 changed files with 9591 additions and 496 deletions

21
.editorconfig Normal file
View File

@@ -0,0 +1,21 @@
root = true
[*]
insert_final_newline = true
charset = utf-8
trim_trailing_whitespace = true
indent_style = space
indent_size = 2
[{Makefile,go.mod,go.sum,*.go,.gitmodules}]
indent_style = tab
indent_size = 4
[*.md]
indent_size = 4
trim_trailing_whitespace = false
eclint_indent_style = unset
[Dockerfile]
indent_size = 4

1
.gitignore vendored
View File

@@ -8,3 +8,4 @@ log
dist
*.log
blacklist.txt
frontend.*

36
app/assistant.go Normal file
View File

@@ -0,0 +1,36 @@
package app
import (
"github.com/gin-gonic/gin"
"go-wechat/common/response"
"go-wechat/config"
"go-wechat/service"
)
// SaveAssistant
// @description: 保存AI助手
// @param ctx
func SaveAssistant(ctx *gin.Context) {
//ctx.String(http.StatusOK, "操作成功")
ctx.Redirect(302, "/assistant.html")
}
// GetAssistants
// @description: 获取AI助手列表
// @param ctx
func GetAssistants(ctx *gin.Context) {
records, err := service.GetAllAiAssistant()
if err != nil {
response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
return
}
response.New(ctx).SetData(records).Success()
}
// GetAiModels
// @description: 获取AI模型列表
// @param ctx
func GetAiModels(ctx *gin.Context) {
response.New(ctx).SetData(config.Conf.Ai.Models).Success()
}

33
app/contact.go Normal file
View File

@@ -0,0 +1,33 @@
package app
import (
"github.com/gin-gonic/gin"
"go-wechat/common/response"
"go-wechat/service"
)
// GetFriends
// @description: 获取好友列表
// @param ctx
func GetFriends(ctx *gin.Context) {
// 取出所有好友列表
friends, _, err := service.GetAllFriend()
if err != nil {
response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
}
response.New(ctx).SetData(friends).Success()
}
// GetGroups
// @description: 获取群列表
// @param ctx
func GetGroups(ctx *gin.Context) {
// 取出所有好友列表
_, groups, err := service.GetAllFriend()
if err != nil {
response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
}
response.New(ctx).SetData(groups).Success()
}

View File

@@ -3,7 +3,7 @@ package app
import (
"github.com/gin-gonic/gin"
"go-wechat/client"
"go-wechat/entity"
"go-wechat/model/entity"
"gorm.io/gorm"
"log"
"net/http"
@@ -16,6 +16,20 @@ type changeStatusParam struct {
UserId string `json:"userId"`
}
// changeUseAiModelParam
// @description: 修改使用的AI模型用的参数集
type changeUseAiModelParam struct {
WxId string `json:"wxid" binding:"required"` // 群Id或微信Id
Model string `json:"model" binding:"required"` // 模型代码
}
// autoClearMembers
// @description: 自动清理群成员
type autoClearMembers struct {
WxId string `json:"wxid" binding:"required"` // 群Id
Days int `json:"days"` // 多少天未发言
}
// ChangeEnableAiStatus
// @description: 修改是否开启AI
// @param ctx
@@ -39,6 +53,49 @@ func ChangeEnableAiStatus(ctx *gin.Context) {
ctx.String(http.StatusOK, "操作成功")
}
// ChangeUseAiModel
// @description: 修改使用的AI模型
// @param ctx
func ChangeUseAiModel(ctx *gin.Context) {
var p changeUseAiModelParam
if err := ctx.ShouldBind(&p); err != nil {
ctx.String(http.StatusBadRequest, "参数错误")
return
}
err := client.MySQL.Model(&entity.Friend{}).
Where("wxid = ?", p.WxId).
Update("`ai_model`", p.Model).Error
if err != nil {
log.Printf("修改【%s】的AI模型失败%s", p.WxId, err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}
// ChangeUseAiAssistant
// @description: 修改使用的AI助手
// @param ctx
func ChangeUseAiAssistant(ctx *gin.Context) {
// 此处复用一下结构体
var p changeUseAiModelParam
if err := ctx.ShouldBind(&p); err != nil {
ctx.String(http.StatusBadRequest, "参数错误")
return
}
err := client.MySQL.Model(&entity.Friend{}).
Where("wxid = ?", p.WxId).
Update("`prompt`", p.Model).Error
if err != nil {
log.Printf("修改【%s】的AI助手失败%s", p.WxId, err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}
// ChangeEnableGroupRankStatus
// @description: 修改是否开启水群排行榜
// @param ctx
@@ -62,6 +119,29 @@ func ChangeEnableGroupRankStatus(ctx *gin.Context) {
ctx.String(http.StatusOK, "操作成功")
}
// ChangeEnableSummaryStatus
// @description: 修改是否开启聊天记录总结
// @param ctx
func ChangeEnableSummaryStatus(ctx *gin.Context) {
var p changeStatusParam
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("`enable_summary`", gorm.Expr(" !`enable_summary`")).Error
if err != nil {
log.Printf("修改开启聊天记录总结失败:%s", err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}
// ChangeEnableWelcomeStatus
// @description: 修改是否开启迎新
// @param ctx
@@ -85,6 +165,29 @@ func ChangeEnableWelcomeStatus(ctx *gin.Context) {
ctx.String(http.StatusOK, "操作成功")
}
// ChangeEnableCommandStatus
// @description: 修改是否开启指令
// @param ctx
func ChangeEnableCommandStatus(ctx *gin.Context) {
var p changeStatusParam
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("`enable_command`", gorm.Expr(" !`enable_command`")).Error
if err != nil {
log.Printf("修改指令启用状态失败:%s", err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}
// ChangeSkipGroupRankStatus
// @description: 修改是否跳过水群排行榜
// @param ctx
@@ -108,3 +211,49 @@ func ChangeSkipGroupRankStatus(ctx *gin.Context) {
ctx.String(http.StatusOK, "操作成功")
}
// ChangeEnableNewsStatus
// @description: 修改是否开启新闻
// @param ctx
func ChangeEnableNewsStatus(ctx *gin.Context) {
var p changeStatusParam
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("`enable_news`", gorm.Expr(" !`enable_news`")).Error
if err != nil {
log.Printf("修改早报启用状态失败:%s", err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}
// AutoClearMembers
// @description: 自动清理群成员
// @param ctx
func AutoClearMembers(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("`clear_member`", p.Days).Error
if err != nil {
log.Printf("修改自动清理群成员阈值失败:%s", err)
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
return
}
ctx.String(http.StatusOK, "操作成功")
}

View File

@@ -1,26 +0,0 @@
package app
import (
"fmt"
"github.com/gin-gonic/gin"
"go-wechat/service"
"net/http"
)
// Index
// @description: 首页
// @param ctx
func Index(ctx *gin.Context) {
var result = gin.H{
"msg": "success",
}
// 取出所有好友列表
friends, groups, err := service.GetAllFriend()
if err != nil {
result["msg"] = fmt.Sprintf("数据获取失败: %s", err.Error())
}
result["friends"] = friends
result["groups"] = groups
// 渲染页面
ctx.HTML(http.StatusOK, "index.html", result)
}

116
app/pages.go Normal file
View File

@@ -0,0 +1,116 @@
package app
import (
"fmt"
"github.com/gin-gonic/gin"
"go-wechat/config"
"go-wechat/service"
"net/http"
)
// Index
// @description: 首页
// @param ctx
func Index(ctx *gin.Context) {
var result = gin.H{
"msg": "success",
}
// 取出所有好友列表
friends, groups, err := service.GetAllFriend()
if err != nil {
result["msg"] = fmt.Sprintf("数据获取失败: %s", err.Error())
}
var in, notIn int
for _, d := range friends {
if d.IsOk {
in++
} else {
notIn++
}
}
result["friendCount"] = in
result["friendWithoutCount"] = notIn
var gin, gnotIn int
for _, d := range groups {
if d.IsOk {
gin++
} else {
gnotIn++
}
}
result["groupCount"] = gin
result["groupWithoutCount"] = gnotIn
result["vnc"] = config.Conf.Wechat.VncUrl
result["isVnc"] = config.Conf.Wechat.VncUrl != ""
result["aiModels"] = config.Conf.Ai.Models
// 渲染页面
ctx.HTML(http.StatusOK, "index.html", result)
}
// Friend
// @description: 好友列表
// @param ctx
func Friend(ctx *gin.Context) {
var result = gin.H{
"msg": "success",
}
// 取出所有好友列表
friends, _, err := service.GetAllFriend()
if err != nil {
result["msg"] = fmt.Sprintf("数据获取失败: %s", err.Error())
}
result["friends"] = friends
result["vnc"] = config.Conf.Wechat.VncUrl
result["aiModels"] = config.Conf.Ai.Models
result["assistant"], _ = service.GetAllAiAssistant()
// 渲染页面
ctx.HTML(http.StatusOK, "friend.html", result)
}
// Group
// @description: 群组列表
// @param ctx
func Group(ctx *gin.Context) {
var result = gin.H{
"msg": "success",
}
// 取出所有好友列表
_, groups, err := service.GetAllFriend()
if err != nil {
result["msg"] = fmt.Sprintf("数据获取失败: %s", err.Error())
}
result["groups"] = groups
result["vnc"] = config.Conf.Wechat.VncUrl
result["aiModels"] = config.Conf.Ai.Models
result["assistant"], _ = service.GetAllAiAssistant()
// 渲染页面
ctx.HTML(http.StatusOK, "group.html", result)
}
// Assistant
// @description: AI角色
// @param ctx
func Assistant(ctx *gin.Context) {
var result = gin.H{
"msg": "success",
}
result["aiModels"] = config.Conf.Ai.Models
result["assistant"], _ = service.GetAllAiAssistant()
// 渲染页面
ctx.HTML(http.StatusOK, "assistant.html", result)
}
// PageNotFound
// @description: 404页面
// @param ctx
func PageNotFound(ctx *gin.Context) {
// 渲染页面
ctx.HTML(http.StatusOK, "404.html", nil)
}

View File

@@ -1,7 +1,7 @@
package current
import (
"go-wechat/model"
"go-wechat/model/model"
plugin "go-wechat/plugin"
)

16
common/response/fail.go Normal file
View File

@@ -0,0 +1,16 @@
package response
// Fail
// @description: 失败响应
// @receiver r
// @param data
// @return err
func (r *Response) Fail() {
if r.msg == "" {
r.msg = "系统错误"
}
if r.code == 0 {
r.code = fail
}
r.Result()
}

43
common/response/page.go Normal file
View File

@@ -0,0 +1,43 @@
package response
// PageData
// @description: 分页数据通用结构体
type PageData[T any] struct {
Current int `json:"current"` // 当前页码
Size int `json:"size"` // 每页数量
Total int64 `json:"total"` // 总数
TotalPage int `json:"totalPage"` // 总页数
Records T `json:"records"` // 返回数据
}
// NewPageData
// @description: 创建分页数据
// @param records any 数据列表
// @param total int64 总数
// @param current int 页码
// @param size int 页数量
// @return data PageData[any] 分页数据
func NewPageData(records any, total int64, current, size int) (data PageData[any]) {
// 处理一下页码、页数量
if current == -1 {
current = 1
size = int(total)
}
// 计算总页码
totalPage := 0
if total > 0 {
upPage := 0
if int(total)%size > 0 {
upPage = 1
}
totalPage = (int(total) / size) + upPage
}
data = PageData[any]{
Current: current,
Size: size,
Total: total,
TotalPage: totalPage,
Records: records,
}
return
}

View File

@@ -0,0 +1,81 @@
package response
import (
"github.com/gin-gonic/gin"
"go-wechat/common/validator"
"net/http"
)
// 定义状态码
const (
fail = http.StatusInternalServerError
success = http.StatusOK
)
// Response
// @description: 返回结果
type Response struct {
ctx *gin.Context
code int
data any
msg string
errMsg string
}
// New
// @description: 返回结果实现
// @param ctx
// @return Response
func New(ctx *gin.Context) *Response {
var r Response
r.ctx = ctx
return &r
}
// SetCode
// @description: 设置状态码
// @receiver r
// @param code
// @return *Response
func (r *Response) SetCode(code int) *Response {
r.code = code
return r
}
// SetData
// @description: 设置返回数据
// @receiver r
// @param data
// @return *Response
func (r *Response) SetData(data any) *Response {
r.data = data
return r
}
// SetMsg
// @description: 设置返回消息
// @receiver r
// @param msg
// @return *Response
func (r *Response) SetMsg(msg string) *Response {
r.msg = msg
return r
}
// SetError
// @description: 设置错误信息
// @receiver r
// @param err
// @return *Response
func (r *Response) SetError(err error) *Response {
if err != nil {
ne := validator.Translate(err)
if ne != nil {
r.errMsg = ne.Error()
} else {
r.errMsg = err.Error()
}
}
return r
}

27
common/response/result.go Normal file
View File

@@ -0,0 +1,27 @@
package response
// Result
// @description: 响应
// @receiver r
// @param code int 状态码
// @param data any 数据
// @param msg string 消息
// @param err string 错误信息
// @return err error 返回数据错误
func (r *Response) Result() {
type resp struct {
Code int `json:"code"`
Data any `json:"data"`
Msg string `json:"message"`
ErrMsg string `json:"errMsg,omitempty"`
}
rd := resp{
r.code,
r.data,
r.msg,
r.errMsg,
}
// 返回数据
r.ctx.JSON(r.code, rd)
}

View File

@@ -0,0 +1,16 @@
package response
// Success
// @description: 成功响应
// @receiver r
// @param data
// @return err
func (r *Response) Success() {
if r.msg == "" {
r.msg = "success"
}
if r.code == 0 {
r.code = success
}
r.Result()
}

View File

@@ -0,0 +1,56 @@
package validator
import (
"errors"
"github.com/gin-gonic/gin/binding"
"github.com/go-playground/locales/zh"
ut "github.com/go-playground/universal-translator"
"github.com/go-playground/validator/v10"
zhTranslations "github.com/go-playground/validator/v10/translations/zh"
"log"
"strings"
)
var (
uni *ut.UniversalTranslator
validate *validator.Validate
trans ut.Translator
)
// Init
// @description: 初始化验证器
func Init() {
//注册翻译器
zhTranslator := zh.New()
uni = ut.New(zhTranslator, zhTranslator)
trans, _ = uni.GetTranslator("zh")
//获取gin的校验器
validate = binding.Validator.Engine().(*validator.Validate)
//注册翻译器
err := zhTranslations.RegisterDefaultTranslations(validate, trans)
if err != nil {
log.Panicf("注册翻译器失败:%v", err)
}
}
// Translate
// @description: 翻译错误信息
// @param err
// @return error
func Translate(err error) error {
errorMsg := make([]string, 0)
var ves validator.ValidationErrors
ok := errors.As(err, &ves)
if !ok {
return err
}
for _, e := range ves {
errorMsg = append(errorMsg, e.Translate(trans))
}
return errors.New(strings.Join(errorMsg, ""))
}

View File

@@ -1,7 +1,28 @@
system:
# 添加新好友或群之后通知给指定的人
newFriendNotify:
enable: true
toUser:
- "wxid_xxx"
# 默认AI等配置
defaultRule:
# 默认是否开启AI
ai: true
# 默认是否开启水群排行榜
chatRank: true
# 默认是否开启聊天记录总结
summary: true
# 默认是否开启新成员加群欢迎
welcome: true
# 每日早报
news: true
# 微信HOOK配置
wechat:
# 微信HOOK接口地址
host: 10.0.0.73:19088
host: 10.0.0.79:19088
# 微信容器映射出来的vnc页面地址没有就不填
# vncUrl: http://192.168.1.175:19087/vnc_lite.html
# 是否在启动的时候自动设置hook服务的回调
autoSetCallback: false
# 回调IP如果是Docker运行本参数必填(填auto表示自动不适用于 docker 环境)如果Docker修改了映射格式为 ip:port
@@ -12,23 +33,44 @@ wechat:
# 数据库
mysql:
drive: mysql # 使用的数据库驱动,支持 mysql、postgres
host: 10.0.0.31
port: 3307
user: wechat
password: wechat123
db: wechat
schema: public # postgres 专用
task:
enable: false
news:
enable: true
cron: '14 11 * * *' # 每天0:30
syncFriends:
enable: true
cron: '*/5 * * * *' # 五分钟一次
enable: false
cron: '*/5 * * * *' # 五分钟一次
groupSummary:
enable: false
cron: '30 0 * * *' # 每天0:30
waterGroup:
enable: false
cron:
yesterday: '30 9 * * *' # 每天9:30
week: '30 9 * * 1' # 每周一9:30
month: '30 9 1 * *' # 每月1号9:30
enable: false
cron:
yesterday: '30 9 * * *' # 每天9:30
week: '30 9 * * 1' # 每周一9:30
month: '30 9 1 * *' # 每月1号9:30
year: '0 9 1 1 *' # 每年1月1号9:30
# MQ配置
mq:
# 是否启用
enable: false
# RabbitMQ配置
rabbitmq:
host: 10.0.0.247
port: 5672
user: wechat
password: wechat123
vhost: wechat
# AI回复
ai:
@@ -36,12 +78,33 @@ ai:
enable: false
# 模型不填默认gpt-3.5-turbo-0613
model: gpt-3.5-turbo-0613
# 群聊总结模型
summaryModel: gpt-4-0613
# OpenAI Api key
apiKey: sk-xxxx
# 接口代理域名不填默认ChatGPT官方地址
baseUrl: https://sxxx
# 人设
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
models:
- name: ChatGPT-4
model: gpt-4
- name: 讯飞星火v3.1
model: SparkDesk-v3.1
- name: 讯飞星火v3.5
model: SparkDesk-v3.5
- name: 月之暗面-8k
model: moonshot-v1-8k
- name: 月之暗面-32k
model: moonshot-v1-32k
- name: 月之暗面-128k
model: moonshot-v1-128k
- name: 跃问
model: StepChat
- name: 豆包Lite-4k
model: Doubao-lite-4k
- name: 豆包Pro-4k
model: Doubao-pro-4k
# 资源配置
# map[k]v结构k 会变成全小写,所以这儿不能用大写字母
@@ -50,6 +113,10 @@ resource:
welcome-new:
type: emotion
path: 58e4150be2bba8f7b71974b10391f9e9
# 给新好友或者群的自我介绍,不配置就不发送
introduce:
type: text
path: "大家好我是一个AI机器人可以直接@我询问你想问的问题。"
# 水群排行榜词云,只能是图片,末尾的`\%s`也是必须的
wordcloud:
type: image

View File

@@ -3,9 +3,18 @@ package config
// ai
// @description: AI配置
type ai struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用AI
Model string `json:"model" yaml:"model"` // 模型
ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key
BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
Personality string `json:"personality" yaml:"personality"` // 人设
Enable bool `json:"enable" yaml:"enable"` // 是否启用AI
Model string `json:"model" yaml:"model"` // 模型
SummaryModel string `json:"summaryModel" yaml:"summaryModel"` // 总结模型
ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key
BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
Personality string `json:"personality" yaml:"personality"` // 人设
Models []aiModel `json:"models" yaml:"models"` // 模型列表
}
// aiModel
// @description: AI模型
type aiModel struct {
Name string `json:"name" yaml:"name"` // 模型名称
Model string `json:"model" yaml:"model"` // 模型代码
}

View File

@@ -6,9 +6,11 @@ var Conf conf
// Config
// @description: 配置
type conf struct {
System system `json:"system" yaml:"system"` // 系统配置
Task task `json:"task" yaml:"task"` // 定时任务配置
MySQL mysql `json:"mysql" yaml:"mysql"` // MySQL 配置
Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
Mq mq `json:"mq" yaml:"mq"` // MQ 配置
Ai ai `json:"ai" yaml:"ai"` // AI配置
Resource map[string]resourceItem `json:"resource" yaml:"resource"` // 资源配置
}

32
config/mq.go Normal file
View File

@@ -0,0 +1,32 @@
package config
import "fmt"
// mq
// @description: MQ配置
type mq struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
RabbitMQ rabbitMq `json:"rabbitmq" yaml:"rabbitmq"` // RabbitMQ配置
}
// rabbitMq
// @description: RabbitMQ配置
type rabbitMq struct {
Host string `json:"host" yaml:"host"` // 主机地址
Port int `json:"port" yaml:"port"` // 端口
User string `json:"user" yaml:"user"` // 用户名
Password string `json:"password" yaml:"password"` // 密码
VHost string `json:"vhost" yaml:"vhost"` // 虚拟主机
}
// GetURL
// @description: 获取MQ连接地址
// @receiver r
// @return string
func (r rabbitMq) GetURL() string {
port := r.Port
if port == 0 {
port = 5672
}
return fmt.Sprintf("amqp://%s:%s@%s:%d/%s", r.User, r.Password, r.Host, port, r.VHost)
}

22
config/system.go Normal file
View File

@@ -0,0 +1,22 @@
package config
// 系统配置
type system struct {
NewFriendNotify newFriendNotify `json:"newFriendNotify" yaml:"newFriendNotify"` // 新好友通知
DefaultRule defaultRule `json:"defaultRule" yaml:"defaultRule"` // 默认规则
}
// 添加新好友或群之后通知给指定的人
type newFriendNotify struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
ToUser []string `json:"toUser" yaml:"toUser"` // 通知给谁
}
// 默认规则
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"` // 是否启用每日早报
}

View File

@@ -3,9 +3,11 @@ package config
// task
// @description: 定时任务
type task struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友
WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
Enable bool `json:"enable" yaml:"enable"` // 是否启用
News syncFriends `json:"news" yaml:"news"` // 每日早报
SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友
WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
GroupSummary syncFriends `json:"groupSummary" yaml:"groupSummary"` // 群聊总结
}
// syncFriends
@@ -28,4 +30,5 @@ type waterGroupCron struct {
Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
Week string `json:"week" yaml:"week"` // 周排行榜
Month string `json:"month" yaml:"month"` // 月排行榜
Year string `json:"year" yaml:"year"` // 年排行榜
}

View File

@@ -6,6 +6,7 @@ import "strings"
// @description: 微信助手
type wechat struct {
Host string `json:"host" yaml:"host"` // 接口地址
VncUrl string `json:"vncUrl" yaml:"vncUrl"` // vnc页面地址
AutoSetCallback bool `json:"autoSetCallback" yaml:"autoSetCallback"` // 是否自动设置回调地址
Callback string `json:"callback" yaml:"callback"` // 回调地址
Forward []string `json:"forward" yaml:"forward"` // 转发地址

View File

@@ -2,7 +2,7 @@ version: '3.9'
services:
wechat:
image: lxh01/wxhelper-docker:3.9.5.81
image: lxh01/wxhelper-docker:3.9.5.81-v11
container_name: gw-wechat
restart: unless-stopped
environment:
@@ -10,7 +10,7 @@ services:
volumes:
- ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
ports:
- "8080:8080"
- "19087:8080"
- "19088:19088"
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
@@ -23,9 +23,6 @@ services:
image: mysql:8
container_name: gw-db
restart: unless-stopped
depends_on:
wechat:
condition: service_healthy
environment:
- MYSQL_ROOT_PASSWORD=wechat
- MYSQL_USER=wechat
@@ -41,6 +38,7 @@ services:
restart: unless-stopped
depends_on:
- mysql
- wechat
volumes:
# 配置文件请参阅项目根目录的config.yaml文件
- ./config/config.yaml:/app/config.yaml

View File

@@ -1,41 +0,0 @@
package entity
import (
"time"
)
// Friend
// @description: 好友列表
type Friend struct {
Wxid string `json:"wxid"` // 微信原始Id
CustomAccount string `json:"customAccount"` // 微信号
Nickname string `json:"nickname"` // 昵称
Pinyin string `json:"pinyin"` // 昵称拼音大写首字母
PinyinAll string `json:"pinyinAll"` // 昵称全拼
EnableAi bool `json:"enableAI" gorm:"type:tinyint(1) default 0 not null"` // 是否使用AI
EnableChatRank bool `json:"enableChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否使用聊天排行
EnableWelcome bool `json:"enableWelcome" gorm:"type:tinyint(1) default 0 not null"` // 是否启用迎新
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常
}
func (Friend) TableName() string {
return "t_friend"
}
// GroupUser
// @description: 群成员
type GroupUser struct {
GroupId string `json:"groupId"` // 群Id
Wxid string `json:"wxid"` // 微信Id
Account string `json:"account"` // 账号
HeadImage string `json:"headImage"` // 头像
Nickname string `json:"nickname"` // 昵称
IsMember bool `json:"isMember" gorm:"type:tinyint(1) default 0 not null"` // 是否群成员
JoinTime time.Time `json:"joinTime"` // 加入时间
LeaveTime *time.Time `json:"leaveTime"` // 离开时间
SkipChatRank bool `json:"skipChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否跳过聊天排行
}
func (GroupUser) TableName() string {
return "t_group_user"
}

18
frontend/.eslintrc.cjs Normal file
View File

@@ -0,0 +1,18 @@
/* eslint-env node */
require('@rushstack/eslint-patch/modern-module-resolution')
module.exports = {
env: {
node: true,
},
root: true,
'extends': [
'plugin:vue/vue3-essential',
'eslint:recommended',
'@vue/eslint-config-typescript',
'@vue/eslint-config-prettier/skip-formatting'
],
parserOptions: {
ecmaVersion: 'latest'
}
}

30
frontend/.gitignore vendored Normal file
View File

@@ -0,0 +1,30 @@
# Logs
logs
*.log
npm-debug.log*
yarn-debug.log*
yarn-error.log*
pnpm-debug.log*
lerna-debug.log*
node_modules
.DS_Store
dist
dist-ssr
coverage
*.local
/cypress/videos/
/cypress/screenshots/
# Editor directories and files
.vscode/*
!.vscode/extensions.json
.idea
*.suo
*.ntvs*
*.njsproj
*.sln
*.sw?
*.tsbuildinfo

View File

@@ -0,0 +1,8 @@
{
"$schema": "https://json.schemastore.org/prettierrc",
"semi": false,
"tabWidth": 2,
"singleQuote": true,
"printWidth": 100,
"trailingComma": "none"
}

7
frontend/.vscode/extensions.json vendored Normal file
View File

@@ -0,0 +1,7 @@
{
"recommendations": [
"Vue.volar",
"dbaeumer.vscode-eslint",
"esbenp.prettier-vscode"
]
}

39
frontend/README.md Normal file
View File

@@ -0,0 +1,39 @@
# frontend
This template should help get you started developing with Vue 3 in Vite.
## Recommended IDE Setup
[VSCode](https://code.visualstudio.com/) + [Volar](https://marketplace.visualstudio.com/items?itemName=Vue.volar) (and disable Vetur).
## Type Support for `.vue` Imports in TS
TypeScript cannot handle type information for `.vue` imports by default, so we replace the `tsc` CLI with `vue-tsc` for type checking. In editors, we need [Volar](https://marketplace.visualstudio.com/items?itemName=Vue.volar) to make the TypeScript language service aware of `.vue` types.
## Customize configuration
See [Vite Configuration Reference](https://vitejs.dev/config/).
## Project Setup
```sh
npm install
```
### Compile and Hot-Reload for Development
```sh
npm run dev
```
### Type-Check, Compile and Minify for Production
```sh
npm run build
```
### Lint with [ESLint](https://eslint.org/)
```sh
npm run lint
```

1
frontend/env.d.ts vendored Normal file
View File

@@ -0,0 +1 @@
/// <reference types="vite/client" />

13
frontend/index.html Normal file
View File

@@ -0,0 +1,13 @@
<!DOCTYPE html>
<html lang="en" class="h-full bg-gray-100">
<head>
<meta charset="UTF-8">
<link rel="icon" href="/favicon.ico">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>一个微信机器人</title>
</head>
<body style="min-height: 911px" class="h-full">
<div id="app"></div>
<script type="module" src="/src/main.ts"></script>
</body>
</html>

5312
frontend/package-lock.json generated Normal file

File diff suppressed because it is too large Load Diff

46
frontend/package.json Normal file
View File

@@ -0,0 +1,46 @@
{
"name": "frontend",
"version": "0.0.0",
"private": true,
"type": "module",
"scripts": {
"dev": "vite",
"build": "run-p type-check \"build-only {@}\" --",
"preview": "vite preview",
"build-only": "vite build",
"type-check": "vue-tsc --build --force",
"lint": "eslint . --ext .vue,.js,.jsx,.cjs,.mjs,.ts,.tsx,.cts,.mts --fix --ignore-path .gitignore",
"format": "prettier --write src/"
},
"dependencies": {
"@headlessui/vue": "^1.7.22",
"@tailwindcss/forms": "^0.5.7",
"axios": "^1.7.2",
"pinia": "^2.1.7",
"qs": "^6.12.2",
"vue": "^3.4.29",
"vue-router": "^4.3.3"
},
"devDependencies": {
"@rushstack/eslint-patch": "^1.8.0",
"@tsconfig/node20": "^20.1.4",
"@types/node": "^20.14.5",
"@types/qs": "^6.9.15",
"@vitejs/plugin-vue": "^5.0.5",
"@vitejs/plugin-vue-jsx": "^4.0.0",
"@vue/eslint-config-prettier": "^9.0.0",
"@vue/eslint-config-typescript": "^13.0.0",
"@vue/tsconfig": "^0.5.1",
"autoprefixer": "^10.4.19",
"daisyui": "^4.12.10",
"eslint": "^8.57.0",
"eslint-plugin-vue": "^9.23.0",
"npm-run-all2": "^6.2.0",
"postcss": "^8.4.39",
"prettier": "^3.2.5",
"tailwindcss": "^3.4.4",
"typescript": "~5.4.0",
"vite": "^5.3.1",
"vue-tsc": "^2.0.21"
}
}

View File

@@ -0,0 +1,6 @@
export default {
plugins: {
tailwindcss: {},
autoprefixer: {},
},
};

BIN
frontend/public/favicon.ico Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 4.2 KiB

10
frontend/src/App.vue Normal file
View File

@@ -0,0 +1,10 @@
<script setup lang="ts">
import MainNav from '@/components/layout/MainNav.vue'
</script>
<template>
<MainNav />
</template>
<style scoped>
</style>

6
frontend/src/api/base.ts Normal file
View File

@@ -0,0 +1,6 @@
// 默认返回数据结构
export type BaseResult<T> = {
code: number;
message: string;
data?: T;
};

View File

@@ -0,0 +1,43 @@
import { http } from '@/utils/http'
import type { BaseResult } from '@/api/base'
/** 通讯录列表返回结果 */
export type ContactItem = {
/** 微信号 */
CustomAccount: string,
/** 昵称 */
Nickname: string,
/** 昵称拼音大写首字母 */
Pinyin: string,
/** 昵称拼音全拼 */
PinyinAll: string,
/** 微信原始Id */
Wxid: string,
/** 最后活跃时间 */
LastActive: Date,
/** 是否使用AI */
EnableAi: boolean,
/** AI模型 */
AiModel: string,
/** AI助手或者自定义提示词 */
Prompt: string,
/** 是否使用聊天排行 */
EnableChatRank: boolean,
/** 是否使用迎新 */
EnableWelcome: boolean,
/** 是否启用指令 */
EnableCommand: boolean,
/** 是否启用总结 */
EnableSummary: boolean,
/** 是否启用新闻 */
EnableNews: boolean,
/** 清理成员配置(多少天未活跃的) */
ClearMember: number,
/** 是否还在通讯库(群聊是要还在群里也算) */
IsOk: boolean
}
/** 获取首页基础信息 */
export const getFriendList = () => {
return http.request<BaseResult<Array<ContactItem>>>('get', '/api/contact/friend')
}

Binary file not shown.

After

Width:  |  Height:  |  Size: 24 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 210 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 123 KiB

View File

@@ -0,0 +1,3 @@
<template>
</template>

View File

@@ -0,0 +1,73 @@
<script setup lang="ts">
import { RouterView } from 'vue-router'
import { Disclosure, DisclosureButton, DisclosurePanel, Menu, MenuButton, MenuItem, MenuItems } from '@headlessui/vue'
defineOptions({
name: 'MainNav'
})
const navigation = [
{ name: '首页', href: '/', current: true },
{ name: '好友列表', href: '/friend', current: false },
{ name: '群组', href: '/group', current: false },
{ name: 'AI角色', href: '/assistant', current: false }
]
// 设置当前选中的导航
const setCurrent = (href: string) => {
navigation.forEach(item => {
item.current = item.href === href
})
}
</script>
<template>
<div class="min-h-full">
<div class="bg-green-600 pb-32">
<Disclosure as="nav" class="border-b border-green-300 border-opacity-25 bg-green-600 lg:border-none">
<div class="mx-auto max-w-7xl px-2 sm:px-4 lg:px-8">
<div
class="relative flex h-16 items-center justify-between lg:border-b lg:border-green-400 lg:border-opacity-25">
<div class="flex items-center px-2 lg:px-0">
<div class="flex-shrink-0">
<img class="block h-8 w-8" src="../../assets/img/logo.png" alt="Your Company" />
</div>
<div class="hidden lg:ml-10 lg:block">
<div class="flex space-x-4">
<a v-for="item in navigation" :key="item.name" :href="item.href"
:class="[item.current ? 'bg-green-700 text-white' : 'text-white hover:bg-green-500 hover:bg-opacity-75', 'rounded-md px-3 py-2 text-sm font-medium']"
:aria-current="item.current ? 'page' : undefined" @click="setCurrent(item.href)">{{ item.name }}</a>
</div>
</div>
</div>
</div>
</div>
</Disclosure>
<header class="py-10">
<div class="mx-auto max-w-7xl px-4 sm:px-6 lg:px-8">
<h1 class="text-3xl font-bold tracking-tight text-white">Dashboard</h1>
</div>
</header>
</div>
<main class="-mt-32">
<div class="mx-auto max-w-7xl px-4 pb-12 sm:px-6 lg:px-8">
<div class="rounded-lg bg-white px-5 py-6 shadow sm:px-6">
<RouterView />
</div>
</div>
</main>
<footer>
<div class="mx-auto max-w-3xl px-4 sm:px-6 lg:max-w-7xl lg:px-8">
<div class="border-t border-gray-200 py-8 text-center text-sm text-gray-500 sm:text-left">
<span class="block sm:inline">本项目完全开源开源地址: </span>
<span class="block sm:inline text-red-500">
<a target="_blank" href="https://gitee.ltd/lxh/go-wxhelper">https://gitee.ltd</a>
</span>
</div>
</div>
</footer>
</div>
</template>

15
frontend/src/main.ts Normal file
View File

@@ -0,0 +1,15 @@
import { createApp } from 'vue'
import { createPinia } from 'pinia'
import App from './App.vue'
import router from './router'
// Tailwind CSS
import "tailwindcss/tailwind.css";
const app = createApp(App)
app.use(createPinia())
app.use(router)
app.mount('#app')

View File

@@ -0,0 +1,31 @@
import { createRouter, createWebHistory } from 'vue-router'
import IndexPage from '@/views/index/index.vue'
const router = createRouter({
history: createWebHistory(import.meta.env.BASE_URL),
routes: [
{
path: '/',
name: 'home',
component: IndexPage
},
{
path: '/friend',
name: 'Friend',
// 这么写是为了实现懒加载
component: () => import('@/views/friend/index.vue')
},
{
path: '/group',
name: 'Group',
component: () => import('@/views/group/index.vue')
},
{
path: '/assistant',
name: 'Assistant',
component: () => import('@/views/assistant/index.vue')
}
]
})
export default router

View File

@@ -0,0 +1,12 @@
import { ref, computed } from 'vue'
import { defineStore } from 'pinia'
export const useCounterStore = defineStore('counter', () => {
const count = ref(0)
const doubleCount = computed(() => count.value * 2)
function increment() {
count.value++
}
return { count, doubleCount, increment }
})

View File

@@ -0,0 +1,136 @@
import Axios, {
type AxiosInstance,
type AxiosRequestConfig,
type CustomParamsSerializer,
type AxiosResponse
} from 'axios'
import type { HttpError, RequestMethods } from './types.d'
import { stringify } from 'qs'
// 相关配置请参考www.axios-js.com/zh-cn/docs/#axios-request-config-1
const defaultConfig: AxiosRequestConfig = {
// 请求超时时间
timeout: 10000,
headers: {
Accept: 'application/json, text/plain, */*',
'Content-Type': 'application/json',
'X-Requested-With': 'XMLHttpRequest'
},
// 数组格式参数序列化https://github.com/axios/axios/issues/5142
paramsSerializer: {
serialize: stringify as unknown as CustomParamsSerializer
}
}
class Http {
constructor() {
this.httpInterceptorsRequest()
this.httpInterceptorsResponse()
}
/** 初始化配置对象 */
private static initConfig: AxiosRequestConfig = {}
/** 保存当前`Axios`实例对象 */
private static axiosInstance: AxiosInstance = Axios.create(defaultConfig)
/** 请求拦截 */
private httpInterceptorsRequest(): void {
Http.axiosInstance.interceptors.request.use(
async (config: AxiosRequestConfig): Promise<any> => {
// 开启进度条动画
// NProgress.start();
// 优先判断post/get等方法是否传入回调否则执行初始化设置等回调
// if (typeof config.beforeRequestCallback === 'function') {
// config.beforeRequestCallback(config)
// return config
// }
// if (Http.initConfig.beforeRequestCallback) {
// Http.initConfig.beforeRequestCallback(config)
// return config
// }
return config
},
(error: HttpError) => {
return Promise.reject(error)
}
)
}
/** 响应拦截 */
private httpInterceptorsResponse(): void {
const instance = Http.axiosInstance
instance.interceptors.response.use(
(response: AxiosResponse) => {
// const $config = response.config
// 关闭进度条动画
// NProgress.done()
// 优先判断post/get等方法是否传入回调否则执行初始化设置等回调
// if (typeof $config.beforeResponseCallback === 'function') {
// $config.beforeResponseCallback(response)
// return response.data
// }
// if (Http.initConfig.beforeResponseCallback) {
// Http.initConfig.beforeResponseCallback(response)
// return response.data
// }
return response.data
},
(error: HttpError) => {
const $error = error
$error.isCancelRequest = Axios.isCancel($error)
// 关闭进度条动画
// NProgress.done()
// 所有的响应异常 区分来源为取消请求/非取消请求
return Promise.reject($error)
}
)
}
/** 通用请求工具函数 */
public request<T>(
method: RequestMethods,
url: string,
param?: AxiosRequestConfig,
axiosConfig?: AxiosRequestConfig
): Promise<T> {
const config = {
method,
url,
...param,
...axiosConfig
} as AxiosRequestConfig
// 单独处理自定义请求/响应回调
return new Promise((resolve, reject) => {
Http.axiosInstance
.request<T>(config)
.then((response: any): void => {
return resolve(response)
})
.catch((error) => {
reject(error)
})
})
}
/** 单独抽离的`post`工具函数 */
public post<T, P>(
url: string,
params?: AxiosRequestConfig<P>,
config?: AxiosRequestConfig
): Promise<T> {
return this.request<T>('post', url, params, config)
}
/** 单独抽离的`get`工具函数 */
public get<T, P>(
url: string,
params?: AxiosRequestConfig<P>,
config?: AxiosRequestConfig
): Promise<T> {
return this.request<T>('get', url, params, config)
}
}
export const http = new Http()

43
frontend/src/utils/http/types.d.ts vendored Normal file
View File

@@ -0,0 +1,43 @@
import type {
Method,
AxiosError,
AxiosResponse,
AxiosRequestConfig
} from "axios";
export type RequestMethods = Extract<
Method,
"get" | "post" | "put" | "delete" | "patch" | "option" | "head"
>;
export interface HttpError extends AxiosError {
isCancelRequest?: boolean;
}
export interface HttpResponse extends AxiosResponse {
config: HttpRequestConfig;
}
export interface HttpRequestConfig extends AxiosRequestConfig {
beforeRequestCallback?: (request: HttpRequestConfig) => void;
beforeResponseCallback?: (response: HttpResponse) => void;
}
export default class Http {
request<T>(
method: RequestMethods,
url: string,
param?: AxiosRequestConfig,
axiosConfig?: HttpRequestConfig
): Promise<T>;
post<T, P>(
url: string,
params?: T,
config?: HttpRequestConfig
): Promise<P>;
get<T, P>(
url: string,
params?: T,
config?: HttpRequestConfig
): Promise<P>;
}

View File

@@ -0,0 +1,12 @@
<script setup lang="ts">
defineOptions({
name: "AssistantPage"
});
</script>
<template>
<main>
<h1>AI角色</h1>
</main>
</template>

View File

@@ -0,0 +1,25 @@
import { onMounted, ref } from 'vue'
import { type ContactItem, getFriendList } from '@/api/contact'
// 起飞
export function useHook() {
const dataList = ref<ContactItem[]>([]);
// 获取数据
async function onSearch() {
const { data } = await getFriendList();
dataList.value = data;
console.log(data);
}
// 挂载时执行
onMounted(() => {
onSearch();
});
return {
dataList,
onSearch
}
}

View File

@@ -0,0 +1,47 @@
<script setup lang="ts">
import { useHook } from '@/views/friend/hook'
defineOptions({
name: "FriendPage"
});
const { dataList, onSearch } = useHook()
</script>
<template>
<ul role="list" class="grid grid-cols-1 gap-x-6 gap-y-8 lg:grid-cols-3 xl:gap-x-8">
<li class="overflow-hidden rounded-xl border border-gray-200" v-for="item in dataList" :key="item.Wxid">
<div class="flex items-center gap-x-4 border-b border-gray-900/5 bg-gray-50 p-6">
<img v-if="item.IsOk" src="@/assets/img/status-ok.png" alt="Tuple" class="h-12 w-12 flex-none rounded-lg bg-white object-cover ring-1 ring-gray-900/10">
<img v-else src="@/assets/img/status-fail.png" alt="Tuple" class="h-12 w-12 flex-none rounded-lg bg-white object-cover ring-1 ring-gray-900/10">
<div class="flex-1">
<div class="text-sm font-medium leading-6 text-gray-900">{{ item.Nickname }}</div>
<span v-if="item.IsOk"
class="inline-flex items-center rounded-md bg-green-50 px-2 py-1 text-xs font-medium text-green-700 ring-1 ring-inset ring-green-600/20">在通讯录</span>
<span v-else
class="inline-flex items-center rounded-md bg-red-50 px-2 py-1 text-xs font-medium text-red-700 ring-1 ring-inset ring-red-600/20">不在通讯录</span>
</div>
</div>
<dl class="-my-3 divide-y divide-gray-100 px-6 py-4 text-sm leading-6">
<div class="flex justify-between gap-x-4 py-3">
<dt class="text-gray-500">原始微信Id<br/>微信号</dt>
<dd>
<div class="text-gray-700">{{ item.Wxid }}</div>
<div class="truncate text-gray-500">{{ item.CustomAccount }}</div>
</dd>
</div>
<div class="flex justify-between gap-x-4 py-3">
<dt class="text-gray-500">最后活跃时间</dt>
<dd class="flex items-start gap-x-2">
<div class="font-medium text-gray-900">
{{ item.LastActive }}
</div>
</dd>
</div>
</dl>
</li>
</ul>
</template>

View File

@@ -0,0 +1,12 @@
<script setup lang="ts">
defineOptions({
name: "GroupPage"
});
</script>
<template>
<main>
<h1>群组</h1>
</main>
</template>

View File

@@ -0,0 +1,12 @@
<script setup lang="ts">
defineOptions({
name: "IndexPage"
});
</script>
<template>
<main>
<h1 class="text-orange-700">首页</h1>
</main>
</template>

View File

@@ -0,0 +1,12 @@
/** @type {import('tailwindcss').Config} */
export default {
content: ['./index.html', './src/**/*.{vue,js,ts,jsx,tsx,md}'],
theme: {
container: {
center: true,
padding: '2rem',
},
},
plugins: [require('daisyui'), require('@tailwindcss/forms'),],
}

View File

@@ -0,0 +1,14 @@
{
"extends": "@vue/tsconfig/tsconfig.dom.json",
"include": ["env.d.ts", "src/**/*", "src/**/*.vue"],
"exclude": ["src/**/__tests__/*"],
"compilerOptions": {
"composite": true,
"tsBuildInfoFile": "./node_modules/.tmp/tsconfig.app.tsbuildinfo",
"baseUrl": ".",
"paths": {
"@/*": ["./src/*"]
}
}
}

11
frontend/tsconfig.json Normal file
View File

@@ -0,0 +1,11 @@
{
"files": [],
"references": [
{
"path": "./tsconfig.node.json"
},
{
"path": "./tsconfig.app.json"
}
]
}

View File

@@ -0,0 +1,19 @@
{
"extends": "@tsconfig/node20/tsconfig.json",
"include": [
"vite.config.*",
"vitest.config.*",
"cypress.config.*",
"nightwatch.conf.*",
"playwright.config.*"
],
"compilerOptions": {
"composite": true,
"noEmit": true,
"tsBuildInfoFile": "./node_modules/.tmp/tsconfig.node.tsbuildinfo",
"module": "ESNext",
"moduleResolution": "Bundler",
"types": ["node"]
}
}

28
frontend/vite.config.ts Normal file
View File

@@ -0,0 +1,28 @@
import { fileURLToPath, URL } from 'node:url'
import { defineConfig } from 'vite'
import vue from '@vitejs/plugin-vue'
import vueJsx from '@vitejs/plugin-vue-jsx'
// https://vitejs.dev/config/
export default defineConfig({
server: {
proxy: {
"/api": {
// target: "https://graduate.frps.ltd",
target: "http://127.0.0.1:8080",
changeOrigin: true
// rewrite: path => path.replace(/^\/api/, "")
}
},
},
plugins: [
vue(),
vueJsx(),
],
resolve: {
alias: {
'@': fileURLToPath(new URL('./src', import.meta.url))
}
}
})

27
go.mod
View File

@@ -3,13 +3,14 @@ module go-wechat
go 1.21
require (
github.com/duke-git/lancet/v2 v2.2.7
github.com/duke-git/lancet/v2 v2.2.8
github.com/fsnotify/fsnotify v1.7.0
github.com/gin-gonic/gin v1.9.1
github.com/go-co-op/gocron v1.36.1
github.com/go-resty/resty/v2 v2.10.0
github.com/sashabaranov/go-openai v1.17.9
github.com/spf13/viper v1.18.0
github.com/go-co-op/gocron v1.37.0
github.com/go-resty/resty/v2 v2.11.0
github.com/rabbitmq/amqp091-go v1.9.0
github.com/sashabaranov/go-openai v1.17.11
github.com/spf13/viper v1.18.2
gorm.io/driver/mysql v1.5.2
gorm.io/gorm v1.25.5
)
@@ -25,7 +26,7 @@ require (
github.com/go-playground/validator/v10 v10.16.0 // indirect
github.com/go-sql-driver/mysql v1.7.1 // indirect
github.com/goccy/go-json v0.10.2 // indirect
github.com/google/uuid v1.4.0 // indirect
github.com/google/uuid v1.5.0 // indirect
github.com/hashicorp/hcl v1.0.0 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
@@ -37,7 +38,7 @@ require (
github.com/mitchellh/mapstructure v1.5.0 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/pelletier/go-toml/v2 v2.1.0 // indirect
github.com/pelletier/go-toml/v2 v2.1.1 // indirect
github.com/robfig/cron/v3 v3.0.1 // indirect
github.com/sagikazarmark/locafero v0.4.0 // indirect
github.com/sagikazarmark/slog-shim v0.1.0 // indirect
@@ -50,13 +51,13 @@ require (
github.com/ugorji/go/codec v1.2.12 // indirect
go.uber.org/atomic v1.11.0 // indirect
go.uber.org/multierr v1.11.0 // indirect
golang.org/x/arch v0.6.0 // indirect
golang.org/x/crypto v0.16.0 // indirect
golang.org/x/exp v0.0.0-20231206192017-f3f8817b8deb // indirect
golang.org/x/net v0.19.0 // indirect
golang.org/x/sys v0.15.0 // indirect
golang.org/x/arch v0.7.0 // indirect
golang.org/x/crypto v0.18.0 // indirect
golang.org/x/exp v0.0.0-20240112132812-db7319d0e0e3 // indirect
golang.org/x/net v0.20.0 // indirect
golang.org/x/sys v0.16.0 // indirect
golang.org/x/text v0.14.0 // indirect
google.golang.org/protobuf v1.31.0 // indirect
google.golang.org/protobuf v1.32.0 // indirect
gopkg.in/ini.v1 v1.67.0 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
)

59
go.sum
View File

@@ -14,8 +14,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/duke-git/lancet/v2 v2.2.7 h1:u9zr6HR+MDUvZEtTlAFtSTIgZfEFsN7cKi27n5weZsw=
github.com/duke-git/lancet/v2 v2.2.7/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
github.com/duke-git/lancet/v2 v2.2.8 h1:wlruXhliDe4zls1e2cYmz4qLc+WtcvrpcCnk1VJdEaA=
github.com/duke-git/lancet/v2 v2.2.8/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=
@@ -26,8 +26,8 @@ github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
github.com/go-co-op/gocron v1.36.1 h1:BnyJ7kOixY/oGW/hkUvuLYOmAHBiY8Mr6kadX2DN1hE=
github.com/go-co-op/gocron v1.36.1/go.mod h1:3L/n6BkO7ABj+TrfSVXLRzsP26zmikL4ISkLQ0O8iNY=
github.com/go-co-op/gocron v1.37.0 h1:ZYDJGtQ4OMhTLKOKMIch+/CY70Brbb1dGdooLEhh7b0=
github.com/go-co-op/gocron v1.37.0/go.mod h1:3L/n6BkO7ABj+TrfSVXLRzsP26zmikL4ISkLQ0O8iNY=
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
@@ -36,20 +36,19 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
github.com/go-playground/validator/v10 v10.16.0 h1:x+plE831WK4vaKHO/jpgUGsvLKIqRRkz6M78GuJAfGE=
github.com/go-playground/validator/v10 v10.16.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
github.com/go-resty/resty/v2 v2.10.0 h1:Qla4W/+TMmv0fOeeRqzEpXPLfTUnR5HZ1+lGs+CkiCo=
github.com/go-resty/resty/v2 v2.10.0/go.mod h1:iiP/OpA0CkcL3IGt1O0+/SIItFUbkkyw5BGXiVdTu+A=
github.com/go-resty/resty/v2 v2.11.0 h1:i7jMfNOJYMp69lq7qozJP+bjgzfAzeOhuGlyDrqxT/8=
github.com/go-resty/resty/v2 v2.11.0/go.mod h1:iiP/OpA0CkcL3IGt1O0+/SIItFUbkkyw5BGXiVdTu+A=
github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI=
github.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
github.com/google/uuid v1.4.0 h1:MtMxsa51/r9yyhkyLsVeVt0B+BGQZzpQiTQ4eHZ8bc4=
github.com/google/uuid v1.4.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU=
github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
@@ -84,12 +83,14 @@ github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
github.com/pelletier/go-toml/v2 v2.1.0 h1:FnwAJ4oYMvbT/34k9zzHuZNrhlz48GB3/s6at6/MHO4=
github.com/pelletier/go-toml/v2 v2.1.0/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc=
github.com/pelletier/go-toml/v2 v2.1.1 h1:LWAJwfNvjQZCFIDKWYQaM62NcYeYViCmWIwmOStowAI=
github.com/pelletier/go-toml/v2 v2.1.1/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc=
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/rabbitmq/amqp091-go v1.9.0 h1:qrQtyzB4H8BQgEuJwhmVQqVHB9O4+MNDJCCAcpc3Aoo=
github.com/rabbitmq/amqp091-go v1.9.0/go.mod h1:+jPrT9iY2eLjRaMSRHUhc3z14E/l85kv/f+6luSD3pc=
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
@@ -100,8 +101,8 @@ github.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6ke
github.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4=
github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
github.com/sashabaranov/go-openai v1.17.9 h1:QEoBiGKWW68W79YIfXWEFZ7l5cEgZBV4/Ow3uy+5hNY=
github.com/sashabaranov/go-openai v1.17.9/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
github.com/sashabaranov/go-openai v1.17.11 h1:XVr00J8JymJVx8Hjbh/5mG0V4PQHRarBU3v7k2x6MR0=
github.com/sashabaranov/go-openai v1.17.11/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=
@@ -110,8 +111,8 @@ github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=
github.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
github.com/spf13/viper v1.18.0 h1:pN6W1ub/G4OfnM+NR9p7xP9R6TltLUzp5JG9yZD3Qg0=
github.com/spf13/viper v1.18.0/go.mod h1:EKmWIqdnk5lOcmR72yw6hS+8OPYcwD0jteitLMVB+yk=
github.com/spf13/viper v1.18.2 h1:LUXCnvUvSM6FXAsj6nnfc8Q2tp1dIgUfY9Kc8GsSOiQ=
github.com/spf13/viper v1.18.2/go.mod h1:EKmWIqdnk5lOcmR72yw6hS+8OPYcwD0jteitLMVB+yk=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
@@ -133,18 +134,20 @@ github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5t
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE=
go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=
go.uber.org/goleak v1.2.1 h1:NBol2c7O1ZokfZ0LEU9K6Whx/KnwvepVetCUhtKja4A=
go.uber.org/goleak v1.2.1/go.mod h1:qlT2yGI9QafXHhZZLxlSuNsMw3FFLxBr+tBRlmO1xH4=
go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
golang.org/x/arch v0.6.0 h1:S0JTfE48HbRj80+4tbvZDYsJ3tGv6BUU3XxyZ7CirAc=
golang.org/x/arch v0.6.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
golang.org/x/arch v0.7.0 h1:pskyeJh/3AmoQ8CPE95vxHLqp1G1GfGNXTmcl9NEKTc=
golang.org/x/arch v0.7.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=
golang.org/x/crypto v0.16.0 h1:mMMrFzRSCF0GvB7Ne27XVtVAaXLrPmgPC7/v0tkwHaY=
golang.org/x/crypto v0.16.0/go.mod h1:gCAAfMLgwOJRpTjQ2zCCt2OcSfYMTeZVSRtQlPC7Nq4=
golang.org/x/exp v0.0.0-20231206192017-f3f8817b8deb h1:c0vyKkb6yr3KR7jEfJaOSv4lG7xPkbN6r52aJz1d8a8=
golang.org/x/exp v0.0.0-20231206192017-f3f8817b8deb/go.mod h1:iRJReGqOEeBhDZGkGbynYwcHlctCvnjTYIamk7uXpHI=
golang.org/x/crypto v0.18.0 h1:PGVlW0xEltQnzFZ55hkuX5+KLyrMYhHld1YHO4AKcdc=
golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg=
golang.org/x/exp v0.0.0-20240112132812-db7319d0e0e3 h1:hNQpMuAJe5CtcUqCXaWga3FHu+kQvCqcsoVaQgSV60o=
golang.org/x/exp v0.0.0-20240112132812-db7319d0e0e3/go.mod h1:idGWGoKP1toJGkd5/ig9ZLuPcZBC3ewk7SzmH0uou08=
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
@@ -153,8 +156,8 @@ golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
golang.org/x/net v0.19.0 h1:zTwKpTd2XuCqf8huc7Fo2iSy+4RHPd10s4KzeTnVr1c=
golang.org/x/net v0.19.0/go.mod h1:CfAk/cbD4CthTvqiEl8NpboMuiuOYsAr/7NOjZJtv1U=
golang.org/x/net v0.20.0 h1:aCL9BSgETF1k+blQaYUBx9hJ9LOGP3gAVemcZlf1Kpo=
golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY=
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -167,8 +170,8 @@ golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=
golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU=
golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
@@ -190,10 +193,8 @@ golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtn
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
google.golang.org/protobuf v1.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8=
google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I=
google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=

View File

@@ -2,7 +2,7 @@ package initialization
import (
"go-wechat/common/current"
"go-wechat/model"
"go-wechat/model/model"
plugin "go-wechat/plugin"
"go-wechat/plugin/plugins"
"go-wechat/service"
@@ -26,13 +26,14 @@ func Plugin() {
// 私聊指令消息
dispatcher.RegisterHandler(func(m *model.Message) bool {
// 私聊消息 或 群聊艾特机器人并且以/开头的消息
return (m.IsPrivateText() || (m.IsAt() && m.CleanContentStartWith("/"))) && service.CheckIsEnableCommand(m.FromUser)
isGroupAt := m.IsAt() && !m.IsAtAll()
return (m.IsPrivateText() || isGroupAt) && m.CleanContentStartWith("/") && service.CheckIsEnableCommand(m.FromUser)
}, plugins.Command)
// AI消息插件
dispatcher.RegisterHandler(func(m *model.Message) bool {
// 群内@或者私聊文字消息
return m.IsAt() || m.IsPrivateText()
return (m.IsAt() && !m.IsAtAll()) || m.IsPrivateText()
}, plugins.AI)
// 欢迎新成员

View File

@@ -4,7 +4,7 @@ import (
"github.com/go-resty/resty/v2"
"go-wechat/common/current"
"go-wechat/config"
"go-wechat/model"
model2 "go-wechat/model/model"
"log"
)
@@ -12,7 +12,7 @@ import (
// @description: 初始化微信机器人信息
func InitWechatRobotInfo() {
// 获取数据
var base model.Response[model.RobotUserInfo]
var base model2.Response[model2.RobotUserInfo]
_, err := resty.New().R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetResult(&base).

26
main.go
View File

@@ -4,6 +4,7 @@ import (
"github.com/gin-gonic/gin"
"go-wechat/config"
"go-wechat/initialization"
"go-wechat/mq"
"go-wechat/router"
"go-wechat/tasks"
"go-wechat/tcpserver"
@@ -20,6 +21,7 @@ func init() {
initialization.InitWechatRobotInfo() // 初始化机器人信息
initialization.Plugin() // 注册插件
tasks.InitTasks() // 初始化定时任务
mq.Init() // 初始化MQ
}
func main() {
@@ -38,11 +40,25 @@ func main() {
// 自定义模板引擎函数
app.SetFuncMap(template.FuncMap{
"checkSwap": func(flag bool) string {
if flag {
return "swap-active"
"codeToChinese": func(code string) string {
switch code {
case "friend":
return "好友列表"
case "group":
return "群组列表"
case "index":
return "首页"
case "assistant":
return "AI角色"
default:
return "其他页面"
}
return ""
},
"boolToChinese": func(flag bool) string {
if flag {
return "是"
}
return "否"
},
})
@@ -56,7 +72,7 @@ func main() {
return
}
// 404直接跳转到首页
ctx.Redirect(302, "/index.html")
ctx.Redirect(302, "/404.html")
})
app.NoMethod(func(ctx *gin.Context) {
ctx.String(http.StatusMethodNotAllowed, "不支持的请求方式")

View File

@@ -0,0 +1,39 @@
package entity
import (
"github.com/google/uuid"
"go-wechat/common/types"
"gorm.io/gorm"
"strings"
)
// AiAssistant
// @description: AI助手表
type AiAssistant struct {
Id string `json:"id" gorm:"type:varchar(32);primarykey"`
CreatedAt types.DateTime `json:"createdAt"`
Name string `json:"name" gorm:"type:varchar(10);not null;comment:'名称'"`
Personality string `json:"personality" gorm:"type:varchar(999);not null;comment:'人设'"`
Model string `json:"model" gorm:"type:varchar(50);not null;comment:'使用的模型'"`
Enable bool `json:"enable" gorm:"type:tinyint(1);not null;default:1;comment:'是否启用'"`
}
// TableName
// @description: 表名
// @receiver AiAssistant
// @return string
func (AiAssistant) TableName() string {
return "t_ai_assistant"
}
// BeforeCreate
// @description: 创建数据库对象之前生成UUID
// @receiver m
// @param *gorm.DB
// @return err
func (m *AiAssistant) BeforeCreate(*gorm.DB) (err error) {
if m.Id == "" {
m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
}
return
}

49
model/entity/friend.go Normal file
View File

@@ -0,0 +1,49 @@
package entity
import (
"time"
)
// Friend
// @description: 好友列表
type Friend struct {
Wxid string `json:"wxid"` // 微信原始Id
CustomAccount string `json:"customAccount"` // 微信号
Nickname string `json:"nickname"` // 昵称
Pinyin string `json:"pinyin"` // 昵称拼音大写首字母
PinyinAll string `json:"pinyinAll"` // 昵称全拼
LastActive time.Time `json:"lastActive"` // 最后活跃时间
EnableAi bool `json:"enableAI" gorm:"type:tinyint(1) default 0 not null"` // 是否使用AI
AiModel string `json:"aiModel"` // AI模型
Prompt string `json:"prompt"` // 提示词
EnableChatRank bool `json:"enableChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否使用聊天排行
EnableWelcome bool `json:"enableWelcome" gorm:"type:tinyint(1) default 0 not null"` // 是否启用迎新
EnableSummary bool `json:"enableSummary" gorm:"type:tinyint(1) default 0 not null"` // 是否启用总结
EnableNews bool `json:"enableNews" gorm:"type:tinyint(1) default 0 not null"` // 是否启用新闻
ClearMember int `json:"clearMember"` // 清理成员配置(多少天未活跃的)
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常
}
func (Friend) TableName() string {
return "t_friend"
}
// GroupUser
// @description: 群成员
type GroupUser struct {
GroupId string `json:"groupId"` // 群Id
Wxid string `json:"wxid"` // 微信Id
Account string `json:"account"` // 账号
HeadImage string `json:"headImage"` // 头像
Nickname string `json:"nickname"` // 昵称
IsMember bool `json:"isMember" gorm:"type:tinyint(1) default 0 not null"` // 是否群成员
IsAdmin bool `json:"isAdmin" gorm:"type:tinyint(1) default 0 not null"` // 是否群主
JoinTime time.Time `json:"joinTime"` // 加入时间
LastActive time.Time `json:"lastActive"` // 最后活跃时间
LeaveTime *time.Time `json:"leaveTime"` // 离开时间
SkipChatRank bool `json:"skipChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否跳过聊天排行
}
func (GroupUser) TableName() string {
return "t_group_user"
}

View File

@@ -2,6 +2,7 @@ package model
import (
"encoding/xml"
"github.com/duke-git/lancet/v2/slice"
"go-wechat/types"
"regexp"
"strings"
@@ -31,6 +32,21 @@ type systemMsgDataXml struct {
Type string `xml:"type,attr"`
}
// atMsgDataXml
// @description: 微信@消息的xml结构
type atMsgDataXml struct {
XMLName xml.Name `xml:"msgsource"`
Text string `xml:",chardata"`
AtUserList string `xml:"atuserlist"`
Silence string `xml:"silence"`
MemberCount string `xml:"membercount"`
Signature string `xml:"signature"`
TmpNode struct {
Text string `xml:",chardata"`
PublisherID string `xml:"publisher-id"`
} `xml:"tmp_node"`
}
// sysMsg
// @description: 消息主体
type sysMsg struct{}
@@ -72,7 +88,13 @@ func (m Message) IsRevokeMsg() bool {
// @receiver m
// @return bool
func (m Message) IsNewUserJoin() bool {
sysFlag := m.Type == types.MsgTypeSys && strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
if m.Type != types.MsgTypeSys {
return false
}
isInvitation := strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
isScanQrCode := strings.Contains(m.Content, "通过扫描") && strings.Contains(m.Content, "加入群聊")
sysFlag := isInvitation || isScanQrCode
if sysFlag {
return true
}
@@ -81,7 +103,7 @@ func (m Message) IsNewUserJoin() bool {
if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
return false
}
return m.Type == types.MsgTypeSys && d.Type == "delchatroommember"
return d.Type == "delchatroommember"
}
// IsAt
@@ -92,6 +114,22 @@ func (m Message) IsAt() bool {
return strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你")
}
// IsAtAll
// @description: 是否是At所有人的消息
// @receiver m
// @return bool
func (m Message) IsAtAll() bool {
// 解析raw里面的xml
var d atMsgDataXml
if err := xml.Unmarshal([]byte(m.Signature), &d); err != nil {
return false
}
// 转换@用户列表为数组
atUserList := strings.Split(d.AtUserList, ",")
// 判断是否包含@所有人
return slice.Contain(atUserList, "notify@all")
}
// IsPrivateText
// @description: 是否是私聊消息
// @receiver m
@@ -125,3 +163,15 @@ func (m Message) CleanContentStartWith(prefix string) bool {
return strings.HasPrefix(content, prefix)
}
// IsInvitationJoinGroup
// @description: 是否是邀请入群消息
// @receiver m
// @return bool
func (m Message) IsInvitationJoinGroup() bool {
if m.Type == types.MsgTypeApp {
// 解析xml
}
return false
}

18
model/model/news.go Normal file
View File

@@ -0,0 +1,18 @@
package model
// MorningPost
// @description: 每日早报返回结构体
type MorningPost struct {
Code int `json:"code"`
Msg string `json:"msg"`
Data struct {
Date string `json:"date"` // 新闻日期
News []string `json:"news"` // 新闻标题文字版
WeiYu string `json:"weiyu"` // 微语,就是一句屁话
Image string `json:"image"` // 早报完整图片
HeadImage string `json:"head_image"` // 早报头部图片
} `json:"data"`
Time int `json:"time"`
Usage int `json:"usage"`
LogId string `json:"log_id"`
}

41
model/vo/friend.go Normal file
View File

@@ -0,0 +1,41 @@
package vo
import (
"go-wechat/common/types"
)
// FriendItem
// @description: 好友列表数据
type FriendItem struct {
CustomAccount string // 微信号
Nickname string // 昵称
Pinyin string // 昵称拼音大写首字母
PinyinAll string // 昵称全拼
Wxid string // 微信原始Id
LastActive types.DateTime // 最后活跃时间
EnableAi bool // 是否使用AI
AiModel string // AI模型
Prompt string // AI助手或者自定义提示词
EnableChatRank bool // 是否使用聊天排行
EnableWelcome bool // 是否使用迎新
EnableCommand bool // 是否启用指令
EnableSummary bool // 是否启用总结
EnableNews bool // 是否启用新闻
ClearMember int // 清理成员配置(多少天未活跃的)
IsOk bool // 是否还在通讯库(群聊是要还在群里也算)
}
// GroupUserItem
// @description: 群成员列表数据
type GroupUserItem struct {
Wxid string `json:"wxid"` // 微信Id
Account string `json:"account"` // 账号
HeadImage string `json:"headImage"` // 头像
Nickname string `json:"nickname"` // 昵称
IsMember bool `json:"isMember" ` // 是否群成员
IsAdmin bool `json:"isAdmin"` // 是否群主
JoinTime types.DateTime `json:"joinTime"` // 加入时间
LastActive types.DateTime `json:"lastActive"` // 最后活跃时间
LeaveTime types.DateTime `json:"leaveTime"` // 离开时间
SkipChatRank bool `json:"skipChatRank" ` // 是否跳过聊天排行
}

8
model/vo/message.go Normal file
View File

@@ -0,0 +1,8 @@
package vo
// TextMessageItem
// @description: 文字消息
type TextMessageItem struct {
Nickname string `json:"nickname"`
Message string `json:"message"`
}

46
mq/handler.go Normal file
View File

@@ -0,0 +1,46 @@
package mq
import (
"encoding/json"
"go-wechat/common/current"
"go-wechat/model/model"
"go-wechat/types"
"log"
"strings"
)
// parse
// @description: 解析消息
// @param msg
func parse(msg []byte) {
var m model.Message
if err := json.Unmarshal(msg, &m); err != nil {
log.Printf("消息解析失败: %v", err)
log.Printf("消息内容: %d -> %v", len(msg), string(msg))
return
}
// 记录原始数据
m.Raw = string(msg)
// 提取出群成员信息
// Sys类型的消息正文不包含微信 Id所以不需要处理
if m.IsGroup() && m.Type != types.MsgTypeSys {
// 群消息,处理一下消息和发信人
groupUser := strings.Split(m.Content, "\n")[0]
groupUser = strings.ReplaceAll(groupUser, ":", "")
// 如果两个id一致说明是系统发的
if m.FromUser != groupUser {
m.GroupUser = groupUser
}
// 用户的操作单独提出来处理一下
m.Content = strings.Join(strings.Split(m.Content, "\n")[1:], "\n")
}
log.Printf("收到新微信消息\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", m.FromUser, m.GroupUser, m.Type, m.Content)
// 插件不为空,开始执行
if p := current.GetRobotMessageHandler(); p != nil {
p(&m)
}
return
}

118
mq/rabbitmq.go Normal file
View File

@@ -0,0 +1,118 @@
package mq
import (
amqp "github.com/rabbitmq/amqp091-go"
"go-wechat/config"
"log"
)
// MQ连接对象
var conn *amqp.Connection
var channel *amqp.Channel
// 交换机名称
const exchangeName = "wechat-message"
// Init
// @description: 初始化MQ
func Init() {
if !config.Conf.Mq.Enable {
log.Println("未启用MQ")
return
}
// 读取MQ连接配置
mqUrl := config.Conf.Mq.RabbitMQ.GetURL()
if mqUrl == "" {
log.Panicf("MQ配置异常")
}
var err error
// 创建MQ连接
if conn, err = amqp.Dial(mqUrl); err != nil {
log.Panicf("RabbitMQ连接失败: %s", err)
}
//获取channel
if channel, err = conn.Channel(); err != nil {
log.Panicf("打开Channel失败: %s", err)
}
log.Println("RabbitMQ连接成功")
go Receive()
log.Println("开始监听消息")
}
// Receive
// @description: 接收消息
func Receive() (err error) {
// 创建交换机
if err = channel.ExchangeDeclare(
exchangeName,
"fanout",
true,
false,
//true表示这个exchange不可以被client用来推送消息仅用来进行exchange和exchange之间的绑定
false,
false,
nil,
); err != nil {
log.Printf("声明Exchange失败: %s", err)
return
}
// 创建队列
var q amqp.Queue
q, err = channel.QueueDeclare(
"", //随机生产队列名称
false,
false,
true,
false,
nil,
)
if err != nil {
log.Printf("无法声明Queue: %s", err)
return
}
// 绑定队列到 exchange 中
err = channel.QueueBind(
q.Name,
//在pub/sub模式下这里的key要为空
"",
exchangeName,
false,
nil)
if err != nil {
log.Printf("绑定队列失败: %s", err)
return
}
// 消费消息
var messages <-chan amqp.Delivery
messages, err = channel.Consume(
q.Name,
"",
false, // 不自动ack手动处理这样即使消费者挂掉消息也不会丢失
false,
false,
false,
nil,
)
if err != nil {
log.Printf("无法使用队列: %s", err)
return
}
for {
msg, ok := <-messages
if !ok {
log.Printf("获取消息失败")
return Receive()
}
log.Printf("收到消息: %s", msg.Body)
parse(msg.Body)
// ACK消息
if err = msg.Ack(true); err != nil {
log.Printf("ACK消息失败: %s", err)
continue
}
}
}

View File

@@ -1,7 +1,7 @@
package plugin
import (
"go-wechat/model"
"go-wechat/model/model"
)
// MessageHandler 消息处理函数

View File

@@ -8,7 +8,7 @@ import (
"go-wechat/client"
"go-wechat/common/current"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/model/entity"
"go-wechat/plugin"
"go-wechat/service"
"go-wechat/types"
@@ -28,9 +28,13 @@ func AI(m *plugin.MessageContext) {
}
// 取出所有启用了AI的好友或群组
var count int64
client.MySQL.Model(&entity.Friend{}).Where("enable_ai IS TRUE").Where("wxid = ?", m.FromUser).Count(&count)
if count < 1 {
var friendInfo entity.Friend
client.MySQL.Where("wxid = ?", m.FromUser).First(&friendInfo)
if friendInfo.Wxid == "" {
return
}
// 判断有没有启用AI
if !friendInfo.EnableAi {
return
}
@@ -42,13 +46,34 @@ func AI(m *plugin.MessageContext) {
m.Content = strings.Replace(m.Content, matches[0], "", 1)
}
// 处理预设角色,默认是配置文件里的,如果数据库配置不为空,则使用数据库配置
prompt := config.Conf.Ai.Personality
var dbPrompt entity.AiAssistant
if friendInfo.Prompt != "" {
// 取出配置的角色
client.MySQL.First(&dbPrompt, "id = ?", friendInfo.Prompt)
if dbPrompt.Id != "" {
prompt = dbPrompt.Personality
}
}
// 配置模型
chatModel := openai.GPT3Dot5Turbo0613
if friendInfo.AiModel != "" {
chatModel = friendInfo.AiModel
} else if dbPrompt.Model != "" {
chatModel = dbPrompt.Model
} else if config.Conf.Ai.Model != "" {
chatModel = config.Conf.Ai.Model
}
// 组装消息体
messages := make([]openai.ChatCompletionMessage, 0)
if config.Conf.Ai.Personality != "" {
if prompt != "" {
// 填充人设
messages = append(messages, openai.ChatCompletionMessage{
Role: openai.ChatMessageRoleSystem,
Content: config.Conf.Ai.Personality,
Content: prompt,
})
}
@@ -91,12 +116,6 @@ func AI(m *plugin.MessageContext) {
Content: m.Content,
})
// 配置模型
chatModel := openai.GPT3Dot5Turbo0613
if config.Conf.Ai.Model != "" {
chatModel = config.Conf.Ai.Model
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {
@@ -117,6 +136,12 @@ func AI(m *plugin.MessageContext) {
return
}
// 返回消息为空
if resp.Choices[0].Message.Content == "" {
utils.SendMessage(m.FromUser, m.GroupUser, "AI似乎抽风了没有告诉我你需要的回答~", 0)
return
}
// 保存一下AI 返回的消息,消息 Id 使用传入 Id 的负数
var replyMessage entity.Message
replyMessage.MsgId = -m.MsgId

View File

@@ -42,6 +42,8 @@ func Command(m *plugin.MessageContext) {
command.LeiGodCmd(m.FromUser, msgArray[1], msgArray[2:]...)
case "/肯德基", "/kfc":
command.KfcCrazyThursdayCmd(m.FromUser)
case "/ai":
command.AiCmd(m.FromUser, m.GroupUser, msgArray[1])
default:
utils.SendMessage(m.FromUser, m.GroupUser, "指令错误", 0)
}

View File

@@ -0,0 +1,65 @@
package command
import (
"fmt"
"go-wechat/client"
"go-wechat/model/entity"
"go-wechat/utils"
"log"
"strings"
)
// AiCmd
// @description: AI指令
// @param userId
// @param groupUserId
// @param cmd
func AiCmd(userId, groupUserId, cmd string) {
// 判断发信人是不是群主
can := false
if strings.Contains(userId, "@chatroom") {
// 判断是不是群主
err := client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", userId).
Where("wxid = ?", groupUserId).
Pluck("is_admin", &can).Error
if err != nil {
log.Printf("查询群主失败: %v", err)
return
}
}
if !can {
utils.SendMessage(userId, groupUserId, "您不是群主,无法使用指令", 0)
return
}
var err error
replyMsg := "操作成功"
switch cmd {
case "enable", "启用", "打开":
err = setAiEnable(userId, true)
case "disable", "停用", "禁用", "关闭":
err = setAiEnable(userId, false)
default:
replyMsg = "指令错误"
}
if err != nil {
log.Printf("AI指令执行失败: %v", err)
replyMsg = fmt.Sprintf("指令执行错误: %v", err)
}
utils.SendMessage(userId, groupUserId, replyMsg, 0)
}
// setAiEnable
// @description: 设置AI启用状态
// @param userId
// @param enable
// @return err
func setAiEnable(userId string, enable bool) (err error) {
// 更新
err = client.MySQL.Model(&entity.Friend{}).
Where("wxid = ?", userId).
Update("enable_ai", enable).Error
return
}

View File

@@ -23,6 +23,13 @@ option: 指令选项,可选值:
#2. 肯德基疯狂星期四文案
/kfc、/肯德基
#3. AI助手
/ai option
option: 指令选项,可选值:
启用: '启用'、'打开'、'enable'
停用: '停用'、'禁用'、'关闭'、'disable'
`
utils.SendMessage(m.FromUser, m.GroupUser, str, 0)

View File

@@ -4,6 +4,7 @@ import (
"github.com/go-resty/resty/v2"
"go-wechat/utils"
"log"
"net/http"
)
// KfcCrazyThursdayCmd
@@ -33,8 +34,9 @@ func kfcApi1() string {
res := resty.New()
resp, err := res.R().
Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
if err != nil {
log.Panicf("KFC接口1文案获取失败: %s", err.Error())
if err != nil || resp.StatusCode() != http.StatusOK {
log.Printf("KFC接口1文案获取失败: %v", err)
return ""
}
log.Printf("KFC接口1文案获取结果: %s", resp.String())
return resp.String()
@@ -58,8 +60,9 @@ func kfcApi2() string {
resp, err := res.R().
SetResult(&resData).
Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
if err != nil {
log.Panicf("KFC接口2文案获取失败: %s", err.Error())
if err != nil || resp.StatusCode() != http.StatusOK {
log.Printf("KFC接口2文案获取失败: %v", err)
return ""
}
log.Printf("KFC接口2文案获取结果: %s", resp.String())
if resData.Data.Msg != "" {
@@ -84,8 +87,9 @@ func kfcApi3() string {
resp, err := res.R().
SetResult(&resData).
Post("https://api.pearktrue.cn/api/kfc")
if err != nil {
log.Panicf("KFC接口3文案获取失败: %s", err.Error())
if err != nil || resp.StatusCode() != http.StatusOK {
log.Printf("KFC接口3文案获取失败: %v", err)
return ""
}
log.Printf("KFC接口3文案获取结果: %s", resp.String())
if resData.Text != "" {

View File

@@ -5,10 +5,10 @@ import (
"errors"
"fmt"
"go-wechat/client"
"go-wechat/entity"
"go-wechat/model"
"go-wechat/model/entity"
"go-wechat/model/model"
"go-wechat/model/vo"
"go-wechat/utils"
"go-wechat/vo"
"gorm.io/gorm"
"log"
"strings"

View File

@@ -1,7 +1,7 @@
package plugins
import (
"go-wechat/entity"
"go-wechat/model/entity"
"go-wechat/plugin"
"go-wechat/service"
"time"

View File

@@ -3,7 +3,7 @@ package plugins
import (
"go-wechat/client"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/model/entity"
"go-wechat/plugin"
"go-wechat/utils"
)

View File

@@ -14,6 +14,8 @@ vim config.yaml # 编辑配置文件,内容如下,最新配置请参考项
wechat:
# 微信HOOK接口地址
host: wechat:19088
# 微信容器映射出来的vnc页面地址没有就不填
vncUrl: http://192.168.1.175:19087/vnc_lite.html
# 是否在启动的时候自动设置hook服务的回调
autoSetCallback: true
# 回调IP如果是Docker运行本参数必填如果Docker修改了映射格式为 ip:port如果使用项目提供的docker-compsoe.yaml文件启动可以填`auto`
@@ -46,7 +48,7 @@ version: '3.9'
services:
wechat:
image: lxh01/wxhelper-docker:3.9.5.81
image: lxh01/wxhelper-docker:3.9.5.81-v11-novnc # 如果不用noVNC网页就删掉后面的-novnc
container_name: gw-wechat
restart: unless-stopped
environment:
@@ -54,8 +56,9 @@ services:
volumes:
- ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
ports:
- "8080:8080"
- "19088:19088"
- "19086:5900" # vnc端口
- "19087:8080" # noVNC端口
- "19088:19088" # 微信HOOK端口
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
interval: 60s
@@ -67,9 +70,6 @@ services:
image: mysql:8
container_name: gw-db
restart: unless-stopped
depends_on:
wechat:
condition: service_healthy
environment:
- MYSQL_ROOT_PASSWORD=wechat
- MYSQL_USER=wechat
@@ -85,6 +85,7 @@ services:
restart: unless-stopped
depends_on:
- mysql
- wechat
volumes:
# 配置文件请参阅项目根目录的config.yaml文件
- ./config/config.yaml:/app/config.yaml
@@ -99,3 +100,7 @@ services:
docker-compose up -d # 老版本
docker compose up -d # 新版本
```
## 注意事项
1. 宿主机必须是`debian`系,因为`wine`的玄学`BUG`,其他系统可能会出现各种问题
2. 登录微信可以用`vnc viewer`连接`5900`端口或者访问`noVNC`端口的`vnc_lite.html`页面进行扫码登录

View File

@@ -9,21 +9,49 @@ import (
// @description: 初始化路由
// @param g
func Init(g *gin.Engine) {
g.GET("/", func(ctx *gin.Context) {
// 重定向到index.html
ctx.Redirect(302, "/index.html")
})
//g.GET("/", func(ctx *gin.Context) {
// // 重定向到index.html
// ctx.Redirect(302, "/index.html")
//})
g.GET("/index.html", app.Index) // 首页
g.GET("/test.html", func(ctx *gin.Context) {
ctx.HTML(200, "test.html", nil)
})
g.GET("/index.html", app.Index) // 首页
g.GET("/friend.html", app.Friend) // 好友列表
g.GET("/group.html", app.Group) // 群组列表
g.GET("/assistant.html", app.Assistant) // AI角色
//g.GET("/404.html", app.PageNotFound) // 群组列表
// 接口
api := g.Group("/api")
api.PUT("/ai/status", app.ChangeEnableAiStatus) // 修改是否开启AI状态
api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态
api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态
api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus) // 修改是否跳过水群排行榜状态
api.GET("/group/users", app.GetGroupUsers) // 获取群成员列表
contact := api.Group("/contact") // 通讯录
{
contact.GET("/friend", app.GetFriends) // 获取好友列表
contact.GET("/group", app.GetGroups) // 获取群列表
profile := contact.Group("/profile") // 配置
{
profile.PUT("/ai/status", app.ChangeEnableAiStatus) // 修改是否开启AI状态
profile.POST("/ai/model", app.ChangeUseAiModel) // 修改使用的AI模型
profile.POST("/ai/assistant", app.ChangeUseAiAssistant) // 修改使用的AI助手
profile.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新
profile.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令
profile.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报
profile.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜
profile.PUT("/summary/status", app.ChangeEnableSummaryStatus) // 修改是否开启群聊总结
profile.PUT("/clearmembers", app.AutoClearMembers) // 自动清理群成员
}
group := contact.Group("/group") // 群组
{
group.GET("/users", app.GetGroupUsers) // 获取群成员列表
group.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus) // 修改是否跳过水群排行榜状态
}
}
system := api.Group("/system") // 系统设置
{
system.GET("/assistant", app.GetAssistants) // 获取AI助手列表
system.POST("/assistant", app.SaveAssistant) // 保存AI助手
system.GET("/models", app.GetAiModels) // 获取AI模型列表
}
}

14
service/aiassistant.go Normal file
View File

@@ -0,0 +1,14 @@
package service
import (
"go-wechat/client"
"go-wechat/model/entity"
)
// GetAllAiAssistant
// @description: 取出所有AI助手
// @return records
func GetAllAiAssistant() (records []entity.AiAssistant, err error) {
err = client.MySQL.Order("created_at DESC").Find(&records).Error
return
}

View File

@@ -2,8 +2,9 @@ package service
import (
"go-wechat/client"
"go-wechat/entity"
"go-wechat/vo"
"go-wechat/model/entity"
"go-wechat/model/vo"
"log"
"strings"
)
@@ -16,10 +17,11 @@ func GetAllFriend() (friends, groups []vo.FriendItem, err error) {
var records []vo.FriendItem
err = client.MySQL.
Table("t_friend AS tf").
Joins("LEFT JOIN t_message AS tm ON tf.wxid = tm.from_user").
Select("tf.*", "MAX(tm.create_at) AS last_active_time").
Group("tf.wxid").
Order("last_active_time DESC").
//Joins("LEFT JOIN t_message AS tm ON tf.wxid = tm.from_user").
//Select("tf.*", "MAX(tm.create_at) AS last_active").
Select("tf.*").
//Group("tf.wxid").
Order("tf.last_active DESC").
Find(&records).Error
if err != nil {
return
@@ -47,7 +49,40 @@ func GetAllEnableAI() (records []entity.Friend, err error) {
// @return records
// @return err
func GetAllEnableChatRank() (records []entity.Friend, err error) {
err = client.MySQL.Where("enable_chat_rank = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
err = client.MySQL.Where("enable_chat_rank = ?", 1).
Where("is_ok IS TRUE").
Where("wxid LIKE '%@chatroom'").
Find(&records).Error
return
}
// GetAllEnableSummary
// @description: 取出所有启用了总结的群组
// @return records
// @return err
func GetAllEnableSummary() (records []entity.Friend, err error) {
err = client.MySQL.Where("enable_summary = ?", 1).
Where("is_ok IS TRUE").
Where("wxid LIKE '%@chatroom'").
Find(&records).Error
return
}
// GetAllEnableNews
// @description: 取出所有启用了新闻的好友或群组
// @return records
// @return err
func GetAllEnableNews() (records []entity.Friend, err error) {
err = client.MySQL.Where("enable_news = ?", 1).Where("is_ok IS TRUE").Find(&records).Error
return
}
// GetAllEnableClearGroup
// @description: 获取所有需要清理成员的群组
// @return records
// @return err
func GetAllEnableClearGroup() (records []entity.Friend, err error) {
err = client.MySQL.Where("clear_member > 0").Where("is_ok IS TRUE").Find(&records).Error
return
}
@@ -60,3 +95,27 @@ func CheckIsEnableCommand(userId string) (flag bool) {
client.MySQL.Model(&entity.Friend{}).Where("enable_command = 1").Where("wxid = ?", userId).Count(&coo)
return coo > 0
}
// updateLastActive
// @description: 更新最后活跃时间
// @param msg
func updateLastActive(msg entity.Message) {
var err error
// 如果是群,更新群成员最后活跃时间
if strings.HasSuffix(msg.FromUser, "@chatroom") {
err = client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", msg.FromUser).
Where("wxid = ?", msg.GroupUser).
Update("last_active", msg.CreateAt).Error
if err != nil {
log.Printf("更新群成员最后活跃时间失败, 错误信息: %v", err)
}
}
// 更新群或者好友活跃时间
err = client.MySQL.Model(&entity.Friend{}).
Where("wxid = ?", msg.FromUser).
Update("last_active", msg.CreateAt).Error
if err != nil {
log.Printf("更新群或者好友活跃时间失败, 错误信息: %v", err)
}
}

View File

@@ -2,7 +2,7 @@ package service
import (
"go-wechat/client"
"go-wechat/vo"
"go-wechat/model/vo"
)
// GetGroupUsersByGroupId
@@ -13,10 +13,9 @@ import (
func GetGroupUsersByGroupId(groupId string) (records []vo.GroupUserItem, err error) {
err = client.MySQL.
Table("t_group_user AS tgu").
Joins("LEFT JOIN t_message AS tm ON tm.from_user = tgu.group_id AND tm.group_user = tgu.wxid").
//Select("tgu.wxid", "tgu.nickname", "tgu.head_image", "tgu.is_member", "tgu.leave_time",
// "tgu.skip_chat_rank", "MAX(tm.create_at) AS last_active_time").
Select("tgu.*", "MAX(tm.create_at) AS last_active_time").
//Joins("LEFT JOIN t_message AS tm ON tm.from_user = tgu.group_id AND tm.group_user = tgu.wxid").
//Select("tgu.*", "MAX(tm.create_at) AS last_active").
Select("tgu.*").
Where("tgu.group_id = ?", groupId).
Group("tgu.group_id, tgu.wxid").
Order("tgu.join_time DESC").

View File

@@ -2,7 +2,8 @@ package service
import (
"go-wechat/client"
"go-wechat/entity"
"go-wechat/model/entity"
"go-wechat/model/vo"
"log"
"os"
"strconv"
@@ -32,4 +33,50 @@ func SaveMessage(msg entity.Message) {
log.Printf("消息入库失败, 错误信息: %v", err)
}
log.Printf("消息入库成功消息Id: %d", msg.MsgId)
// 更新最后活跃时间
// 只处理收到的消息
if msg.MsgId > 1 {
go updateLastActive(msg)
}
}
// GetTextMessagesById
// @description: 根据群id或者用户Id获取消息
// @param id
// @return records
// @return err
func GetTextMessagesById(id string) (records []vo.TextMessageItem, err error) {
// APP消息类型
appMsgList := []string{"57", "4", "5", "6"}
// 这个查询子句抽出来写,方便后续扩展
selectStr := `CASE
WHEN tm.type = 49 THEN
CASE
WHEN EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) = '57' THEN
EXTRACTVALUE ( tm.content, "/msg/appmsg/title" )
WHEN EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) = '5' THEN
CONCAT("网页分享消息,标题: ", EXTRACTVALUE (tm.content, "/msg/appmsg/title"), ",描述:", EXTRACTVALUE (tm.content, "/msg/appmsg/des"))
WHEN EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) = '4' THEN
CONCAT("网页分享消息,标题: ", EXTRACTVALUE (tm.content, "/msg/appmsg/title"), ",描述:", EXTRACTVALUE (tm.content, "/msg/appmsg/des"))
WHEN EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) = '6' THEN
CONCAT("文件消息,文件名: ", EXTRACTVALUE (tm.content, "/msg/appmsg/title"))
ELSE EXTRACTVALUE ( tm.content, "/msg/appmsg/des" )
END ELSE tm.content
END`
tx := client.MySQL.
Table("`t_message` AS tm").
Joins("LEFT JOIN t_group_user AS tgu ON tm.group_user = tgu.wxid AND tgu.group_id = tm.from_user").
Select("tgu.nickname", selectStr+" AS message").
Where("tm.`from_user` = ?", id).
Where(`(tm.type = 1 OR ( tm.type = 49 AND EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) IN (?) ))`, appMsgList).
Where("DATE ( tm.create_at ) = DATE ( CURDATE() - INTERVAL 1 DAY )").
Where("tm.content NOT LIKE '#昨日水群排行榜%'").
Where("tm.content NOT LIKE '#昨日消息总结%'").
Order("tm.create_at ASC")
err = tx.Find(&records).Error
return
}

View File

@@ -0,0 +1,68 @@
package cleargroupuser
import (
"fmt"
"go-wechat/client"
"go-wechat/model/entity"
"go-wechat/service"
"go-wechat/utils"
"log"
"strings"
)
// ClearGroupUser
// @description: 清理群成员
func ClearGroupUser() {
groups, err := service.GetAllEnableClearGroup()
if err != nil {
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
return
}
for _, group := range groups {
// 获取需要清理的群成员Id
members := getNeedDeleteMembers(group.Wxid, group.ClearMember)
memberCount := len(members)
log.Printf("群[%s(%s)]需要清理的成员数量: %d", group.Nickname, group.Wxid, memberCount)
if memberCount < 1 {
continue
}
var memberMap = make(map[string]string)
var deleteIds = make([]string, 0)
for _, member := range members {
deleteIds = append(deleteIds, member.Wxid)
// 昵称为空取id后4位
if member.Nickname == "" {
member.Nickname = "无名氏_" + member.Wxid[len(member.Wxid)-4:]
}
memberMap[member.Nickname] = member.LastActive.Format("2006-01-02 15:04:05")
}
// 调用接口
utils.DeleteGroupMember(group.Wxid, strings.Join(deleteIds, ","), 0)
// 发送通知到群里
ms := make([]string, 0)
for k, v := range memberMap {
ms = append(ms, fmt.Sprintf("昵称:%s\n最后活跃时间%s", k, v))
}
msg := fmt.Sprintf("#清理群成员\n\n很遗憾地通知各位就在刚刚有%d名群友因活跃度不够暂时离开了我们希望还健在的群友引以为戒、保持活跃\n\n详细信息: \n%s",
memberCount, strings.Join(ms, "\n"))
utils.SendMessage(group.Wxid, "", msg, 0)
}
}
// getNeedDeleteMembers
// @description: 获取需要删除的群成员
// @param groupId 群Id
// @param days 需要清理的未活跃的天数
// @return members
func getNeedDeleteMembers(groupId string, days int) (members []entity.GroupUser) {
err := client.MySQL.Model(&entity.GroupUser{}).Where("group_id = ?", groupId).
Where("is_member IS TRUE").
Where("DATEDIFF( NOW(), last_active ) >= ?", days).
Order("last_active DESC").
Find(&members).Error
if err != nil {
log.Printf("获取需要清理的群成员失败, 错误信息: %v", err)
}
return
}

View File

@@ -6,8 +6,9 @@ import (
"go-wechat/client"
"go-wechat/common/constant"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/model"
"go-wechat/model/entity"
model2 "go-wechat/model/model"
"go-wechat/utils"
"gorm.io/gorm"
"log"
"slices"
@@ -23,7 +24,7 @@ var hc = resty.New()
// Sync
// @description: 同步好友列表
func Sync() {
var base model.Response[[]model.FriendItem]
var base model2.Response[[]model2.FriendItem]
resp, err := hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
@@ -40,6 +41,9 @@ func Sync() {
nowIds := []string{}
// 新增的成员,用于通知给指定的人
var newItmes = make(map[string]string)
for _, friend := range base.Data {
if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") {
continue
@@ -60,23 +64,46 @@ func Sync() {
if count == 0 {
// 新增
err = tx.Create(&entity.Friend{
CustomAccount: friend.CustomAccount,
Nickname: friend.Nickname,
Pinyin: friend.Pinyin,
PinyinAll: friend.PinyinAll,
Wxid: friend.Wxid,
IsOk: true,
CustomAccount: friend.CustomAccount,
Nickname: friend.Nickname,
Pinyin: friend.Pinyin,
PinyinAll: friend.PinyinAll,
Wxid: friend.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,
ClearMember: 0,
LastActive: time.Now().Local(),
}).Error
if err != nil {
log.Printf("新增好友失败: %s", err.Error())
continue
}
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)
}
}
} else {
pm := map[string]any{
"nickname": friend.Nickname,
"custom_account": friend.CustomAccount,
"pinyin": friend.Pinyin,
"pinyin_all": friend.PinyinAll,
"is_ok": true,
}
err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Updates(pm).Error
if err != nil {
@@ -91,8 +118,35 @@ func Sync() {
}
}
// 通知有新成员
if len(newItmes) > 0 && config.Conf.System.NewFriendNotify.Enable {
// 组装成一句话
msg := []string{"#新好友通知\n"}
for wxId, nickname := range newItmes {
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)
}
}
}
// 清理不在列表中的好友
err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Update("is_ok", false).Error
clearPm := map[string]any{
"is_ok": false,
"enable_chat_rank": false,
"enable_welcome": false,
"enable_summary": false,
"enable_news": false,
"clear_member": false,
"enable_ai": false,
}
err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Updates(clearPm).Error
if err != nil {
log.Printf("清理好友失败: %s", err.Error())
}
log.Println("同步好友列表完成")
}
@@ -101,7 +155,7 @@ func Sync() {
// @description: 同步群成员
// @param gid
func syncGroupUsers(tx *gorm.DB, gid string) {
var baseResp model.Response[model.GroupUser]
var baseResp model2.Response[model2.GroupUser]
// 组装参数
param := map[string]any{
@@ -149,13 +203,15 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
if count == 0 {
// 新增
err = tx.Create(&entity.GroupUser{
GroupId: gid,
Account: cp.Account,
HeadImage: cp.HeadImage,
Nickname: cp.Nickname,
Wxid: cp.Wxid,
IsMember: true,
JoinTime: time.Now().Local(),
GroupId: gid,
Account: cp.Account,
HeadImage: cp.HeadImage,
Nickname: cp.Nickname,
Wxid: cp.Wxid,
IsMember: true,
IsAdmin: wxid == baseResp.Data.Admin,
JoinTime: time.Now().Local(),
LastActive: time.Now().Local(),
}).Error
if err != nil {
log.Printf("新增群成员失败: %s", err.Error())
@@ -168,6 +224,7 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
"head_image": cp.HeadImage,
"nickname": cp.Nickname,
"is_member": true,
"is_admin": wxid == baseResp.Data.Admin,
"leave_time": nil,
}
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Updates(pm).Error
@@ -185,8 +242,8 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
// @param wxid
// @return ent
// @return err
func getContactProfile(wxid string) (ent model.ContactProfile, err error) {
var baseResp model.Response[model.ContactProfile]
func getContactProfile(wxid string) (ent model2.ContactProfile, err error) {
var baseResp model2.Response[model2.ContactProfile]
// 组装参数
param := map[string]any{

36
tasks/news/news.go Normal file
View File

@@ -0,0 +1,36 @@
package news
import (
"fmt"
"go-wechat/service"
"go-wechat/utils"
"log"
"strings"
"time"
)
// DailyNews
// @description: 每日新闻
func DailyNews() {
groups, err := service.GetAllEnableNews()
if err != nil {
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
return
}
news := utils.NewsUtil().MorningPost()
if len(news) == 0 {
log.Println("每日早报获取失败")
return
}
newsStr := fmt.Sprintf("#每日早报\n\n又是新的一天了让我们康康一觉醒来世界又发生了哪些变化~\n\n%s", strings.Join(news, "\n"))
// 循环发送新闻
for _, group := range groups {
// 发送消息
utils.SendMessage(group.Wxid, "", newsStr, 0)
// 休眠一秒,防止频繁发送
time.Sleep(time.Second)
}
}

88
tasks/summary/summary.go Normal file
View File

@@ -0,0 +1,88 @@
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"
)
// AiSummary
// @description: AI总结群聊记录
func AiSummary() {
groups, err := service.GetAllEnableSummary()
if err != nil {
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
return
}
for _, group := range groups {
// 获取对话记录
var records []vo.TextMessageItem
if records, err = service.GetTextMessagesById(group.Wxid); err != nil {
log.Printf("获取群[%s]对话记录失败, 错误信息: %v", group.Wxid, err)
continue
}
if len(records) < 100 {
log.Printf("群[%s]对话记录不足100条跳过总结", group.Wxid)
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,
},
)
if err != nil {
log.Printf("群聊记录总结失败: %v", err.Error())
continue
}
// 返回消息为空
if resp.Choices[0].Message.Content == "" {
continue
}
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)
}
}

View File

@@ -3,7 +3,10 @@ package tasks
import (
"github.com/go-co-op/gocron"
"go-wechat/config"
"go-wechat/tasks/cleargroupuser"
"go-wechat/tasks/friends"
"go-wechat/tasks/news"
"go-wechat/tasks/summary"
"go-wechat/tasks/watergroup"
"log"
"time"
@@ -32,6 +35,15 @@ func InitTasks() {
if config.Conf.Task.WaterGroup.Cron.Month != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Month).Do(watergroup.Month)
}
if config.Conf.Task.WaterGroup.Cron.Year != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Year).Do(watergroup.Year)
}
}
// 群聊总结
if config.Conf.Task.GroupSummary.Enable {
log.Printf("群聊总结任务已启用,执行表达式: %s", config.Conf.Task.GroupSummary.Cron)
_, _ = s.Cron(config.Conf.Task.GroupSummary.Cron).Do(summary.AiSummary)
}
// 更新好友列表
@@ -40,6 +52,14 @@ func InitTasks() {
_, _ = s.Cron(config.Conf.Task.SyncFriends.Cron).Do(friends.Sync)
}
// 每日早报
if config.Conf.Task.News.Enable {
_, _ = s.Cron(config.Conf.Task.News.Cron).Do(news.DailyNews)
}
// 每天0点检查一次处理清理群成员
_, _ = s.Cron("0 0 * * *").Do(cleargroupuser.ClearGroupUser)
// 开启定时任务
s.StartAsync()
log.Println("定时任务初始化成功")

View File

@@ -2,7 +2,9 @@ package watergroup
import (
"fmt"
"go-wechat/client"
"go-wechat/config"
"go-wechat/model/entity"
"go-wechat/service"
"go-wechat/utils"
"log"
@@ -55,14 +57,48 @@ func dealMonth(gid string) {
log.Printf("上月群[%s]无对话记录", gid)
return
}
// 计算消息总数
var msgCount int64
for _, v := range records {
msgCount += v.Count
// 查询群成员总数
var groupUsers int64
err = client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", gid).
Where("is_member IS TRUE").
Count(&groupUsers).Error
if err != nil {
log.Printf("查询群成员总数失败, 错误信息: %v", err)
}
// 计算活跃度
showActivity := err == nil && groupUsers > 0
activity := "0.00"
if groupUsers > 0 {
activity = fmt.Sprintf("%.2f", (float64(len(records))/float64(groupUsers))*100)
}
// 计算消息总数、中位数、前十位消息总数
var msgCount, medianCount, topTenCount int64
for idx, v := range records {
msgCount += v.Count
if idx == (len(records)/2)-1 {
medianCount = v.Count
}
if len(records) > 10 && idx < 10 {
topTenCount += v.Count
}
}
// 计算活跃用户人均消息条数
avgMsgCount := int(float64(msgCount) / float64(len(records)))
// 组装消息总数推送信息
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ %s本群 %d 位朋友共产生 %d 条发言", monthStr, len(records), msgCount))
if showActivity {
m := fmt.Sprintf("🎭 活跃度: %s%%,人均消息条数: %d中位数: %d", activity, avgMsgCount, medianCount)
// 计算前十占比
if topTenCount > 0 {
m += fmt.Sprintf(",前十名占比: %.2f%%", float64(topTenCount)/float64(msgCount)*100)
}
notifyMsgs = append(notifyMsgs, m)
}
notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
notifyMsgs = append(notifyMsgs, " ")

View File

@@ -18,7 +18,7 @@ type rankUser struct {
// @return err
func getRankData(groupId, date string) (rank []rankUser, err error) {
tx := client.MySQL.Table("t_message AS tm").
Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id AND tgu.skip_chat_rank = 0").
Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id AND tgu.skip_chat_rank = 0 AND is_member = 1").
Select("tm.group_user", "tgu.nickname", "count( 1 ) AS `count`").
Where("tm.from_user = ?", groupId).
Where("tm.type < 10000").
@@ -32,7 +32,9 @@ func getRankData(groupId, date string) (rank []rankUser, err error) {
case "week":
tx.Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1")
case "month":
tx.Where("PERIOD_DIFF(date_format(now(), '%Y%m'), date_format(create_at, '%Y%m')) = 1")
tx.Where("PERIOD_DIFF(date_format(now(), '%Y%m'), date_format(tm.create_at, '%Y%m')) = 1")
case "year":
tx.Where("YEAR(tm.create_at) = YEAR(NOW()) - 1")
}
// 查询指定时间段全部数据

View File

@@ -2,7 +2,9 @@ package watergroup
import (
"fmt"
"go-wechat/client"
"go-wechat/config"
"go-wechat/model/entity"
"go-wechat/service"
"go-wechat/utils"
"log"
@@ -54,14 +56,48 @@ func dealWeek(gid string) {
log.Printf("上周群[%s]无对话记录", gid)
return
}
// 计算消息总数
var msgCount int64
for _, v := range records {
msgCount += v.Count
// 查询群成员总数
var groupUsers int64
err = client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", gid).
Where("is_member IS TRUE").
Count(&groupUsers).Error
if err != nil {
log.Printf("查询群成员总数失败, 错误信息: %v", err)
}
// 计算活跃度
showActivity := err == nil && groupUsers > 0
activity := "0.00"
if groupUsers > 0 {
activity = fmt.Sprintf("%.2f", (float64(len(records))/float64(groupUsers))*100)
}
// 计算消息总数、中位数、前十位消息总数
var msgCount, medianCount, topTenCount int64
for idx, v := range records {
msgCount += v.Count
if idx == (len(records)/2)-1 {
medianCount = v.Count
}
if len(records) > 10 && idx < 10 {
topTenCount += v.Count
}
}
// 计算活跃用户人均消息条数
avgMsgCount := int(float64(msgCount) / float64(len(records)))
// 组装消息总数推送信息
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ 上周本群 %d 位朋友共产生 %d 条发言", len(records), msgCount))
if showActivity {
m := fmt.Sprintf("🎭 活跃度: %s%%,人均消息条数: %d中位数: %d", activity, avgMsgCount, medianCount)
// 计算前十占比
if topTenCount > 0 {
m += fmt.Sprintf(",前十名占比: %.2f%%", float64(topTenCount)/float64(msgCount)*100)
}
notifyMsgs = append(notifyMsgs, m)
}
notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
notifyMsgs = append(notifyMsgs, " ")

Some files were not shown because too many files have changed in this diff Show More