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,65 @@
+import { type ActionFunctionArgs, json } from "@remix-run/server-runtime";
+import { type TaskRun } from "@trigger.dev/database";
+import { z } from "zod";
+import { prisma } from "~/db.server";
+import { runStore } from "~/v3/runStore.server";
+import { logger } from "~/services/logger.server";
+import { requireAdminApiRequest } from "~/services/personalAccessToken.server";
+import { getRunsReplicationGlobal } from "~/services/runsReplicationGlobal.server";
+import { runsReplicationInstance } from "~/services/runsReplicationInstance.server";
+import { FINAL_RUN_STATUSES } from "~/v3/taskStatus";
+
+const Body = z.object({
+  runIds: z.array(z.string()),
+});
+
+const MAX_BATCH_SIZE = 50;
+
+export async function action({ request }: ActionFunctionArgs) {
+  await requireAdminApiRequest(request);
+
+  try {
+    const body = await request.json();
+    const { runIds } = Body.parse(body);
+
+    logger.info("Backfilling runs", { runIds });
+
+    const runs: TaskRun[] = [];
+    for (let i = 0; i < runIds.length; i += MAX_BATCH_SIZE) {
+      const batch = runIds.slice(i, i + MAX_BATCH_SIZE);
+      const batchRuns = await runStore.findRuns(
+        {
+          where: {
+            id: { in: batch },
+            status: {
+              in: FINAL_RUN_STATUSES,
+            },
+          },
+        },
+        prisma
+      );
+      runs.push(...batchRuns);
+    }
+
+    const service = getRunsReplicationGlobal() ?? runsReplicationInstance;
+    if (!service) {
+      throw new Error("Runs replication instance not found");
+    }
+
+    await service.backfill(
+      runs.map((run) => ({
+        ...run,
+        masterQueue: run.workerQueue,
+      }))
+    );
+
+    logger.info("Backfilled runs", { runs });
+
+    return json({
+      success: true,
+      runCount: runs.length,
+    });
+  } catch (error) {
+    return json({ error: error instanceof Error ? error.message : error }, { status: 400 });
+  }
+}