chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,91 @@
+import { parseWithZod } from "@conform-to/zod";
+import type { ActionFunction } from "@remix-run/node";
+import { json } from "@remix-run/node";
+import { assertExhaustive } from "@trigger.dev/core/utils";
+import { z } from "zod";
+import { prisma } from "~/db.server";
+import { redirectWithErrorMessage, redirectWithSuccessMessage } from "~/models/message.server";
+import { logger } from "~/services/logger.server";
+import { requireUserId } from "~/services/session.server";
+import { sanitizeRedirectPath } from "~/utils";
+import { runStore } from "~/v3/runStore.server";
+import { findBatchRunIdForUser } from "~/v3/services/batchRunAccess.server";
+import { ResumeBatchRunService } from "~/v3/services/resumeBatchRun.server";
+
+export const checkCompletionSchema = z.object({
+  redirectUrl: z.string(),
+});
+
+const ParamSchema = z.object({
+  batchId: z.string(),
+});
+
+export const action: ActionFunction = async ({ request, params }) => {
+  // Require a logged-in user; org membership is checked below before resuming.
+  const userId = await requireUserId(request);
+  const { batchId } = ParamSchema.parse(params);
+
+  const formData = await request.formData();
+  const submission = parseWithZod(formData, { schema: checkCompletionSchema });
+
+  if (submission.status !== "success") {
+    return json(submission.reply());
+  }
+
+  // Keep the post-action redirect same-origin.
+  const safeRedirectUrl = sanitizeRedirectPath(submission.value.redirectUrl);
+
+  // Only act on a batch in an org the caller belongs to. Accepts either the
+  // friendlyId or the internal id; both forms stay org-scoped.
+  const ownedBatchRunId = await findBatchRunIdForUser(prisma, runStore, batchId, userId);
+
+  if (!ownedBatchRunId) {
+    return redirectWithErrorMessage(safeRedirectUrl, request, "Batch not found");
+  }
+
+  try {
+    const resumeBatchRunService = new ResumeBatchRunService();
+    // Resume by the resolved internal id: the service looks up strictly by
+    // `{ id }`, so passing a friendlyId param would resolve to nothing.
+    const resumeResult = await resumeBatchRunService.call(ownedBatchRunId);
+
+    let message: string | undefined;
+
+    switch (resumeResult) {
+      case "ERROR": {
+        throw "Unknown error during batch completion check";
+      }
+      case "ALREADY_COMPLETED": {
+        message = "Batch already completed.";
+        break;
+      }
+      case "COMPLETED": {
+        message = "Batch completed and parent tasks resumed.";
+        break;
+      }
+      case "PENDING": {
+        message = "Child runs still in progress. Please try again later.";
+        break;
+      }
+      default: {
+        assertExhaustive(resumeResult);
+      }
+    }
+
+    return redirectWithSuccessMessage(safeRedirectUrl, request, message);
+  } catch (error) {
+    if (error instanceof Error) {
+      logger.error("Failed to check batch completion", {
+        error: {
+          name: error.name,
+          message: error.message,
+          stack: error.stack,
+        },
+      });
+      return redirectWithErrorMessage(safeRedirectUrl, request, error.message);
+    } else {
+      logger.error("Failed to check batch completion", { error });
+      return redirectWithErrorMessage(safeRedirectUrl, request, "Unknown error");
+    }
+  }
+};