import { Hono } from '@hono/hono';
import { HTTPException } from '@hono/hono/http-exception';

import type { HonoOptions } from '@hono/hono/hono-base';
import type { DittoEnv } from './DittoEnv.ts';

/**
 * Ditto base route class.
 * Ensures that required variables are set for type safety.
 */
export class DittoRoute extends Hono<DittoEnv> {
  constructor(opts: HonoOptions<DittoEnv> = {}) {
    super(opts);

    this.use((c, next) => {
      this.assertVars(c.var);
      return next();
    });
  }

  private assertVars(vars: Partial<DittoEnv['Variables']>): DittoEnv['Variables'] {
    if (!vars.db) this.throwMissingVar('db');
    if (!vars.conf) this.throwMissingVar('conf');
    if (!vars.relay) this.throwMissingVar('relay');
    if (!vars.signal) this.throwMissingVar('signal');
    if (!vars.requestId) this.throwMissingVar('requestId');

    return {
      ...vars,
      db: vars.db,
      conf: vars.conf,
      relay: vars.relay,
      signal: vars.signal,
      requestId: vars.requestId,
    };
  }

  private throwMissingVar(name: string): never {
    throw new HTTPException(500, { message: `Missing required variable: ${name}` });
  }
}