import type { PrismaClient } from "~/db.server";
import { prisma } from "~/db.server";
import { workerQueue } from "../worker.server";
import { requestUrl } from "~/utils/requestUrl.server";
import { RuntimeEnvironmentType } from "@trigger.dev/database";

export class HandleHttpSourceService {
  #prismaClient: PrismaClient;

  constructor(prismaClient: PrismaClient = prisma) {
    this.#prismaClient = prismaClient;
  }

  public async call(id: string, request: Request) {
    const url = requestUrl(request);

    const triggerSource = await this.#prismaClient.triggerSource.findUnique({
      where: { id },
      include: {
        endpoint: true,
        environment: true,
        secretReference: true,
      },
    });

    if (!triggerSource) {
      return { status: 404 };
    }

    if (!triggerSource.active) {
      return { status: 200 };
    }

    if (!triggerSource.interactive) {
      await this.#prismaClient.$transaction(async (tx) => {
        // Create a request delivery and then enqueue it to be delivered
        const delivery = await tx.httpSourceRequestDelivery.create({
          data: {
            sourceId: triggerSource.id,
            endpointId: triggerSource.endpointId,
            environmentId: triggerSource.environmentId,
            url: url.href,
            method: request.method,
            headers: Object.fromEntries(request.headers),
            body: ["POST", "PUT", "PATCH"].includes(request.method)
              ? Buffer.from(await request.arrayBuffer())
              : undefined,
          },
        });

        await workerQueue.enqueue(
          "deliverHttpSourceRequest",
          {
            id: delivery.id,
          },
          {
            queueName: `deliver:${triggerSource.id}`,
            tx,
            maxAttempts:
              triggerSource.environment.type === RuntimeEnvironmentType.DEVELOPMENT ? 1 : undefined,
          }
        );
      });

      return { status: 200 };
    }

    // TODO: implement interactive webhooks

    return { status: 200 };
  }
}