refactor(api): extract timeline allocation procedure support

packages/api/src/router/timeline-allocation-mutations.ts

@@ -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 };
     }),
 };