123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
// Copyright 2022 Connor Logan. All rights reserved. MIT License.
// TODO: Add support for arrays of handlers  // TODO: Don't throw errors when using advanced URLPattern syntax (just use the// unknown type on the client when they use it instead)  
import { http } from "./deps.ts";import { requestData, NO_MATCH } from "./http.ts";
import type { RouterRequest, RouterShape, Handler } from "./client.ts";
/** * Handler that routes requests to Rpcs (endpoints) or other Stacks (routers). */export interface Stack<S extends RouterShape = RouterShape> {  (req: RouterRequest<S>, connInfo: http.ConnInfo): Promise<Response>;  /**   * The routes specified when this Stack was constructed.   */  readonly routes: S;}
const nextPathGroupName = "__nextPath";
/** * Constructs a new Stack handler using the provided routes object. Keys can be * a subset of the URLPattern syntax when group capturing is desired. See the * documentation for more information about how Stack routing works. TODO: the * documentation about how Stack routing works */export function stack<S extends RouterShape>(routes: S): Stack<S> {  // Stack routes can only use some of the features of URLPattern. If attempts  // are made to use features that aren't supported, throw an error  for (const [k, _] of Object.entries(routes)) {    const split = k.split("/").filter(v => !!v);
    // Trailing wildcard needs to be let through    if (split[split.length-1] === "*") {      split.pop();    }
    // Make sure the stack route is simple    for (const s of split) {      if (        !s.match(/^:[a-zA-Z_$][a-zA-Z_$0-9]$/) &&        !s.match(/^[^:*?(){}]*$/) // TODO: I don't think this is fully correct      ) {        throw new SyntaxError(          `"${k}" isn't a valid stack route. Stack routes only support basic path segments and named path segments (groups). Non-trailing wildcards, RegExp, optionals, and other advanced URLPattern syntax is not supported`,        );      }    }  }
  const handlers: Record<string, Handler | Handler[]> = {};  for (const k of Object.keys(routes)) {    const v = routes[k];    if (v && typeof v === "object" && !Array.isArray(v)) {      handlers[k] = stack(v);    } else if (v) {      handlers[k] = v;    }  }
  // Sort the handlers like this:  //   1. Solo wildcards are always last  //   2. By path depth. Paths with more path segments get tested sooner.  //      (Trailing wildcards don't affect path depth)  //   3. For two paths that have the same depth, index order is used (lower  //      index === higher priority)  const paths = Object.keys(handlers);  const sortedPaths: string[] = paths.sort((a, b) => {    if (a === b) {      return 0;    }        // #1    if (a === "*" || a === "/*") {      return 1;    }    if (b === "*" || b === "/*") {      return -1;    }
    // #2    if (a.endsWith("/*")) {      a = a.slice(0, a.length - 2);    }    if (b.endsWith("/*")) {      b = b.slice(0, b.length - 2);    }    const la = a.split("/").filter(v => !!v).length;    const lb = b.split("/").filter(v => !!v).length;    if (la !== lb) {      return lb - la;    }
    // #3    return paths.indexOf(a) - paths.indexOf(b);  });
  const patterns = new Map<URLPattern, Handler | Handler[]>();  for (const op of sortedPaths) {    let p = "/" + op.split("/").filter(v => !!v).join("/");
    // Sometimes the path might already end in a wildcard. If it does, remove it    // before adding the path capture wildcard    if (p.endsWith("/*")) {      p = p.slice(0, p.length - 2);    }    p = `${p}/:${nextPathGroupName}*/*?`;    patterns.set(new URLPattern(p, "http://_._"), handlers[op]);  }
  const stackHandler = async (    req: Request,    conn: http.ConnInfo,  ): Promise<Response> => {    const data = requestData(req);    if (data instanceof Response) { // Handle malformed path redirect      return data;    }
    for (const [pattern, handler] of patterns.entries()) {      const match = pattern.exec(data.path, "http://_._");      if (!match) {        continue;      }
      const groups = { ...data.groups, ...match.pathname.groups };      const path = `/${groups[nextPathGroupName] || ""}`;      delete groups[nextPathGroupName];
      const oPath = data.path;      const oGroups = data.groups;
      // The data object is only created once for every request, therefore      // modifications to the data object will be preserved across handling      // contexts      Object.assign(data, { path, groups });
      try {        if (Array.isArray(handler)) {          for (const h of handler) {            try {              return await h(req, conn);            } catch (e) {              if (e === NO_MATCH) {                continue;              }            }          }          throw NO_MATCH;        }        return await handler(req, conn);      } catch (e) {        // Keep looking for matches if it's the NO_MATCH error. All other errors        // bubble        if (e === NO_MATCH) {          continue;        }        throw e;      } finally {        // Before moving on, put the path and groups back to their original        // state        Object.assign(req, { path: oPath, groups: oGroups });      }    }
    // When nothing matches, throw NO_MATCH    throw NO_MATCH;  };
  return Object.assign(stackHandler, { routes });}