using Admin.NET.Plugin.AiDOP.Entity;
using Admin.NET.Plugin.AiDOP.Infrastructure;
using Admin.NET.Plugin.ApprovalFlow;
using Admin.NET.Plugin.ApprovalFlow.Service;

namespace Admin.NET.Plugin.AiDOP.Service;

/// <summary>
/// 智慧运营改善计划审批回调。
/// </summary>
public class SmartOpsImprovementBizHandler : IFlowBizHandler, ITransient
{
    public const string BizTypeCode = "SMART_OPS_IMPROVEMENT";

    public string BizType => BizTypeCode;

    private readonly SqlSugarRepository<AdoSmartOpsImprovementPlan> _rep;

    public SmartOpsImprovementBizHandler(SqlSugarRepository<AdoSmartOpsImprovementPlan> rep)
    {
        _rep = rep;
    }

    public async Task OnFlowStarted(long bizId, long instanceId)
    {
        var plan = await _rep.GetByIdAsync(bizId);
        if (plan == null) return;
        await _rep.AsUpdateable()
            .SetColumns(x => x.Status == "approving")
            .SetColumns(x => x.FlowInstanceId == instanceId)
            .SetColumns(x => x.UpdateTime == DateTime.Now)
            .Where(x => x.Id == bizId && x.TenantId == plan.TenantId)
            .ExecuteCommandAsync();
    }

    public async Task OnFlowCompleted(long bizId, long instanceId, FlowInstanceStatusEnum finalStatus, long? lastApproverId)
    {
        var plan = await _rep.GetByIdAsync(bizId);
        if (plan == null) return;
        var status = SmartOpsImprovementApprovalRules.MapCompletedStatus(finalStatus);
        await _rep.AsUpdateable()
            .SetColumns(x => x.Status == status)
            .SetColumns(x => x.UpdateTime == DateTime.Now)
            .Where(x => x.Id == bizId && x.TenantId == plan.TenantId)
            .ExecuteCommandAsync();
    }

    public async Task<Dictionary<string, object>> GetBizData(long bizId)
    {
        var plan = await _rep.GetByIdAsync(bizId);
        if (plan == null) return new Dictionary<string, object>();
        var route = SmartOpsImprovementApprovalRules.ResolveRoute(plan);
        return new Dictionary<string, object>
        {
            ["moduleCode"] = plan.ModuleCode,
            ["metricCode"] = plan.MetricCode ?? "",
            ["problemLevel"] = plan.ProblemLevel,
            ["problemName"] = plan.ProblemName,
            ["problemDept"] = plan.ProblemDept ?? "",
            ["problemSeverity"] = SmartOpsImprovementApprovalRules.NormalizeSeverity(plan.ProblemSeverity),
            ["isCrossDept"] = plan.IsCrossDept ? 1 : 0,
            ["route"] = route,
            ["status"] = plan.Status,
        };
    }
}