Compare commits
91 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
55d219b364 | ||
|
|
a4234532cd | ||
|
|
3275a64852 | ||
|
|
21505702e6 | ||
|
|
45db832dad | ||
|
|
b806ade655 | ||
|
|
f39f46bfbf | ||
|
|
5819ac3c04 | ||
|
|
c4964a9e21 | ||
|
|
c0dcffce9d | ||
|
|
b14dbe0d1d | ||
|
|
b024600ef0 | ||
|
|
bc2893fad1 | ||
|
|
a098da39ee | ||
|
|
42339e3c51 | ||
|
|
20aeeefb3c | ||
|
|
08ffd4de6c | ||
|
|
2238e23c8d | ||
|
|
0f506e5afc | ||
|
|
44c45d11f2 | ||
|
|
50e91680bb | ||
|
|
b2598f2406 | ||
|
|
6818b10f4a | ||
|
|
727b06e143 | ||
|
|
c0c3864a8e | ||
|
|
8c2ab9376c | ||
|
|
9c7e93660d | ||
|
|
4cc50718e2 | ||
|
|
534f7a0ec8 | ||
|
|
16838ff80f | ||
|
|
ce37a269e1 | ||
|
|
aa78f3940c | ||
|
|
1c55900291 | ||
|
|
386e0dd6d3 | ||
|
|
9bec70319b | ||
|
|
bd33759b80 | ||
|
|
45109b02ef | ||
|
|
e8523c1e3e | ||
|
|
765e0ead7f | ||
|
|
e7117d227b | ||
|
|
448ac83514 | ||
|
|
5024fb4b05 | ||
|
|
1fca021b38 | ||
|
|
70fc7add78 | ||
|
|
ab5a8092f7 | ||
|
|
21a6f75ed3 | ||
|
|
daa36f308b | ||
|
|
e78f6a6d6e | ||
|
|
bb92c67069 | ||
|
|
542a2d5ef6 | ||
|
|
c13771be22 | ||
|
|
982ddef0da | ||
|
|
99447d332d | ||
|
|
3478f4f9e2 | ||
|
|
b3ed0fcc6f | ||
|
|
3da8b327d0 | ||
|
|
0262fdd9f7 | ||
|
|
f775f1d67d | ||
|
|
43f9d590b7 | ||
|
|
22c7d3683b | ||
|
|
d589a9615e | ||
|
|
13205e445f | ||
|
|
a414a98a51 | ||
|
|
d21089ab69 | ||
|
|
a274f085f8 | ||
|
|
8ddc40cc01 | ||
|
|
c0c810d02e | ||
|
|
905dab5ab8 | ||
|
|
ddb0db0b6a | ||
|
|
8bac050a02 | ||
|
|
a548af9de2 | ||
|
|
64c3c9c78b | ||
|
|
0f7cf5515d | ||
|
|
a2d84dea08 | ||
|
|
d3f8a59390 | ||
|
|
165fefdb48 | ||
|
|
22474efc57 | ||
|
|
797821e2ed | ||
|
|
79cbeb6ea5 | ||
|
|
49705a03a8 | ||
|
|
40e97608d3 | ||
|
|
fe7bc02c08 | ||
|
|
bb2c4e919e | ||
|
|
0ea189fb82 | ||
|
|
778db8e349 | ||
|
|
4b7ad4126c | ||
|
|
1413181bd4 | ||
|
|
a11ba51246 | ||
|
|
ea4262adf0 | ||
|
|
a9f6c9ff0d | ||
|
|
244bff5714 |
21
.editorconfig
Normal file
21
.editorconfig
Normal 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
1
.gitignore
vendored
@@ -8,3 +8,4 @@ log
|
||||
dist
|
||||
*.log
|
||||
blacklist.txt
|
||||
frontend.*
|
||||
|
||||
36
app/assistant.go
Normal file
36
app/assistant.go
Normal 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
33
app/contact.go
Normal 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()
|
||||
}
|
||||
128
app/friend.go
128
app/friend.go
@@ -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
|
||||
@@ -131,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, "操作成功")
|
||||
}
|
||||
|
||||
28
app/index.go
28
app/index.go
@@ -1,28 +0,0 @@
|
||||
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())
|
||||
}
|
||||
result["friends"] = friends
|
||||
result["groups"] = groups
|
||||
result["vnc"] = config.Conf.Wechat.VncUrl
|
||||
// 渲染页面
|
||||
ctx.HTML(http.StatusOK, "index.html", result)
|
||||
}
|
||||
116
app/pages.go
Normal file
116
app/pages.go
Normal 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)
|
||||
}
|
||||
@@ -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
16
common/response/fail.go
Normal 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
43
common/response/page.go
Normal 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
|
||||
}
|
||||
81
common/response/response.go
Normal file
81
common/response/response.go
Normal 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
27
common/response/result.go
Normal 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)
|
||||
}
|
||||
16
common/response/success.go
Normal file
16
common/response/success.go
Normal 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()
|
||||
}
|
||||
56
common/validator/validator.go
Normal file
56
common/validator/validator.go
Normal 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, ";"))
|
||||
}
|
||||
84
config.yaml
84
config.yaml
@@ -1,9 +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
|
||||
# vncUrl: http://192.168.1.175:19087/vnc_lite.html
|
||||
# 是否在启动的时候自动设置hook服务的回调
|
||||
autoSetCallback: false
|
||||
# 回调IP,如果是Docker运行,本参数必填(填auto表示自动,不适用于 docker 环境),如果Docker修改了映射,格式为 ip:port
|
||||
@@ -14,24 +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: false
|
||||
cron: '*/5 * * * *' # 五分钟一次
|
||||
enable: false
|
||||
cron: '*/5 * * * *' # 五分钟一次
|
||||
groupSummary:
|
||||
enable: false
|
||||
cron: '30 0 * * *' # 每天0:30
|
||||
waterGroup:
|
||||
enable: true
|
||||
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
|
||||
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:
|
||||
@@ -39,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 会变成全小写,所以这儿不能用大写字母
|
||||
@@ -53,6 +113,10 @@ resource:
|
||||
welcome-new:
|
||||
type: emotion
|
||||
path: 58e4150be2bba8f7b71974b10391f9e9
|
||||
# 给新好友或者群的自我介绍,不配置就不发送
|
||||
introduce:
|
||||
type: text
|
||||
path: "大家好,我是一个AI机器人,可以直接@我询问你想问的问题。"
|
||||
# 水群排行榜词云,只能是图片,末尾的`\%s`也是必须的
|
||||
wordcloud:
|
||||
type: image
|
||||
|
||||
19
config/ai.go
19
config/ai.go
@@ -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"` // 模型代码
|
||||
}
|
||||
|
||||
@@ -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
32
config/mq.go
Normal 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
22
config/system.go
Normal 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"` // 是否启用每日早报
|
||||
}
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
18
frontend/.eslintrc.cjs
Normal file
18
frontend/.eslintrc.cjs
Normal 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
30
frontend/.gitignore
vendored
Normal 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
|
||||
8
frontend/.prettierrc.json
Normal file
8
frontend/.prettierrc.json
Normal 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
7
frontend/.vscode/extensions.json
vendored
Normal file
@@ -0,0 +1,7 @@
|
||||
{
|
||||
"recommendations": [
|
||||
"Vue.volar",
|
||||
"dbaeumer.vscode-eslint",
|
||||
"esbenp.prettier-vscode"
|
||||
]
|
||||
}
|
||||
39
frontend/README.md
Normal file
39
frontend/README.md
Normal 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
1
frontend/env.d.ts
vendored
Normal file
@@ -0,0 +1 @@
|
||||
/// <reference types="vite/client" />
|
||||
13
frontend/index.html
Normal file
13
frontend/index.html
Normal 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
5312
frontend/package-lock.json
generated
Normal file
File diff suppressed because it is too large
Load Diff
46
frontend/package.json
Normal file
46
frontend/package.json
Normal 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"
|
||||
}
|
||||
}
|
||||
6
frontend/postcss.config.js
Normal file
6
frontend/postcss.config.js
Normal file
@@ -0,0 +1,6 @@
|
||||
export default {
|
||||
plugins: {
|
||||
tailwindcss: {},
|
||||
autoprefixer: {},
|
||||
},
|
||||
};
|
||||
BIN
frontend/public/favicon.ico
Normal file
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
10
frontend/src/App.vue
Normal 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
6
frontend/src/api/base.ts
Normal file
@@ -0,0 +1,6 @@
|
||||
// 默认返回数据结构
|
||||
export type BaseResult<T> = {
|
||||
code: number;
|
||||
message: string;
|
||||
data?: T;
|
||||
};
|
||||
43
frontend/src/api/contact.ts
Normal file
43
frontend/src/api/contact.ts
Normal 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')
|
||||
}
|
||||
BIN
frontend/src/assets/img/logo.png
Normal file
BIN
frontend/src/assets/img/logo.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 24 KiB |
BIN
frontend/src/assets/img/status-fail.png
Normal file
BIN
frontend/src/assets/img/status-fail.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 210 KiB |
BIN
frontend/src/assets/img/status-ok.png
Normal file
BIN
frontend/src/assets/img/status-ok.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 123 KiB |
3
frontend/src/components/contact/index.vue
Normal file
3
frontend/src/components/contact/index.vue
Normal file
@@ -0,0 +1,3 @@
|
||||
<template>
|
||||
|
||||
</template>
|
||||
73
frontend/src/components/layout/MainNav.vue
Normal file
73
frontend/src/components/layout/MainNav.vue
Normal 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
15
frontend/src/main.ts
Normal 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')
|
||||
31
frontend/src/router/index.ts
Normal file
31
frontend/src/router/index.ts
Normal 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
|
||||
12
frontend/src/stores/counter.ts
Normal file
12
frontend/src/stores/counter.ts
Normal 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 }
|
||||
})
|
||||
136
frontend/src/utils/http/index.ts
Normal file
136
frontend/src/utils/http/index.ts
Normal 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
43
frontend/src/utils/http/types.d.ts
vendored
Normal 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>;
|
||||
}
|
||||
12
frontend/src/views/assistant/index.vue
Normal file
12
frontend/src/views/assistant/index.vue
Normal file
@@ -0,0 +1,12 @@
|
||||
<script setup lang="ts">
|
||||
|
||||
defineOptions({
|
||||
name: "AssistantPage"
|
||||
});
|
||||
</script>
|
||||
|
||||
<template>
|
||||
<main>
|
||||
<h1>AI角色</h1>
|
||||
</main>
|
||||
</template>
|
||||
25
frontend/src/views/friend/hook.tsx
Normal file
25
frontend/src/views/friend/hook.tsx
Normal 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
|
||||
}
|
||||
}
|
||||
47
frontend/src/views/friend/index.vue
Normal file
47
frontend/src/views/friend/index.vue
Normal 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>
|
||||
12
frontend/src/views/group/index.vue
Normal file
12
frontend/src/views/group/index.vue
Normal file
@@ -0,0 +1,12 @@
|
||||
<script setup lang="ts">
|
||||
|
||||
defineOptions({
|
||||
name: "GroupPage"
|
||||
});
|
||||
</script>
|
||||
|
||||
<template>
|
||||
<main>
|
||||
<h1>群组</h1>
|
||||
</main>
|
||||
</template>
|
||||
12
frontend/src/views/index/index.vue
Normal file
12
frontend/src/views/index/index.vue
Normal file
@@ -0,0 +1,12 @@
|
||||
<script setup lang="ts">
|
||||
|
||||
defineOptions({
|
||||
name: "IndexPage"
|
||||
});
|
||||
</script>
|
||||
|
||||
<template>
|
||||
<main>
|
||||
<h1 class="text-orange-700">首页</h1>
|
||||
</main>
|
||||
</template>
|
||||
12
frontend/tailwind.config.js
Normal file
12
frontend/tailwind.config.js
Normal 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'),],
|
||||
}
|
||||
|
||||
14
frontend/tsconfig.app.json
Normal file
14
frontend/tsconfig.app.json
Normal 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
11
frontend/tsconfig.json
Normal file
@@ -0,0 +1,11 @@
|
||||
{
|
||||
"files": [],
|
||||
"references": [
|
||||
{
|
||||
"path": "./tsconfig.node.json"
|
||||
},
|
||||
{
|
||||
"path": "./tsconfig.app.json"
|
||||
}
|
||||
]
|
||||
}
|
||||
19
frontend/tsconfig.node.json
Normal file
19
frontend/tsconfig.node.json
Normal 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
28
frontend/vite.config.ts
Normal 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))
|
||||
}
|
||||
}
|
||||
})
|
||||
1
go.mod
1
go.mod
@@ -8,6 +8,7 @@ require (
|
||||
github.com/gin-gonic/gin v1.9.1
|
||||
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
|
||||
|
||||
33
go.sum
33
go.sum
@@ -14,8 +14,6 @@ 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=
|
||||
@@ -28,8 +26,6 @@ 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=
|
||||
@@ -40,8 +36,6 @@ 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=
|
||||
@@ -49,12 +43,9 @@ github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrt
|
||||
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=
|
||||
@@ -92,14 +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=
|
||||
@@ -110,8 +101,6 @@ 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=
|
||||
@@ -122,8 +111,6 @@ 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=
|
||||
@@ -147,22 +134,18 @@ 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/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-20231206192017-f3f8817b8deb h1:c0vyKkb6yr3KR7jEfJaOSv4lG7xPkbN6r52aJz1d8a8=
|
||||
golang.org/x/exp v0.0.0-20231206192017-f3f8817b8deb/go.mod h1:iRJReGqOEeBhDZGkGbynYwcHlctCvnjTYIamk7uXpHI=
|
||||
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=
|
||||
@@ -173,8 +156,6 @@ 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=
|
||||
@@ -189,8 +170,6 @@ 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=
|
||||
@@ -214,10 +193,6 @@ 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=
|
||||
|
||||
@@ -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"
|
||||
|
||||
@@ -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
26
main.go
@@ -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, "不支持的请求方式")
|
||||
|
||||
39
model/entity/aiassistant.go
Normal file
39
model/entity/aiassistant.go
Normal 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
|
||||
}
|
||||
@@ -15,8 +15,12 @@ type Friend struct {
|
||||
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"` // 是否正常
|
||||
}
|
||||
|
||||
@@ -88,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
|
||||
}
|
||||
@@ -97,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
|
||||
@@ -157,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
18
model/model/news.go
Normal 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"`
|
||||
}
|
||||
@@ -15,9 +15,13 @@ type FriendItem struct {
|
||||
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 // 是否还在通讯库(群聊是要还在群里也算)
|
||||
}
|
||||
|
||||
8
model/vo/message.go
Normal file
8
model/vo/message.go
Normal 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
46
mq/handler.go
Normal 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
118
mq/rabbitmq.go
Normal 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
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@@ -1,7 +1,7 @@
|
||||
package plugin
|
||||
|
||||
import (
|
||||
"go-wechat/model"
|
||||
"go-wechat/model/model"
|
||||
)
|
||||
|
||||
// MessageHandler 消息处理函数
|
||||
|
||||
@@ -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"
|
||||
@@ -46,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,
|
||||
})
|
||||
}
|
||||
|
||||
@@ -95,14 +116,6 @@ func AI(m *plugin.MessageContext) {
|
||||
Content: m.Content,
|
||||
})
|
||||
|
||||
// 配置模型
|
||||
chatModel := openai.GPT3Dot5Turbo0613
|
||||
if friendInfo.AiModel != "" {
|
||||
chatModel = friendInfo.AiModel
|
||||
} else if config.Conf.Ai.Model != "" {
|
||||
chatModel = config.Conf.Ai.Model
|
||||
}
|
||||
|
||||
// 默认使用AI回复
|
||||
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
|
||||
if config.Conf.Ai.BaseUrl != "" {
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
|
||||
65
plugin/plugins/command/ai.go
Normal file
65
plugin/plugins/command/ai.go
Normal 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
|
||||
}
|
||||
@@ -23,6 +23,13 @@ option: 指令选项,可选值:
|
||||
|
||||
#2. 肯德基疯狂星期四文案
|
||||
/kfc、/肯德基
|
||||
|
||||
#3. AI助手
|
||||
/ai option
|
||||
option: 指令选项,可选值:
|
||||
启用: '启用'、'打开'、'enable'
|
||||
停用: '停用'、'禁用'、'关闭'、'disable'
|
||||
|
||||
`
|
||||
utils.SendMessage(m.FromUser, m.GroupUser, str, 0)
|
||||
|
||||
|
||||
@@ -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 != "" {
|
||||
|
||||
@@ -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"
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
package plugins
|
||||
|
||||
import (
|
||||
"go-wechat/entity"
|
||||
"go-wechat/model/entity"
|
||||
"go-wechat/plugin"
|
||||
"go-wechat/service"
|
||||
"time"
|
||||
|
||||
@@ -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"
|
||||
)
|
||||
|
||||
15
readme.md
15
readme.md
@@ -48,7 +48,7 @@ version: '3.9'
|
||||
|
||||
services:
|
||||
wechat:
|
||||
image: lxh01/wxhelper-docker:3.9.5.81-v11
|
||||
image: lxh01/wxhelper-docker:3.9.5.81-v11-novnc # 如果不用noVNC网页,就删掉后面的-novnc
|
||||
container_name: gw-wechat
|
||||
restart: unless-stopped
|
||||
environment:
|
||||
@@ -56,8 +56,9 @@ services:
|
||||
volumes:
|
||||
- ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
|
||||
ports:
|
||||
- "19087:8080"
|
||||
- "19088:19088"
|
||||
- "19086:5900" # vnc端口
|
||||
- "19087:8080" # noVNC端口
|
||||
- "19088:19088" # 微信HOOK端口
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
|
||||
interval: 60s
|
||||
@@ -69,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
|
||||
@@ -87,6 +85,7 @@ services:
|
||||
restart: unless-stopped
|
||||
depends_on:
|
||||
- mysql
|
||||
- wechat
|
||||
volumes:
|
||||
# 配置文件请参阅项目根目录的config.yaml文件
|
||||
- ./config/config.yaml:/app/config.yaml
|
||||
@@ -101,3 +100,7 @@ services:
|
||||
docker-compose up -d # 老版本
|
||||
docker compose up -d # 新版本
|
||||
```
|
||||
|
||||
## 注意事项
|
||||
1. 宿主机必须是`debian`系,因为`wine`的玄学`BUG`,其他系统可能会出现各种问题
|
||||
2. 登录微信可以用`vnc viewer`连接`5900`端口或者访问`noVNC`端口的`vnc_lite.html`页面进行扫码登录
|
||||
@@ -9,22 +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("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态
|
||||
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
14
service/aiassistant.go
Normal 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
|
||||
}
|
||||
@@ -2,8 +2,8 @@ package service
|
||||
|
||||
import (
|
||||
"go-wechat/client"
|
||||
"go-wechat/entity"
|
||||
"go-wechat/vo"
|
||||
"go-wechat/model/entity"
|
||||
"go-wechat/model/vo"
|
||||
"log"
|
||||
"strings"
|
||||
)
|
||||
@@ -49,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
|
||||
}
|
||||
|
||||
|
||||
@@ -2,7 +2,7 @@ package service
|
||||
|
||||
import (
|
||||
"go-wechat/client"
|
||||
"go-wechat/vo"
|
||||
"go-wechat/model/vo"
|
||||
)
|
||||
|
||||
// GetGroupUsersByGroupId
|
||||
|
||||
@@ -2,7 +2,8 @@ package service
|
||||
|
||||
import (
|
||||
"go-wechat/client"
|
||||
"go-wechat/entity"
|
||||
"go-wechat/model/entity"
|
||||
"go-wechat/model/vo"
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
@@ -39,3 +40,43 @@ func SaveMessage(msg entity.Message) {
|
||||
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
|
||||
}
|
||||
|
||||
68
tasks/cleargroupuser/cleargroupuser.go
Normal file
68
tasks/cleargroupuser/cleargroupuser.go
Normal 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
|
||||
}
|
||||
@@ -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,14 +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,
|
||||
IsAdmin: wxid == baseResp.Data.Admin,
|
||||
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())
|
||||
@@ -187,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
36
tasks/news/news.go
Normal 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
88
tasks/summary/summary.go
Normal 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)
|
||||
}
|
||||
}
|
||||
@@ -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"
|
||||
@@ -37,12 +40,26 @@ func InitTasks() {
|
||||
}
|
||||
}
|
||||
|
||||
// 群聊总结
|
||||
if config.Conf.Task.GroupSummary.Enable {
|
||||
log.Printf("群聊总结任务已启用,执行表达式: %s", config.Conf.Task.GroupSummary.Cron)
|
||||
_, _ = s.Cron(config.Conf.Task.GroupSummary.Cron).Do(summary.AiSummary)
|
||||
}
|
||||
|
||||
// 更新好友列表
|
||||
if config.Conf.Task.SyncFriends.Enable {
|
||||
log.Printf("更新好友列表任务已启用,执行表达式: %s", config.Conf.Task.SyncFriends.Cron)
|
||||
_, _ = 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("定时任务初始化成功")
|
||||
|
||||
@@ -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, " ")
|
||||
|
||||
@@ -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,7 @@ 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")
|
||||
}
|
||||
|
||||
@@ -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, " ")
|
||||
|
||||
@@ -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,11 +56,37 @@ func dealYear(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, "亲爱的群友们,新年已经悄悄来临,让我们一起迎接这充满希望和美好的时刻。在这个特殊的日子里,我要向你们致以最真挚的祝福。")
|
||||
@@ -70,6 +98,14 @@ func dealYear(gid string) {
|
||||
notifyMsgs = append(notifyMsgs, fmt.Sprintf("祝福你们新年快乐!让我们一起迎接%d年的到来!", time.Now().Local().Year()))
|
||||
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, " ")
|
||||
|
||||
@@ -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"
|
||||
@@ -56,14 +58,48 @@ func dealYesterday(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
Reference in New Issue
Block a user