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 语义）。独立货架页 CRUD。
/// 多租户隔离：所有读写显式限定当前请求租户；父库位存在性按 tenant+domain+location 校验；不依赖全局 AOP。
/// </summary>
[ApiController]
[Route("api/s0/warehouse/location-shelves")]
[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)
    {
        if (!AdoS0TenantScope.TryResolveRequired(out var tenantId, out var tenantError)) return tenantError!;
        (q.Page, q.PageSize) = PagingGuard.Normalize(q.Page, q.PageSize);

        var query = _rep.AsQueryable()
            .Where(x => x.TenantId == tenantId)
            .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, tenantId);

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

    [HttpGet("{id:long}")]
    public async Task<IActionResult> GetAsync(long id)
    {
        if (!AdoS0TenantScope.TryResolveRequired(out var tenantId, out var tenantError)) return tenantError!;

        var item = await _rep.AsQueryable().Where(x => x.Id == id && x.TenantId == tenantId).FirstAsync();
        if (item == null) return NotFound();
        await ApplyLocationDescrAsync(new List<AdoS0LocationShelfMaster> { item }, tenantId);
        return Ok(item);
    }

    [HttpPost]
    public async Task<IActionResult> CreateAsync([FromBody] AdoS0LocationShelfUpsertDto dto)
    {
        if (!AdoS0TenantScope.TryResolveRequired(out var tenantId, out var tenantError)) return tenantError!;

        // 唯一性：当前租户 + domain + location + invShelf
        if (await _rep.IsAnyAsync(x => x.TenantId == tenantId && x.DomainCode == dto.DomainCode && x.Location == dto.Location && x.InvShelf == dto.InvShelf))
            return AdoS0ApiErrors.Conflict(AdoS0ErrorCodes.DuplicateCode, "货架编码已存在");

        // 父库位存在性：必须为当前租户下同 domain+location 的库位（不得为他租户库位建货架）
        if (!await _refChecker.LocationExistsInTenantAsync(tenantId, dto.DomainCode, dto.Location))
            return AdoS0ApiErrors.InvalidReference(AdoS0ErrorCodes.ReferenceNotFound,
                $"库位编码 '{dto.Location}' 不存在于当前租户库位主数据");

        var entity = new AdoS0LocationShelfMaster
        {
            TenantId = tenantId,
            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)
    {
        if (!AdoS0TenantScope.TryResolveRequired(out var tenantId, out var tenantError)) return tenantError!;

        var entity = await _rep.AsQueryable().Where(x => x.Id == id && x.TenantId == tenantId).FirstAsync();
        if (entity == null) return NotFound();

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

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

        // entity.TenantId 保持不变（== tenantId）；不接受前端修改 tenant
        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)
    {
        if (!AdoS0TenantScope.TryResolveRequired(out var tenantId, out var tenantError)) return tenantError!;

        var item = await _rep.AsQueryable().Where(x => x.Id == id && x.TenantId == tenantId).FirstAsync();
        if (item == null) return NotFound();
        await _rep.DeleteAsync(item);
        return Ok(new { message = "删除成功" });
    }

    /// <summary>
    /// 补显库位说明 + 「库位:货架」拼接。关联 LocationMaster 时限定当前租户，避免同编码库位跨租户补错描述。
    /// </summary>
    private async Task ApplyLocationDescrAsync(List<AdoS0LocationShelfMaster> shelves, long tenantId)
    {
        if (shelves.Count == 0) return;

        var domainCodes = shelves.Select(s => s.DomainCode).Distinct().ToList();
        var locations = await _locRep.AsQueryable()
            .Where(l => l.TenantId == tenantId && 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}";
        }
    }
}