refactor(api): extract timeline project query and read helpers

2026-03-31 16:01:35 +02:00
parent fda6bcab74
commit 803de725ad
7 changed files with 357 additions and 288 deletions
@@ -1,5 +1,4 @@
 import { TRPCError } from "@trpc/server";
-import type { Allocation } from "@capakraken/shared";
 import {
  buildTimelineProjectAssignmentConflicts,
  type TimelineProjectAssignmentConflict,
@@ -87,79 +86,6 @@ export function buildTimelineProjectContextSummary(input: {
  };
 }

-export function buildTimelineShiftValidationBookings(
-  bookings: Array<{
-    id: string;
-    resourceId: string | null;
-    projectId: string | null;
-    startDate: Date;
-    endDate: Date;
-    hoursPerDay: number;
-    status: string;
-  }>,
-) {
-  return bookings
-    .filter(
-      (
-        booking,
-      ): booking is typeof booking & { resourceId: string; projectId: string } =>
-        booking.resourceId !== null && booking.projectId !== null,
-    )
-    .map((booking) => ({
-      id: booking.id,
-      resourceId: booking.resourceId,
-      projectId: booking.projectId,
-      startDate: booking.startDate,
-      endDate: booking.endDate,
-      hoursPerDay: booking.hoursPerDay,
-      status: booking.status,
-    }));
-}
-
-export function buildTimelineBudgetStatusAllocations(
-  bookings: Array<{
-    status: string;
-    dailyCostCents: number;
-    startDate: Date;
-    endDate: Date;
-    hoursPerDay: number;
-  }>,
-): Pick<
-  Allocation,
-  "status" | "dailyCostCents" | "startDate" | "endDate" | "hoursPerDay"
->[] {
-  return bookings.map((booking) => ({
-    status: booking.status as Allocation["status"],
-    dailyCostCents: booking.dailyCostCents,
-    startDate: booking.startDate,
-    endDate: booking.endDate,
-    hoursPerDay: booking.hoursPerDay,
-  }));
-}
-
-export function buildTimelineBudgetStatusResponse<TBudgetStatus extends object>(input: {
-  project: {
-    name: string;
-    shortCode: string | null;
-    budgetCents: number;
-  };
-  budgetStatus: TBudgetStatus;
-  totalAllocations: number;
-}): TBudgetStatus & {
-  projectName: string;
-  projectCode: string;
-  totalAllocations: number;
-  budgetCents: number;
-} {
-  return {
-    ...input.budgetStatus,
-    projectName: input.project.name,
-    projectCode: input.project.shortCode ?? "",
-    totalAllocations: input.totalAllocations,
-    budgetCents: input.project.budgetCents,
-  };
-}
-
 export function buildTimelineProjectContextResponse<
  TProject,
  TAllocation extends { resource?: { id: string } | null },
@@ -303,37 +229,3 @@ export async function loadTimelineProjectContextDetailArtifacts(
    assignmentConflicts,
  };
 }
-
-export function buildTimelineShiftPreviewDetailResponse<TPreview>(input: {
-  project: {
-    id: string;
-    name: string;
-    shortCode: string | null;
-    status: string;
-    responsiblePerson: string | null;
-    startDate: Date;
-    endDate: Date;
-  };
-  requestedShift: {
-    newStartDate: Date;
-    newEndDate: Date;
-  };
-  preview: TPreview;
-}) {
-  return {
-    project: {
-      id: input.project.id,
-      name: input.project.name,
-      shortCode: input.project.shortCode,
-      status: input.project.status,
-      responsiblePerson: input.project.responsiblePerson,
-      startDate: fmtDate(input.project.startDate),
-      endDate: fmtDate(input.project.endDate),
-    },
-    requestedShift: {
-      newStartDate: fmtDate(input.requestedShift.newStartDate),
-      newEndDate: fmtDate(input.requestedShift.newEndDate),
-    },
-    preview: input.preview,
-  };
-}
@@ -0,0 +1,60 @@
+import type { Prisma } from "@capakraken/db";
+import { findUniqueOrThrow } from "../db/helpers.js";
+import type { ShiftDbClient } from "./timeline-read-shared.js";
+
+export const projectShiftContextSelect = {
+  id: true,
+  budgetCents: true,
+  winProbability: true,
+  startDate: true,
+  endDate: true,
+} as const satisfies Prisma.ProjectSelect;
+
+export const timelineProjectContextSelect = {
+  id: true,
+  name: true,
+  shortCode: true,
+  orderType: true,
+  budgetCents: true,
+  winProbability: true,
+  status: true,
+  startDate: true,
+  endDate: true,
+  staffingReqs: true,
+} as const satisfies Prisma.ProjectSelect;
+
+export const timelineShiftPreviewProjectSelect = {
+  id: true,
+  name: true,
+  shortCode: true,
+  status: true,
+  responsiblePerson: true,
+  startDate: true,
+  endDate: true,
+} as const satisfies Prisma.ProjectSelect;
+
+export const timelineBudgetStatusProjectSelect = {
+  id: true,
+  name: true,
+  shortCode: true,
+  budgetCents: true,
+  winProbability: true,
+  startDate: true,
+  endDate: true,
+} as const satisfies Prisma.ProjectSelect;
+
+export async function findTimelineProjectOrThrow<TSelect extends Prisma.ProjectSelect>(
+  db: Pick<ShiftDbClient, "project">,
+  input: {
+    projectId: string;
+    select: TSelect;
+  },
+): Promise<Prisma.ProjectGetPayload<{ select: TSelect }>> {
+  return findUniqueOrThrow(
+    db.project.findUnique({
+      where: { id: input.projectId },
+      select: input.select,
+    }),
+    "Project",
+  ) as Promise<Prisma.ProjectGetPayload<{ select: TSelect }>>;
+}
@@ -0,0 +1,109 @@
+import type { Allocation } from "@capakraken/shared";
+import { fmtDate } from "./timeline-read-shared.js";
+
+export function buildTimelineShiftValidationBookings(
+  bookings: Array<{
+    id: string;
+    resourceId: string | null;
+    projectId: string | null;
+    startDate: Date;
+    endDate: Date;
+    hoursPerDay: number;
+    status: string;
+  }>,
+) {
+  return bookings
+    .filter(
+      (
+        booking,
+      ): booking is typeof booking & { resourceId: string; projectId: string } =>
+        booking.resourceId !== null && booking.projectId !== null,
+    )
+    .map((booking) => ({
+      id: booking.id,
+      resourceId: booking.resourceId,
+      projectId: booking.projectId,
+      startDate: booking.startDate,
+      endDate: booking.endDate,
+      hoursPerDay: booking.hoursPerDay,
+      status: booking.status,
+    }));
+}
+
+export function buildTimelineBudgetStatusAllocations(
+  bookings: Array<{
+    status: string;
+    dailyCostCents: number;
+    startDate: Date;
+    endDate: Date;
+    hoursPerDay: number;
+  }>,
+): Pick<
+  Allocation,
+  "status" | "dailyCostCents" | "startDate" | "endDate" | "hoursPerDay"
+>[] {
+  return bookings.map((booking) => ({
+    status: booking.status as Allocation["status"],
+    dailyCostCents: booking.dailyCostCents,
+    startDate: booking.startDate,
+    endDate: booking.endDate,
+    hoursPerDay: booking.hoursPerDay,
+  }));
+}
+
+export function buildTimelineBudgetStatusResponse<TBudgetStatus extends object>(input: {
+  project: {
+    name: string;
+    shortCode: string | null;
+    budgetCents: number;
+  };
+  budgetStatus: TBudgetStatus;
+  totalAllocations: number;
+}): TBudgetStatus & {
+  projectName: string;
+  projectCode: string;
+  totalAllocations: number;
+  budgetCents: number;
+} {
+  return {
+    ...input.budgetStatus,
+    projectName: input.project.name,
+    projectCode: input.project.shortCode ?? "",
+    totalAllocations: input.totalAllocations,
+    budgetCents: input.project.budgetCents,
+  };
+}
+
+export function buildTimelineShiftPreviewDetailResponse<TPreview>(input: {
+  project: {
+    id: string;
+    name: string;
+    shortCode: string | null;
+    status: string;
+    responsiblePerson: string | null;
+    startDate: Date;
+    endDate: Date;
+  };
+  requestedShift: {
+    newStartDate: Date;
+    newEndDate: Date;
+  };
+  preview: TPreview;
+}) {
+  return {
+    project: {
+      id: input.project.id,
+      name: input.project.name,
+      shortCode: input.project.shortCode,
+      status: input.project.status,
+      responsiblePerson: input.project.responsiblePerson,
+      startDate: fmtDate(input.project.startDate),
+      endDate: fmtDate(input.project.endDate),
+    },
+    requestedShift: {
+      newStartDate: fmtDate(input.requestedShift.newStartDate),
+      newEndDate: fmtDate(input.requestedShift.newEndDate),
+    },
+    preview: input.preview,
+  };
+}
@@ -2,38 +2,37 @@ import { listAssignmentBookings } from "@capakraken/application";
 import { computeBudgetStatus } from "@capakraken/engine";
 import { ShiftProjectSchema } from "@capakraken/shared";
 import { z } from "zod";
-import { findUniqueOrThrow } from "../db/helpers.js";
 import { getAnonymizationDirectory } from "../lib/anonymization.js";
 import { controllerProcedure } from "../trpc.js";
 import { loadProjectPlanningReadModel } from "./project-planning-read-model.js";
 import {
-  buildTimelineBudgetStatusAllocations,
-  buildTimelineBudgetStatusResponse,
-  buildTimelineShiftValidationBookings,
-  buildTimelineShiftPreviewDetailResponse,
  buildTimelineProjectContextDetailResponse,
  buildTimelineProjectContextResponse,
  loadTimelineProjectContextDetailArtifacts,
 } from "./timeline-project-context-support.js";
+import {
+  buildTimelineBudgetStatusAllocations,
+  buildTimelineBudgetStatusResponse,
+  buildTimelineShiftPreviewDetailResponse,
+  buildTimelineShiftValidationBookings,
+} from "./timeline-project-read-support.js";
+import {
+  findTimelineProjectOrThrow,
+  projectShiftContextSelect,
+  timelineBudgetStatusProjectSelect,
+  timelineProjectContextSelect,
+  timelineShiftPreviewProjectSelect,
+} from "./timeline-project-query-support.js";
 import { buildTimelineShiftPlan } from "./timeline-shift-planning.js";
 import { getAssignmentResourceIds, ShiftDbClient } from "./timeline-read-shared.js";
 import { buildTimelineProjectShiftValidation } from "./timeline-shift-support.js";

 export async function loadProjectShiftContext(db: ShiftDbClient, projectId: string) {
  const [project, planningRead] = await Promise.all([
-    findUniqueOrThrow(
-      db.project.findUnique({
-        where: { id: projectId },
-        select: {
-          id: true,
-          budgetCents: true,
-          winProbability: true,
-          startDate: true,
-          endDate: true,
-        },
-      }),
-      "Project",
-    ),
+    findTimelineProjectOrThrow(db, {
+      projectId,
+      select: projectShiftContextSelect,
+    }),
    loadProjectPlanningReadModel(db, { projectId, activeOnly: true }),
  ]);

@@ -65,24 +64,10 @@ export async function loadProjectShiftContext(db: ShiftDbClient, projectId: stri

 export async function loadTimelineProjectContext(db: ShiftDbClient, projectId: string) {
  const [project, planningRead] = await Promise.all([
-    findUniqueOrThrow(
-      db.project.findUnique({
-        where: { id: projectId },
-        select: {
-          id: true,
-          name: true,
-          shortCode: true,
-          orderType: true,
-          budgetCents: true,
-          winProbability: true,
-          status: true,
-          startDate: true,
-          endDate: true,
-          staffingReqs: true,
-        },
-      }),
-      "Project",
-    ),
+    findTimelineProjectOrThrow(db, {
+      projectId,
+      select: timelineProjectContextSelect,
+    }),
    loadProjectPlanningReadModel(db, {
      projectId,
      activeOnly: true,
@@ -198,21 +183,10 @@ export const timelineProjectReadProcedures = {
    .input(ShiftProjectSchema)
    .query(async ({ ctx, input }) => {
      const [project, preview] = await Promise.all([
-        findUniqueOrThrow(
-          ctx.db.project.findUnique({
-            where: { id: input.projectId },
-            select: {
-              id: true,
-              name: true,
-              shortCode: true,
-              status: true,
-              responsiblePerson: true,
-              startDate: true,
-              endDate: true,
-            },
-          }),
-          "Project",
-        ),
+        findTimelineProjectOrThrow(ctx.db, {
+          projectId: input.projectId,
+          select: timelineShiftPreviewProjectSelect,
+        }),
        previewTimelineProjectShift(ctx.db, input),
      ]);

@@ -229,21 +203,10 @@ export const timelineProjectReadProcedures = {
  getBudgetStatus: controllerProcedure
    .input(z.object({ projectId: z.string() }))
    .query(async ({ ctx, input }) => {
-      const project = await findUniqueOrThrow(
-        ctx.db.project.findUnique({
-          where: { id: input.projectId },
-          select: {
-            id: true,
-            name: true,
-            shortCode: true,
-            budgetCents: true,
-            winProbability: true,
-            startDate: true,
-            endDate: true,
-          },
-        }),
-        "Project",
-      );
+      const project = await findTimelineProjectOrThrow(ctx.db, {
+        projectId: input.projectId,
+        select: timelineBudgetStatusProjectSelect,
+      });

      const bookings = await listAssignmentBookings(ctx.db, {
        projectIds: [project.id],