.gitea/workflows/build-image.yaml

@@ -1,54 +0,0 @@
-name: BuildImage
-
-on:
-  push:
-#    branches:
-#      - main
-    tags:
-      - '*'
-
-jobs:
-  build-image:
-    runs-on: ubuntu-latest
-    container:
-      # 使用这个镜像，不然Docker无法打包镜像
-      image: catthehacker/ubuntu:act-latest
-
-    steps:
-      - name: Setup Golang
-        uses: actions/setup-go@v3
-        with:
-          go-version: '>=1.21.0'
-          cache: false
-
-      - name: Checkout Code
-        uses: actions/checkout@v3
-
-      - name: Gen Tags
-        id: gen_tags
-        uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
-        with:
-          images: gitee.ltd/lxh/go-wxhelper
-          tags: |
-            type=ref,event=branch
-            type=ref,event=tag
-
-      - name: Print Tags
-        run: |
-          echo "${{ steps.gen_tags.outputs.tags }}"
-          echo "----------------- labels -----------------"
-          echo "${{ steps.meta.outputs.labels }}"
-
-      - name: Login to Repository
-        uses: docker/login-action@v2
-        with:
-          registry: gitee.ltd
-          username: ${{ secrets.USERNAME }}
-          password: ${{ secrets.PASSWORD }}
-
-      - name: Build image
-        uses: docker/build-push-action@v2
-        with:
-          push: true
-          tags: ${{ steps.gen_tags.outputs.tags }}
-          labels: ${{ steps.gen_tags.outputs.labels }}

.gitignore

@@ -1,10 +1,10 @@
 .idea
 vendor
 logs
-*.exe
+*build*
-*.pprof
+debug
-cache
-log
 dist
-*.log
+
-blacklist.txt
+*_deleted*
+
+config.yaml

Dockerfile

@@ -1,21 +1,18 @@
 FROM golang:alpine as builder
+
 WORKDIR /builder
 COPY . .
-
+RUN go mod download
-#ENV GO111MODULE=on
+RUN go build -o app
-#ENV GOPROXY=https://goproxy.cn,direct
+RUN ls -lh && chmod +x ./app
-
-RUN go version
-RUN go mod download && go build -o wxhelper
-RUN ls -lh && chmod -R +x ./*
 
 FROM code.hyxc1.com/open/alpine:3.16.0 as runner
 LABEL org.opencontainers.image.authors="lxh@cxh.cn"
 
-EXPOSE 19099
+# 定义一下版本号
-EXPOSE 8080
+ARG APP_VER
+ENV APP_VER=${APP_VER}
 
 WORKDIR /app
-COPY --from=builder /builder/wxhelper ./wxhelper
+COPY --from=builder /builder/app ./app
-COPY --from=builder /builder/views ./views
+CMD ./app
-CMD ./wxhelper

api/admin/aiassistant/delete.go

@@ -0,0 +1,25 @@
+package aiassistant
+
+import (
+	"github.com/gin-gonic/gin"
+	"unicode/utf8"
+	"wechat-robot/pkg/response"
+	aiAssistantService "wechat-robot/service/aiassistant"
+)
+
+// DeleteById
+// @description: 删除AI助手
+// @param ctx
+func DeleteById(ctx *gin.Context) {
+	var id = ctx.Param("id")
+	if utf8.RuneCountInString(id) != 32 {
+		response.New(ctx).SetMsg("参数错误").Fail()
+		return
+	}
+	// 删除数据
+	if err := aiAssistantService.DeleteById(id); err != nil {
+		response.New(ctx).SetMsg("删除失败").SetError(err).Fail()
+	} else {
+		response.New(ctx).SetMsg("删除成功").Success()
+	}
+}

api/admin/aiassistant/save.go

@@ -0,0 +1,25 @@
+package aiassistant
+
+import (
+	"github.com/gin-gonic/gin"
+	"wechat-robot/model/param/aiassistant"
+	"wechat-robot/pkg/response"
+	aiAssistantService "wechat-robot/service/aiassistant"
+)
+
+// Save
+// @description: 保存AI助手
+// @param ctx
+func Save(ctx *gin.Context) {
+	var p aiassistant.Save
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+	// 保存数据
+	if err := aiAssistantService.Save(p); err != nil {
+		response.New(ctx).SetMsg("保存失败").SetError(err).Fail()
+		return
+	}
+	response.New(ctx).SetMsg("保存成功").Success()
+}

api/admin/aiassistant/select.go

@@ -0,0 +1,26 @@
+package aiassistant
+
+import (
+	"github.com/gin-gonic/gin"
+	"wechat-robot/model/param/aiassistant"
+	"wechat-robot/pkg/response"
+	aiAssistantService "wechat-robot/service/aiassistant"
+)
+
+// GetAll
+// @description: 获取所有AI助手
+// @param ctx
+func GetAll(ctx *gin.Context) {
+	var p aiassistant.GetAll
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	records, err := aiAssistantService.GetAll(p)
+	if err != nil {
+		response.New(ctx).SetMsg("获取失败").SetError(err).Fail()
+		return
+	}
+	response.New(ctx).SetData(records).Success()
+}

api/admin/login/captcha.go

@@ -0,0 +1,34 @@
+package login
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/mojocn/base64Captcha"
+	"wechat-robot/model/vo/login"
+	"wechat-robot/pkg/captcha"
+	"wechat-robot/pkg/response"
+)
+
+// GetImgCaptcha
+// @description: 获取图片验证码
+// @param ctx
+func GetImgCaptcha(ctx *gin.Context) {
+	store := new(captcha.RedisStore)
+	math := base64Captcha.DriverMath{
+		Height: 60,
+		Width:  240,
+		Fonts:  []string{"ApothecaryFont.ttf"},
+	}
+
+	driver := math.ConvertFonts()
+
+	// 验证码生成器
+	c := base64Captcha.NewCaptcha(driver, store)
+	id, imgStr, _, err := c.Generate()
+	if err != nil {
+		response.New(ctx).SetMsg("验证码生成失败").Fail()
+		return
+	}
+	// 数据返回
+	result := login.CaptchaCode{Id: id, Img: imgStr}
+	response.New(ctx).SetData(result).Success()
+}

api/admin/login/login.go

@@ -0,0 +1,120 @@
+package login
+
+import (
+	"context"
+	"encoding/json"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"net/url"
+	"wechat-robot/internal/redis"
+	"wechat-robot/model/param/login"
+	"wechat-robot/pkg/auth"
+	"wechat-robot/pkg/captcha"
+	"wechat-robot/pkg/response"
+	userService "wechat-robot/service/adminuser"
+)
+
+// Login
+// @description: 登录
+// @param ctx
+// @return err
+func Login(ctx *gin.Context) {
+	log.Debugf("收到登录请求")
+	var p login.GetTokenWithPassword
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	// 验证验证码是否正确
+	if !new(captcha.RedisStore).Verify(p.VerifyId, p.VerifyCode, true) {
+		response.New(ctx).SetMsg("验证码错误").Fail()
+		return
+	}
+
+	// 重写参数
+	ctx.Request.Form = url.Values{
+		"username":   {p.Username},
+		"password":   {p.Password},
+		"scope":      {"ALL"},
+		"grant_type": {"password"},
+	}
+	// 参数解析成功，进行登录
+	if err := auth.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
+		log.Errorf("登录失败：%s", err.Error())
+		response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
+		return
+	}
+
+	if ctx.Writer.Status() == http.StatusOK {
+		go userService.UpdateLastLoginInfo(p.Username, ctx.ClientIP())
+	}
+}
+
+// Refresh
+// @description: 刷新Token
+// @param ctx
+func Refresh(ctx *gin.Context) {
+	var p login.RefreshToken
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	// 取出用户Id
+	userId := auth.GetUserIdWithRefreshToken(p.RefreshToken)
+
+	// 重写参数
+	ctx.Request.Form = url.Values{
+		"refresh_token": {p.RefreshToken},
+		"grant_type":    {"refresh_token"},
+	}
+
+	// 刷新Token
+	if err := auth.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
+		log.Errorf("Token数据返回失败: %v", err.Error())
+		response.New(ctx).SetMsg("系统错误").Fail()
+	}
+
+	// 登录成功才更新登录时间
+	if ctx.Writer.Status() == http.StatusOK {
+		// 登录成功，更新登录时间和IP
+		go userService.UpdateLastLoginInfo(userId, ctx.ClientIP())
+	}
+}
+
+// Logout
+// @description: 退出登录
+// @param ctx
+func Logout(ctx *gin.Context) {
+	log.Debug("退出登录啦")
+	// Token字符串前缀
+	const bearerSchema string = "Bearer "
+	// 取出Token
+	tokenHeader := ctx.GetHeader("Authorization")
+	tokenStr := tokenHeader[len(bearerSchema):]
+	// 取出原始RedisKey
+	baseDataId, err := redis.Client.Get(context.Background(), "oauth:token:"+tokenStr).Result()
+	if err != nil {
+		response.New(ctx).SetMsg("Token信息获取失败").Fail()
+		return
+	}
+	baseDataStr, err := redis.Client.Get(context.Background(), "oauth:token:"+baseDataId).Result()
+	if err != nil {
+		response.New(ctx).SetMsg("Token信息获取失败").Fail()
+		return
+	}
+	// 转换数据为Map
+	tokenData := make(map[string]interface{})
+	if err = json.Unmarshal([]byte(baseDataStr), &tokenData); err != nil {
+		response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
+		return
+	}
+	// 删除Redis缓存的数据
+	redis.Client.Del(context.Background(), "oauth:token:"+baseDataId)
+	redis.Client.Del(context.Background(), "oauth:token:"+tokenData["Access"].(string))
+	redis.Client.Del(context.Background(), "oauth:token:"+tokenData["Refresh"].(string))
+
+	response.New(ctx).Success()
+}

api/admin/menu/menu.go

@@ -0,0 +1 @@
+package menu

api/admin/menu/save.go

@@ -0,0 +1,24 @@
+package menu
+
+import (
+	"github.com/gin-gonic/gin"
+	menuParam "wechat-robot/model/param/menu"
+	"wechat-robot/pkg/response"
+	"wechat-robot/service/menu"
+)
+
+// Save
+// @description: 保存菜单
+// @param ctx
+func Save(ctx *gin.Context) {
+	var p menuParam.Save
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetError(err).Fail()
+		return
+	}
+	if err := menu.Save(p); err != nil {
+		response.New(ctx).SetError(err).Fail()
+		return
+	}
+	response.New(ctx).Success()
+}

api/admin/menu/user.go

@@ -0,0 +1,23 @@
+package menu
+
+import (
+	"github.com/gin-gonic/gin"
+	"wechat-robot/pkg/response"
+	"wechat-robot/service/menu"
+)
+
+// GetUserMenuTree
+// @description: 获取用户权限内的菜单树
+// @param ctx
+func GetUserMenuTree(ctx *gin.Context) {
+	// 获取用户Id
+	userId := ctx.GetString("userId")
+	// 获取菜单树
+	tree, err := menu.GetUserMenus(userId)
+	if err != nil {
+		response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
+		return
+	}
+	// 返回数据
+	response.New(ctx).SetData(tree).Success()
+}

api/admin/robot/delete.go

@@ -0,0 +1,25 @@
+package robot
+
+import (
+	"github.com/gin-gonic/gin"
+	"unicode/utf8"
+	"wechat-robot/pkg/response"
+	robotService "wechat-robot/service/robot"
+)
+
+// DeleteById
+// @description: 删除机器人
+// @param ctx
+func DeleteById(ctx *gin.Context) {
+	var id = ctx.Param("id")
+	if utf8.RuneCountInString(id) != 32 {
+		response.New(ctx).SetMsg("参数错误").Fail()
+		return
+	}
+	// 删除数据
+	if err := robotService.DeleteById(id); err != nil {
+		response.New(ctx).SetMsg("删除失败").SetError(err).Fail()
+	} else {
+		response.New(ctx).SetMsg("删除成功").Success()
+	}
+}

api/admin/robot/save.go

@@ -0,0 +1,31 @@
+package robot
+
+import (
+	"github.com/gin-gonic/gin"
+	robotParam "wechat-robot/model/param/robot"
+	"wechat-robot/pkg/response"
+	robotService "wechat-robot/service/robot"
+)
+
+// Save
+// @description: 保存机器人
+// @param ctx
+func Save(ctx *gin.Context) {
+	var p robotParam.Save
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	// 参数校验
+	if p.Id == "" && (p.HookApi == "" || p.Version == 0) {
+		response.New(ctx).SetMsg("参数错误").Fail()
+		return
+	}
+
+	if err := robotService.Save(p); err != nil {
+		response.New(ctx).SetMsg("保存失败").SetError(err).Fail()
+		return
+	}
+	response.New(ctx).SetMsg("保存成功").Success()
+}

api/admin/robot/select.go

@@ -0,0 +1,26 @@
+package robot
+
+import (
+	"github.com/gin-gonic/gin"
+	robotParam "wechat-robot/model/param/robot"
+	"wechat-robot/pkg/response"
+	robotService "wechat-robot/service/robot"
+)
+
+// GetAll
+// @description: 获取所有机器人
+// @param ctx
+func GetAll(ctx *gin.Context) {
+	var p robotParam.GetAll
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	records, err := robotService.GetAll(p)
+	if err != nil {
+		response.New(ctx).SetMsg("数据获取失败").SetError(err).Fail()
+		return
+	}
+	response.New(ctx).SetData(records).Success()
+}

api/admin/role/role.go

@@ -0,0 +1,27 @@
+package role
+
+import (
+	"github.com/gin-gonic/gin"
+	roleParam "wechat-robot/model/param/role"
+	"wechat-robot/pkg/response"
+	roleService "wechat-robot/service/role"
+)
+
+// GetAll
+// @description: 获取所有角色
+// @param ctx
+func GetAll(ctx *gin.Context) {
+	var p roleParam.GetAll
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
+		return
+	}
+
+	records, err := roleService.GetAll(p)
+	if err != nil {
+		response.New(ctx).SetMsg("获取所有角色失败").SetError(err).Fail()
+		return
+	}
+
+	response.New(ctx).SetData(records).Success()
+}

api/callback/hook.go

@@ -0,0 +1,21 @@
+package callback
+
+import (
+	"gitee.ltd/lxh/logger/log"
+	"github.com/gin-gonic/gin"
+	"wechat-robot/model/param/callback"
+	"wechat-robot/pkg/response"
+)
+
+// RobotHookNotify
+// @description: 机器人HOOK后回调
+// @param ctx
+func RobotHookNotify(ctx *gin.Context) {
+	var param callback.RobotHook
+	if err := ctx.ShouldBind(&param); err != nil {
+		response.New(ctx).SetError(err).Fail()
+		return
+	}
+
+	log.Debugf("机器人已启动，Id: %s", param.Robot)
+}

app/assistant.go

@@ -1,14 +0,0 @@
-package app
-
-import (
-	"github.com/gin-gonic/gin"
-)
-
-// SaveAssistant
-// @description: 保存AI助手
-// @param ctx
-func SaveAssistant(ctx *gin.Context) {
-
-	//ctx.String(http.StatusOK, "操作成功")
-	ctx.Redirect(302, "/assistant.html")
-}

app/friend.go

@@ -1,305 +0,0 @@
-package app
-
-import (
-	"github.com/gin-gonic/gin"
-	"go-wechat/client"
-	"go-wechat/model/entity"
-	"gorm.io/gorm"
-	"log"
-	"net/http"
-)
-
-// changeStatusParam
-// @description: 修改状态用的参数集
-type changeStatusParam struct {
-	WxId   string `json:"wxId" binding:"required"`
-	UserId string `json:"userId"`
-}
-
-// changeUseAiModelParam
-// @description: 修改使用的AI模型用的参数集
-type changeUseAiModelParam struct {
-	WxId  string `json:"wxid" binding:"required"` // 群Id或微信Id
-	Model string `json:"model"`                   // 模型代码
-}
-
-// autoClearMembers
-// @description: 自动清理群成员
-type autoClearMembers struct {
-	WxId string `json:"wxid" binding:"required"` // 群Id
-	Days int    `json:"days"`                    // 多少天未发言
-}
-
-// ChangeEnableAiStatus
-// @description: 修改是否开启AI
-// @param ctx
-func ChangeEnableAiStatus(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_ai`", gorm.Expr(" !`enable_ai`")).Error
-	if err != nil {
-		log.Printf("修改是否开启AI失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	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
-func ChangeEnableGroupRankStatus(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_chat_rank`", gorm.Expr(" !`enable_chat_rank`")).Error
-	if err != nil {
-		log.Printf("修改开启水群排行榜失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	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
-func ChangeEnableWelcomeStatus(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_welcome`", gorm.Expr(" !`enable_welcome`")).Error
-	if err != nil {
-		log.Printf("修改开启迎新失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	ctx.String(http.StatusOK, "操作成功")
-}
-
-// ChangeEnableCommandStatus
-// @description: 修改是否开启指令
-// @param ctx
-func ChangeEnableCommandStatus(ctx *gin.Context) {
-	var p changeStatusParam
-	if err := ctx.ShouldBindJSON(&p); err != nil {
-		ctx.String(http.StatusBadRequest, "参数错误")
-		return
-	}
-	log.Printf("待修改的群Id：%s", p.WxId)
-
-	err := client.MySQL.Model(&entity.Friend{}).
-		Where("wxid = ?", p.WxId).
-		Update("`enable_command`", gorm.Expr(" !`enable_command`")).Error
-	if err != nil {
-		log.Printf("修改指令启用状态失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	ctx.String(http.StatusOK, "操作成功")
-}
-
-// ChangeSkipGroupRankStatus
-// @description: 修改是否跳过水群排行榜
-// @param ctx
-func ChangeSkipGroupRankStatus(ctx *gin.Context) {
-	var p changeStatusParam
-	if err := ctx.ShouldBindJSON(&p); err != nil {
-		ctx.String(http.StatusBadRequest, "参数错误")
-		return
-	}
-	log.Printf("待修改的群Id：%s -> %s", p.WxId, p.UserId)
-
-	err := client.MySQL.Model(&entity.GroupUser{}).
-		Where("group_id = ?", p.WxId).
-		Where("wxid = ?", p.UserId).
-		Update("`skip_chat_rank`", gorm.Expr(" !`skip_chat_rank`")).Error
-	if err != nil {
-		log.Printf("修改跳过水群排行榜失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	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, "操作成功")
-}
-
-// ChangeEnableHotTopStatus
-// @description: 修改是否开启热搜
-// @param ctx
-func ChangeEnableHotTopStatus(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_hot_top`", gorm.Expr(" !`enable_hot_top`")).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, "操作成功")
-}
-
-// ChangeAiFreeLimit
-// @description: 修改AI免费次数
-// @param ctx
-func ChangeAiFreeLimit(ctx *gin.Context) {
-	var p autoClearMembers
-	if err := ctx.ShouldBindJSON(&p); err != nil {
-		ctx.String(http.StatusBadRequest, "参数错误")
-		return
-	}
-	log.Printf("待修改的Id：%s", p.WxId)
-
-	err := client.MySQL.Model(&entity.Friend{}).
-		Where("wxid = ?", p.WxId).
-		Update("`ai_free_limit`", p.Days).Error
-	if err != nil {
-		log.Printf("修改AI免费次数失败：%s", err)
-		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
-		return
-	}
-
-	ctx.String(http.StatusOK, "操作成功")
-}

app/group.go

@@ -1,43 +0,0 @@
-package app
-
-import (
-	"github.com/gin-gonic/gin"
-	"go-wechat/service"
-	"net/http"
-)
-
-type getGroupUser struct {
-	GroupId string `json:"groupId" form:"groupId" binding:"required"` // 群Id
-}
-
-// GetGroupUsers
-// @description: 获取群成员列表
-// @param ctx
-func GetGroupUsers(ctx *gin.Context) {
-	var p getGroupUser
-	if err := ctx.ShouldBind(&p); err != nil {
-		ctx.String(http.StatusBadRequest, "参数错误")
-		return
-	}
-	// 查询数据
-	records, err := service.GetGroupUsersByGroupId(p.GroupId)
-	if err != nil {
-		ctx.String(http.StatusInternalServerError, "查询失败: %s", err.Error())
-		return
-	}
-
-	result := map[string]any{
-		"records": records,
-	}
-	// 循环数据，统计健在成员
-	var isOkCount int
-	for _, record := range records {
-		if record.IsMember {
-			isOkCount++
-		}
-	}
-	result["isOkCount"] = isOkCount
-
-	// 暂时先就这样写着，跑通了再改
-	ctx.JSON(http.StatusOK, result)
-}

app/other.go

@@ -1,56 +0,0 @@
-package app
-
-import (
-	"fmt"
-	"github.com/gin-gonic/gin"
-	"go-wechat/model/vo"
-	"go-wechat/service"
-	"go-wechat/utils"
-	"log"
-	"net/http"
-)
-
-// SendAiSummary
-// @description: 发送AI摘要
-// @param ctx
-func SendAiSummary(ctx *gin.Context) {
-	// 获取群Id
-	groupId := ctx.Query("id")
-	if groupId == "" {
-		ctx.String(http.StatusForbidden, "群Id不能为空")
-		return
-	}
-
-	// 取出群名称
-	groupInfo, err := service.GetFriendInfoById(groupId)
-	if err != nil {
-		ctx.String(http.StatusInternalServerError, "获取群信息失败")
-		return
-	}
-
-	// 获取对话记录
-	var records []vo.TextMessageItem
-	if records, err = service.GetTextMessagesById(groupId); err != nil {
-		log.Printf("获取群[%s]对话记录失败, 错误信息: %v", groupId, err)
-		ctx.String(http.StatusInternalServerError, "获取群对话记录失败")
-		return
-	}
-	if len(records) < 10 {
-		ctx.String(http.StatusForbidden, "群对话记录不足10条，建议自己看")
-		return
-	}
-
-	// 组装对话记录为字符串
-	var replyMsg string
-	replyMsg, err = utils.GetAiSummary(groupInfo.Nickname, records)
-	if err != nil {
-		log.Printf("群聊记录总结失败: %v", err.Error())
-		ctx.String(http.StatusInternalServerError, "群聊消息总结失败，错误信息: "+err.Error())
-		return
-	}
-
-	replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了，让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg)
-	log.Printf("群[%s]对话记录总结成功，总结内容: %s", groupInfo.Nickname, replyMsg)
-	_ = utils.SendMessage(groupId, "", replyMsg, 0)
-	ctx.String(http.StatusOK, "操作完成")
-}

app/pages.go

@@ -1,137 +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())
-	}
-	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)
-}
-
-// ManageWithGroupAdmin
-// @description: 群组管理（管理员可用）
-// @param ctx
-func ManageWithGroupAdmin(ctx *gin.Context) {
-	// 取出id
-	id := ctx.Query("id")
-	if id == "" {
-		ctx.HTML(http.StatusOK, "404.html", nil)
-		return
-	}
-
-	var result = gin.H{
-		"msg": "success",
-	}
-	result["info"], _ = service.GetFriendInfoById(id)
-	result["aiModels"] = config.Conf.Ai.Models
-	result["assistant"], _ = service.GetAllAiAssistant()
-	// 渲染页面
-	ctx.HTML(http.StatusOK, "manager-one.html", result)
-}
-
-// PageNotFound
-// @description: 404页面
-// @param ctx
-func PageNotFound(ctx *gin.Context) {
-	// 渲染页面
-	ctx.HTML(http.StatusOK, "404.html", nil)
-}

client/mysql.go

@@ -1,38 +0,0 @@
-package client
-
-import (
-	"go-wechat/config"
-	"gorm.io/driver/mysql"
-	"gorm.io/gorm"
-	"gorm.io/gorm/logger"
-	"log"
-	"os"
-	"strconv"
-)
-
-// MySQL MySQL客户端
-var MySQL *gorm.DB
-
-func InitMySQLClient() {
-	// 创建连接对象
-	mysqlConfig := mysql.Config{
-		DSN:                     config.Conf.MySQL.GetDSN(),
-		DontSupportRenameIndex:  true, // 重命名索引时采用删除并新建的方式
-		DontSupportRenameColumn: true, // 用 `change` 重命名列
-	}
-
-	// gorm 配置
-	gormConfig := gorm.Config{}
-	// 是否开启调试模式
-	if flag, _ := strconv.ParseBool(os.Getenv("GORM_DEBUG")); flag {
-		gormConfig.Logger = logger.Default.LogMode(logger.Info)
-	}
-
-	conn, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
-	if err != nil {
-		log.Panicf("初始化MySQL连接失败, 错误信息: %v", err)
-	} else {
-		log.Println("MySQL连接成功")
-	}
-	MySQL = conn
-}

common/constant/wxid.go

@@ -1,6 +0,0 @@
-package constant
-
-var SpecialId = []string{"filehelper", "newsapp", "fmessage", "weibo", "qqmail", "tmessage", "qmessage", "qqsync",
-	"floatbottle", "lbsapp", "shakeapp", "medianote", "qqfriend", "readerapp", "blogapp", "facebookapp", "masssendapp",
-	"meishiapp", "feedsapp", "voip", "blogappweixin", "weixin", "brandsessionholder", "weixinreminder", "officialaccounts",
-	"notification_messages", "wxitil", "userexperience_alarm", "notification_messages", "exmail_tool"}

common/current/robot.go

@@ -1,44 +0,0 @@
-package current
-
-import (
-	"go-wechat/model/dto"
-	"go-wechat/plugin"
-)
-
-// robotInfo
-// @description: 机器人信息
-type robotInfo struct {
-	info           dto.RobotUserInfo
-	MessageHandler plugin.MessageHandler // 启用的插件
-}
-
-// 当前接入的机器人信息
-var ri robotInfo
-
-// SetRobotInfo
-// @description: 设置机器人信息
-// @param info
-func SetRobotInfo(info dto.RobotUserInfo) {
-	ri.info = info
-}
-
-// GetRobotInfo
-// @description: 获取机器人信息
-// @return dto.RobotUserInfo
-func GetRobotInfo() dto.RobotUserInfo {
-	return ri.info
-}
-
-// GetRobotMessageHandler
-// @description: 获取机器人插件信息
-// @return robotInfo
-func GetRobotMessageHandler() plugin.MessageHandler {
-	return ri.MessageHandler
-}
-
-// SetRobotMessageHandler
-// @description: 设置机器人插件信息
-// @param handler
-func SetRobotMessageHandler(handler plugin.MessageHandler) {
-	ri.MessageHandler = handler
-}

config.yaml

@@ -1,146 +0,0 @@
-system:
-  # 每日新闻接口 Token
-  # 获取地址: https://admin.alapi.cn/api_manager/token_manager
-  alApiToken: xxx
-  # urlc.cn的Token，用来生成短链接
-  urlcApiToken: xxx
-  # 系统访问域名，必须是包括 http[s]:// 的完整域名
-  domain: https://wechat.abc.com
-  # 添加新好友或群之后通知给指定的人
-  newFriendNotify:
-    enable: true
-    toUser:
-      - "wxid_xxx"
-  # 默认AI等配置
-  defaultRule:
-    # 默认是否开启AI
-    ai: true
-    # 默认是否开启水群排行榜
-    chatRank: true
-    # 默认是否开启聊天记录总结
-    summary: true
-    # 默认是否开启新成员加群欢迎
-    welcome: true
-    # 每日早报
-    news: true
-    # 热榜
-    hotTop: true
-    # 每天免费 AI 次数限制
-    aiFreeLimit: 50
-
-# 微信HOOK配置
-wechat:
-  # 微信HOOK接口地址
-  host: 10.0.0.79:19098
-  # 微信容器映射出来的vnc页面地址，没有就不填
-  #  vncUrl: http://192.168.1.175:19087/vnc_lite.html
-  # 是否在启动的时候自动设置hook服务的回调
-  autoSetCallback: false
-  # 回调IP，如果是Docker运行，本参数必填(填auto表示自动，不适用于 docker 环境)，如果Docker修改了映射，格式为 ip:port
-  callback: 10.0.0.51
-  # 转发到其他地址
-  forward:
-#    - 10.0.0.247:4299
-
-# 数据库
-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: false
-    cron: '14 11 * * *' # 每天0:30
-  hotTop:
-    enable: true
-    cron: '0 */1 * * *' # 每小时一次
-  syncFriends: # 逻辑已修改，每天大半夜同步一次即可(主要是用以修正数据)，平时会根据收到的消息针对性同步
-    enable: false
-    cron: '0 4 * * *' # 每天4:00
-  groupSummary:
-    enable: false
-    cron: '30 0 * * *' # 每天0:30
-  waterGroup:
-    enable: false
-    cron:
-      yesterday: '30 9 * * *' # 每天9:30
-      week: '30 9 * * 1' # 每周一9:30
-      month: '30 9 1 * *' # 每月1号9:30
-      year: '0 9 1 1 *' # 每年1月1号9:30
-
-# MQ配置
-mq:
-  # 是否启用
-  enable: false
-  # RabbitMQ配置
-  rabbitmq:
-    host: 10.0.0.31
-    port: 5672
-    user: wechat
-    password: wechat123
-    vhost: wechat
-
-# AI回复
-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: 默认
-    - name: ChatGPT-4
-      model: gpt-4
-      canManager: false
-    - 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
-      canManager: true
-    - name: 豆包Pro-4k
-      model: Doubao-pro-4k
-      canManager: true
-
-# 资源配置
-# map[k]v结构，k 会变成全小写，所以这儿不能用大写字母
-resource:
-  # 欢迎新成员表情包
-  welcome-new:
-    type: emotion
-    path: 58e4150be2bba8f7b71974b10391f9e9
-  # 给新好友或者群的自我介绍，不配置就不发送
-  introduce:
-    type: text
-    path: "大家好，我是一个AI机器人，可以直接@我询问你想问的问题。"
-  # 水群排行榜词云，只能是图片，末尾的`\%s`也是必须的
-  wordcloud:
-    type: image
-    path: D:\Share\wordcloud\%s
-  # 邀请入群消息回复
-  invitation-join-group:
-    type: text
-    path: "您的邀请消息已收到啦，正在通知我的主人来同意请求。在我加群之后将会进行初始化操作，直到收到我主动发送的消息就是初始化完成咯，在那之前请耐心等待喔~"

config.yaml.example

@@ -0,0 +1,15 @@
+# 数据库配置
+database:
+  type: mysql # 使用的数据库类型，可选 mysql | postgresql
+  host: mysql # 数据库地址(使用docker-compose启动可以不用改)
+  port: 3306 # 数据库端口
+  username: pixiu # 数据库用户名
+  password: pixiu # 数据库密码
+  database: pixiu # 数据库名
+
+# Redis配置
+redis:
+  host: redis # Redis地址(使用docker-compose启动可以不用改)
+  port: 6379 # Redis端口
+  password: mNhgeSk32fUf69C6
+  db: 0

config/ai.go

@@ -1,21 +0,0 @@
-package config
-
-// ai
-// @description: AI配置
-type ai struct {
-	Enable       bool      `json:"enable" yaml:"enable"`             // 是否启用AI
-	Model        string    `json:"dto" yaml:"dto"`                   // 模型
-	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:"dto" yaml:"dto"`               // 模型代码
-	CanManager bool   `json:"canManager" yaml:"canManager"` // 是否可以在manager页面中选择
-}

config/config.go

@@ -6,11 +6,7 @@ var Conf conf
 // Config
 // @description: 配置
 type conf struct {
-	System   system                  `json:"system" yaml:"system"`     // 系统配置
+	Database db    `json:"database" yaml:"database"` // 数据库 配置
-	Task     task                    `json:"task" yaml:"task"`         // 定时任务配置
+	Redis    redis `json:"redis" yaml:"redis"`       // Redis 配置
-	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"` // 资源配置
 }

config/db.go

@@ -0,0 +1,32 @@
+package config
+
+import "fmt"
+
+// db
+// @description: 数据库配置
+type db struct {
+	Type     string `json:"type" yaml:"type" mapstructure:"type"`             // 数据库类型
+	Host     string `json:"host" yaml:"host" mapstructure:"host"`             // 数据库地址
+	Port     int    `json:"port" yaml:"port" mapstructure:"port"`             // 数据库端口
+	Username string `json:"username" yaml:"username" mapstructure:"username"` // 数据库用户名
+	Password string `json:"password" yaml:"password" mapstructure:"password"` // 数据库密码
+	Database string `json:"database" yaml:"database" mapstructure:"database"` // 数据库名称
+}
+
+// GetMysqlDSN
+// @description: 获取MySQL连接DSN
+// @receiver d
+// @return string
+func (d db) GetMysqlDSN() string {
+	return fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		d.Username, d.Password, d.Host, d.Port, d.Database)
+}
+
+// GetPostgreSQLDSN
+// @description: 获取PostgreSQL连接DSN
+// @receiver d
+// @return string
+func (d db) GetPostgreSQLDSN() string {
+	return fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai",
+		d.Host, d.Username, d.Password, d.Database, d.Port)
+}

config/mq.go

@@ -5,7 +5,6 @@ import "fmt"
 // mq
 // @description: MQ配置
 type mq struct {
-	Enable   bool     `json:"enable" yaml:"enable"`     // 是否启用
 	RabbitMQ rabbitMq `json:"rabbitmq" yaml:"rabbitmq"` // RabbitMQ配置
 }
 

config/mysql.go

@@ -1,24 +0,0 @@
-package config
-
-import (
-	"fmt"
-)
-
-// mysql
-// @description: MySQL配置
-type mysql struct {
-	Host     string `mapstructure:"host" yaml:"host"`         // 主机
-	Port     int    `mapstructure:"port" yaml:"port"`         // 端口
-	User     string `mapstructure:"user" yaml:"user"`         // 用户名
-	Password string `mapstructure:"password" yaml:"password"` // 密码
-	Db       string `mapstructure:"db" yaml:"db"`             // 数据库名称
-}
-
-// GetDSN
-// @description: 返回 MySQL 连接字符串
-// @receiver c
-// @return string
-func (c mysql) GetDSN() string {
-	return fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=utf8mb4&parseTime=True&loc=Local",
-		c.User, c.Password, c.Host, c.Port, c.Db)
-}

config/redis.go

@@ -0,0 +1,15 @@
+package config
+
+import "fmt"
+
+// Redis配置
+type redis struct {
+	Host     string `mapstructure:"host" yaml:"host"`         // 主机
+	Port     int    `mapstructure:"port" yaml:"port"`         // 端口
+	Password string `mapstructure:"password" yaml:"password"` // 密码
+	Db       int    `mapstructure:"db" yaml:"db"`             // 数据库名称
+}
+
+func (r redis) GetDSN() string {
+	return fmt.Sprintf("%s:%v", r.Host, r.Port)
+}

config/resource.go

@@ -1,8 +0,0 @@
-package config
-
-// resourceItem
-// @description: 资源项
-type resourceItem struct {
-	Type string `json:"type" yaml:"type"` // 类型
-	Path string `json:"path" yaml:"path"` // 路径
-}

config/system.go

@@ -1,26 +0,0 @@
-package config
-
-// 系统配置
-type system struct {
-	Domain          string          `json:"domain" yaml:"domain"`                   // 域名
-	AlApiToken      string          `json:"alApiToken" yaml:"alApiToken"`           // AL API Token
-	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"`               // 是否启用每日早报
-	HotTop      bool `json:"hotTop" yaml:"hotTop"`           // 是否启用热门话题
-	AiFreeLimit int  `json:"aiFreeLimit" yaml:"aiFreeLimit"` // AI免费次数
-}

config/task.go

@@ -1,35 +0,0 @@
-package config
-
-// task
-// @description: 定时任务
-type task struct {
-	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"` // 群聊总结
-	HotTop       syncFriends `json:"hotTop" yaml:"hotTop"`             // 热搜排行榜
-}
-
-// syncFriends
-// @description: 同步好友
-type syncFriends struct {
-	Enable bool   `json:"enable" yaml:"enable"` // 是否启用
-	Cron   string `json:"cron" yaml:"cron"`     // 定时任务表达式
-}
-
-// waterGroup
-// @description: 水群排行榜
-type waterGroup struct {
-	Enable bool           `json:"enable" yaml:"enable"` // 是否启用
-	Cron   waterGroupCron `json:"cron" yaml:"cron"`     // 定时任务表达式
-}
-
-// waterGroupCron
-// @description: 水群排行榜定时任务
-type waterGroupCron struct {
-	Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
-	Week      string `json:"week" yaml:"week"`           // 周排行榜
-	Month     string `json:"month" yaml:"month"`         // 月排行榜
-	Year      string `json:"year" yaml:"year"`           // 年排行榜
-}

config/wechat.go

@@ -1,35 +0,0 @@
-package config
-
-import "strings"
-
-// wxHelper
-// @description: 微信助手
-type wechat struct {
-	Host            string   `json:"host" yaml:"host"`                       // 接口地址
-	VncUrl          string   `json:"vncUrl" yaml:"vncUrl"`                   // vnc页面地址
-	AutoSetCallback bool     `json:"autoSetCallback" yaml:"autoSetCallback"` // 是否自动设置回调地址
-	Callback        string   `json:"callback" yaml:"callback"`               // 回调地址
-	Forward         []string `json:"forward" yaml:"forward"`                 // 转发地址
-}
-
-// Check
-// @description: 检查配置是否可用
-// @receiver w
-// @return bool
-func (w wechat) Check() bool {
-	if w.Host == "" {
-		return false
-	}
-	if w.AutoSetCallback && w.Callback == "" {
-		return false
-	}
-	return true
-}
-
-func (w wechat) GetURL(uri string) string {
-	host := w.Host
-	if !strings.HasPrefix(w.Host, "http://") {
-		host = "http://" + w.Host
-	}
-	return host + uri
-}

docker-compose.yaml

@@ -1,46 +0,0 @@
-version: '3.9'
-
-services:
-  wechat:
-    image: lxh01/wxhelper-docker:3.9.5.81-v11
-    container_name: gw-wechat
-    restart: unless-stopped
-    environment:
-      - WINEDEBUG=fixme-all
-    volumes:
-      - ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
-    ports:
-      - "19087:8080"
-      - "19088:19088"
-    healthcheck:
-      test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
-      interval: 60s
-      timeout: 10s
-      retries: 5
-
-
-  mysql:
-    image: mysql:8
-    container_name: gw-db
-    restart: unless-stopped
-    environment:
-      - MYSQL_ROOT_PASSWORD=wechat
-      - MYSQL_USER=wechat
-      - MYSQL_PASSWORD=wechat
-      - MYSQL_DATABASE=wechat
-    volumes:
-      - ./data/db:/var/lib/mysql
-
-
-  wxhelper:
-    image: gitee.ltd/lxh/go-wxhelper:latest
-    container_name: gw-service
-    restart: unless-stopped
-    depends_on:
-      - mysql
-      - wechat
-    volumes:
-      # 配置文件请参阅项目根目录的config.yaml文件
-      - ./config/config.yaml:/app/config.yaml
-    ports:
-      - "19099:19099"

go.mod

@@ -1,65 +1,110 @@
-module go-wechat
+module wechat-robot
 
 go 1.21
 
 require (
-	github.com/duke-git/lancet/v2 v2.3.1
+	gitee.ltd/lxh/logger v1.0.15
 	github.com/fsnotify/fsnotify v1.7.0
-	github.com/gin-gonic/gin v1.10.0
+	github.com/gin-gonic/gin v1.9.1
 	github.com/go-co-op/gocron v1.37.0
-	github.com/go-resty/resty/v2 v2.13.1
+	github.com/go-oauth2/oauth2/v4 v4.5.2
+	github.com/go-oauth2/redis/v4 v4.1.1
+	github.com/go-playground/locales v0.14.1
+	github.com/go-playground/universal-translator v0.18.1
+	github.com/go-playground/validator/v10 v10.17.0
+	github.com/go-redis/redis/v8 v8.11.5
+	github.com/go-resty/resty/v2 v2.11.0
 	github.com/google/uuid v1.6.0
-	github.com/rabbitmq/amqp091-go v1.10.0
+	github.com/mojocn/base64Captcha v1.3.6
-	github.com/sashabaranov/go-openai v1.26.3
+	github.com/rabbitmq/amqp091-go v1.2.0
-	github.com/spf13/viper v1.19.0
+	github.com/spf13/viper v1.18.2
-	gorm.io/driver/mysql v1.5.7
+	golang.org/x/crypto v0.18.0
-	gorm.io/gorm v1.25.11
+	gorm.io/driver/mysql v1.5.2
+	gorm.io/driver/postgres v1.5.4
+	gorm.io/gorm v1.25.6
+	gorm.io/plugin/soft_delete v1.2.1
 )
 
 require (
-	filippo.io/edwards25519 v1.1.0 // indirect
+	github.com/beorn7/perks v1.0.1 // indirect
-	github.com/bytedance/sonic v1.11.9 // indirect
+	github.com/bytedance/sonic v1.10.2 // indirect
-	github.com/bytedance/sonic/loader v0.1.1 // indirect
+	github.com/caarlos0/env/v6 v6.10.1 // indirect
-	github.com/cloudwego/base64x v0.1.4 // indirect
+	github.com/cespare/xxhash/v2 v2.2.0 // indirect
-	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect
-	github.com/gabriel-vasile/mimetype v1.4.4 // indirect
+	github.com/chenzhuoyu/iasm v0.9.1 // indirect
+	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
+	github.com/gabriel-vasile/mimetype v1.4.3 // indirect
 	github.com/gin-contrib/sse v0.1.0 // indirect
-	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-kit/kit v0.13.0 // indirect
-	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-kit/log v0.2.1 // indirect
-	github.com/go-playground/validator/v10 v10.22.0 // indirect
+	github.com/go-logfmt/logfmt v0.6.0 // indirect
-	github.com/go-sql-driver/mysql v1.8.1 // indirect
+	github.com/go-sql-driver/mysql v1.7.1 // indirect
-	github.com/goccy/go-json v0.10.3 // indirect
+	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/gogo/protobuf v1.3.2 // indirect
+	github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
+	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
+	github.com/golang/protobuf v1.5.3 // indirect
+	github.com/golang/snappy v0.0.4 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9 // indirect
+	github.com/jackc/pgx/v5 v5.5.2 // indirect
+	github.com/jackc/puddle/v2 v2.2.1 // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
+	github.com/jpillora/backoff v1.0.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/klauspost/cpuid/v2 v2.2.8 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.6 // indirect
-	github.com/leodido/go-urn v1.4.0 // indirect
+	github.com/leodido/go-urn v1.3.0 // indirect
+	github.com/lixh00/loki-client-go v1.0.1 // indirect
 	github.com/magiconair/properties v1.8.7 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f // indirect
+	github.com/natefinch/lumberjack v2.0.0+incompatible // indirect
+	github.com/pelletier/go-toml/v2 v2.1.1 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
+	github.com/prometheus/client_golang v1.18.0 // indirect
+	github.com/prometheus/client_model v0.5.0 // indirect
+	github.com/prometheus/common v0.46.0 // indirect
+	github.com/prometheus/procfs v0.12.0 // indirect
+	github.com/prometheus/prometheus v1.8.2-0.20201028100903-3245b3267b24 // indirect
 	github.com/robfig/cron/v3 v3.0.1 // indirect
-	github.com/sagikazarmark/locafero v0.6.0 // indirect
+	github.com/sagikazarmark/locafero v0.4.0 // indirect
 	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
 	github.com/sourcegraph/conc v0.3.0 // indirect
 	github.com/spf13/afero v1.11.0 // indirect
 	github.com/spf13/cast v1.6.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
+	github.com/tidwall/btree v1.7.0 // indirect
+	github.com/tidwall/buntdb v1.3.0 // indirect
+	github.com/tidwall/gjson v1.17.0 // indirect
+	github.com/tidwall/grect v0.1.4 // indirect
+	github.com/tidwall/match v1.1.1 // indirect
+	github.com/tidwall/pretty v1.2.1 // indirect
+	github.com/tidwall/rtred v0.1.2 // indirect
+	github.com/tidwall/tinyqueue v0.1.1 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.2.12 // indirect
 	go.uber.org/atomic v1.11.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
-	golang.org/x/arch v0.8.0 // indirect
+	go.uber.org/zap v1.26.0 // indirect
-	golang.org/x/crypto v0.25.0 // indirect
+	golang.org/x/arch v0.7.0 // indirect
-	golang.org/x/exp v0.0.0-20240707233637-46b078467d37 // indirect
+	golang.org/x/exp v0.0.0-20240119083558-1b970713d09a // indirect
-	golang.org/x/net v0.27.0 // indirect
+	golang.org/x/image v0.15.0 // indirect
-	golang.org/x/sys v0.22.0 // indirect
+	golang.org/x/net v0.20.0 // indirect
-	golang.org/x/text v0.16.0 // indirect
+	golang.org/x/oauth2 v0.16.0 // indirect
-	google.golang.org/protobuf v1.34.2 // indirect
+	golang.org/x/sync v0.6.0 // indirect
+	golang.org/x/sys v0.16.0 // indirect
+	golang.org/x/text v0.14.0 // indirect
+	google.golang.org/appengine v1.6.8 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20240125205218-1f4bbc51befe // indirect
+	google.golang.org/grpc v1.61.0 // indirect
+	google.golang.org/protobuf v1.32.0 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

initialization/plugin.go

@@ -1,60 +0,0 @@
-package initialization
-
-import (
-	"go-wechat/common/current"
-	"go-wechat/model/dto"
-	plugin "go-wechat/plugin"
-	"go-wechat/plugin/plugins"
-	"go-wechat/service"
-)
-
-// Plugin
-// @description: 初始化插件
-func Plugin() {
-	// 定义一个处理器
-	dispatcher := plugin.NewMessageMatchDispatcher()
-	// 设置为异步处理
-	dispatcher.SetAsync(true)
-
-	// 注册插件
-
-	// 保存消息进数据库
-	dispatcher.RegisterHandler(func(*dto.Message) bool {
-		return true
-	}, plugins.SaveToDb)
-
-	// 通知邀请入群消息到配置用户
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		flag, _ := m.IsInvitationJoinGroup()
-		return flag && !m.IsGroup()
-	}, plugins.NotifyInvitationJoinGroup)
-	// 被移除群聊通知到配置用户
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		return m.IsRemoveFromChatroom()
-	}, plugins.NotifyRemoveFromChatroom)
-	// 响应好友添加成功消息
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		return m.IsNewFriendAdd() || m.IsJoinToGroup() || m.IsOldFriendBack()
-	}, plugins.ReplyNewFriend)
-
-	// 私聊指令消息
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		// 私聊消息 或 群聊艾特机器人并且以/开头的消息
-		isGroupAt := m.IsAt() && !m.IsAtAll()
-		return (m.IsPrivateText() || isGroupAt) && m.CleanContentStartWith("/") && service.CheckIsEnableCommand(m.FromUser)
-	}, plugins.Command)
-
-	// AI消息插件
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		// 群内@或者私聊文字消息
-		return (m.IsAt() && !m.IsAtAll()) || m.IsPrivateText()
-	}, plugins.AI)
-
-	// 欢迎新成员
-	dispatcher.RegisterHandler(func(m *dto.Message) bool {
-		return m.IsNewUserJoin()
-	}, plugins.WelcomeNew)
-
-	// 注册消息处理器
-	current.SetRobotMessageHandler(plugin.DispatchMessage(dispatcher))
-}

initialization/wechat.go

@@ -1,31 +0,0 @@
-package initialization
-
-import (
-	"github.com/go-resty/resty/v2"
-	"go-wechat/common/current"
-	"go-wechat/config"
-	"go-wechat/model/dto"
-	"log"
-)
-
-// InitWechatRobotInfo
-// @description: 初始化微信机器人信息
-func InitWechatRobotInfo() {
-	// 获取数据
-	var base dto.Response[dto.RobotUserInfo]
-	_, err := resty.New().R().
-		SetHeader("Content-Type", "application/json;chartset=utf-8").
-		SetResult(&base).
-		Post(config.Conf.Wechat.GetURL("/api/userInfo"))
-	if err != nil {
-		log.Printf("获取机器人信息失败: %s", err.Error())
-		return
-	}
-
-	log.Printf("机器人Id: %s", base.Data.WxId)
-	log.Printf("机器人微信号: %s", base.Data.Account)
-	log.Printf("机器人名称: %s", base.Data.Name)
-
-	// 设置为单例
-	current.SetRobotInfo(base.Data)
-}

internal/database/database.go

@@ -0,0 +1,49 @@
+package database
+
+import (
+	"gitee.ltd/lxh/logger"
+	"gitee.ltd/lxh/logger/log"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"os"
+	"strconv"
+	"wechat-robot/config"
+)
+
+// Client 客户端
+var Client *gorm.DB
+
+// Init
+// @description: 初始化数据库连接
+func Init() {
+	var dialector gorm.Dialector
+	switch config.Conf.Database.Type {
+	case "mysql":
+		// MySQL
+		dialector = mysql.Open(config.Conf.Database.GetMysqlDSN())
+	case "postgresql":
+		// PostgreSQL
+		dialector = postgres.Open(config.Conf.Database.GetPostgreSQLDSN())
+	default:
+		log.Panic("未配置数据库或数据库类型不支持")
+	}
+
+	// gorm 配置
+	gormConfig := gorm.Config{
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
+	// 是否开启调试模式
+	if flag, _ := strconv.ParseBool(os.Getenv("GORM_DEBUG")); flag {
+		gormConfig.Logger = logger.DefaultGormLogger()
+	}
+
+	// 初始化连接
+	conn, err := gorm.Open(dialector, &gormConfig)
+	if err != nil {
+		log.Panicf("数据库连接初始化失败: %v", err)
+	} else {
+		log.Debug("数据库连接初始化成功")
+	}
+	Client = conn
+}

internal/initialize/adminuser.go

@@ -0,0 +1,72 @@
+package initialize
+
+import (
+	"errors"
+	"gitee.ltd/lxh/logger/log"
+	"gorm.io/gorm"
+	"wechat-robot/internal/database"
+	"wechat-robot/model/entity"
+	"wechat-robot/utils"
+)
+
+// initDefaultAdminUser
+// @description: 初始化默认后台用户
+func initDefaultAdminUser() {
+	// 如果数据库没有有效用户，初始化一个默认账号
+	var count int64
+	if err := database.Client.Model(entity.AdminUser{}).Count(&count).Error; err != nil {
+		log.Panicf("初始化默认账号失败: %s", err.Error())
+	}
+	if count > 0 {
+		return
+	}
+
+	// 开启事务
+	var err error
+	tx := database.Client.Begin()
+	defer func() {
+		if recover() != nil {
+			tx.Rollback()
+		} else {
+			if err != nil {
+				tx.Rollback()
+			} else {
+				tx.Commit()
+			}
+		}
+	}()
+	// 判断角色是否存在
+	var role entity.Role
+	if err = tx.Where("code = ?", "admin").First(&role).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		log.Panicf("初始化默认账号失败: %s", err.Error())
+		return
+	}
+	// 如果角色不存在，创建角色
+	if role.Id == "" {
+		role.Name = "超级管理员"
+		role.Code = "admin"
+		role.Describe = "系统默认超级管理员"
+		if err = tx.Create(&role).Error; err != nil {
+			log.Panicf("初始化默认账号失败: %s", err.Error())
+			return
+		}
+	}
+
+	// 创建用户
+	var adminUser entity.AdminUser
+	adminUser.Username = "admin"
+	adminUser.Password = "admin123"
+	utils.PasswordUtils().HashPassword(&adminUser.Password)
+	if err = tx.Create(&adminUser).Error; err != nil {
+		log.Panicf("初始化默认账号失败: %s", err.Error())
+		return
+	}
+	// 绑定角色
+	var userRole entity.AdminUserRole
+	userRole.RoleId = role.Id
+	userRole.UserId = adminUser.Id
+	if err = tx.Create(&userRole).Error; err != nil {
+		log.Panicf("初始化默认账号失败: %s", err.Error())
+		return
+	}
+}

internal/initialize/config.go

@@ -1,19 +1,20 @@
-package initialization
+package initialize
 
 import (
+	"gitee.ltd/lxh/logger/log"
 	"github.com/fsnotify/fsnotify"
 	"github.com/spf13/viper"
-	"go-wechat/client"
+	"wechat-robot/config"
-	"go-wechat/config"
+	"wechat-robot/internal/database"
-	"log"
+	"wechat-robot/internal/redis"
 )
 
 // 配置管理工具
 var vp *viper.Viper
 
-// InitConfig
+// Config
 // @description: 初始化配置
-func InitConfig() {
+func initConfig() {
 	vp = viper.New()
 	vp.AddConfigPath(".")      // 设置配置文件路径
 	vp.SetConfigName("config") // 设置配置文件名
@@ -26,27 +27,21 @@ func InitConfig() {
 	if err := vp.Unmarshal(&config.Conf); err != nil {
 		log.Panicf("配置文件解析失败: %v", err)
 	}
-	log.Printf("配置文件解析完成: %+v", config.Conf)
+	log.Debugf("配置文件解析完成: %+v", config.Conf)
-	if !config.Conf.Wechat.Check() {
-		log.Panicf("微信HOOK配置缺失")
-	}
 	// 初始化数据库连接
-	client.InitMySQLClient()
+	database.Init()
-	//redis.Init()
+	redis.Init()
 
 	// 下面的代码是配置变动之后自动刷新的
 	vp.WatchConfig()
 	vp.OnConfigChange(func(e fsnotify.Event) {
 		// 绑定配置文件
 		if err := vp.Unmarshal(&config.Conf); err != nil {
-			log.Printf("配置文件更新失败: %v", err)
+			log.Errorf("配置文件更新失败: %v", err)
 		} else {
-			if !config.Conf.Wechat.Check() {
-				log.Panicf("微信HOOK配置缺失")
-			}
 			// 初始化数据库连接
-			client.InitMySQLClient()
+			database.Init()
-			//redis.Init()
+			redis.Init()
 		}
 	})
 }

internal/initialize/datatable.go

@@ -0,0 +1,28 @@
+package initialize
+
+import (
+	"gitee.ltd/lxh/logger/log"
+	"wechat-robot/internal/database"
+	"wechat-robot/model/entity"
+)
+
+// databaseTable
+// @description: 初始化数据库表
+func databaseTable() {
+	tables := []any{
+		new(entity.AdminUser),     // 用户表
+		new(entity.Menu),          // 菜单表
+		new(entity.Role),          // 角色表
+		new(entity.RoleMenu),      // 角色菜单表
+		new(entity.AdminUserRole), // 用户角色表
+		new(entity.SystemConfig),  // 系统配置表
+		new(entity.Robot),         // 机器人表
+		new(entity.AiAssistant),   // AI助手表
+		new(entity.Message),       // 微信消息表
+	}
+
+	// 同步表结构
+	if err := database.Client.AutoMigrate(tables...); err != nil {
+		log.Panicf("初始化数据库表失败: %v", err)
+	}
+}

internal/initialize/init.go

@@ -0,0 +1,18 @@
+package initialize
+
+import (
+	"wechat-robot/internal/tasks"
+	"wechat-robot/mq"
+	"wechat-robot/pkg/auth"
+)
+
+// InitSystem
+// @description: 初始化系统
+func InitSystem() {
+	initConfig()            // 初始化配置
+	databaseTable()         // 初始化数据库表
+	initDefaultAdminUser()  // 初始化默认管理员用户
+	auth.InitOAuth2Server() // 初始化OAuth2服务
+	tasks.StartScheduled()  // 启动定时任务
+	mq.Init()               // 初始化MQ
+}

internal/message/handler.go

@@ -0,0 +1,56 @@
+package message
+
+import (
+	"encoding/json"
+	"log"
+	"strings"
+	"time"
+	"wechat-robot/model/entity"
+	"wechat-robot/model/robot"
+	"wechat-robot/pkg/types"
+	"wechat-robot/service/message"
+)
+
+// Message
+// @description: 处理消息
+// @param msg
+// @return err
+func Message(msg []byte) (err error) {
+	var m robot.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机器人Id: %s\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", m.ToUser, m.FromUser, m.GroupUser, m.Type, m.Content)
+
+	// 消息入库
+	var ent entity.Message
+	ent.MsgId = m.MsgId
+	ent.Timestamp = m.CreateTime
+	ent.MessageTime = time.Unix(int64(m.CreateTime), 0)
+	ent.Content = m.Content
+	ent.FromUser = m.FromUser
+	ent.GroupUser = m.GroupUser
+	ent.ToUser = m.ToUser
+	ent.Type = m.Type
+	ent.DisplayFullContent = m.DisplayFullContent
+	ent.Raw = m.Raw
+	err = message.Save(ent)
+	return
+}

internal/orm/page.go

@@ -0,0 +1,30 @@
+package orm
+
+import "gorm.io/gorm"
+
+// Page
+// @description: 分页组件
+// @param current
+// @param size
+// @return func(db *gorm.DB) *gorm.DB
+func Page(current, size int) func(db *gorm.DB) *gorm.DB {
+	// 如果页码是-1，就不分页
+	if current == -1 {
+		return func(db *gorm.DB) *gorm.DB {
+			return db
+		}
+	}
+	// 分页
+	return func(db *gorm.DB) *gorm.DB {
+		if current == 0 {
+			current = 1
+		}
+		if size < 1 {
+			size = 10
+		}
+		// 计算偏移量
+		offset := (current - 1) * size
+		// 返回组装结果
+		return db.Offset(offset).Limit(size)
+	}
+}

internal/orm/sort.go

@@ -0,0 +1,148 @@
+package orm
+
+import (
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"gorm.io/gorm"
+	"slices"
+)
+
+// @title checkHasDeletedAt
+// @description 检查指定表是否有id_del字段
+// @param tx *gorm.DB "已开启的事务对象"
+// @param tableName string "表名"
+// @return bool "是否包含"
+func checkHasDeletedAt(tx *gorm.DB, tableName string) bool {
+	var columns []string
+	// SQL语句
+	sql := fmt.Sprintf("select COLUMN_NAME from information_schema.COLUMNS where table_name = '%s'", tableName)
+	err := tx.Raw(sql).Scan(&columns).Error
+	if err != nil {
+		log.Errorf("查询表字段失败: %v", err.Error())
+		return false
+	}
+	return slices.Contains(columns, "id_del")
+}
+
+// @title checkHasUpdatedAt
+// @description 检查指定表是否有updated_at字段
+// @param tx *gorm.DB "已开启的事务对象"
+// @param tableName string "表名"
+// @return bool "是否包含"
+func checkHasUpdatedAt(tx *gorm.DB, tableName string) bool {
+	var columns []string
+	// SQL语句
+	sql := fmt.Sprintf("select COLUMN_NAME from information_schema.COLUMNS where table_name = '%s'", tableName)
+	err := tx.Raw(sql).Scan(&columns).Error
+	if err != nil {
+		log.Errorf("查询表字段失败: %v", err.Error())
+		return false
+	}
+	return slices.Contains(columns, "updated_at")
+}
+
+// UpdateSortBefore
+// @description 更新之前处理序号
+// @param tx *gorm.DB "已开启的事务对象"
+// @param model any "模型对象"
+// @return error "错误信息"
+func UpdateSortBefore(tx *gorm.DB, tableName, id string, sort int, param string) (err error) {
+	// 查出原来的排序号
+	var oldSort int
+	err = tx.Table(tableName).Select("`sort`").Where("id = ?", id).Scan(&oldSort).Error
+	if err != nil {
+		log.Errorf("查询老数据失败: %v", err.Error())
+		return
+	}
+	// 如果相等，啥都不干
+	if oldSort == sort {
+		return nil
+	}
+
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, tableName)
+
+	// 处理排序
+	// 如果老的排序号小于新的，(老, 新]之间的排序号都要-1
+	// 如果老的大于新的，[老, 新)排序号-1
+	if oldSort < sort {
+		// 老的小于新的，[老, 新) + 1
+		sel := tx.Table(tableName).
+			Where("sort <= ? AND sort > ?", sort, oldSort)
+		if hasDeletedAt {
+			sel.Where("id_del = 0")
+		}
+		if param != "" {
+			sel.Where(param) // 自定义条件
+		}
+		err = sel.Update("sort", gorm.Expr("sort - 1")).Error
+	} else {
+		// 老的大于新的，[新, 老) + 1
+		sel := tx.Table(tableName).
+			Where("sort >= ? AND sort < ?", sort, oldSort)
+		if hasDeletedAt {
+			sel.Where("id_del = 0")
+		}
+		if param != "" {
+			sel.Where(param) // 自定义条件
+		}
+		err = sel.Update("sort", gorm.Expr("sort + 1")).Error
+	}
+	return
+}
+
+// CreateSortBefore
+// @description 新建之前处理序号
+// @param tx *gorm.DB "已开启的事务对象"
+// @param model any "模型对象"
+// @return error "错误信息"
+func CreateSortBefore(tx *gorm.DB, tableName string, sort int, param string) (err error) {
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, tableName)
+
+	// 处理排序,如果没有传，就会是在最前面
+	sel := tx.Table(tableName).Where("sort >= ?", sort)
+	if hasDeletedAt {
+		sel.Where("id_del = 0")
+	}
+	if param != "" {
+		sel.Where(param)
+	}
+	err = sel.Update("sort", gorm.Expr("sort + 1")).Error
+	if err != nil {
+		log.Errorf("处理前置排序失败：%v", err)
+	}
+	return
+}
+
+// DealSortAfter
+// @description 处理序号之后
+// @param tx *gorm.DB "已开启的事务对象"
+// @param modelName string "表名"
+// @return error "错误信息"
+func DealSortAfter(tx *gorm.DB, modelName, param string) (err error) {
+	// 保存成功，刷新排序
+	if param != "" {
+		param = " AND " + param
+	}
+
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, modelName)
+	if hasDeletedAt {
+		param += " AND id_del = 0"
+	}
+
+	// 如果有更新时间字段，也更新一下值
+	updateParam := ""
+	if checkHasUpdatedAt(tx, modelName) {
+		updateParam = ",updated_at = NOW()"
+	}
+
+	sql := fmt.Sprintf("UPDATE %s a, (SELECT (@i := @i + 1) i, id FROM %s WHERE 1=1 %s order by sort ASC) i, "+
+		"(SELECT @i := 0) ir SET a.sort = i.i %s WHERE a.id = i.id", modelName, modelName, param, updateParam)
+	err = tx.Exec(sql).Error
+	if err != nil {
+		log.Errorf("刷新排序失败: %v", err.Error())
+	}
+	return
+}

internal/redis/redis.go

@@ -0,0 +1,28 @@
+package redis
+
+import (
+	"context"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/go-redis/redis/v8"
+	"wechat-robot/config"
+)
+
+var Client *redis.Client
+
+// Init
+// @description: 初始化redis客户端
+func Init() {
+	conf := config.Conf.Redis
+	// 初始化连接
+	conn := redis.NewClient(&redis.Options{
+		Addr:     conf.GetDSN(),
+		Password: conf.Password,
+		DB:       conf.Db,
+	})
+	if err := conn.Ping(context.Background()).Err(); err != nil {
+		log.Panicf("Redis连接初始化失败: %v", err)
+	} else {
+		log.Debug("Redis连接初始化成功")
+	}
+	Client = conn
+}

internal/tasks/tasks.go

@@ -0,0 +1,20 @@
+package tasks
+
+import (
+	"github.com/go-co-op/gocron"
+	"log"
+	"time"
+)
+
+var Scheduler *gocron.Scheduler
+
+// StartScheduled
+// @description: 启动定时任务
+func StartScheduled() {
+	// 定时任务发送消息
+	Scheduler = gocron.NewScheduler(time.Local)
+
+	// 开启定时任务
+	Scheduler.StartAsync()
+	log.Println("定时任务初始化成功")
+}

main.go

@@ -1,98 +1,49 @@
 package main
 
 import (
+	"gitee.ltd/lxh/logger"
+	"gitee.ltd/lxh/logger/log"
 	"github.com/gin-gonic/gin"
-	"go-wechat/config"
+	"wechat-robot/internal/initialize"
-	"go-wechat/initialization"
+	"wechat-robot/pkg/validator"
-	"go-wechat/mq"
+	"wechat-robot/router/admin"
-	"go-wechat/router"
+	"wechat-robot/router/callback"
-	"go-wechat/tasks"
+	"wechat-robot/router/middleware"
-	"go-wechat/tcpserver"
-	"go-wechat/utils"
-	"html/template"
-	"log"
-	"net/http"
-	"os"
-	"strconv"
-	"strings"
-	"time"
 )
 
+// init
+// @description: 初始系统
 func init() {
-	initialization.InitConfig()          // 初始化配置
+	// 初始化日志工具
-	initialization.InitWechatRobotInfo() // 初始化机器人信息
+	logger.InitLogger(logger.LogConfig{Mode: logger.Dev, LokiEnable: false, FileEnable: true})
-	initialization.Plugin()              // 注册插件
+	// 初始化系统
-	tasks.InitTasks()                    // 初始化定时任务
+	initialize.InitSystem()
-	mq.Init()                            // 初始化MQ
-
-	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
-		log.Printf("已设置环境变量DONT_SEND，不发送消息")
-	}
-	if flag, _ := strconv.ParseBool(os.Getenv("DONT_SAVE")); flag {
-		log.Printf("已设置环境变量DONT_SAVE，不保存消息")
-	}
 }
 
+// main
+// @description: 启动入口
 func main() {
-	// 如果启用了自动配置回调，就设置一下
-	if config.Conf.Wechat.AutoSetCallback {
-		utils.ClearCallback()
-		time.Sleep(500 * time.Millisecond) // 休眠五百毫秒再设置
-		utils.SetCallback(config.Conf.Wechat.Callback)
-	}
-
 	// 启动TCP服务
-	go tcpserver.Start()
+	//go tcpserver.Start()
 
-	// 启动HTTP服务
+	// 注册参数绑定错误信息翻译器
+	validator.Init()
 	app := gin.Default()
 
-	// 自定义模板引擎函数
+	// 开启自定义请求方式不允许处理函数
-	app.SetFuncMap(template.FuncMap{
+	app.HandleMethodNotAllowed = true
-		"codeToChinese": func(code string) string {
+	// 处理请求方式不对
-			switch code {
+	app.NoMethod(middleware.NoMethodHandler())
-			case "friend":
-				return "好友列表"
-			case "group":
-				return "群组列表"
-			case "index":
-				return "首页"
-			case "assistant":
-				return "AI角色"
-			default:
-				return "其他页面"
-			}
-		},
-		"boolToChinese": func(flag bool) string {
-			if flag {
-				return "是"
-			}
-			return "否"
-		},
-		"checkIsGroup": func(str string) bool {
-			// 如果id包含@chatroom就是群组
-			return strings.HasSuffix(str, "@chatroom")
-		},
-	})
-
-	app.LoadHTMLGlob("views/*.html")
-	app.Static("/assets", "./views/static")
-	app.StaticFile("/favicon.ico", "./views/wechat.ico")
 	// 404返回数据
-	app.NoRoute(func(ctx *gin.Context) {
+	app.NoRoute(middleware.NoRouteHandler())
-		if strings.HasPrefix(ctx.Request.URL.Path, "/api") {
+
-			ctx.String(404, "接口不存在")
+	// 初始化接口路由
+	admin.InitRoute(app.Group("/admin/v1"))    // 后台接口
+	callback.InitRoute(app.Group("/callback")) // 回调接口
+
+	// 启动服务
+	if err := app.Run(":8080"); err != nil {
+		log.Errorf("服务启动失败：%v", err)
 		return
 	}
-		// 404直接跳转到首页
-		ctx.Redirect(302, "/404.html")
-	})
-	app.NoMethod(func(ctx *gin.Context) {
-		ctx.String(http.StatusMethodNotAllowed, "不支持的请求方式")
-	})
-	// 初始化路由
-	router.Init(app)
-	if err := app.Run(":8080"); err != nil {
-		log.Panicf("服务启动失败：%v", err)
-	}
 }

model/dto/leigod.go

@@ -1,73 +0,0 @@
-package dto
-
-// LeiGodLoginResp
-// @description: 雷神登录返回
-type LeiGodLoginResp struct {
-	LoginInfo struct {
-		AccountToken string `json:"account_token"` // Token
-		ExpiryTime   string `json:"expiry_time"`   // 有效期
-		NnToken      string `json:"nn_token"`
-	} `json:"login_info"`
-	UserInfo struct {
-		Nickname   string `json:"nickname"`
-		Email      string `json:"email"`
-		Mobile     string `json:"mobile"`
-		Avatar     string `json:"avatar"`
-		RegionCode int    `json:"region_code"`
-	} `json:"user_info"`
-}
-
-// LeiGodUserInfoResp
-// @description: 雷神用户信息返回
-type LeiGodUserInfoResp struct {
-	UserPauseTime         int    `json:"user_pause_time"`
-	Nickname              string `json:"nickname"`
-	Email                 string `json:"email"`
-	CountryCode           string `json:"country_code"`
-	Mobile                string `json:"mobile"`
-	UserName              string `json:"user_name"`
-	MasterAccount         string `json:"master_account"`
-	Birthday              string `json:"birthday"`
-	PublicIp              string `json:"public_ip"`
-	Sex                   string `json:"sex"`
-	LastLoginTime         string `json:"last_login_time"`
-	LastLoginIp           string `json:"last_login_ip"`
-	PauseStatus           string `json:"pause_status"`    // 暂停状态
-	PauseStatusId         int    `json:"pause_status_id"` // 暂停状态 0未暂停1已暂停
-	LastPauseTime         string `json:"last_pause_time"` // 最后一次暂停时间
-	VipLevel              string `json:"vip_level"`
-	Avatar                string `json:"avatar"`
-	AvatarNew             string `json:"avatar_new"`
-	PackageId             string `json:"package_id"`
-	IsSwitchPackage       int    `json:"is_switch_package"`
-	PackageTitle          string `json:"package_title"`
-	PackageLevel          string `json:"package_level"`
-	BillingType           string `json:"billing_type"`
-	Lang                  string `json:"lang"`
-	StopedRemaining       string `json:"stoped_remaining"`
-	ExpiryTime            string `json:"expiry_time"`      // 剩余时长
-	ExpiryTimeSamp        int    `json:"expiry_time_samp"` // 剩余时长秒数
-	Address               string `json:"address"`
-	MobileContactType     string `json:"mobile_contact_type"`
-	MobileContactNumber   string `json:"mobile_contact_number"`
-	MobileContactTitle    string `json:"mobile_contact_title"`
-	RegionCode            int    `json:"region_code"`
-	IsPayUser             string `json:"is_pay_user"`
-	WallLogSwitch         string `json:"wall_log_switch"`
-	IsSetAdminPass        int    `json:"is_set_admin_pass"`
-	ExpiredExperienceTime string `json:"expired_experience_time"`
-	ExperienceExpiryTime  string `json:"experience_expiry_time"`
-	ExperienceTime        int    `json:"experience_time"`
-	FirstInvoiceDiscount  int    `json:"first_invoice_discount"`
-	NnNumber              string `json:"nn_number"`
-	UserSignature         string `json:"user_signature"`
-	MobileExpiryTime      string `json:"mobile_expiry_time"`
-	MobileExpiryTimeSamp  int    `json:"mobile_expiry_time_samp"`
-	MobilePauseStatus     int    `json:"mobile_pause_status"`
-	BlackExpiredTime      string `json:"black_expired_time"`
-	MobileExperienceTime  string `json:"mobile_experience_time"`
-	SuperTime             string `json:"super_time"`
-	NowDate               string `json:"now_date"`
-	NowTimeSamp           int    `json:"now_time_samp"`
-	UserEarnMinutes       string `json:"user_earn_minutes"`
-}

model/dto/news.go

@@ -1,38 +0,0 @@
-package dto
-
-// 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"`
-}
-
-// HotTop
-// @description: 热搜排行榜返回结构体
-type HotTop struct {
-	Success    bool             `json:"success"`     // 是否成功
-	Name       string           `json:"name"`        // 渠道
-	Subtitle   string           `json:"subtitle"`    // 副标题
-	UpdateTime string           `json:"update_time"` // 更新时间
-	Data       []HotTopDataItem `json:"data"`        // 数据
-}
-
-// HotTopDataItem
-// @description: 热搜排行榜数据项
-type HotTopDataItem struct {
-	Index    int    `json:"index"`    // 排行
-	Title    string `json:"title"`    // 标题
-	Hot      string `json:"hot"`      // 热度
-	Url      string `json:"url"`      // 链接
-	MobilUrl string `json:"mobilUrl"` // 手机端链接
-}

model/dto/urlc.go

@@ -1,8 +0,0 @@
-package dto
-
-// ShortUrlResponse
-// @description: 短链接返回结构体
-type ShortUrlResponse struct {
-	Error int    `json:"error"` // 错误码，1表示发生错误，0则正常
-	Short string `json:"short"` // 短链接
-}

model/entity/adminuser.go

@@ -0,0 +1,23 @@
+package entity
+
+import "wechat-robot/pkg/types"
+
+// AdminUser
+// @description: 用户表
+type AdminUser struct {
+	types.BaseDbModel
+	Username    string          `json:"username" gorm:"index:deleted,unique;type:varchar(255); not null; comment:'登录账号'"`
+	Password    string          `json:"password" gorm:"type:varchar(255); comment:'密码'"`
+	Email       *string         `json:"email" gorm:"type:varchar(255); comment:'邮箱'"`
+	IsVerified  bool            `json:"isVerified" gorm:"type:tinyint(1); not null; default:0; comment:'是否验证邮箱'"`
+	LastLoginAt *types.DateTime `json:"lastLoginAt" gorm:"comment:'最后登录时间'"`
+	LastLoginIp *string         `json:"lastLoginIp" gorm:"type:varchar(255); comment:'最后登录ip'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver User
+// @return string
+func (AdminUser) TableName() string {
+	return "t_admin_user"
+}

model/entity/aiassistant.go

@@ -1,20 +1,14 @@
 package entity
 
-import (
+import "wechat-robot/pkg/types"
-	"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"`
+	types.BaseDbModel
-	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:"dto" gorm:"type:varchar(50);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:'是否启用'"`
 }
 
@@ -25,15 +19,3 @@ type AiAssistant struct {
 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
-}

model/entity/friend.go

@@ -1,54 +0,0 @@
-package entity
-
-import (
-	"go-wechat/common/types"
-	"time"
-)
-
-// Friend
-// @description: 好友列表
-type Friend struct {
-	Wxid           string         `json:"wxid"`                                                     // 微信原始Id
-	CustomAccount  string         `json:"customAccount"`                                            // 微信号
-	Nickname       string         `json:"nickname"`                                                 // 昵称
-	Pinyin         string         `json:"pinyin"`                                                   // 昵称拼音大写首字母
-	PinyinAll      string         `json:"pinyinAll"`                                                // 昵称全拼
-	LastActive     types.DateTime `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"`     // 是否启用新闻
-	EnableHotTop   bool           `json:"enableHotTop" gorm:"type:tinyint(1) default 0 not null"`   // 是否启用热榜新闻
-	ClearMember    int            `json:"clearMember"`                                              // 清理成员配置（多少天未活跃的）
-	IsOk           bool           `json:"isOk" gorm:"type:tinyint(1) default 0 not null"`           // 是否正常
-	UsedTokens     int            `json:"usedTokens"`                                               // 已使用的AI Token数量
-	AiFreeLimit    int            `json:"aiFreeLimit"`                                              // AI免费次数
-	AiUsedToday    int            `json:"aiUsedToday"`                                              // 今日已使用的AI次数
-}
-
-func (Friend) TableName() string {
-	return "t_friend"
-}
-
-// GroupUser
-// @description: 群成员
-type GroupUser struct {
-	GroupId      string     `json:"groupId"`                                                // 群Id
-	Wxid         string     `json:"wxid"`                                                   // 微信Id
-	Account      string     `json:"account"`                                                // 账号
-	HeadImage    string     `json:"headImage"`                                              // 头像
-	Nickname     string     `json:"nickname"`                                               // 昵称
-	IsMember     bool       `json:"isMember" gorm:"type:tinyint(1) default 0 not null"`     // 是否群成员
-	IsAdmin      bool       `json:"isAdmin" gorm:"type:tinyint(1) default 0 not null"`      // 是否群主
-	JoinTime     time.Time  `json:"joinTime"`                                               // 加入时间
-	LastActive   time.Time  `json:"lastActive"`                                             // 最后活跃时间
-	LeaveTime    *time.Time `json:"leaveTime"`                                              // 离开时间
-	SkipChatRank bool       `json:"skipChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否跳过聊天排行
-}
-
-func (GroupUser) TableName() string {
-	return "t_group_user"
-}

model/entity/hottop.go

@@ -1,40 +0,0 @@
-package entity
-
-import (
-	"github.com/google/uuid"
-	"go-wechat/common/types"
-	"gorm.io/gorm"
-	"strings"
-)
-
-// HotTop
-// @description: 热榜数据
-type HotTop struct {
-	Id        string         `json:"id" gorm:"type:varchar(32);primarykey"`
-	CreatedAt types.DateTime `json:"createdAt"`
-	Channel   string         `json:"channel"`   // 渠道
-	Title     string         `json:"title"`     // 标题
-	Hot       string         `json:"hot"`       // 热度
-	Url       string         `json:"url"`       // 链接
-	MobileUrl string         `json:"mobileUrl"` // 手机端链接
-}
-
-// TableName
-// @description: 表名
-// @receiver HotTop
-// @return string
-func (HotTop) TableName() string {
-	return "t_hot_top"
-}
-
-// BeforeCreate
-// @description: 创建数据库对象之前生成UUID
-// @receiver m
-// @param *gorm.DB
-// @return err
-func (m *HotTop) BeforeCreate(*gorm.DB) (err error) {
-	if m.Id == "" {
-		m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
-	}
-	return
-}

model/entity/menu.go

@@ -0,0 +1,24 @@
+package entity
+
+import "wechat-robot/pkg/types"
+
+// Menu
+// @description: 菜单表
+type Menu struct {
+	types.BaseDbModel
+	Type     types.MenuType `json:"type" gorm:"type:enum('MENU','BUTTON'); default:'MENU'; not null; comment:'菜单类型(菜单或按钮)'"`
+	Name     string         `json:"name" gorm:"type:varchar(255);not null;comment:'页面组件名称'"`
+	Path     string         `json:"path" gorm:"type:varchar(255);comment:'路径'"`
+	Title    string         `json:"title" gorm:"type:varchar(255);not null;comment:'菜单标题'"`
+	Icon     string         `json:"icon" gorm:"type:varchar(255);comment:'菜单图标'"`
+	Sort     int            `json:"sort" gorm:"type:int(3);not null;comment:'排序值(数字越小约靠前)'"`
+	ParentId *string        `json:"parentId" gorm:"type:varchar(32);comment:'父级菜单ID(0表示顶级菜单)'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver Menu
+// @return string
+func (Menu) TableName() string {
+	return "t_menu"
+}

model/entity/message.go

@@ -1,23 +1,24 @@
 package entity
 
 import (
-	"go-wechat/types"
 	"time"
+	"wechat-robot/pkg/types"
 )
 
 // Message
 // @description: 消息数据库结构体
 type Message struct {
-	MsgId              int64             `gorm:"primaryKey"` // 消息Id
+	types.BaseDbModelWithReal
-	CreateTime         int               // 发送时间戳
+	MsgId              int64             `gorm:"index:idx_msg_id,unique;type:bigint;not null;comment:'微信Id'"`      // 消息Id
-	CreateAt           time.Time         // 发送时间
+	Timestamp          int               `gorm:"type:bigint;not null;comment:'消息时间戳'"`                             // 发送时间戳
-	Type               types.MessageType // 消息类型
+	MessageTime        time.Time         `gorm:"type:datetime;not null;comment:'消息时间'"`                            // 发送时间
-	Content            string            // 内容
+	Type               types.MessageType `gorm:"type:int;not null;comment:'消息类型'"`                                 // 消息类型
-	DisplayFullContent string            // 显示的完整内容
+	Content            string            `gorm:"type:longtext;not null;comment:'消息内容'"`                            // 内容
-	FromUser           string            // 发送者
+	DisplayFullContent string            `gorm:"type:longtext;not null;comment:'显示的完整内容'"`                         // 显示的完整内容
-	GroupUser          string            // 群成员
+	FromUser           string            `gorm:"type:varchar(100);not null;comment:'发送者'"`                         // 发送者
-	ToUser             string            // 接收者
+	GroupUser          string            `gorm:"type:varchar(100);comment:'群成员'"`                                  // 群成员
-	Raw                string            // 原始通知字符串
+	ToUser             string            `gorm:"index:idx_msg_id,unique;type:varchar(100);not null;comment:'接收者'"` // 接收者
+	Raw                string            `gorm:"type:longtext;not null;comment:'原始通知字符串'"`                         // 原始通知字符串
 }
 
 func (Message) TableName() string {

model/entity/plugindata.go

@@ -1,13 +0,0 @@
-package entity
-
-// PluginData
-// @description: 插件数据
-type PluginData struct {
-	UserId     string `json:"userId"`     // 用户Id
-	PluginCode string `json:"pluginCode"` // 插件编码
-	Data       string `json:"data"`       // 数据
-}
-
-func (PluginData) TableName() string {
-	return "t_plugin_data"
-}

model/entity/robot.go

@@ -0,0 +1,30 @@
+package entity
+
+import "wechat-robot/pkg/types"
+
+// Robot
+// @description: 机器人信息
+type Robot struct {
+	types.BaseDbModel
+	WxId            string `json:"wxid" gorm:"index:deleted,unique;column:wxid;type:varchar(255);not null;comment:'微信Id'"` // 微信Id
+	Account         string `json:"account" gorm:"type:varchar(255);comment:'微信号'"`                                         // 微信号
+	Nickname        string `json:"name" gorm:"type:varchar(255);comment:'昵称'"`                                             // 昵称
+	Avatar          string `json:"avatar" gorm:"type:varchar(255);comment:'头像'"`                                           // 头像
+	Mobile          string `json:"mobile" gorm:"type:varchar(255);comment:'手机号'"`                                          // 手机
+	CurrentDataPath string `json:"currentDataPath" gorm:"type:varchar(255);comment:'当前数据目录'"`                              // 当前数据目录,登录的账号目录
+	DataSavePath    string `json:"dataSavePath" gorm:"type:varchar(255);comment:'微信保存目录'"`                                 // 微信保存目录
+	DbKey           string `json:"dbKey" gorm:"type:varchar(255);comment:'数据库的SQLCipher的加密key'"`                           // 数据库的SQLCipher的加密key，可以使用该key配合decrypt.py解密数据库
+	HookApi         string `json:"hookApi" gorm:"type:varchar(255);comment:'hook接口地址'"`                                    // hook接口地址
+	Remark          string `json:"remark" gorm:"type:varchar(255);comment:'备注'"`                                           // 备注
+	Version         int    `json:"version" gorm:"type:int;comment:'版本号'"`                                                  // 版本号
+	VncUrl          string `json:"vncUrl" gorm:"type:varchar(255);comment:'vnc地址'"`                                        // vnc地址
+	Tag             string `json:"tag" gorm:"type:varchar(255);comment:'标签'"`                                              // 标签
+}
+
+// TableName
+// @description: 表名
+// @receiver Robot
+// @return string
+func (Robot) TableName() string {
+	return "t_robot"
+}

model/entity/role.go

@@ -0,0 +1,56 @@
+package entity
+
+import "wechat-robot/pkg/types"
+
+// Role
+// @description: 角色表
+type Role struct {
+	types.BaseDbModel
+	Name     string `json:"name" gorm:"type:varchar(20) not null comment '角色名称'"`
+	Code     string `json:"code" gorm:"type:varchar(20) not null comment '角色代码'"`
+	Describe string `json:"describe" gorm:"type:varchar(20) not null comment '描述'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver Role
+// @return string
+func (Role) TableName() string {
+	return "t_role"
+}
+
+// =====================================================================================================================
+
+// RoleMenu
+// @description: 角色菜单表
+type RoleMenu struct {
+	types.BaseDbModelWithReal
+	RoleId string `json:"roleId" gorm:"index:idx_rm_only,unique;type:varchar(32);not null;comment:'角色id'"`
+	MenuId string `json:"menuId" gorm:"index:idx_rm_only,unique;type:varchar(32);not null;comment:'菜单id'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver RoleMenu
+// @return string
+func (RoleMenu) TableName() string {
+	return "t_role_menu"
+}
+
+// =====================================================================================================================
+
+// AdminUserRole
+// @description: 管理员角色表
+type AdminUserRole struct {
+	types.BaseDbModelWithReal
+	RoleId string `json:"roleId" gorm:"index:idx_aur_only,unique;type:varchar(32);not null;comment:'角色id'"`
+	UserId string `json:"menuId" gorm:"index:idx_aur_only,unique;type:varchar(32);not null;comment:'管理员id'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver AdminUserRole
+// @return string
+func (AdminUserRole) TableName() string {
+	return "t_admin_user_role"
+}

model/entity/systemconfig.go

@@ -0,0 +1,19 @@
+package entity
+
+import "wechat-robot/pkg/types"
+
+// SystemConfig
+// @description: 系统配置
+type SystemConfig struct {
+	types.BaseDbModelWithReal
+	Code    string `json:"code" gorm:"type:varchar(255);not null;comment:'配置编码'"`
+	Content string `json:"content" gorm:"type:text;comment:'配置内容'"`
+}
+
+// TableName
+// @description: 表名
+// @receiver SystemConfig
+// @return string
+func (SystemConfig) TableName() string {
+	return "t_system_config"
+}

model/param/aiassistant/save.go

@@ -0,0 +1,11 @@
+package aiassistant
+
+// Save
+// @description: 保存AI助手入参
+type Save struct {
+	Id          string `json:"id" form:"id"`                                      // Id
+	Name        string `json:"name" form:"name" binding:"required"`               // 名称
+	Personality string `json:"personality" form:"personality" binding:"required"` // 人设
+	Model       string `json:"model" form:"model" binding:"required"`             // 使用的模型
+	Enable      bool   `json:"enable" form:"enable"`                              // 是否启用
+}

model/param/aiassistant/select.go

@@ -0,0 +1,7 @@
+package aiassistant
+
+// GetAll
+// @description: 获取所有AI助手
+type GetAll struct {
+	Keyword string `json:"keyword" form:"keyword"` // 关键字
+}

model/param/callback/hook.go

@@ -0,0 +1,8 @@
+package callback
+
+// RobotHook
+// @description: 机器人HOOK后回调
+type RobotHook struct {
+	Robot  string `json:"robot" form:"robot"`   // 机器人
+	Status string `json:"status" form:"status"` // 状态
+}

model/param/login/login.go

@@ -0,0 +1,24 @@
+package login
+
+// GetTokenWithPassword
+// @description: 账号密码登录
+type GetTokenWithPassword struct {
+	VerifyId   string `json:"verifyId" form:"verifyId" binding:"required"`     // 验证Id
+	VerifyCode string `json:"verifyCode" form:"verifyCode" binding:"required"` // 验证码
+	Username   string `json:"username" form:"username" binding:"required"`     // 用户名
+	Password   string `json:"password" form:"password" binding:"required"`     // 密码
+}
+
+// RefreshToken
+// @description: 刷新Token入参
+type RefreshToken struct {
+	RefreshToken string `json:"refresh_token" form:"refresh_token" binding:"required"` // 刷新Token
+}
+
+// Register
+// @description: 注册账号
+type Register struct {
+	Username       string `json:"username" form:"username" binding:"required"` // 用户名
+	Password       string `json:"password" form:"password" binding:"required"` // 密码
+	InvitationCode string `json:"invitationCode" form:"invitationCode"`        // 邀请码
+}

model/param/menu/save.go

@@ -0,0 +1,16 @@
+package menu
+
+import "wechat-robot/pkg/types"
+
+// Save
+// @description: 保存菜单入参
+type Save struct {
+	Id       string         `json:"id" form:"id" comment:"菜单Id"`
+	Type     types.MenuType `json:"type" form:"type" comment:"菜单类型(菜单或按钮)" validate:"oneof=MENU BUTTON"`
+	Name     string         `json:"name" form:"name" comment:"页面组件名称"`
+	Path     string         `json:"path" form:"path" comment:"访问路径"`
+	Title    string         `json:"title" form:"title" comment:"菜单标题"`
+	Icon     string         `json:"icon" form:"icon" comment:"菜单图标"`
+	Sort     int            `json:"sort" form:"sort" comment:"排序值(数字越小越靠前)"`
+	ParentId string         `json:"parentId" form:"parentId" comment:"父级菜单ID"`
+}

model/param/robot/robot.go

@@ -0,0 +1,13 @@
+package robot
+
+import "wechat-robot/pkg/wxhelper"
+
+// Save
+// @description: 保存机器人
+type Save struct {
+	Id      string                 `json:"id" form:"id"`           // id
+	HookApi string                 `json:"hookApi" form:"hookApi"` // hook接口地址
+	Version wxhelper.WeChatVersion `json:"version" form:"version"` // 版本
+	VncUrl  string                 `json:"vncUrl" form:"vncUrl"`   // vnc地址
+	Remark  string                 `json:"remark" form:"remark"`   // 备注
+}

model/param/robot/select.go

@@ -0,0 +1,8 @@
+package robot
+
+// GetAll
+// @description: 查询所有机器人入参
+type GetAll struct {
+	Keyword string `json:"keyword" form:"keyword"` // 关键字
+	Tag     string `json:"tag" form:"tag"`         // 标签
+}

model/param/role/select.go

@@ -0,0 +1,8 @@
+package role
+
+// GetAll
+// @description: 获取所有角色
+type GetAll struct {
+	Keyword string `json:"keyword" form:"keyword"` // 关键词
+	Code    string `json:"code" form:"code"`       // 角色代码
+}

model/robot/chatroom.go

@@ -0,0 +1,20 @@
+package robot
+
+// ChatRoomDetailInfo
+// @description: 群聊详情
+type ChatRoomDetailInfo struct {
+	Admin      string `json:"admin"`      // 群主微信
+	ChatRoomId string `json:"chatRoomId"` // 群Id
+	Notice     string `json:"notice"`     // 群公告
+	Xml        string `json:"xml"`        // 不知道是啥玩意儿
+}
+
+// GroupUser
+// @description: 群成员返回结果
+type GroupUser struct {
+	Admin          string `json:"admin"`          // 群主微信
+	AdminNickname  string `json:"adminNickname"`  // 群主昵称
+	ChatRoomId     string `json:"chatRoomId"`     // 群Id
+	MemberNickname string `json:"memberNickname"` // 成员昵称 `^G`切割
+	Members        string `json:"members"`        // 成员Id `^G`切割
+}

model/robot/friend.go

@@ -1,4 +1,4 @@
-package dto
+package robot
 
 // FriendItem
 // @description: 好友列表数据
@@ -12,17 +12,7 @@ type FriendItem struct {
 	Reserved2     int    `json:"reserved2"`     // 未知
 	Type          int    `json:"type"`          // 类型
 	VerifyFlag    int    `json:"verifyFlag"`    // 未知
-	Wxid          string `json:"wxid"`          // 微信原始Id
+	WxId          string `json:"wxid"`          // 微信原始Id
-}
-
-// GroupUser
-// @description: 群成员返回结果
-type GroupUser struct {
-	Admin          string `json:"admin"`          // 群主微信
-	AdminNickname  string `json:"adminNickname"`  // 群主昵称
-	ChatRoomId     string `json:"chatRoomId"`     // 群Id
-	MemberNickname string `json:"memberNickname"` // 成员昵称 `^G`切割
-	Members        string `json:"members"`        // 成员Id `^G`切割
 }
 
 // ContactProfile
@@ -32,5 +22,5 @@ type ContactProfile struct {
 	HeadImage string `json:"headImage"` // 头像
 	Nickname  string `json:"nickname"`  // 昵称
 	V3        string `json:"v3"`        // v3
-	Wxid      string `json:"wxid"`      // 微信Id
+	WxId      string `json:"wxid"`      // 微信Id
 }

model/robot/message.go

@@ -1,11 +1,11 @@
-package dto
+package robot
 
 import (
 	"encoding/xml"
-	"github.com/duke-git/lancet/v2/slice"
-	"go-wechat/types"
 	"regexp"
+	"slices"
 	"strings"
+	"wechat-robot/pkg/types"
 )
 
 // Message
@@ -32,39 +32,6 @@ type systemMsgDataXml struct {
 	Type   string `xml:"type,attr"`
 }
 
-// appMsgDataXml
-// @description: 微信app消息的xml结构
-type appMsgDataXml struct {
-	XMLName xml.Name `xml:"msg"`
-	Text    string   `xml:",chardata"`
-	AppMsg  struct {
-		Text      string `xml:",chardata"`
-		Appid     string `xml:"appid,attr"`
-		SdkVer    string `xml:"sdkver,attr"`
-		Title     string `xml:"title"`
-		Des       string `xml:"des"`
-		Action    string `xml:"action"`
-		Type      string `xml:"type"`
-		ShowType  string `xml:"showtype"`
-		Content   string `xml:"content"`
-		URL       string `xml:"url"`
-		ThumbUrl  string `xml:"thumburl"`
-		LowUrl    string `xml:"lowurl"`
-		AppAttach struct {
-			Text     string `xml:",chardata"`
-			TotalLen string `xml:"totallen"`
-			AttachId string `xml:"attachid"`
-			FileExt  string `xml:"fileext"`
-		} `xml:"appattach"`
-		ExtInfo string `xml:"extinfo"`
-	} `xml:"appmsg"`
-	AppInfo struct {
-		Text    string `xml:",chardata"`
-		Version string `xml:"version"`
-		AppName string `xml:"appname"`
-	} `xml:"appinfo"`
-}
-
 // atMsgDataXml
 // @description: 微信@消息的xml结构
 type atMsgDataXml struct {
@@ -121,13 +88,7 @@ func (m Message) IsRevokeMsg() bool {
 // @receiver m
 // @return bool
 func (m Message) IsNewUserJoin() bool {
-	if m.Type != types.MsgTypeSys {
+	sysFlag := m.Type == types.MsgTypeSys && strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
-		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
 	}
@@ -136,7 +97,7 @@ func (m Message) IsNewUserJoin() bool {
 	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
 		return false
 	}
-	return d.Type == "delchatroommember"
+	return m.Type == types.MsgTypeSys && d.Type == "delchatroommember"
 }
 
 // IsAt
@@ -160,11 +121,7 @@ func (m Message) IsAtAll() bool {
 	// 转换@用户列表为数组
 	atUserList := strings.Split(d.AtUserList, ",")
 	// 判断是否包含@所有人
-	if slice.Contain(atUserList, "notify@all") {
+	return slices.Contains(atUserList, "notify@all")
-		return true
-	}
-	// 数据格式变动，再检查一下字符串是否包含 @所有人 字样
-	return m.IsAt() && strings.Contains(m.Content, "@所有人")
 }
 
 // IsPrivateText
@@ -200,81 +157,3 @@ func (m Message) CleanContentStartWith(prefix string) bool {
 
 	return strings.HasPrefix(content, prefix)
 }
-
-// IsInvitationJoinGroup
-// @description: 是否是邀请入群消息
-// @receiver m
-// @return bool 是否是邀请入群消息
-// @return string 邀请入群消息内容
-func (m Message) IsInvitationJoinGroup() (flag bool, str string) {
-	if m.Type == types.MsgTypeApp {
-		// 解析xml
-		var md appMsgDataXml
-		if err := xml.Unmarshal([]byte(m.Content), &md); err != nil {
-			return
-		}
-		flag = md.AppMsg.Type == "5" && md.AppMsg.Title == "邀请你加入群聊"
-		str = strings.ReplaceAll(md.AppMsg.Des, "，进入可查看详情。", "")
-		return
-	}
-	return
-}
-
-// IsNewFriendAdd
-// @description: 是否是新好友添加消息
-// @receiver m
-// @return flag
-func (m Message) IsNewFriendAdd() (flag bool) {
-	if m.Type != types.MsgTypeSys {
-		return
-	}
-	return strings.HasPrefix(m.Content, "你已添加了") && strings.HasSuffix(m.Content, "，现在可以开始聊天了。")
-}
-
-// IsOldFriendBack
-// @description: 是否是老好友回归消息
-// @receiver m
-// @return flag
-func (m Message) IsOldFriendBack() (flag bool) {
-	if m.Type != types.MsgTypeSys {
-		return
-	}
-	return m.Content == "以上是打招呼的内容"
-}
-
-// IsJoinToGroup
-// @description: 是否是加入群聊消息
-// @receiver m
-// @return flag
-func (m Message) IsJoinToGroup() (flag bool) {
-	if m.Type != types.MsgTypeSys {
-		return
-	}
-	flag = strings.Contains(m.Content, "\"邀请你加入了群聊，群聊参与人还有：")
-	if flag {
-		return
-	}
-	return strings.Contains(m.Content, "\"邀请你和\"") && strings.Contains(m.Content, "\"加入了群聊")
-}
-
-// IsRemoveFromChatroom
-// @description: 是否是被移出群聊消息
-// @receiver m
-// @return flag
-func (m Message) IsRemoveFromChatroom() (flag bool) {
-	if m.Type != types.MsgTypeSys {
-		return
-	}
-	return strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊")
-}
-
-// IsChangeGroupName
-// @description: 是否是修改群名称消息
-// @receiver m
-// @return flag
-func (m Message) IsChangeGroupName() (flag bool) {
-	if m.Type != types.MsgTypeSys {
-		return
-	}
-	return strings.HasPrefix(m.Content, "\"修改群名为“")
-}

model/robot/response.go

@@ -1,4 +1,4 @@
-package dto
+package robot
 
 // Response
 // @description: 基础返回结构体

model/robot/userinfo.go

@@ -1,8 +1,8 @@
-package dto
+package robot
 
-// RobotUserInfo
+// UserInfo
 // @description: 机器人用户信息
-type RobotUserInfo struct {
+type UserInfo struct {
 	WxId            string `json:"wxid"`            // 微信Id
 	Account         string `json:"account"`         // 微信号
 	Name            string `json:"name"`            // 昵称

model/vo/friend.go

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

model/vo/leigod.go

@@ -1,8 +0,0 @@
-package vo
-
-// LeiGodAccount
-// @description: 雷神账号
-type LeiGodAccount struct {
-	Account  string `json:"account"`  // 账号
-	Password string `json:"password"` // 密码
-}

model/vo/login/captcha.go

@@ -0,0 +1,8 @@
+package login
+
+// CaptchaCode
+// @description: 获取图形验证码返回结果
+type CaptchaCode struct {
+	Id  string `json:"id"`  // 验证ID
+	Img string `json:"img"` // 图片Base64字符串
+}

model/vo/menu/menu.go

@@ -0,0 +1,21 @@
+package menu
+
+// Item
+// @description:
+type Item struct {
+	Id       string   `json:"id"`                 // 菜单Id
+	Path     string   `json:"path"`               // 访问路径
+	Name     string   `json:"name"`               // 路由名字（对应不要重复，和页面组件的name保持一致）必填
+	Meta     ItemMeta `json:"meta"`               // 元数据
+	Children []Item   `json:"children,omitempty"` // 子菜单
+}
+
+// ItemMeta
+// @description: 菜单元数据
+type ItemMeta struct {
+	Title string   `json:"title"`          // 标题
+	Icon  string   `json:"icon"`           // 图标
+	Rank  int      `json:"rank,omitempty"` // 排序
+	Roles []string `json:"roles"`          // 当前菜单所属的角色代码
+	Auths []string `json:"auths"`          // 当前菜单包含的按钮，如果传入了用户Id，返回的将会是权限内所有的按钮
+}

model/vo/message.go

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

mq/handler.go

@@ -1,46 +0,0 @@
-package mq
-
-import (
-	"encoding/json"
-	"go-wechat/common/current"
-	"go-wechat/model/dto"
-	"go-wechat/types"
-	"log"
-	"strings"
-)
-
-// parse
-// @description: 解析消息
-// @param msg
-func parse(msg []byte) {
-	var m dto.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
-}

mq/rabbitmq.go

@@ -1,9 +1,10 @@
 package mq
 
 import (
+	"gitee.ltd/lxh/logger/log"
 	amqp "github.com/rabbitmq/amqp091-go"
-	"go-wechat/config"
+	"wechat-robot/config"
-	"log"
+	"wechat-robot/internal/message"
 )
 
 // MQ连接对象
@@ -16,10 +17,6 @@ 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 == "" {
@@ -36,14 +33,23 @@ func Init() {
 	if channel, err = conn.Channel(); err != nil {
 		log.Panicf("打开Channel失败: %s", err)
 	}
-	log.Println("RabbitMQ连接成功")
+	log.Debug("RabbitMQ连接成功")
 	go Receive()
-	log.Println("开始监听消息")
+	log.Debug("开始监听消息")
 }
 
 // Receive
 // @description: 接收消息
-func Receive() (err error) {
+func Receive(retry ...int) (err error) {
+	var retryCount int
+	if len(retry) > 0 {
+		retryCount = retry[0]
+	}
+	// 重试次数超过100次，退出
+	if retryCount > 100 {
+		log.Panicf("获取消息失败次数过多")
+	}
+
 	// 创建交换机
 	if err = channel.ExchangeDeclare(
 		exchangeName,
@@ -55,7 +61,7 @@ func Receive() (err error) {
 		false,
 		nil,
 	); err != nil {
-		log.Printf("声明Exchange失败: %s", err)
+		log.Errorf("声明Exchange失败: %s", err)
 		return
 	}
 	// 创建队列
@@ -69,7 +75,7 @@ func Receive() (err error) {
 		nil,
 	)
 	if err != nil {
-		log.Printf("无法声明Queue: %s", err)
+		log.Errorf("无法声明Queue: %s", err)
 		return
 	}
 	// 绑定队列到 exchange 中
@@ -81,7 +87,7 @@ func Receive() (err error) {
 		false,
 		nil)
 	if err != nil {
-		log.Printf("绑定队列失败: %s", err)
+		log.Errorf("绑定队列失败: %s", err)
 		return
 	}
 
@@ -97,20 +103,23 @@ func Receive() (err error) {
 		nil,
 	)
 	if err != nil {
-		log.Printf("无法使用队列: %s", err)
+		log.Errorf("无法使用队列: %s", err)
 		return
 	}
 	for {
 		msg, ok := <-messages
 		if !ok {
-			log.Printf("获取消息失败")
+			log.Errorf("获取消息失败")
-			return Receive()
+			return Receive(retryCount + 1)
+		}
+		log.Debugf("收到消息: %s", msg.Body)
+		if err = message.Message(msg.Body); err != nil {
+			log.Errorf("处理消息失败: %s", err)
+			continue
 		}
-		log.Printf("收到消息: %s", msg.Body)
-		parse(msg.Body)
 		// ACK消息
 		if err = msg.Ack(true); err != nil {
-			log.Printf("ACK消息失败: %s", err)
+			log.Errorf("ACK消息失败: %s", err)
 			continue
 		}
 	}

pkg/auth/auth.go

@@ -0,0 +1,60 @@
+package auth
+
+import (
+	"github.com/go-oauth2/oauth2/v4"
+	"github.com/go-oauth2/oauth2/v4/manage"
+	"github.com/go-oauth2/oauth2/v4/models"
+	"github.com/go-oauth2/oauth2/v4/server"
+	"github.com/go-oauth2/oauth2/v4/store"
+	od "github.com/go-oauth2/redis/v4"
+	"time"
+	"wechat-robot/internal/redis"
+	"wechat-robot/pkg/auth/handle"
+)
+
+var (
+	OAuthServer *server.Server // 后台OAuth2服务
+)
+
+// InitOAuth2Server
+// @description: 初始化后台OAuth2服务
+func InitOAuth2Server() {
+	manager := manage.NewDefaultManager()
+	// 配置信息
+	cfg := &manage.Config{
+		AccessTokenExp:    time.Hour * 24,      // 访问令牌过期时间
+		RefreshTokenExp:   time.Hour * 24 * 30, // 更新令牌过期时间
+		IsGenerateRefresh: true,                // 是否生成新的更新令牌
+	}
+	// 设置密码模式的配置参数
+	manager.SetPasswordTokenCfg(cfg)
+
+	manager.MapTokenStorage(od.NewRedisStoreWithCli(redis.Client, "oauth:token:"))
+	// 生成Token方式
+	manager.MapAccessGenerate(handle.NewAccessGenerate())
+
+	// 配置客户端
+	// 配置客户端
+	clientStore := store.NewClientStore()
+	_ = clientStore.Set("admin", &models.Client{
+		ID:     "admin",
+		Secret: "kycgfS4sBjx2rPVD",
+	})
+	manager.MapClientStorage(clientStore)
+
+	srv := server.NewServer(server.NewConfig(), manager)
+	// 设置密码登录模式处理逻辑
+	srv.SetPasswordAuthorizationHandler(handle.LoginWithPassword)
+	// 允许密码模式、刷新Token
+	srv.SetAllowedGrantType(oauth2.PasswordCredentials, oauth2.Refreshing)
+	// 客户端ID和授权模式检查
+	//srv.SetClientAuthorizedHandler(handle.CheckClient)
+	// 自定义响应Token的扩展字段
+	srv.SetExtensionFieldsHandler(handle.ExtensionFields)
+	// 自定义返回数据接口
+	srv.SetResponseTokenHandler(handle.ResponseToken)
+	// 自定义内部错误处理
+	srv.SetInternalErrorHandler(handle.InternalErrorHandler)
+
+	OAuthServer = srv
+}

pkg/auth/handle/oauth2.go

@@ -0,0 +1,139 @@
+package handle
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/go-oauth2/oauth2/v4"
+	"github.com/go-oauth2/oauth2/v4/errors"
+	"net/http"
+	"time"
+	"wechat-robot/internal/database"
+	"wechat-robot/model/entity"
+	userService "wechat-robot/service/adminuser"
+	"wechat-robot/service/role"
+	"wechat-robot/utils"
+)
+
+// LoginWithPassword
+// @description: 账号密码登录模式
+// @param _
+// @param clientId
+// @param userId
+// @param password
+// @return userID
+// @return err
+func LoginWithPassword(_ context.Context, clientId, username, password string) (userId string, err error) {
+	log.Debugf("[%v]处理登录请求，账号：%s --> %s", clientId, username, password)
+
+	// 取出用户信息
+	var userInfo entity.AdminUser
+	userInfo, err = userService.GetUserWithLogin(username)
+	userId = userInfo.Id
+	if err != nil {
+		log.Errorf("获取用户信息失败: %v", err.Error())
+		err = errors.New("账号不存在")
+		return
+	}
+	// 校验密码
+	if !utils.PasswordUtils().ComparePassword(userInfo.Password, password) {
+		err = errors.New("密码错误")
+		return
+	}
+	return
+}
+
+// ExtensionFields
+// @description: 自定义响应Token的扩展字段
+// @param _
+// @return fieldsValue
+func ExtensionFields(ti oauth2.TokenInfo) (fieldsValue map[string]any) {
+	fieldsValue = map[string]any{}
+	fieldsValue["license"] = "Made By Lixunhuan"
+	fieldsValue["developer"] = "https://lxh.io"
+	fieldsValue["expires"] = time.Now().Local().Add(ti.GetAccessExpiresIn()).Format("2006/01/02 15:04:05")
+
+	// 缓存用户的accessToken，为后续删除做准备
+	//key := fmt.Sprintf("oauth:token:token:%s", ti.GetUserID())
+	//if err := redis.Client.Set(context.Background(), key, ti.GetAccess(), time.Hour*24*30).Err(); err != nil {
+	//	log.Errorf("缓存用户的accessToken失败: %v", err.Error())
+	//}
+
+	// 填充用户信息
+	var ui entity.AdminUser
+	_ = database.Client.Take(&ui, "id = ?", ti.GetUserID()).Error
+	fieldsValue["username"] = ui.Username
+	fieldsValue["id"] = ui.Id
+	// 获取用户的角色代码
+	fieldsValue["roles"] = role.GetCodesByUserId(ti.GetUserID())
+	return
+}
+
+// ResponseToken
+// @description: 返回Token生成结果
+// @param w http.ResponseWriter 写入响应
+// @param data map[string]any 响应数据
+// @param header http.Header 响应头
+// @param statusCode ...int 响应状态
+// @return error 错误信息
+func ResponseToken(w http.ResponseWriter, data map[string]any, header http.Header, statusCode ...int) error {
+	log.Debugf("返回Token原始数据: %+v", data)
+	type response struct {
+		Code int            `json:"code"`
+		Data map[string]any `json:"data"`
+		Msg  string         `json:"message"`
+	}
+
+	status := http.StatusOK
+	msg := "login success"
+	if len(statusCode) > 0 && statusCode[0] > 0 {
+		status = statusCode[0]
+		msg = fmt.Sprintf("%v", data["error_description"])
+		// 处理特殊返回 - 刷新Token到期了
+		switch data["error"] {
+		case "invalid_grant":
+			msg = "登录已过期，请重新授权登录"
+		case "invalid_request":
+			msg = "登录参数错误"
+		case "invalid_client":
+			msg = "客户端验证失败"
+		default:
+			log.Errorf("收到未定义的登录错误: %v", data["error_description"])
+		}
+		data = nil
+	}
+
+	res := response{
+		Code: status,
+		Msg:  msg,
+		Data: data,
+	}
+
+	jsonBytes, err := json.Marshal(res)
+	if err != nil {
+		return err
+	}
+	w.Header().Set("Content-Type", "application/json;charset=UTF-8")
+	w.Header().Set("Cache-Control", "no-store")
+	w.Header().Set("Pragma", "no-cache")
+
+	for key := range header {
+		w.Header().Set(key, header.Get(key))
+	}
+
+	w.WriteHeader(status)
+	_, err = w.Write(jsonBytes)
+	if err != nil {
+		log.Errorf("返回Token失败: %v", err.Error())
+		return err
+	}
+	return err
+}
+
+// InternalErrorHandler 自定义内部错误处理
+func InternalErrorHandler(err error) (re *errors.Response) {
+	re = errors.NewResponse(err, http.StatusUnauthorized)
+	re.Description = err.Error()
+	return
+}

pkg/auth/handle/token_gen.go

@@ -0,0 +1,78 @@
+package handle
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/go-oauth2/oauth2/v4"
+	"github.com/google/uuid"
+	"strings"
+	"wechat-robot/internal/redis"
+)
+
+func NewAccessGenerate() *AccessGenerate {
+	return &AccessGenerate{}
+}
+
+type AccessGenerate struct {
+}
+
+// Token
+// @description: 手动实现Token生成，直接生成UUID，替换掉自带的那个憨得一批的长长的字符串
+// @receiver ag
+// @param ctx context.Context 上下文
+// @param data *oauth2.GenerateBasic 生成基础数据
+// @param isGenRefresh bool 是否生成RefreshToken
+// @return string AccessToken 令牌
+// @return string RefreshToken 刷新令牌
+// @return error 错误信息
+func (ag *AccessGenerate) Token(ctx context.Context, data *oauth2.GenerateBasic, isGenRefresh bool) (string, string, error) {
+	u, _ := uuid.NewUUID()
+	access := strings.ReplaceAll(u.String(), "-", "")
+
+	refresh := ""
+	if isGenRefresh {
+		u, _ = uuid.NewUUID()
+		refresh = strings.ReplaceAll(u.String(), "-", "")
+	}
+	// 生成新的，清理掉旧的
+	ag.clearOldToken(ctx, data.UserID)
+	// 返回结果
+	return access, refresh, nil
+}
+
+// clearOldToken
+// @description: 清理掉旧的Token和RefreshToken
+// @receiver ag
+// @param ctx context.Context 上下文
+// @param userId string 用户ID
+func (ag *AccessGenerate) clearOldToken(ctx context.Context, userId string) {
+	key := fmt.Sprintf("oauth:token:token:%s", userId)
+	accessToken, err := redis.Client.Get(context.Background(), key).Result()
+	if err != nil {
+		//log.Errorf("获取缓存用户的accessToken失败: %v", err.Error())
+		return
+	}
+	if accessToken != "" {
+		// 老的Token
+		var baseKey string
+		baseKey, err = redis.Client.Get(ctx, "oauth:token:"+accessToken).Result()
+		if err != nil {
+			return
+		}
+		// 老Token详细数据
+		var dataStr string
+		dataStr, err = redis.Client.Get(ctx, "oauth:token:"+baseKey).Result()
+		if err != nil {
+			return
+		}
+		var m map[string]interface{}
+		if err = json.Unmarshal([]byte(dataStr), &m); err != nil {
+			return
+		}
+		// 删除AccessToken等信息
+		redis.Client.Del(ctx, fmt.Sprintf("oauth:token:%v", m["Access"]))
+		redis.Client.Del(ctx, fmt.Sprintf("oauth:token:%v", m["Refresh"]))
+		redis.Client.Del(ctx, "oauth:token:"+baseKey)
+	}
+}

pkg/auth/utils.go

@@ -0,0 +1,39 @@
+package auth
+
+import (
+	"context"
+	"encoding/json"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/go-oauth2/oauth2/v4/models"
+	"wechat-robot/internal/redis"
+)
+
+// GetUserIdWithRefreshToken
+// @description: 根据refreshToken获取userId
+// @param refreshToken string refreshToken
+// @return userId string userId
+func GetUserIdWithRefreshToken(refreshToken string) (userId string) {
+	// 取出真实保存token信息的key
+	realKey, err := redis.Client.Get(context.Background(), "oauth:token:"+refreshToken).Result()
+	if err != nil {
+		log.Errorf("根据refreshToken获取realKey失败: %v", err)
+		return
+	}
+	// 取出缓存信息
+	var (
+		tiStr     string       // 保存的原始字符串
+		tokenInfo models.Token // Token结构体
+	)
+	tiStr, err = redis.Client.Get(context.Background(), "oauth:token:"+realKey).Result()
+	if err != nil {
+		log.Errorf("获取tokenInfo失败: %v", err)
+		return
+	}
+	// 反序列化
+	if err = json.Unmarshal([]byte(tiStr), &tokenInfo); err != nil {
+		log.Errorf("tokenInfo反序列化失败: %v", err)
+		return
+	}
+	// 返回userId
+	return tokenInfo.GetUserID()
+}

pkg/captcha/captcha.go

@@ -0,0 +1,66 @@
+package captcha
+
+import (
+	"context"
+	"gitee.ltd/lxh/logger/log"
+	"time"
+	"wechat-robot/internal/redis"
+)
+
+var ctx = context.Background()
+
+type RedisStore struct{}
+
+// 验证码缓存前缀
+const cachePrefix = "captcha:img:"
+
+// Set
+// @description: 缓存验证码信息
+// @receiver r
+// @param id string 验证码ID
+// @param value string 验证码答案
+// @return error 错误信息
+func (r RedisStore) Set(id string, value string) error {
+	key := cachePrefix + id
+	err := redis.Client.Set(ctx, key, value, time.Minute).Err()
+	if err != nil {
+		log.Errorf("写入缓存验证码失败: %v", err.Error())
+		return err
+	}
+	return nil
+}
+
+// Get
+// @description: 获取验证码信息
+// @receiver r
+// @param id string 验证码ID
+// @param clear
+// @return string
+func (r RedisStore) Get(id string, clear bool) string {
+	key := cachePrefix + id
+	val, err := redis.Client.Get(ctx, key).Result()
+	if err != nil {
+		log.Errorf("读取缓存验证码失败: %v", err.Error())
+		return ""
+	}
+	if clear {
+		err = redis.Client.Del(ctx, key).Err()
+		if err != nil {
+			log.Errorf("读取缓存验证码失败: %v", err.Error())
+			return ""
+		}
+	}
+	return val
+}
+
+// Verify
+// @description: 验证验证码
+// @receiver r
+// @param id string 验证码ID
+// @param answer string 验证码答案
+// @param clear bool 是否清除验证码
+// @return bool 是否验证成功
+func (r RedisStore) Verify(id, answer string, clear bool) bool {
+	v := r.Get(id, clear)
+	return v == answer
+}

pkg/response/fail.go

@@ -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()
+}

pkg/response/page.go

@@ -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
+}

pkg/response/response.go

@@ -0,0 +1,77 @@
+package response
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"wechat-robot/pkg/validator"
+)
+
+// 定义状态码
+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 {
+		err = validator.Translate(err)
+		r.errMsg = err.Error()
+	}
+	return r
+}

pkg/response/result.go

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

pkg/response/success.go

@@ -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()
+}

pkg/types/date.go

@@ -0,0 +1,111 @@
+package types
+
+import (
+	"database/sql/driver"
+	"fmt"
+	"strings"
+	"time"
+)
+
+// 默认时间格式
+const dateFormat = "2006-01-02"
+
+// Date 自定义时间类型
+type Date time.Time
+
+// Scan implements the Scanner interface.
+func (dt *Date) Scan(value any) error {
+	// mysql 内部日期的格式可能是 2006-01-02 15:04:05 +0800 CST 格式，所以检出的时候还需要进行一次格式化
+	tTime, _ := time.Parse("2006-01-02 15:04:05 +0800 CST", value.(time.Time).String())
+	*dt = Date(tTime)
+	return nil
+}
+
+// Value implements the driver Valuer interface.
+func (dt Date) Value() (driver.Value, error) {
+	// 0001-01-01 00:00:00 属于空值，遇到空值解析成 null 即可
+	if dt.String() == "0001-01-01" {
+		return nil, nil
+	}
+	return []byte(dt.Format(dateFormat)), nil
+}
+
+// 用于 fmt.Println 和后续验证场景
+func (dt Date) String() string {
+	return dt.Format(dateFormat)
+}
+
+// Format 格式化
+func (dt Date) Format(fm string) string {
+	return time.Time(dt).Format(fm)
+}
+
+// After 时间比较
+func (dt *Date) After(now time.Time) bool {
+	return time.Time(*dt).After(now)
+}
+
+// Before 时间比较
+func (dt *Date) Before(now time.Time) bool {
+	return time.Time(*dt).Before(now)
+}
+
+// IBefore 时间比较
+func (dt *Date) IBefore(now Date) bool {
+	return dt.Before(time.Time(now))
+}
+
+// SubTime 对比
+func (dt Date) SubTime(t time.Time) time.Duration {
+	return dt.ToTime().Sub(t)
+}
+
+// Sub 对比
+func (dt Date) Sub(t Date) time.Duration {
+	return dt.ToTime().Sub(t.ToTime())
+}
+
+// ToTime 转换为golang的时间类型
+func (dt Date) ToTime() time.Time {
+	return time.Time(dt)
+}
+
+// IsNil 是否为空值
+func (dt Date) IsNil() bool {
+	return dt.Format(dateFormat) == "0001-01-01"
+}
+
+// Unix 实现Unix函数
+func (dt Date) Unix() int64 {
+	return dt.ToTime().Unix()
+}
+
+// ========  序列化 JSON ========
+
+// MarshalJSON 时间到字符串
+func (dt Date) MarshalJSON() ([]byte, error) {
+	// 过滤掉空数据
+	if dt.IsNil() {
+		return []byte("\"\""), nil
+	}
+	output := fmt.Sprintf(`"%s"`, dt.Format(dateFormat))
+	return []byte(output), nil
+}
+
+// UnmarshalJSON 字符串到时间
+func (dt *Date) UnmarshalJSON(b []byte) error {
+	if len(b) == 2 {
+		*dt = Date{}
+		return nil
+	}
+	// 解析指定的格式
+	var now time.Time
+	var err error
+	if strings.HasPrefix(string(b), "\"") {
+		now, err = time.ParseInLocation(`"`+dateFormat+`"`, string(b), time.Local)
+	} else {
+		now, err = time.ParseInLocation(dateFormat, string(b), time.Local)
+	}
+	*dt = Date(now)
+	return err
+}

pkg/types/datetime.go

@@ -55,7 +55,7 @@ func (dt DateTime) Value() (dv driver.Value, err error) {
 
 // 用于 fmt.Println 和后续验证场景
 func (dt DateTime) String() string {
-	return dt.Format("2006-01-02 15:04:05")
+	return dt.Format(dateTimeFormat)
 }
 
 // Format 格式化

pkg/types/menu.go

@@ -0,0 +1,22 @@
+package types
+
+type MenuType string
+
+const (
+	MenuTypeMenu   = MenuType("MENU")   // 菜单
+	MenuTypeButton = MenuType("BUTTON") // 按钮
+)
+
+// 状态对应的描述
+var menuTypeMap = map[MenuType]string{
+	MenuTypeMenu:   "菜单",
+	MenuTypeButton: "按钮",
+}
+
+// 处理为看得懂的状态
+func (s MenuType) String() string {
+	if str, ok := menuTypeMap[s]; ok {
+		return str
+	}
+	return string(s)
+}

pkg/types/model.go

@@ -0,0 +1,42 @@
+package types
+
+import (
+	"github.com/google/uuid"
+	"gorm.io/gorm"
+	"gorm.io/plugin/soft_delete"
+	"strings"
+)
+
+// BaseDbModel
+// @description: 带逻辑删除的数据库通用字段
+type BaseDbModel struct {
+	Id        string                `json:"id" gorm:"type:varchar(32);primarykey"`
+	CreatedAt DateTime              `json:"createdAt"`
+	UpdatedAt DateTime              `json:"updatedAt"`
+	DeletedAt int64                 `json:"-" gorm:"index:deleted; default:0"`
+	IsDel     soft_delete.DeletedAt `json:"-" gorm:"softDelete:flag,DeletedAtField:DeletedAt; index:deleted; default:0; type:tinyint(1)"`
+}
+
+// BeforeCreate 创建数据库对象之前生成UUID
+func (m *BaseDbModel) BeforeCreate(*gorm.DB) (err error) {
+	if m.Id == "" {
+		m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
+	}
+	return
+}
+
+// =====================================================================================================================
+
+// BaseDbModelWithReal
+// @description: 不带逻辑删除的数据库通用字段
+type BaseDbModelWithReal struct {
+	Id        string   `json:"id" gorm:"type:varchar(32);primarykey"`
+	CreatedAt DateTime `json:"createdAt"`
+	UpdatedAt DateTime `json:"updatedAt"`
+}
+
+// BeforeCreate 创建数据库对象之前生成UUID
+func (m *BaseDbModelWithReal) BeforeCreate(*gorm.DB) (err error) {
+	m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
+	return
+}