import type { ServedJSContext } from "jsr:@classic/js@0";
import type { Context } from "jsr:@hono/hono@^4.4.8";
import type { Env, Input, MiddlewareHandler } from "jsr:@hono/hono@^4.4.8/types";
import { createContext } from "./render.ts";
import type { Segment } from "./router.ts";
import type { JSXContextAPI, JSXContextInit } from "./types.ts";

const honoContext = createContext<Context>("hono");

export const $hono = (use: JSXContextAPI): Context => use(honoContext);

export const classicRouter = <E extends Env, P extends string, I extends Input>(
  router: Segment<never, never, undefined>,
  { context = () => [], ...opts }: {
    js: ServedJSContext;
    context?: (c: Context<E, P, I>) => JSXContextInit<unknown>[];
  },
): MiddlewareHandler<E, P, I> =>
async (c, next) =>
  await router.fetch(c.req.raw, {
    context: [honoContext.init(c), ...context(c)],
    ...opts,
  }) ??
    next();