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": "规则删除成功"}) }