import { createTRPCRateLimit } from "@fantasy/trpc-rate-limit";
import {initTRPC} from "@trpc/server";

const root = initTRPC.create();

const {
  procedure
} = root;

const rateLimiter = createTRPCRateLimiter({
  root,
	windowMs: 15 * 60 * 1000, // 15 minutes
	max: 100, // Limit each IP to 100 requests per `window` (here, per 15 minutes)
	headers: true, // Return rate limit info in the `RateLimit-*` headers
});


export const rateLimitedProcedure = procedure
  .use(rateLimiter.middleware());

/**
 * Inner function for `createContext` where we create the context.
 * This is useful for testing when we don't want to mock Next.js' request/response
 */
export async function createContextInner(_opts: CreateContextOptions) {
  return {};
}

export type Context = trpc.inferAsyncReturnType<typeof createContextInner>;

/**
 * Creates context for an incoming request
 * @link https://trpc.io/docs/context
 */
export async function createContext(
  opts: trpcNext.CreateNextContextOptions,
): Promise<Context> {
  // for API-response caching see https://trpc.io/docs/caching
  await rateLimiter.validate({
    info: opts.info,
  })

  return await createContextInner({});
}