dbgate/packages/api/src/utility/useController.js

const _ = require('lodash');
const express = require('express');

/**
 * @param {string} route
 */
module.exports = function useController(app, route, controller) {
  const router = express.Router();

  if (controller._init) {
    console.log(`Calling init controller for controller ${route}`);
    try {
      controller._init();
    } catch (err) {
      console.log(`Error initializing controller, exiting application`, err);
      process.exit(1);
    }
  }

  for (const key of _.keys(controller)) {
    const obj = controller[key];
    if (!_.isFunction(obj)) continue;
    const meta = controller[`${key}_meta`];
    if (!meta) continue;

    let method = 'get';
    let raw = false;
    let rawParams = false;

    if (_.isString(meta)) {
      method = meta;
    }
    if (_.isPlainObject(meta)) {
      method = meta.method;
      raw = meta.raw;
      rawParams = meta.rawParams;
    }

    const route = `/${_.kebabCase(key)}`;
    if (raw) {
      router[method](route, controller[key]);
    } else {
      router[method](route, async (req, res) => {
        // if (controller._init && !controller._init_called) {
        //   await controller._init();
        //   controller._init_called = true;
        // }
        try {
          let params = [{ ...req.body, ...req.query }];
          if (rawParams) params = [req, res];
          const data = await controller[key](...params);
          res.json(data);
        } catch (e) {
          console.log(e);
          res.status(500).json({ error: e.message });
        }
      });
    }
  }

  app.use(route, router);
};