Initial commit

lib/shot-status.ts

@@ -0,0 +1,53 @@
+import { ShotStatus, TaskStatus } from "@prisma/client";
+import { db } from "@/lib/db";
+import type { PrismaClient } from "@prisma/client";
+
+type TxClient = Omit<
+  PrismaClient,
+  "$connect" | "$disconnect" | "$on" | "$transaction" | "$use" | "$extends"
+>;
+
+/**
+ * Derive the shot status from its tasks.
+ * Priority order (highest → lowest):
+ *   CHANGES on any task     → REVISIONS
+ *   INTERNAL_REVIEW / CLIENT_REVIEW on any task → IN_REVIEW
+ *   TODO / IN_PROGRESS on any task → IN_PROGRESS
+ *   All tasks DONE          → COMPLETE
+ *   No tasks                → WAITING
+ */
+export function deriveShotStatus(
+  tasks: { status: TaskStatus }[]
+): ShotStatus {
+  if (tasks.length === 0) return "WAITING";
+  if (tasks.some((t) => t.status === "CHANGES")) return "REVISIONS";
+  if (tasks.some((t) => t.status === "INTERNAL_REVIEW" || t.status === "CLIENT_REVIEW"))
+    return "IN_REVIEW";
+  if (tasks.some((t) => t.status === "TODO" || t.status === "IN_PROGRESS"))
+    return "IN_PROGRESS";
+  if (tasks.every((t) => t.status === "DONE")) return "COMPLETE";
+  return "WAITING";
+}
+
+/**
+ * Query all tasks for a shot, derive the correct ShotStatus, and persist it.
+ * Accepts an optional Prisma transaction client (tx) for use inside transactions.
+ */
+export async function recalcShotStatus(
+  shotId: string,
+  tx?: TxClient
+): Promise<void> {
+  const client = tx ?? db;
+
+  const tasks = await client.task.findMany({
+    where: { shotId },
+    select: { status: true },
+  });
+
+  const newStatus = deriveShotStatus(tasks);
+
+  await client.shot.update({
+    where: { id: shotId },
+    data: { status: newStatus },
+  });
+}