v3/packages/api/router.ts

import { addCORS } from './utils/api-tools';
import { uiPublicURI } from './utils/config';

export type Handler = (request: Request) => Promise<Response> | Response;

type RoutingTree = {
  [method: string]: {
    [path: string]: Handler;
  };
};

type Fallbacks = {
  root: Handler;
  404: Handler;
  500: Handler;
};

export class Router {
  private routingTree: RoutingTree = {};
  private fallbacks: Fallbacks = {
    root: this.respondToRoot,
    404: this.notFound,
    500: this.serverError,
  };

  private uiURL = new URL(uiPublicURI);

  addFallback(which: keyof Fallbacks, handler: Handler) {
    this.fallbacks[which] = handler;
  }

  add(method: string, rootPath: string, handler: Handler) {
    const lowerMethod = method.toLowerCase();

    if (!this.routingTree[lowerMethod]) {
      this.routingTree[lowerMethod] = {};
    }

    this.routingTree[lowerMethod][rootPath] = handler;
  }

  async handle(request: Request): Promise<Response> {
    const url = new URL(request.url);

    if (url.pathname === '/' || url.pathname === '') {
      return this.fallbacks.root(request);
    }
    const lowerMethod = request.method.toLowerCase();
    const rootPath = url.pathname.split('/')[1];
    const handler = this.routingTree[lowerMethod]?.[rootPath];

    if (handler) {
      try {
        const response = await handler(request);
        return response;
      } catch (e) {
        console.error(e);
        return this.fallbacks[500](request);
      }
    }

    if (lowerMethod === 'options') {
      return new Response(null, addCORS({}));
    }

    return this.fallbacks[404](request);
  }

  private respondToRoot(): Response {
    return new Response('Hi there!');
  }

  private notFound(): Response {
    return new Response(JSON.stringify({ error: 'not_found' }), {
      status: 404,
    });
  }

  private serverError(): Response {
    return new Response(JSON.stringify({ error: 'internal_server_error' }), {
      status: 500,
    });
  }
}