wehub-resource-sync
80 lines
2.7 KiB
TypeScript
80 lines
2.7 KiB
TypeScript
import type { TypedResponse } from "@remix-run/server-runtime";
|
|
import { json } from "@remix-run/server-runtime";
|
|
import type { WaitForDurationResponseBody } from "@trigger.dev/core/v3";
|
|
import { WaitForDurationRequestBody } from "@trigger.dev/core/v3";
|
|
import { RunId } from "@trigger.dev/core/v3/isomorphic";
|
|
|
|
import { z } from "zod";
|
|
import { prisma } from "~/db.server";
|
|
import { logger } from "~/services/logger.server";
|
|
import { createActionApiRoute } from "~/services/routeBuilders/apiBuilder.server";
|
|
import { resolveIdempotencyKeyTTL } from "~/utils/idempotencyKeys.server";
|
|
import { engine } from "~/v3/runEngine.server";
|
|
import { runStore } from "~/v3/runStore.server";
|
|
|
|
const { action } = createActionApiRoute(
|
|
{
|
|
body: WaitForDurationRequestBody,
|
|
params: z.object({
|
|
runFriendlyId: z.string(),
|
|
}),
|
|
method: "POST",
|
|
},
|
|
async ({ authentication, body, params }): Promise<TypedResponse<WaitForDurationResponseBody>> => {
|
|
const { runFriendlyId } = params;
|
|
const runId = RunId.toId(runFriendlyId);
|
|
|
|
try {
|
|
const run = await runStore.findRun(
|
|
{
|
|
id: runId,
|
|
runtimeEnvironmentId: authentication.environment.id,
|
|
},
|
|
prisma
|
|
);
|
|
|
|
if (!run) {
|
|
throw new Response("You don't have permissions for this run", { status: 401 });
|
|
}
|
|
|
|
const idempotencyKeyExpiresAt = body.idempotencyKeyTTL
|
|
? resolveIdempotencyKeyTTL(body.idempotencyKeyTTL)
|
|
: undefined;
|
|
|
|
const { waitpoint } = await engine.createDateTimeWaitpoint({
|
|
// Co-locate the waitpoint with the run that blocks on it (run-ops split): a run-ops run lives
|
|
// on the dedicated DB, but the minted waitpoint id is always a cuid, so without the run id
|
|
// the waitpoint would route to the control-plane DB and the block edge would never resolve.
|
|
runId: run.id,
|
|
projectId: authentication.environment.project.id,
|
|
environmentId: authentication.environment.id,
|
|
completedAfter: body.date,
|
|
idempotencyKey: body.idempotencyKey,
|
|
idempotencyKeyExpiresAt: idempotencyKeyExpiresAt,
|
|
});
|
|
|
|
const _waitResult = await engine.blockRunWithWaitpoint({
|
|
runId: run.id,
|
|
waitpoints: waitpoint.id,
|
|
projectId: authentication.environment.project.id,
|
|
organizationId: authentication.environment.organization.id,
|
|
});
|
|
|
|
return json({
|
|
waitUntil: body.date,
|
|
waitpoint: {
|
|
id: waitpoint.friendlyId,
|
|
},
|
|
});
|
|
} catch (error) {
|
|
logger.error("Failed to wait for duration dev", {
|
|
environmentId: authentication.environment.id,
|
|
error,
|
|
});
|
|
throw error;
|
|
}
|
|
}
|
|
);
|
|
|
|
export { action };
|