import { type NextRequest } from "next/server";
import {
	ChatHistory,
	ChatHistoryZod,
	convertChatHistoryList,
	SourcesFromApi,
} 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 jsonRequest = (await req.json()) as {
		chatHistory: ChatHistory[];
		sources: SourcesFromApi[] | undefined;
	};
	const { chatHistory, sources } = jsonRequest;

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

	const validated = 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);

	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,
				sources,
			}),
		},
	);

	if (sourcesOnly == "true") {
		const data = (await resp.json()) as SourcesFromApi;
		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 resp;
}