AiDOPWarehouse/server/Plugins/Admin.NET.Plugin.AiDOP/Controllers/S0/Warehouse/AdoS0TaskAssignmentsController.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 物料状态任务指派（wms_rwzp 语义）
/// </summary>
[ApiController]
[Route("api/s0/warehouse/task-assignments")]
[AllowAnonymous]
[NonUnify]
public class AdoS0TaskAssignmentsController : ControllerBase
{
    private readonly SqlSugarRepository<AdoS0TaskAssignment> _rep;

    public AdoS0TaskAssignmentsController(SqlSugarRepository<AdoS0TaskAssignment> rep)
    {
        _rep = rep;
    }

    [HttpGet]
    public async Task<IActionResult> GetPagedAsync([FromQuery] AdoS0TaskAssignmentQueryDto 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.Rwlx), x => x.Rwlx == q.Rwlx)
            .WhereIF(!string.IsNullOrWhiteSpace(q.Zt), x => x.Zt == q.Zt)
            .WhereIF(!string.IsNullOrWhiteSpace(q.Sqr), x => x.Sqr != null && x.Sqr.Contains(q.Sqr!))
            .WhereIF(!string.IsNullOrWhiteSpace(q.Keyword),
                x => (x.Wlbm != null && x.Wlbm.Contains(q.Keyword!)) || (x.Sqr != null && x.Sqr.Contains(q.Keyword!)));

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

        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);
        return item == null ? NotFound() : Ok(item);
    }

    [HttpPost]
    public async Task<IActionResult> CreateAsync([FromBody] AdoS0TaskAssignmentUpsertDto dto)
    {
        var entity = new AdoS0TaskAssignment
        {
            CompanyRefId = dto.CompanyRefId,
            FactoryRefId = dto.FactoryRefId,
            DomainCode = dto.DomainCode ?? string.Empty,
            Tcrq = dto.Tcrq,
            Sqr = dto.Sqr,
            Rwlx = dto.Rwlx,
            Wlbm = dto.Wlbm,
            Sl = dto.Sl,
            Rqpc = dto.Rqpc,
            Yskw = dto.Yskw,
            Mdkw = dto.Mdkw,
            Xqsj = dto.Xqsj,
            Clr = dto.Clr,
            Bz = dto.Bz,
            Zt = dto.Zt,
            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] AdoS0TaskAssignmentUpsertDto dto)
    {
        var entity = await _rep.GetByIdAsync(id);
        if (entity == null) return NotFound();

        entity.CompanyRefId = dto.CompanyRefId;
        entity.FactoryRefId = dto.FactoryRefId;
        entity.DomainCode = dto.DomainCode ?? string.Empty;
        entity.Tcrq = dto.Tcrq;
        entity.Sqr = dto.Sqr;
        entity.Rwlx = dto.Rwlx;
        entity.Wlbm = dto.Wlbm;
        entity.Sl = dto.Sl;
        entity.Rqpc = dto.Rqpc;
        entity.Yskw = dto.Yskw;
        entity.Mdkw = dto.Mdkw;
        entity.Xqsj = dto.Xqsj;
        entity.Clr = dto.Clr;
        entity.Bz = dto.Bz;
        entity.Zt = dto.Zt;
        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 = "删除成功" });
    }
}