import { appRouter } from "./trpc/router";
// TODO: stop using universal-middleware and directly integrate server middlewares instead and/or use vike-server https://vike.dev/server. (Bati generates boilerplates that use universal-middleware https://github.com/magne4000/universal-middleware to make Bati's internal logic easier. This is temporary and will be removed soon.)
import {
  type Get,
  type UniversalHandler,
  env as getEnv,
} from "@universal-middleware/core";
import { fetchRequestHandler } from "@trpc/server/adapters/fetch";
import { getDb, getDbClient } from "../database/postgres";
import { getOpenrouter } from "./provider.js";
import { env as processEnv } from "./env.js";
import { getToken } from "@auth/core/jwt";

export const trpcHandler = ((endpoint) => (request, context, runtime) => {
  return fetchRequestHandler({
    endpoint,
    req: request,
    router: appRouter,
    async createContext({ req, resHeaders }) {
      const env = getEnv(runtime);
      const dbClient = getDbClient(
        (env.POSTGRES_CONNECTION_STRING ||
          processEnv.POSTGRES_CONNECTION_STRING) as string
      );
      const db = getDb(
        (env.POSTGRES_CONNECTION_STRING ||
          processEnv.POSTGRES_CONNECTION_STRING) as string
      );
      const openrouter = getOpenrouter(
        (env.OPENROUTER_API_KEY || processEnv.OPENROUTER_API_KEY) as string
      );
      const jwt = await getToken({ req: request, secret: "buginoo" });
      return {
        ...context,
        ...runtime,
        req,
        resHeaders,
        dbClient,
        db,
        openrouter,
        jwt,
      };
    },
    allowMethodOverride: true,
  });
}) satisfies Get<[endpoint: string], UniversalHandler>;