package repository

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

// ExecutionLogRepository 定义了与执行日志交互的接口。
// 这为服务层提供了一个清晰的契约，并允许在测试中轻松地进行模拟。
type ExecutionLogRepository interface {
	CreatePlanExecutionLog(log *models.PlanExecutionLog) error
	UpdatePlanExecutionLog(log *models.PlanExecutionLog) error
	CreateTaskExecutionLogsInBatch(logs []*models.TaskExecutionLog) error
	UpdateTaskExecutionLog(log *models.TaskExecutionLog) error
	FindTaskExecutionLogByID(id uint) (*models.TaskExecutionLog, error)
}

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

// NewExecutionLogRepository 创建一个新的执行日志仓库。
// 它接收一个 GORM DB 实例作为依赖。
func NewExecutionLogRepository(db *gorm.DB) ExecutionLogRepository {
	return &executionLogRepository{db: db}
}

// CreatePlanExecutionLog 为一次计划执行创建一条新的日志条目。
func (r *executionLogRepository) CreatePlanExecutionLog(log *models.PlanExecutionLog) error {
	return r.db.Create(log).Error
}

// UpdatePlanExecutionLog 使用 Updates 方法更新一个计划执行日志。
// GORM 的 Updates 传入 struct 时，只会更新非零值字段。
// 在这里，我们期望传入的对象一定包含一个有效的 ID。
func (r *executionLogRepository) UpdatePlanExecutionLog(log *models.PlanExecutionLog) error {
	return r.db.Updates(log).Error
}

// CreateTaskExecutionLogsInBatch 在一次数据库调用中创建多个任务执行日志条目。
// 这是“预写日志”步骤的关键。
func (r *executionLogRepository) CreateTaskExecutionLogsInBatch(logs []*models.TaskExecutionLog) error {
	// GORM 的 Create 传入一个切片指针会执行批量插入。
	return r.db.Create(&logs).Error
}

// UpdateTaskExecutionLog 使用 Updates 方法更新一个任务执行日志。
// GORM 的 Updates 传入 struct 时，只会更新非零值字段。
// 这种方式代码更直观，上层服务可以直接修改模型对象后进行保存。
func (r *executionLogRepository) UpdateTaskExecutionLog(log *models.TaskExecutionLog) error {
	return r.db.Updates(log).Error
}

// FindTaskExecutionLogByID 根据 ID 查找单个任务执行日志。
// 它会预加载关联的 Task 信息。
func (r *executionLogRepository) FindTaskExecutionLogByID(id uint) (*models.TaskExecutionLog, error) {
	var log models.TaskExecutionLog
	// 使用 Preload("Task") 来确保关联的任务信息被一并加载
	err := r.db.Preload("Task").First(&log, id).Error
	if err != nil {
		return nil, err
	}
	return &log, nil
}