chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,99 @@
|
||||
import type { ActionFunctionArgs } from "@remix-run/server-runtime";
|
||||
import { json } from "@remix-run/server-runtime";
|
||||
import { FinalizeDeploymentRequestBody } from "@trigger.dev/core/v3";
|
||||
import { z } from "zod";
|
||||
import { authenticateApiRequest } from "~/services/apiAuth.server";
|
||||
import { logger } from "~/services/logger.server";
|
||||
import { ServiceValidationError } from "~/v3/services/baseService.server";
|
||||
import { FinalizeDeploymentV2Service } from "~/v3/services/finalizeDeploymentV2.server";
|
||||
|
||||
const ParamsSchema = z.object({
|
||||
deploymentId: z.string(),
|
||||
});
|
||||
|
||||
export async function action({ request, params }: ActionFunctionArgs) {
|
||||
// Ensure this is a POST request
|
||||
if (request.method.toUpperCase() !== "POST") {
|
||||
return { status: 405, body: "Method Not Allowed" };
|
||||
}
|
||||
|
||||
const parsedParams = ParamsSchema.safeParse(params);
|
||||
|
||||
if (!parsedParams.success) {
|
||||
return json({ error: "Invalid params" }, { status: 400 });
|
||||
}
|
||||
|
||||
// Next authenticate the request
|
||||
const authenticationResult = await authenticateApiRequest(request);
|
||||
|
||||
if (!authenticationResult) {
|
||||
logger.info("Invalid or missing api key", { url: request.url });
|
||||
return json({ error: "Invalid or Missing API key" }, { status: 401 });
|
||||
}
|
||||
|
||||
const authenticatedEnv = authenticationResult.environment;
|
||||
|
||||
const { deploymentId } = parsedParams.data;
|
||||
|
||||
const rawBody = await request.json();
|
||||
const body = FinalizeDeploymentRequestBody.safeParse(rawBody);
|
||||
|
||||
if (!body.success) {
|
||||
return json({ error: "Invalid body", issues: body.error.issues }, { status: 400 });
|
||||
}
|
||||
|
||||
try {
|
||||
// Create a text stream chain
|
||||
const stream = new TransformStream();
|
||||
const encoder = new TextEncoderStream();
|
||||
const writer = stream.writable.getWriter();
|
||||
|
||||
const service = new FinalizeDeploymentV2Service();
|
||||
|
||||
// Chain the streams: stream -> encoder -> response
|
||||
const response = new Response(stream.readable.pipeThrough(encoder), {
|
||||
headers: {
|
||||
"Content-Type": "text/event-stream",
|
||||
"Cache-Control": "no-cache",
|
||||
Connection: "keep-alive",
|
||||
},
|
||||
});
|
||||
|
||||
const pingInterval = setInterval(() => {
|
||||
writer.write("event: ping\ndata: {}\n\n");
|
||||
}, 10000); // 10 seconds
|
||||
|
||||
service
|
||||
.call(authenticatedEnv, deploymentId, body.data, writer)
|
||||
.then(async () => {
|
||||
clearInterval(pingInterval);
|
||||
|
||||
await writer.write(`event: complete\ndata: ${JSON.stringify({ id: deploymentId })}\n\n`);
|
||||
await writer.close();
|
||||
})
|
||||
.catch(async (error) => {
|
||||
let errorMessage;
|
||||
|
||||
if (error instanceof ServiceValidationError) {
|
||||
errorMessage = { error: error.message };
|
||||
} else {
|
||||
logger.error("Error finalizing deployment", { error });
|
||||
errorMessage = { error: "Internal server error" };
|
||||
}
|
||||
|
||||
clearInterval(pingInterval);
|
||||
|
||||
await writer.write(`event: error\ndata: ${JSON.stringify(errorMessage)}\n\n`);
|
||||
await writer.close();
|
||||
});
|
||||
|
||||
return response;
|
||||
} catch (error) {
|
||||
if (error instanceof ServiceValidationError) {
|
||||
return json({ error: error.message }, { status: 400 });
|
||||
}
|
||||
|
||||
logger.error("Error finalizing deployment", { error });
|
||||
return json({ error: "Internal server error" }, { status: 500 });
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user