dbgate/api/src/utility/useController.js

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

module.exports = function useController(app, route, controller) {
  const router = express.Router();

  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 }];
          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);
};