package rule

import (
	"github.com/gin-gonic/gin"
	"github.com/ssdomei232/goodBaby/api/user"
	"github.com/ssdomei232/goodBaby/handler/db"
	"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
	}

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