123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223
/** * tool/router.ts * An flex router for fen * @author DominicMing */
import { Controller, IContext } from "../server.ts";
/** * extract params from route with number * @param target * @param template */export function extractParams(target: string, template: string) {  let t1 = target.split("/"),    t2 = template.split("/");  const result = {} as {    [key: string]: string;  };
  t2.map((el, i) => {    if (el.startsWith(":")) {      result[el.substring(1)] = decodeURIComponent(t1[i]);    }  });
  return result;}
/** * Route type used in use method */interface IRoute {  [path: string]: {    [method: string]: (context: IContext) => Promise<void>;  };}
/** * router himself */export class Router {  constructor(name = "") {    this.name = name;  }
  pool = new Map<string, any>();
  name = "";
  appendRoute(method: string, route: string, controller: Controller) {    let pool = this.pool;    let _route = route.startsWith("/") ? route : "/" + route;    let routeArr = _route.split("/");    let routerName = this.name;
    for (let r of routeArr) {      if (r) {        if (r.startsWith(":")) {          r = ":";        }
        if (!pool.has(r)) {          pool.set(r, new Map());        }
        pool = pool.get(r);      }    }
    pool.set("__" + method + "__", {      route: _route,      originalRoute: _route,      method,      controller,      name: routerName,    });  }
  private getRoute(path: string, method: string) {    let pool = this.pool;    let pathArr = path.split("/");
    for (const p of pathArr) {      if (p) {        if (pool.has(p)) {          pool = pool.get(p);        } else if (pool.has(":")) {          pool = pool.get(":");        } else {          return null;        }      }    }
    if (pool.has("__" + method + "__")) {      return pool.get("__" + method + "__");    }
    return null;  }
  controller = async (context: IContext) => {    let r = this.getRoute(context.path, context.method);
    if (r) {      const { route, controller, name } = r;      const params = extractParams(context.path, route);
      context.data.set("router", {        route,        params,        name,      });
      await controller(context);    } else {      context.throw(404, "Not Found Route");    }  };
  use(route: IRoute) {    for (const r in route) {      for (const method in route[r]) {        if (route[r].hasOwnProperty(method)) {          this.appendRoute(method.toUpperCase(), r, route[r][method]);        }      }    }
    return this;  }
  merge(route: string, router: Router) {    let pool = this.pool;    let _route = route.startsWith("/") ? route.slice(1) : route;    const routeArr = _route.split("/");
    for (let r of routeArr) {      if (r) {        if (r.startsWith(":")) {          r = ":";        }
        if (!pool.has(r)) {          pool.set(r, new Map());        }
        pool = pool.get(r);      }    }
    const changeQ: Map<string, any>[] = [];
    for (const [key, val] of router.pool.entries()) {      pool.set(key, val);      if (val instanceof Map) {        changeQ.push(val);      } else if (val) {        val.route = route + val.route;      }    }
    while (changeQ.length > 0) {      let change = changeQ.pop();
      if (!change) {        continue;      }
      for (const [key, val] of change.entries()) {        if (val instanceof Map) {          changeQ.push(val);        } else if (val) {          val.route = route + val.route;        }      }    }
    return this;  }
  get(route: string, controller: Controller) {    this.appendRoute("GET", route, controller);    return this;  }
  post(route: string, controller: Controller) {    this.appendRoute("POST", route, controller);    return this;  }
  head(route: string, controller: Controller) {    this.appendRoute("HEAD", route, controller);    return this;  }
  put(route: string, controller: Controller) {    this.appendRoute("PUT", route, controller);    return this;  }
  delete(route: string, controller: Controller) {    this.appendRoute("DELETE", route, controller);    return this;  }
  connect(route: string, controller: Controller) {    this.appendRoute("CONNECT", route, controller);    return this;  }
  options(route: string, controller: Controller) {    this.appendRoute("OPTION", route, controller);    return this;  }
  trace(route: string, controller: Controller) {    this.appendRoute("TRACE", route, controller);    return this;  }}