107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
package rule
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/ssdomei232/goodBaby/api/user"
|
|
"github.com/ssdomei232/goodBaby/handler/db"
|
|
"github.com/ssdomei232/goodBaby/internal/ruleConfigChecker"
|
|
"github.com/ssdomei232/goodBaby/model"
|
|
)
|
|
|
|
// 获取用户的所有规则
|
|
func HandleGetAllRules(c *gin.Context) {
|
|
gormDB, err := db.GetGormDB()
|
|
if err != nil {
|
|
c.JSON(500, gin.H{"code": 500, "data": "获取规则失败"})
|
|
return
|
|
}
|
|
|
|
userInfo, err := user.GetUserInfoByGinCtx(c)
|
|
if err != nil {
|
|
c.JSON(401, gin.H{"code": 401, "data": "获取用户信息失败"})
|
|
return
|
|
}
|
|
|
|
var rules []model.Rule
|
|
result := gormDB.Where("uid = ?", userInfo.ID).Find(&rules)
|
|
if result.Error != nil {
|
|
c.JSON(500, gin.H{"code": 500, "data": "获取规则失败"})
|
|
return
|
|
}
|
|
|
|
c.JSON(200, gin.H{"code": 200, "data": rules})
|
|
}
|
|
|
|
// 创建新规则
|
|
func HandleCreateRule(c *gin.Context) {
|
|
gormDB, err := db.GetGormDB()
|
|
if err != nil {
|
|
c.JSON(500, gin.H{"code": 500, "data": "服务器内部错误"})
|
|
return
|
|
}
|
|
|
|
userInfo, err := user.GetUserInfoByGinCtx(c)
|
|
if err != nil {
|
|
c.JSON(401, gin.H{"code": 401, "data": "获取用户信息失败"})
|
|
return
|
|
}
|
|
|
|
var newRule model.Rule
|
|
if err := c.BindJSON(&newRule); err != nil {
|
|
c.JSON(400, gin.H{"code": 400, "data": "输入参数错误"})
|
|
return
|
|
}
|
|
|
|
if newRule.Type == "" {
|
|
c.JSON(400, gin.H{"code": 400, "data": "规则类型不能为空"})
|
|
return
|
|
}
|
|
|
|
if newRule.Name == "" {
|
|
c.JSON(400, gin.H{"code": 400, "data": "规则名称不能为空"})
|
|
return
|
|
}
|
|
|
|
// 规则校验
|
|
validatorRegistry := ruleConfigChecker.InitValidatorRegistry()
|
|
if err := validatorRegistry.Validate(newRule.Type, newRule.ConfigJson); err != nil {
|
|
c.JSON(400, gin.H{"code": 400, "data": fmt.Sprintf("规则配置验证失败: %s", err.Error())})
|
|
return
|
|
}
|
|
|
|
newRule.UID = userInfo.ID
|
|
result := gormDB.Create(&newRule)
|
|
if result.Error != nil {
|
|
c.JSON(500, gin.H{"code": 500, "data": "创建规则失败"})
|
|
return
|
|
}
|
|
|
|
c.JSON(200, gin.H{"code": 200, "data": "规则创建成功"})
|
|
}
|
|
|
|
// 根据删除规则
|
|
func HandleDeleteRule(c *gin.Context) {
|
|
userInfo, err := user.GetUserInfoByGinCtx(c)
|
|
if err != nil {
|
|
c.JSON(401, gin.H{"code": 401, "data": "获取用户信息失败"})
|
|
return
|
|
}
|
|
|
|
ruleID, err := strconv.Atoi(c.Param("ruleID"))
|
|
if err != nil {
|
|
c.JSON(400, gin.H{"code": 400, "data": "规则 ID 格式错误"})
|
|
return
|
|
}
|
|
|
|
err = DeleteRuleByID(uint(ruleID), userInfo.ID)
|
|
if err != nil {
|
|
c.JSON(500, gin.H{"code": 500, "data": "删除规则失败"})
|
|
return
|
|
}
|
|
|
|
c.JSON(200, gin.H{"code": 200, "data": "规则删除成功"})
|
|
}
|