package repository

import (
	"time"

	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

// PendingTaskRepository 定义了与待执行任务队列交互的接口。
type PendingTaskRepository interface {
	CreatePendingTasksInBatch(tasks []*models.PendingTask) error
	ClaimNextDueTask() (*models.TaskExecutionLog, error)
}

// pendingTaskRepository 是使用 GORM 的具体实现。
type pendingTaskRepository struct {
	db *gorm.DB
}

// NewPendingTaskRepository 创建一个新的待执行任务队列仓库。
// 它接收一个 GORM DB 实例作为依赖。
func NewPendingTaskRepository(db *gorm.DB) PendingTaskRepository {
	return &pendingTaskRepository{db: db}
}

// CreatePendingTasksInBatch 在一次数据库调用中创建多个待执行任务条目。
func (r *pendingTaskRepository) CreatePendingTasksInBatch(tasks []*models.PendingTask) error {
	return r.db.Create(&tasks).Error
}

// ClaimNextDueTask 以原子方式认领下一个到期任务。
// 它在一个事务中完成三件事：
// 1. 查找并锁定一个到期的待办任务。
// 2. 从待办队列中将其删除。
// 3. 将其在执行日志表中的状态更新为 'running'。
// 最后返回更新后的执行日志对象，作为执行的唯一凭证。
func (r *pendingTaskRepository) ClaimNextDueTask() (*models.TaskExecutionLog, error) {
	var log models.TaskExecutionLog

	err := r.db.Transaction(func(tx *gorm.DB) error {
		var pendingTask models.PendingTask

		// 1. 查找并用 FOR UPDATE 锁定一个到期的待办任务
		if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).
			Where("execute_at <= ?", time.Now()).
			Order("execute_at ASC").
			First(&pendingTask).Error; err != nil {
			return err // 没找到是正常情况，事务将回滚
		}

		// 2. 在同一个事务中，立刻硬删除这个待办任务
		if err := tx.Unscoped().Delete(&pendingTask).Error; err != nil {
			return err
		}

		// 3. 在同一个事务中，更新其在日志表中的孪生兄弟的状态
		updates := map[string]interface{}{
			"status":     models.ExecutionStatusStarted,
			"started_at": time.Now(),
		}
		if err := tx.Model(&models.TaskExecutionLog{}).Where("id = ?", pendingTask.TaskExecutionLogID).Updates(updates).Error; err != nil {
			return err
		}

		// 4. 获取更新后的完整日志对象，以返回给调用者
		if err := tx.First(&log, pendingTask.TaskExecutionLogID).Error; err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		// 如果错误是 `gorm.ErrRecordNotFound`，它仅表示当前没有到期的任务。
		// 服务层应该优雅地处理这种情况，而不是将其视为需要立即处理的严重错误。
		return nil, err
	}

	return &log, nil
}