123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
import type { HandlerFunc } from "./types.ts";import type { Context } from "./context.ts";
import { Node } from "./vendor/https/deno.land/x/router/mod.ts";import { hasTrailingSlash, NotFoundHandler } from "./util.ts";
export class Router {  trees: Record<string, Node> = {};
  add(method: string, path: string, h: HandlerFunc): void {    if (path[0] !== "/") {      path = `/${path}`;    }
    if (hasTrailingSlash(path)) {      path = path.slice(0, path.length - 1);    }
    let root = this.trees[method];    if (!root) {      root = new Node();      this.trees[method] = root;    }
    root.add(path, h);  }
  find(method: string, c: Context): HandlerFunc {    const node = this.trees[method];    let path = c.path;    if (hasTrailingSlash(path)) {      path = path.slice(0, path.length - 1);    }    let h: HandlerFunc | undefined;    if (node) {      const [handle, params] = node.find(path);      if (params) {        for (const [k, v] of params) {          c.params[k] = v;        }      }
      if (handle) {        h = handle as HandlerFunc;      }    }
    return h ?? NotFoundHandler;  }}