refactor(api): extract timeline allocation procedure support

2026-03-31 17:45:54 +02:00
parent a3fb95ae07
commit 109bf70699
3 changed files with 289 additions and 74 deletions
@@ -1,7 +1,3 @@
-import {
-  buildSplitAllocationReadModel,
-  createAssignment,
-} from "@capakraken/application";
 import type { PrismaClient } from "@capakraken/db";
 import {
  AllocationStatus,
@@ -9,18 +5,14 @@ import {
  UpdateAllocationHoursSchema,
 } from "@capakraken/shared";
 import { z } from "zod";
-import {
-  emitAllocationCreated,
-  emitAllocationUpdated,
-} from "../sse/event-bus.js";
+import { emitAllocationUpdated } from "../sse/event-bus.js";
 import { managerProcedure, requirePermission } from "../trpc.js";
 import {
-  assertTimelineDateRangeValid,
-  buildTimelineQuickAssignAssignmentInput,
-  validateTimelineAllocationDateRanges,
-} from "./timeline-allocation-mutation-support.js";
+  createTimelineBatchQuickAssignments,
+  createTimelineQuickAssignment,
+  shiftTimelineAllocations,
+} from "./timeline-allocation-procedure-support.js";
 import { applyTimelineInlineAllocationUpdate } from "./timeline-allocation-inline-support.js";
-import { applyTimelineBatchAllocationShift } from "./timeline-allocation-shift-support.js";

 export const timelineAllocationMutationProcedures = {
  updateAllocationInline: managerProcedure
@@ -61,30 +53,10 @@ export const timelineAllocationMutationProcedures = {
    )
    .mutation(async ({ ctx, input }) => {
      requirePermission(ctx, PermissionKey.MANAGE_ALLOCATIONS);
-      assertTimelineDateRangeValid(input.startDate, input.endDate);
-
-      const allocation = await ctx.db.$transaction(async (tx) => {
-        const assignment = await createAssignment(
-          tx as unknown as Parameters<typeof createAssignment>[0],
-          buildTimelineQuickAssignAssignmentInput({
-            ...input,
-            source: "quickAssign",
-          }),
-        );
-
-        return buildSplitAllocationReadModel({
-          demandRequirements: [],
-          assignments: [assignment],
-        }).allocations[0]!;
+      return createTimelineQuickAssignment(ctx.db as PrismaClient, {
+        ...input,
+        source: "quickAssign",
      });
-
-      emitAllocationCreated({
-        id: allocation.id,
-        projectId: allocation.projectId,
-        resourceId: allocation.resourceId,
-      });
-
-      return allocation;
    }),

  batchQuickAssign: managerProcedure
@@ -110,32 +82,12 @@ export const timelineAllocationMutationProcedures = {
    )
    .mutation(async ({ ctx, input }) => {
      requirePermission(ctx, PermissionKey.MANAGE_ALLOCATIONS);
-      validateTimelineAllocationDateRanges(input.assignments);
-
-      const results = await ctx.db.$transaction(async (tx) => {
-        const created = [];
-        for (const assignment of input.assignments) {
-          const createdAssignment = await createAssignment(
-            tx as unknown as Parameters<typeof createAssignment>[0],
-            buildTimelineQuickAssignAssignmentInput({
-              ...assignment,
-              source: "batchQuickAssign",
-            }),
-          );
-          created.push(createdAssignment);
-        }
-        return created;
+      return createTimelineBatchQuickAssignments(ctx.db as PrismaClient, {
+        assignments: input.assignments.map((assignment) => ({
+          ...assignment,
+          source: "batchQuickAssign" as const,
+        })),
      });
-
-      for (const assignment of results) {
-        emitAllocationCreated({
-          id: assignment.id,
-          projectId: assignment.projectId,
-          resourceId: assignment.resourceId,
-        });
-      }
-
-      return { count: results.length };
    }),

  batchShiftAllocations: managerProcedure
@@ -148,22 +100,10 @@ export const timelineAllocationMutationProcedures = {
    )
    .mutation(async ({ ctx, input }) => {
      requirePermission(ctx, PermissionKey.MANAGE_ALLOCATIONS);
-
-      const results = await applyTimelineBatchAllocationShift({
-        db: ctx.db as PrismaClient,
+      return shiftTimelineAllocations(ctx.db as PrismaClient, {
        allocationIds: input.allocationIds,
        daysDelta: input.daysDelta,
        mode: input.mode,
      });
-
-      for (const allocation of results) {
-        emitAllocationUpdated({
-          id: allocation.id,
-          projectId: allocation.projectId,
-          resourceId: allocation.resourceId,
-        });
-      }
-
-      return { count: results.length };
    }),
 };
@@ -0,0 +1,95 @@
+import {
+  buildSplitAllocationReadModel,
+  createAssignment,
+} from "@capakraken/application";
+import type { PrismaClient } from "@capakraken/db";
+import {
+  emitAllocationCreated,
+  emitAllocationUpdated,
+} from "../sse/event-bus.js";
+import {
+  assertTimelineDateRangeValid,
+  buildTimelineQuickAssignAssignmentInput,
+  validateTimelineAllocationDateRanges,
+} from "./timeline-allocation-mutation-support.js";
+import { applyTimelineBatchAllocationShift } from "./timeline-allocation-shift-support.js";
+
+export async function createTimelineQuickAssignment(
+  db: PrismaClient,
+  input: Parameters<typeof buildTimelineQuickAssignAssignmentInput>[0],
+) {
+  assertTimelineDateRangeValid(input.startDate, input.endDate);
+
+  const allocation = await db.$transaction(async (tx) => {
+    const assignment = await createAssignment(
+      tx as unknown as Parameters<typeof createAssignment>[0],
+      buildTimelineQuickAssignAssignmentInput(input),
+    );
+
+    return buildSplitAllocationReadModel({
+      demandRequirements: [],
+      assignments: [assignment],
+    }).allocations[0]!;
+  });
+
+  emitAllocationCreated({
+    id: allocation.id,
+    projectId: allocation.projectId,
+    resourceId: allocation.resourceId,
+  });
+
+  return allocation;
+}
+
+export async function createTimelineBatchQuickAssignments(
+  db: PrismaClient,
+  input: {
+    assignments: Array<Parameters<typeof buildTimelineQuickAssignAssignmentInput>[0]>;
+  },
+) {
+  validateTimelineAllocationDateRanges(input.assignments);
+
+  const results = await db.$transaction(async (tx) => {
+    const created = [];
+    for (const assignment of input.assignments) {
+      const createdAssignment = await createAssignment(
+        tx as unknown as Parameters<typeof createAssignment>[0],
+        buildTimelineQuickAssignAssignmentInput(assignment),
+      );
+      created.push(createdAssignment);
+    }
+    return created;
+  });
+
+  for (const assignment of results) {
+    emitAllocationCreated({
+      id: assignment.id,
+      projectId: assignment.projectId,
+      resourceId: assignment.resourceId,
+    });
+  }
+
+  return { count: results.length };
+}
+
+export async function shiftTimelineAllocations(
+  db: PrismaClient,
+  input: Omit<Parameters<typeof applyTimelineBatchAllocationShift>[0], "db">,
+) {
+  const results = await applyTimelineBatchAllocationShift({
+    db,
+    allocationIds: input.allocationIds,
+    daysDelta: input.daysDelta,
+    mode: input.mode,
+  });
+
+  for (const allocation of results) {
+    emitAllocationUpdated({
+      id: allocation.id,
+      projectId: allocation.projectId,
+      resourceId: allocation.resourceId,
+    });
+  }
+
+  return { count: results.length };
+}