feature: Phase3 - Task templates + generation on send

backend/src/modules/cases/lab-case-task.generator.ts

@@ -0,0 +1,91 @@
+import { LabTaskStatus, Prisma } from '@prisma/client';
+import { normalizeTeeth } from '../treatments/treatment.utils';
+
+type TransactionClient = Prisma.TransactionClient;
+
+export async function generateLabCaseTasks(
+  tx: TransactionClient,
+  labCaseId: string,
+): Promise<number> {
+  const existingCount = await tx.labCaseTask.count({ where: { labCaseId } });
+  if (existingCount > 0) {
+    return 0;
+  }
+
+  const labCase = await tx.labCase.findUnique({
+    where: { id: labCaseId },
+    include: {
+      details: {
+        include: {
+          detail: {
+            select: { id: true, treatmentType: true, teeth: true },
+          },
+        },
+      },
+    },
+  });
+
+  if (!labCase?.details.length) {
+    return 0;
+  }
+
+  const treatmentTypeCodes = [...new Set(labCase.details.map((d) => d.detail.treatmentType))];
+
+  const labDependentTypes = await tx.treatmentType.findMany({
+    where: { code: { in: treatmentTypeCodes }, labDependent: true },
+    select: { id: true, code: true },
+  });
+
+  if (labDependentTypes.length === 0) {
+    return 0;
+  }
+
+  const labDependentCodes = new Set(labDependentTypes.map((t) => t.code));
+
+  const workflowSteps = await tx.treatmentWorkflowStep.findMany({
+    where: { treatmentTypeId: { in: labDependentTypes.map((t) => t.id) } },
+    orderBy: [{ treatmentTypeId: 'asc' }, { stepOrder: 'asc' }],
+    include: { treatmentType: { select: { code: true } } },
+  });
+
+  const stepsByTypeCode = new Map<string, { stepOrder: number; label: string }[]>();
+  for (const step of workflowSteps) {
+    const code = step.treatmentType.code;
+    const list = stepsByTypeCode.get(code) ?? [];
+    list.push({ stepOrder: step.stepOrder, label: step.label });
+    stepsByTypeCode.set(code, list);
+  }
+
+  const taskRows: Prisma.LabCaseTaskCreateManyInput[] = [];
+
+  for (const link of labCase.details) {
+    const detail = link.detail;
+    if (!labDependentCodes.has(detail.treatmentType)) {
+      continue;
+    }
+
+    const teeth = normalizeTeeth(detail.teeth);
+    const typeSteps = stepsByTypeCode.get(detail.treatmentType) ?? [];
+
+    for (const tooth of teeth) {
+      for (const step of typeSteps) {
+        taskRows.push({
+          labCaseId,
+          treatmentDetailId: detail.id,
+          tooth,
+          treatmentType: detail.treatmentType,
+          stepOrder: step.stepOrder,
+          stepLabel: step.label,
+          status: LabTaskStatus.PENDING,
+        });
+      }
+    }
+  }
+
+  if (taskRows.length === 0) {
+    return 0;
+  }
+
+  await tx.labCaseTask.createMany({ data: taskRows });
+  return taskRows.length;
+}