import type { Method } from './method';
import { Route } from './route';
import type { Path, Pattern, Resolver } from './types';

export type Routes = Map<Method, Set<Route>>;

export class Router {
  #routes: Routes;

  constructor() {
    this.#routes = new Map<Method, Set<Route>>();
  }

  public add(
    method: Method,
    pattern: Pattern,
    resolver: Resolver,
    ...resolvers: Resolver[]
  ): this {
    const _resolvers = [resolver, ...resolvers];

    if (!this.#routes.has(method)) {
      this.#routes.set(method, new Set<Route>());
    }

    const routes = this.#routes.get(method)!;

    for (const _resolver of _resolvers) {
      routes.add(new Route(pattern, _resolver));
    }

    return this;
  }

  public get(method: string, path: Path): Route | null {
    for (const [_method, routes] of this.#routes) {
      if (_method === method) {
        for (const route of routes) {
          if (route.matches(path)) {
            return route;
          }
        }
      }
    }
    return null;
  }
}