import { ActionFunctionArgs, 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 });
  }
}
