import type { TSchema } from "@sinclair/typebox";
import { TypeCompiler } from "@sinclair/typebox/compiler";
import { initTRPC, TRPCError } from "@trpc/server";
import type { getDb } from "../database/postgres";
import type { getOpenrouter } from "../pages/chat/provider";

/**
 * Initialization of tRPC backend
 * Should be done only once per backend!
 */
const t = initTRPC
  .context<
    object & {
      db: ReturnType<typeof getDb>;
      openrouter: ReturnType<typeof getOpenrouter>;
    }
  >()
  .create(/*{
  sse: {
    maxDurationMs: 5 * 60 * 1_000, // 5 minutes
    ping: {
      enabled: true,
      intervalMs: 3_000,
    },
    client: {
      reconnectAfterInactivityMs: 5_000,
    },
  },
}*/);

/**
 * Export reusable router and procedure helpers
 * that can be used throughout the router
 */
export const router = t.router;
export const publicProcedure = t.procedure;

/**
 * Generate a TRPC-compatible validator function given a Typebox schema.
 * This was copied from [https://github.com/sinclairzx81/typebox/blob/6cfcdc02cc813af2f1be57407c771fc4fadfc34a/example/trpc/readme.md].
 * @param schema A Typebox schema
 * @returns A TRPC-compatible validator function
 */
export function Validator<T extends TSchema>(schema: T) {
  const check = TypeCompiler.Compile(schema);
  return (value: unknown) => {
    if (check.Check(value)) return value;
    const err = check.Errors(value).First();
    throw new TRPCError({
      message: `${err?.message} for ${err?.path}`,
      code: "BAD_REQUEST",
    });
  };
}

export const createCallerFactory = t.createCallerFactory;