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

namespace MES.Service.service.BasicData;

public class MesProjectManager : Repository<MesProject>
{
    public bool Save(ErpProject unit)
    {
        var entity = GetMesProject(unit);

        return UseTransaction(db =>
        {
            switch (unit.Type)
            {
                case "0":
                    if (UpdateSupplierStatus(db, entity, "A"))
                        return 1;
                    break;
                case "1":
                    if (UpdateSupplierStatus(db, entity, "B"))
                        return 1;
                    break;
                case "3":
                    if (DeleteSupplier(entity))
                        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<ErpProject> suppliers)
    {
        var result = suppliers.Select(Save).ToList();
        return result.All(b => b);
    }

    private bool UpdateSupplierStatus(SqlSugarScope db,
        List<MesProject> supplierId,
        string status)
    {
        var ids = supplierId.Select(it => it.Id).ToArray();

        var result = db.Updateable<MesProject>()
            .SetColumns(s => s.ForbidStatus == status)
            .Where(s => ids.Contains(s.Id)).ExecuteCommand();

        if (result > 0)
            return true;

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

    private bool DeleteSupplier(List<MesProject> supplierId)
    {
        // 提取所有需要的 MesId 组合 
        var ids = supplierId.Select(customer => customer.Id.ToString())
            .ToList();

        // 使用这些 MesId 去查询 MesLinkU9 表，找到对应的 U9Id
        var u9Ids = Db.Queryable<MesLinkU9>()
            .Where(u =>
                ids.Contains(u.MesId) && u.TableType == "MES_PROJECT")
            .Select(u => u.U9Id)
            .Distinct() // 应用 Distinct 去重
            .ToList();

        // 根据 U9Id 找到所有对应的 MesId，进行删除
        // 注意：这里假设您确实想要删除这些记录。如果不确定，请先查询以验证结果。
        var idsToDelete = Db.Queryable<MesLinkU9>()
            .Where(u => u9Ids.Contains(u.U9Id) && u.TableType == "MES_PROJECT")
            .Select(u => u.MesId)
            .Distinct() // 应用 Distinct 去重
            .ToList();

        var deleteByIds = Db.Deleteable<MesProject>().In(idsToDelete)
            .ExecuteCommand();
        // if (deleteByIds > 0)
        //     return true;
        return true;
        //throw new NotImplementedException("删除失败");
    }

    private List<MesProject> GetMesProject(ErpProject supplier)
    {
        List<MesProject> list = new();
        foreach (var se in supplier.FUseOrgId)
        {
            var exists = Db.Queryable<MesLinkU9>().Any(u =>
                u.U9Id == supplier.ID && u.OrgId == se.FUseOrgId &&
                u.TableType == "MES_PROJECT");

            decimal mesId = 0;
            if (exists)
            {
                //获取mesid
                mesId = Convert.ToDecimal(Db.Queryable<MesLinkU9>()
                    .Where(u =>
                        u.U9Id == supplier.ID && u.OrgId == se.FUseOrgId &&
                        u.TableType == "MES_PROJECT")
                    .Select(u => u.MesId) // 选择 MesId 字段
                    .First());
            }
            else
            {
                var entity = new MesLinkU9
                {
                    CreateDate = DateTime.Now,
                    MesId = mesId.ToString(),
                    U9Id = supplier.ID,
                    OrgId = se.FUseOrgId,
                    TableType = "MES_PROJECT"
                };
                mesId = Db.Insertable(entity).ExecuteReturnIdentity();
            }

            var mesSupplier = new MesProject
            {
                Id = mesId,
                ProjectCode = supplier.Code,
                ProjectName = supplier.Name,
                ProjectType = supplier.ProjectType,
                Description = supplier.Description,
                OperatorsId = supplier.OperatorsID,
                DocumentStatus = supplier.FDocumentStatus,
                ForbidStatus = supplier.FForbidStatus,
                UseOrgId = se.FUseOrgId,
            };

            list.Add(mesSupplier);
        }

        return list;
    }

    private bool InsertOrUpdate(SqlSugarScope db, List<MesProject> entity)
    {
        DeleteSupplier(entity);
        
        var insert = db.Insertable(entity).ExecuteCommand();
        return insert > 0;
    }
}