using OrpaonVision.Core.Results;
using OrpaonVision.Model.Training;

namespace OrpaonVision.Core.Training;

/// <summary>
/// 训练任务仓储接口。
/// </summary>
public interface ITrainingTaskStore
{
    /// <summary>
    /// 创建训练任务。
    /// </summary>
    Task<Result<TrainingTaskModel>> CreateAsync(TrainingTaskModel task);

    /// <summary>
    /// 更新训练任务。
    /// </summary>
    Task<Result<TrainingTaskModel>> UpdateAsync(TrainingTaskModel task);

    /// <summary>
    /// 删除训练任务。
    /// </summary>
    Task<Result> DeleteAsync(Guid id);

    /// <summary>
    /// 根据ID获取训练任务。
    /// </summary>
    Task<Result<TrainingTaskModel?>> GetByIdAsync(Guid id);

    /// <summary>
    /// 获取训练任务分页列表。
    /// </summary>
    Task<Result<(IReadOnlyList<TrainingTaskModel> tasks, int totalCount)>> GetPagedListAsync(
        int pageIndex, 
        int pageSize, 
        TrainingTaskStatus? status = null, 
        string? keyword = null,
        Guid? assignedToId = null);

    /// <summary>
    /// 获取用户的训练任务列表。
    /// </summary>
    Task<Result<IReadOnlyList<TrainingTaskModel>>> GetUserTasksAsync(Guid userId, TrainingTaskStatus? status = null);

    /// <summary>
    /// 获取任务统计信息。
    /// </summary>
    Task<Result<TrainingTaskStatistics>> GetStatisticsAsync();

    /// <summary>
    /// 更新任务状态。
    /// </summary>
    Task<Result<TrainingTaskModel>> UpdateStatusAsync(Guid id, TrainingTaskStatus status, string operatedBy);

    /// <summary>
    /// 更新任务进度。
    /// </summary>
    Task<Result<TrainingTaskModel>> UpdateProgressAsync(Guid id, int currentEpoch, double currentLoss, double currentMap, double progress);

    /// <summary>
    /// 分配任务给用户。
    /// </summary>
    Task<Result<TrainingTaskModel>> AssignTaskAsync(Guid id, Guid assignedToId, string assignedToName, string assignedBy);

    /// <summary>
    /// 完成训练任务。
    /// </summary>
    Task<Result<TrainingTaskModel>> CompleteTaskAsync(Guid id, string outputModelPath, double finalLoss, double finalMap, string completedBy);

    /// <summary>
    /// 失败训练任务。
    /// </summary>
    Task<Result<TrainingTaskModel>> FailTaskAsync(Guid id, string errorMessage, string failedBy);

    /// <summary>
    /// 检查任务名称是否存在。
    /// </summary>
    Task<Result<bool>> NameExistsAsync(string name, Guid? excludeId = null);
}

/// <summary>
/// 训练任务统计信息。
/// </summary>
public sealed class TrainingTaskStatistics
{
    /// <summary>
    /// 总任务数。
    /// </summary>
    public int TotalTasks { get; set; }

    /// <summary>
    /// 草稿任务数。
    /// </summary>
    public int DraftTasks { get; set; }

    /// <summary>
    /// 等待中任务数。
    /// </summary>
    public int PendingTasks { get; set; }

    /// <summary>
    /// 运行中任务数。
    /// </summary>
    public int RunningTasks { get; set; }

    /// <summary>
    /// 已暂停任务数。
    /// </summary>
    public int PausedTasks { get; set; }

    /// <summary>
    /// 已完成任务数。
    /// </summary>
    public int CompletedTasks { get; set; }

    /// <summary>
    /// 已失败任务数。
    /// </summary>
    public int FailedTasks { get; set; }

    /// <summary>
    /// 已取消任务数。
    /// </summary>
    public int CancelledTasks { get; set; }

    /// <summary>
    /// 今日完成任务数。
    /// </summary>
    public int TodayCompletedTasks { get; set; }

    /// <summary>
    /// 本周完成任务数。
    /// </summary>
    public int ThisWeekCompletedTasks { get; set; }

    /// <summary>
    /// 本月完成任务数。
    /// </summary>
    public int ThisMonthCompletedTasks { get; set; }

    /// <summary>
    /// 平均训练时长（分钟）。
    /// </summary>
    public double AverageTrainingMinutes { get; set; }

    /// <summary>
    /// 成功率（百分比）。
    /// </summary>
    public double SuccessRate { get; set; }
}