goodBaby/api/timer/timer.go

package timer

import (
	"time"

	"github.com/gin-gonic/gin"
	"github.com/ssdomei232/goodBaby/api/response"
	"github.com/ssdomei232/goodBaby/api/user"
	"github.com/ssdomei232/goodBaby/handler/db"
	"github.com/ssdomei232/goodBaby/model"
	"gorm.io/gorm"
)

// HandleGetAllTimers 获取当前用户的所有 Timer
func HandleGetAllTimers(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	timers := []model.Timer{}
	if err := gormDB.Where("uid = ?", userInfo.ID).Order("id DESC").Find(&timers).Error; err != nil {
		response.ServerError(c, "获取 Timer 失败")
		return
	}

	response.OK(c, timers)
}

// HandleGetTimer 获取单个 Timer 及其关联规则数量
func HandleGetTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	timerID, err := parseID(c.Param("timerID"))
	if err != nil {
		response.BadRequest(c, "Timer ID 格式错误")
		return
	}

	timer, err := findTimer(timerID, userInfo.ID)
	if err != nil {
		response.NotFound(c, "Timer 不存在")
		return
	}

	ruleCount, err := countRules(timerID, userInfo.ID)
	if err != nil {
		response.ServerError(c, "获取关联规则失败")
		return
	}

	response.OK(c, gin.H{"timer": timer, "rule_count": ruleCount})
}

// HandleCreateTimer 创建 Timer
func HandleCreateTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	var req model.TimerRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.BadRequest(c, "输入参数错误")
		return
	}

	if err := req.Validate(); err != nil {
		response.FromError(c, err, "创建 Timer 失败")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	now := time.Now().Unix()
	timer := model.Timer{
		UID:                 userInfo.ID,
		Name:                req.Name,
		Description:         req.Description,
		Enabled:             boolOr(req.Enabled, true),
		SignDerationSeconds: req.SignDerationSeconds,
		RemindTimeSeconds:   req.RemindTimeSeconds,
		// 创建即视为完成一次签到，从当前时间开始计时
		LastSign: now,
		CreateAt: now,
	}

	if err := gormDB.Create(&timer).Error; err != nil {
		response.ServerError(c, "创建 Timer 失败")
		return
	}

	response.OK(c, timer)
}

// HandleEditTimer 编辑 Timer
func HandleEditTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	timerID, err := parseID(c.Param("timerID"))
	if err != nil {
		response.BadRequest(c, "Timer ID 格式错误")
		return
	}

	timer, err := findTimer(timerID, userInfo.ID)
	if err != nil {
		response.NotFound(c, "Timer 不存在")
		return
	}

	var req model.TimerRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.BadRequest(c, "输入参数错误")
		return
	}

	if err := req.Validate(); err != nil {
		response.FromError(c, err, "更新 Timer 失败")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	timer.Name = req.Name
	timer.Description = req.Description
	timer.Enabled = boolOr(req.Enabled, timer.Enabled)
	timer.SignDerationSeconds = req.SignDerationSeconds
	timer.RemindTimeSeconds = req.RemindTimeSeconds

	if err := gormDB.Save(timer).Error; err != nil {
		response.ServerError(c, "更新 Timer 失败")
		return
	}

	response.OK(c, timer)
}

// HandleSignTimer 签到：重置计时并解除已触发状态
func HandleSignTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	timerID, err := parseID(c.Param("timerID"))
	if err != nil {
		response.BadRequest(c, "Timer ID 格式错误")
		return
	}

	timer, err := findTimer(timerID, userInfo.ID)
	if err != nil {
		response.NotFound(c, "Timer 不存在")
		return
	}

	if err := signTimer(timer); err != nil {
		response.ServerError(c, "签到失败")
		return
	}

	response.OK(c, timer)
}

// HandleSignAll 一键签到当前用户的所有启用中的 Timer
func HandleSignAll(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	now := time.Now().Unix()
	result := gormDB.Model(&model.Timer{}).
		Where("uid = ? AND enabled = ?", userInfo.ID, true).
		Updates(map[string]any{
			"last_sign":   now,
			"last_remind": 0,
			"triggered":   false,
		})
	if result.Error != nil {
		response.ServerError(c, "签到失败")
		return
	}

	response.OK(c, gin.H{"signed": result.RowsAffected, "last_sign": now})
}

// HandleDeleteTimer 删除 Timer
//
// Timer 被删除后，挂在它下面的规则就没有触发来源了，因此一并删除。
func HandleDeleteTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	timerID, err := parseID(c.Param("timerID"))
	if err != nil {
		response.BadRequest(c, "Timer ID 格式错误")
		return
	}

	if _, err := findTimer(timerID, userInfo.ID); err != nil {
		response.NotFound(c, "Timer 不存在")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	err = gormDB.Transaction(func(tx *gorm.DB) error {
		if err := tx.Where("timer_id = ? AND uid = ?", timerID, userInfo.ID).Delete(&model.Rule{}).Error; err != nil {
			return err
		}
		return tx.Where("id = ? AND uid = ?", timerID, userInfo.ID).Delete(&model.Timer{}).Error
	})
	if err != nil {
		response.ServerError(c, "删除 Timer 失败")
		return
	}

	response.OK(c, "Timer 删除成功")
}

// HandleCheckDeleteTimer 删除前查询会被一并删除的规则
func HandleCheckDeleteTimer(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	timerID, err := parseID(c.Param("timerID"))
	if err != nil {
		response.BadRequest(c, "Timer ID 格式错误")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	rules := []model.Rule{}
	if err := gormDB.Where("timer_id = ? AND uid = ?", timerID, userInfo.ID).Find(&rules).Error; err != nil {
		response.ServerError(c, "获取相关规则失败")
		return
	}

	response.OK(c, rules)
}