pmcelhaney / counterfact / 3596104599

import nodePath from "node:path";

import { pathToFileURL } from "node:url";

import yaml from "js-yaml";

import Koa from "koa";

import bodyParser from "koa-bodyparser";

import { koaSwagger } from "koa2-swagger-ui";

import Handlebars from "handlebars";

import { readFile } from "../util/read-file.js";

import { counterfact } from "./counterfact.js";

// eslint-disable-next-line no-underscore-dangle

const __dirname = nodePath.dirname(new URL(import.meta.url).pathname);

const DEFAULT_PORT = 3100;

Handlebars.registerHelper("escape_route", (route) =>

  route.replace(/[^\w/]/gu, "-")

);

function openapi(openApiPath, url) {

  return async (ctx, next) => {

    if (ctx.URL.pathname === "/counterfact/openapi") {

      const openApiDocument = await yaml.load(await readFile(openApiPath));

      openApiDocument.servers ??= [];

      openApiDocument.servers.unshift({

        description: "Counterfact",

        url,

      });

      // OpenApi 2 support:

      openApiDocument.host = url;

      // eslint-disable-next-line require-atomic-updates

      ctx.body = yaml.dump(openApiDocument);

      return;

    }

    // eslint-disable-next-line node/callback-return

    await next();

  };

}

function page(pathname, templateName, locals) {

  return async (ctx, next) => {

    const render = Handlebars.compile(

      await readFile(

        nodePath.join(__dirname, `../client/${templateName}.html.hbs`)

      )

    );

    if (ctx.URL.pathname === pathname) {

      // eslint-disable-next-line require-atomic-updates

      ctx.body = await render(locals);

      return;

    }

    // eslint-disable-next-line node/callback-return

    await next();

  };

}

export async function start({

  basePath = process.cwd(),

  port = DEFAULT_PORT,

  openApiPath = nodePath.join(basePath, "../openapi.yaml"),

}) {

  const app = new Koa();

  const { koaMiddleware, contextRegistry, registry } = await counterfact(

    basePath,

    openApiPath

  );

  app.use(openapi(openApiPath, `//localhost:${port}`));

  app.use(

    koaSwagger({

      routePrefix: "/counterfact/swagger",

      swaggerOptions: {

        url: "/counterfact/openapi",

      },

    })

  );

  app.use(

    page("/counterfact/", "index", {

      basePath,

      routes: registry.routes,

      methods: ["get", "post", "put", "delete", "patch"],

      openApiPath,

      openApiHref: openApiPath.includes("://")

        ? openApiPath

        : pathToFileURL(openApiPath).href,

    })

  );

  app.use(

    page("/counterfact/rapidoc", "rapi-doc", {

      basePath,

      routes: registry.routes,

    })

  );

  app.use(bodyParser());

  app.use(koaMiddleware);

  app.listen(port);

  return { contextRegistry };

}