@mainframe/bot-sdk/lib/router.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.createRouter = undefined;

var _koaBodyparser = require('koa-bodyparser');

var _koaBodyparser2 = _interopRequireDefault(_koaBodyparser);

var _koaRouter = require('koa-router');

var _koaRouter2 = _interopRequireDefault(_koaRouter);

var _log = require('./log');

var _log2 = _interopRequireDefault(_log);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new Promise(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then(function (value) { step("next", value); }, function (err) { step("throw", err); }); } } return step("next"); }); }; }

const createRouter = exports.createRouter = (handlers, context) => {
  (0, _log2.default)('create router with handlers: %s', Object.keys(handlers).join(', '));

  const router = new _koaRouter2.default();
  router.use((0, _koaBodyparser2.default)({ enableTypes: ['json'] }));

  if (handlers.enable) {
    router.post('/enable', ctx => {
      handlers.enable && handlers.enable(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.disable) {
    router.post('/disable', ctx => {
      handlers.disable && handlers.disable(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.conversation_added) {
    router.post('/conversation_added', ctx => {
      handlers.conversation_added && handlers.conversation_added(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.conversation_removed) {
    router.post('/conversation_removed', ctx => {
      handlers.conversation_removed && handlers.conversation_removed(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.edit_subscription) {
    router.post('/edit_subscription', ctx => {
      handlers.edit_subscription && handlers.edit_subscription(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.delete_subscription) {
    router.post('/delete_subscription', ctx => {
      handlers.delete_subscription && handlers.delete_subscription(ctx.request.body || {}, context);
      ctx.status = 204;
    });
  }

  if (handlers.post) {
    router.post('/post', (() => {
      var _ref = _asyncToGenerator(function* (ctx) {
        const params = ctx.request.body || {};

        let missingParam;
        if (!params.user_id) {
          missingParam = 'user_id';
        } else if (!params.data) {
          missingParam = 'data';
        } else if (!params.context) {
          missingParam = 'context';
        }

        if (missingParam) {
          ctx.status = 400;
          ctx.body = `Missing ${missingParam}`;
          return;
        }

        try {
          const postResult = handlers.post ? handlers.post(params, context) : { success: false };
          const payload = postResult && postResult.then ? yield postResult : postResult;
          ctx.body = payload || { success: true };
        } catch (err) {
          (0, _log2.default)('error handling post request', params, err);
          ctx.status = 500;
        }
      });

      return function (_x) {
        return _ref.apply(this, arguments);
      };
    })());
  }

  return router;
};