import { type NextRequest } from "next/server";
import {
  ChatHistory,
  ChatHistoryZod,
  convertChatHistoryList,
} from "@repo/shared-types";
import { ensureAuth } from "../ensureAuth";
import { z } from "zod";

export const runtime = "edge";

export async function POST(req: NextRequest) {
  const session = await ensureAuth(req);

  if (!session) {
    return new Response("Unauthorized", { status: 401 });
  }

  if (!process.env.BACKEND_SECURITY_KEY) {
    return new Response("Missing BACKEND_SECURITY_KEY", { status: 500 });
  }

  const url = new URL(req.url);

  const query = url.searchParams.get("q");
  const spaces = url.searchParams.get("spaces");

  const sourcesOnly = url.searchParams.get("sourcesOnly") ?? "false";

  const chatHistory = await req.json();

  if (!query || query.trim.length < 0) {
    return new Response(JSON.stringify({ message: "Invalid query" }), {
      status: 400,
    });
  }

  const validated = z
    .object({ chatHistory: z.array(ChatHistoryZod) })
    .safeParse(chatHistory ?? []);

  if (!validated.success) {
    return new Response(
      JSON.stringify({
        message: "Invalid chat history",
        error: validated.error,
      }),
      { status: 400 },
    );
  }

  const modelCompatible = await convertChatHistoryList(
    validated.data.chatHistory,
  );

  const resp = await fetch(
    `${process.env.BACKEND_BASE_URL}/api/chat?query=${query}&user=${session.user.id}&sourcesOnly=${sourcesOnly}&spaces=${spaces}`,
    {
      headers: {
        Authorization: `Bearer ${process.env.BACKEND_SECURITY_KEY}`,
        "Content-Type": "application/json",
      },
      method: "POST",
      body: JSON.stringify({
        chatHistory: modelCompatible,
      }),
    },
  );

  console.log("sourcesOnly", sourcesOnly);

  if (sourcesOnly == "true") {
    const data = await resp.json();
    console.log("data", data);
    return new Response(JSON.stringify(data), { status: 200 });
  }

  if (resp.status !== 200 || !resp.ok) {
    const errorData = await resp.text();
    console.log(errorData);
    return new Response(
      JSON.stringify({ message: "Error in CF function", error: errorData }),
      { status: resp.status },
    );
  }

  return new Response(resp.body, { status: 200 });
}