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,116 @@
+import { type ActionFunctionArgs, json } from "@remix-run/server-runtime";
+import { type CompleteWaitpointTokenResponseBody, stringifyIO } from "@trigger.dev/core/v3";
+import { WaitpointId } from "@trigger.dev/core/v3/isomorphic";
+import { z } from "zod";
+import {
+  $replica,
+  type PrismaReplicaClient,
+  runOpsNewReplica,
+  runOpsSplitReadEnabled,
+} from "~/db.server";
+import { env } from "~/env.server";
+import { processWaitpointCompletionPacket } from "~/runEngine/concerns/waitpointCompletionPacket.server";
+import { resolveWaitpointThroughReadThrough } from "~/runEngine/concerns/resolveWaitpointThroughReadThrough.server";
+import { verifyHttpCallbackHash } from "~/services/httpCallback.server";
+import { logger } from "~/services/logger.server";
+import { controlPlaneResolver } from "~/v3/runOpsMigration/controlPlaneResolver.server";
+import { engine } from "~/v3/runEngine.server";
+
+const paramsSchema = z.object({
+  waitpointFriendlyId: z.string(),
+  hash: z.string(),
+});
+
+export async function action({ request, params }: ActionFunctionArgs) {
+  if (request.method.toUpperCase() !== "POST") {
+    return json({ error: "Method not allowed" }, { status: 405, headers: { Allow: "POST" } });
+  }
+
+  const contentLength = request.headers.get("content-length");
+  if (!contentLength) {
+    return json({ error: "Content-Length header is required" }, { status: 411 });
+  }
+
+  if (parseInt(contentLength) > env.TASK_PAYLOAD_MAXIMUM_SIZE) {
+    return json({ error: "Request body too large" }, { status: 413 });
+  }
+
+  const { waitpointFriendlyId, hash } = paramsSchema.parse(params);
+  const waitpointId = WaitpointId.toId(waitpointFriendlyId);
+
+  try {
+    // Resolve wherever the waitpoint resides. The env is resolved below from the row; residency
+    // is classified off the waitpoint id, so env "" is fine. Fan-out reads the run-ops replica
+    // first, then the control-plane replica so both a co-located and a standalone token resolve,
+    // gated on the URL-presence read gate so the fan-out spans both DBs independent of the mint flag.
+    const waitpoint = await resolveWaitpointThroughReadThrough({
+      waitpointId,
+      environmentId: "",
+      read: (client: PrismaReplicaClient) =>
+        client.waitpoint.findFirst({
+          where: {
+            id: waitpointId,
+          },
+          select: { id: true, status: true, environmentId: true },
+        }),
+      deps: {
+        newClient: runOpsNewReplica,
+        legacyReplica: $replica,
+        splitEnabled: runOpsSplitReadEnabled,
+      },
+    });
+
+    if (!waitpoint) {
+      return json({ error: "Waitpoint not found" }, { status: 404 });
+    }
+
+    const environment = await controlPlaneResolver.resolveAuthenticatedEnv(waitpoint.environmentId);
+
+    if (!environment) {
+      return json({ error: "Waitpoint not found" }, { status: 404 });
+    }
+
+    if (
+      !verifyHttpCallbackHash(
+        waitpoint.id,
+        hash,
+        environment.parentEnvironment?.apiKey ?? environment.apiKey
+      )
+    ) {
+      return json({ error: "Invalid URL, hash doesn't match" }, { status: 401 });
+    }
+
+    if (waitpoint.status === "COMPLETED") {
+      return json<CompleteWaitpointTokenResponseBody>({
+        success: true,
+      });
+    }
+
+    // If the request body is not valid JSON, return an empty object
+    const body = await request.json().catch(() => ({}));
+
+    const stringifiedData = await stringifyIO(body);
+    const finalData = await processWaitpointCompletionPacket(
+      stringifiedData,
+      environment,
+      `${WaitpointId.toFriendlyId(waitpointId)}/http-callback`
+    );
+
+    const _result = await engine.completeWaitpoint({
+      id: waitpointId,
+      output: finalData.data
+        ? { type: finalData.dataType, value: finalData.data, isError: false }
+        : undefined,
+    });
+
+    return json<CompleteWaitpointTokenResponseBody>(
+      {
+        success: true,
+      },
+      { status: 200 }
+    );
+  } catch (error) {
+    logger.error("Failed to complete HTTP callback", { error });
+    throw json({ error: "Failed to complete HTTP callback" }, { status: 500 });
+  }
+}