shoc-backend/SeaHaven.DataServices/Helpers/WorkOrderBoardProjection.cs

using Data.SeaHavenIndustries;
using Data.SeaHavenIndustries.Enums;
using Microsoft.EntityFrameworkCore;
using SeaHaven.DataServices.Interfaces;

namespace SeaHaven.DataServices.Helpers
{
    public static class WorkOrderBoardProjection
    {
        public static async Task<IReadOnlyList<WorkOrderBoardRawRow>> ProjectRowsAsync(
            ApplicationDbContext context,
            IQueryable<WorkOrder> query,
            bool isUnscheduled)
        {
            var rows = await query
                .Select(w => new
                {
                    w.Id,
                    w.InternalWONumber,
                    w.RescheduleCount,
                    w.CarriedOver,
                    w.IsAddOn,
                    w.AvetaRequired,
                    HasAvetaDocument = w.workOrderAttachments!.Any(a =>
                        (a.IsDeleted == null || a.IsDeleted == false)
                        && a.Category == WorkOrderMediaCategory.Aveta),
                    w.OriginalDate,
                    w.OriginalWeek,
                    w.WorkOrderType,
                    w.SiteCode,
                    LocationName = w.Locations != null ? w.Locations.Name : null,
                    WoPocName = w.PocName,
                    WoPocPhone = w.PocPhone,
                    WoPocNotes = w.PocNotes,
                    ContactPoc = w.WorkOrderContacts!
                        .OrderBy(c => c.Id)
                        .Select(c => new
                        {
                            Name = c.POC != null
                                ? ((c.POC.FirstName ?? "") + " " + (c.POC.LastName ?? "")).Trim()
                                : null,
                            PhoneNumber = c.POC != null ? c.POC.PhoneNumber : null,
                            c.Notes
                        })
                        .FirstOrDefault(),
                    w.LifecycleStatus,
                    w.LegacyStatus,
                    w.Status,
                    w.AssignTo,
                    w.AssignToUser!.FirstName,
                    w.AssignToUser!.LastName,
                    w.AssignToUser!.Initials,
                    w.AssignToUser!.Color,
                    w.DueDate,
                    w.ScheduledDate,
                    w.ScheduledStart,
                    w.ScheduledEnd,
                    w.TargetWeek,
                    w.ScheduleWeekOnly,
                    PrimaryDispatchStatus = w.PrimaryDispatch != null ? w.PrimaryDispatch.Status : null,
                    VendorId = w.PrimaryDispatch != null ? (int?)w.PrimaryDispatch.VendorId : null,
                    VendorName = w.PrimaryDispatch != null && w.PrimaryDispatch.Vendor != null
                        ? w.PrimaryDispatch.Vendor.CompanyName
                        : null,
                    TechName = w.PrimaryDispatch != null && w.PrimaryDispatch.Vendor != null
                        ? w.PrimaryDispatch.Vendor.ContactName
                        : null,
                    DispatchTechPhone = w.PrimaryDispatch != null ? w.PrimaryDispatch.TechPhone : null,
                    WoTechPhone = w.TechPhone,
                    VendorPhone = w.PrimaryDispatch != null && w.PrimaryDispatch.Vendor != null
                        ? w.PrimaryDispatch.Vendor.Phone
                        : null,
                    DispatchApptDate = w.PrimaryDispatch != null ? w.PrimaryDispatch.ScheduledDate : null,
                    w.Trade,
                    w.Problem,
                    w.ServiceNotes,
                    w.ExtraServices,
                    w.AdditionalContacts,
                    w.DocStatus,
                    w.CompletedDate,
                    w.FlagColor,
                    w.PrimaryDispatchId,
                    w.RowVersion,
                    DispatchRowVersion = w.PrimaryDispatch != null ? w.PrimaryDispatch.RowVersion : null,
                    // SH-188: aggregate Pending uplifts on every live dispatch tied to this WO
                    // (primary WorkOrderId and multi-WO DispatchWorkOrders links).
                    PendingUpliftCount = context.DispatchUpliftRequests.Count(u =>
                        u.Status == "Pending"
                        && (u.IsDeleted == null || u.IsDeleted == false)
                        && u.Dispatch != null
                        && (u.Dispatch.IsDeleted == null || u.Dispatch.IsDeleted == false)
                        && (
                            u.Dispatch.WorkOrderId == w.Id
                            || u.Dispatch.DispatchWorkOrders!.Any(link => link.WorkOrderId == w.Id))),
                    HasUplift = context.DispatchUpliftRequests.Any(u =>
                        (u.IsDeleted == null || u.IsDeleted == false)
                        && u.Dispatch != null
                        && (u.Dispatch.IsDeleted == null || u.Dispatch.IsDeleted == false)
                        && (
                            u.Dispatch.WorkOrderId == w.Id
                            || u.Dispatch.DispatchWorkOrders!.Any(link => link.WorkOrderId == w.Id))),
                    PrimaryUpliftStatus = context.DispatchUpliftRequests
                        .Where(u =>
                            (u.IsDeleted == null || u.IsDeleted == false)
                            && u.Dispatch != null
                            && (u.Dispatch.IsDeleted == null || u.Dispatch.IsDeleted == false)
                            && (
                                u.Dispatch.WorkOrderId == w.Id
                                || u.Dispatch.DispatchWorkOrders!.Any(link => link.WorkOrderId == w.Id)))
                        .OrderByDescending(u => u.CreatedDate)
                        .Select(u => u.Status)
                        .FirstOrDefault(),
                    PrimaryUpliftAmount = context.DispatchUpliftRequests
                        .Where(u =>
                            (u.IsDeleted == null || u.IsDeleted == false)
                            && u.Dispatch != null
                            && (u.Dispatch.IsDeleted == null || u.Dispatch.IsDeleted == false)
                            && (
                                u.Dispatch.WorkOrderId == w.Id
                                || u.Dispatch.DispatchWorkOrders!.Any(link => link.WorkOrderId == w.Id)))
                        .OrderByDescending(u => u.CreatedDate)
                        .Select(u => (decimal?)u.RequestedNTE)
                        .FirstOrDefault()
                })
                .ToListAsync();

            return rows.Select(w =>
            {
                var pocName = !string.IsNullOrWhiteSpace(w.WoPocName) ? w.WoPocName : w.ContactPoc?.Name;
                var pocPhone = !string.IsNullOrWhiteSpace(w.WoPocPhone) ? w.WoPocPhone : w.ContactPoc?.PhoneNumber;
                var pocNotes = !string.IsNullOrWhiteSpace(w.WoPocNotes) ? w.WoPocNotes : w.ContactPoc?.Notes;
                var techPhone = !string.IsNullOrWhiteSpace(w.DispatchTechPhone)
                    ? w.DispatchTechPhone
                    : (!string.IsNullOrWhiteSpace(w.WoTechPhone) ? w.WoTechPhone : w.VendorPhone);
                var (vendorId, vendorName) = WorkOrderBoardDispatchAssignment.LiveVendor(
                    w.VendorId,
                    w.VendorName,
                    w.PrimaryDispatchStatus);

                return new WorkOrderBoardRawRow(
                    w.Id,
                    w.InternalWONumber,
                    w.RescheduleCount,
                    w.CarriedOver,
                    w.IsAddOn,
                    w.WorkOrderType,
                    w.SiteCode,
                    w.LocationName,
                    pocName,
                    pocPhone,
                    pocNotes,
                    w.LifecycleStatus,
                    w.LegacyStatus ?? w.Status,
                    w.AssignTo,
                    w.FirstName,
                    w.LastName,
                    w.Initials,
                    w.Color,
                    w.DueDate,
                    w.ScheduledDate,
                    w.ScheduledStart,
                    w.ScheduledEnd,
                    w.TargetWeek,
                    w.ScheduleWeekOnly,
                    vendorId,
                    vendorName,
                    w.TechName,
                    techPhone,
                    w.DispatchApptDate,
                    w.Trade ?? w.Problem,
                    w.Problem,
                    w.ServiceNotes,
                    w.ExtraServices,
                    w.AdditionalContacts,
                    w.DocStatus,
                    w.CompletedDate,
                    w.FlagColor,
                    w.PrimaryDispatchId,
                    w.RowVersion,
                    w.DispatchRowVersion,
                    w.PendingUpliftCount,
                    w.HasUplift,
                    w.PrimaryUpliftStatus,
                    w.PrimaryUpliftAmount,
                    w.PrimaryDispatchStatus,
                    w.AvetaRequired,
                    w.HasAvetaDocument,
                    w.OriginalDate,
                    w.OriginalWeek,
                    isUnscheduled);
            }).ToList();
        }
    }
}