AiDOPWarehouse/server/Plugins/Admin.NET.Plugin.AiDOP/Controllers/S0/Warehouse/AdoS0LocationShelvesController.cs

using Admin.NET.Plugin.AiDOP.Dto.S0.Warehouse;
using Admin.NET.Plugin.AiDOP.Entity.S0.Warehouse;
using Admin.NET.Plugin.AiDOP.Infrastructure;

namespace Admin.NET.Plugin.AiDOP.Controllers.S0.Warehouse;

/// <summary>
/// S0 货架主数据（LocationShelfMaster 语义）
/// </summary>
[ApiController]
[Route("api/s0/warehouse/location-shelves")]
[AllowAnonymous]
[NonUnify]
public class AdoS0LocationShelvesController : ControllerBase
{
    private readonly SqlSugarRepository<AdoS0LocationShelfMaster> _rep;
    private readonly SqlSugarRepository<AdoS0LocationMaster> _locRep;
    private readonly AdoS0ReferenceChecker _refChecker;

    public AdoS0LocationShelvesController(
        SqlSugarRepository<AdoS0LocationShelfMaster> rep,
        SqlSugarRepository<AdoS0LocationMaster> locRep,
        AdoS0ReferenceChecker refChecker)
    {
        _rep = rep;
        _locRep = locRep;
        _refChecker = refChecker;
    }

    [HttpGet]
    public async Task<IActionResult> GetPagedAsync([FromQuery] AdoS0LocationShelfQueryDto q)
    {
        (q.Page, q.PageSize) = PagingGuard.Normalize(q.Page, q.PageSize);

        var query = _rep.AsQueryable()
            .WhereIF(q.CompanyRefId.HasValue, x => x.CompanyRefId == q.CompanyRefId!.Value)
            .WhereIF(q.FactoryRefId.HasValue, x => x.FactoryRefId == q.FactoryRefId!.Value)
            .WhereIF(!string.IsNullOrWhiteSpace(q.DomainCode), x => x.DomainCode == q.DomainCode)
            .WhereIF(!string.IsNullOrWhiteSpace(q.Location), x => x.Location == q.Location)
            .WhereIF(!string.IsNullOrWhiteSpace(q.Keyword),
                x => x.InvShelf.Contains(q.Keyword!) || x.Location.Contains(q.Keyword!));

        var total = await query.CountAsync();
        var list = await query
            .OrderBy(x => x.Location)
            .OrderBy(x => x.InvShelf)
            .Skip((q.Page - 1) * q.PageSize)
            .Take(q.PageSize)
            .ToListAsync();

        await ApplyLocationDescrAsync(list);

        return Ok(new { total, page = q.Page, pageSize = q.PageSize, list });
    }

    [HttpGet("{id:long}")]
    public async Task<IActionResult> GetAsync(long id)
    {
        var item = await _rep.GetByIdAsync(id);
        if (item == null) return NotFound();
        await ApplyLocationDescrAsync(new List<AdoS0LocationShelfMaster> { item });
        return Ok(item);
    }

    [HttpPost]
    public async Task<IActionResult> CreateAsync([FromBody] AdoS0LocationShelfUpsertDto dto)
    {
        if (await _rep.IsAnyAsync(x => x.FactoryRefId == dto.FactoryRefId && x.Location == dto.Location && x.InvShelf == dto.InvShelf))
            return AdoS0ApiErrors.Conflict(AdoS0ErrorCodes.DuplicateCode, "货架编码已存在");

        if (!await _refChecker.LocationExistsAsync(dto.Location))
            return AdoS0ApiErrors.InvalidReference(AdoS0ErrorCodes.ReferenceNotFound,
                $"库位编码 '{dto.Location}' 不存在于库位主数据");

        var entity = new AdoS0LocationShelfMaster
        {
            CompanyRefId = dto.CompanyRefId,
            FactoryRefId = dto.FactoryRefId,
            DomainCode = dto.DomainCode ?? string.Empty,
            Location = dto.Location,
            InvShelf = dto.InvShelf,
            Descr = dto.Descr,
            Area = dto.Area,
            CreateUser = dto.CreateUser,
            CreateTime = DateTime.Now
        };

        await _rep.AsInsertable(entity).ExecuteReturnEntityAsync();
        return Ok(entity);
    }

    [HttpPut("{id:long}")]
    public async Task<IActionResult> UpdateAsync(long id, [FromBody] AdoS0LocationShelfUpsertDto dto)
    {
        var entity = await _rep.GetByIdAsync(id);
        if (entity == null) return NotFound();

        if (await _rep.IsAnyAsync(x => x.Id != id && x.FactoryRefId == dto.FactoryRefId && x.Location == dto.Location && x.InvShelf == dto.InvShelf))
            return AdoS0ApiErrors.Conflict(AdoS0ErrorCodes.DuplicateCode, "货架编码已存在");

        if (!await _refChecker.LocationExistsAsync(dto.Location))
            return AdoS0ApiErrors.InvalidReference(AdoS0ErrorCodes.ReferenceNotFound,
                $"库位编码 '{dto.Location}' 不存在于库位主数据");

        entity.CompanyRefId = dto.CompanyRefId;
        entity.FactoryRefId = dto.FactoryRefId;
        entity.DomainCode = dto.DomainCode ?? string.Empty;
        entity.Location = dto.Location;
        entity.InvShelf = dto.InvShelf;
        entity.Descr = dto.Descr;
        entity.Area = dto.Area;
        entity.UpdateUser = dto.UpdateUser;
        entity.UpdateTime = DateTime.Now;

        await _rep.AsUpdateable(entity).ExecuteCommandAsync();
        return Ok(entity);
    }

    [HttpDelete("{id:long}")]
    public async Task<IActionResult> DeleteAsync(long id)
    {
        var item = await _rep.GetByIdAsync(id);
        if (item == null) return NotFound();
        await _rep.DeleteAsync(item);
        return Ok(new { message = "删除成功" });
    }

    private async Task ApplyLocationDescrAsync(List<AdoS0LocationShelfMaster> shelves)
    {
        if (shelves.Count == 0) return;

        var domainCodes = shelves.Select(s => s.DomainCode).Distinct().ToList();
        var locations = await _locRep.AsQueryable()
            .Where(l => domainCodes.Contains(l.DomainCode))
            .ToListAsync();

        foreach (var shelf in shelves)
        {
            var loc = locations.Find(l =>
                string.Equals(l.DomainCode, shelf.DomainCode, StringComparison.OrdinalIgnoreCase)
                && string.Equals(l.Location, shelf.Location, StringComparison.OrdinalIgnoreCase));
            shelf.LocationDescr = loc?.Descr;
            shelf.KwhjName = $"{shelf.Location}:{shelf.InvShelf}";
        }
    }
}