using MES.Service.DB;
using MES.Service.Dto.webApi;
using MES.Service.Modes;
using SqlSugar;

namespace MES.Service.service.BasicData;

public class MesUnitZhManager : Repository<MesUnitZh>
{
    public bool Save(ErpUnitZh unit)
    {
        var entity = GetMesUnitZh(unit);

        return UseTransaction(db =>
        {
            switch (unit.Type)
            {
                case "0":
                    if (UpdateUnitZhStatus(db, entity.Id, "A"))
                        return 1;
                    break;
                case "1":
                    if (UpdateUnitZhStatus(db, entity.Id, "B"))
                        return 1;
                    break;
                case "3":
                    if (DeleteUnitZh(db, entity.Id))
                        return 1;
                    break;
                case "2":
                case "4":
                    if (InsertOrUpdate(db, entity))
                        return 1;
                    break;
                default:
                    throw new ArgumentNullException(
                        $"type没有{unit.Type}这个类型的参数");
            }

            throw new NotImplementedException("操作失败");
        }) > 0;
    }

    public bool SaveList(List<ErpUnitZh> units)
    {
        var list = units.Select(GetMesUnitZh).ToList();
        var groupBy = list.GroupBy(s => s.Type)
            .ToDictionary(g => g.Key, g => g.ToList());

        return UseTransaction(db =>
        {
            foreach (var unitGroup in groupBy)
                switch (unitGroup.Key)
                {
                    case "0":
                        if (!UpdateUnitZhStatusBatch(db, unitGroup.Value, "A"))
                            throw new NotImplementedException("启用失败");
                        break;
                    case "1":
                        if (!UpdateUnitZhStatusBatch(db, unitGroup.Value, "B"))
                            throw new NotImplementedException("禁用失败");
                        break;
                    case "3":
                        if (!DeleteUnitZhBatch(db, unitGroup.Value))
                            throw new NotImplementedException("删除失败");
                        break;
                    case "2":
                    case "4":
                        if (!InsertOrUpdateBatch(db, unitGroup.Value))
                            throw new NotImplementedException("同步失败");
                        break;
                    default:
                        throw new ArgumentNullException(
                            $"type没有{unitGroup.Key}这个类型的参数");
                }

            return 1;
        }) > 0;
    }

    private bool UpdateUnitZhStatus(SqlSugarScope db, decimal unitId,
        string status)
    {
        // 注意：如果MesUnitZh表没有Fforbidstatus字段，需要根据实际情况调整
        // 这里可能需要通过其他方式更新状态，或者暂时注释掉
        var result = db.Updateable<MesUnitZh>()
            .Where(s => s.Id == unitId).ExecuteCommand();

        if (result > 0)
            return true;

        throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败");
    }

    private bool InsertUnitZh(SqlSugarScope db, MesUnitZh entity)
    {
        var insert = db.Insertable(entity).ExecuteCommand();
        if (insert > 0)
            return true;

        throw new NotImplementedException("插入失败");
    }

    private bool DeleteUnitZh(SqlSugarScope db, decimal unitId)
    {
        var deleteById = db.Deleteable<MesUnitZh>().In(unitId).ExecuteCommand();
        if (deleteById > 0)
            return true;

        throw new NotImplementedException("删除失败");
    }

    private MesUnitZh GetMesUnitZh(ErpUnitZh unit)
    {
        return new MesUnitZh
        {
            Id = string.IsNullOrEmpty(unit.FDWA010) ? 0 : Convert.ToDecimal(unit.FDWA010),
            Dwa001 = string.IsNullOrEmpty(unit.FDWA001) ? null : Convert.ToDecimal(unit.FDWA001),
            Dwa002 = unit.FDWA002,
            Dwa003 = unit.FDWA003,
            Dwa004 = unit.FDWA004,
            Dwa005 = string.IsNullOrEmpty(unit.FDWA005) ? null : Convert.ToDecimal(unit.FDWA005),
            Dwa006 = unit.FDWA006,
            Dwa007 = unit.FDWA007,
            Dwa008 = string.IsNullOrEmpty(unit.FDWA008) ? null : Convert.ToDecimal(unit.FDWA008),
            Dwa009 = unit.FDWA009,
            Erpid = string.IsNullOrEmpty(unit.FDWA010) ? null : Convert.ToDecimal(unit.FDWA010),
            Type = unit.Type
        };
    }

    private bool UpdateUnitZhStatusBatch(SqlSugarScope db, List<MesUnitZh> unitList,
        string status)
    {
        var ids = unitList.Select(it => it.Id).ToArray();
        var result = db.Updateable<MesUnitZh>()
            .Where(s => ids.Contains(s.Id)).ExecuteCommand();

        if (result > 0)
            return true;

        throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败");
    }

    private bool InsertUnitZhBatch(SqlSugarScope db, List<MesUnitZh> unitList)
    {
        var insertRange = db.Insertable(unitList).ExecuteCommand();
        if (insertRange > 0)
            return true;

        throw new NotImplementedException("插入失败");
    }

    private bool DeleteUnitZhBatch(SqlSugarScope db, List<MesUnitZh> unitList)
    {
        var ids = unitList.Select(it => it.Id).ToArray();
        var deleteByIds = db.Deleteable<MesUnitZh>().In(ids).ExecuteCommand();
        return deleteByIds > 0;
    }

    /// <summary>
    /// 插入或更新的方法：先删除再插入
    /// </summary>
    private bool InsertOrUpdate(SqlSugarScope db, MesUnitZh entity)
    {
        db.Deleteable<MesUnitZh>()
            .Where(s => s.Id == entity.Id)
            .ExecuteCommand();

        var insert = db.Insertable(entity).ExecuteCommand();
        return insert > 0;
    }

    private bool InsertOrUpdateBatch(SqlSugarScope db, List<MesUnitZh> unitList)
    {
        foreach (var entity in unitList)
            if (!InsertOrUpdate(db, entity))
                return false;

        return true;
    }
}