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

  router.post('/enable', ctx => {
    handlers.enable && handlers.enable(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/disable', ctx => {
    handlers.disable && handlers.disable(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/conversation_added', ctx => {
    handlers.conversation_added && handlers.conversation_added(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/conversation_removed', ctx => {
    handlers.conversation_removed && handlers.conversation_removed(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/edit_subscription', ctx => {
    handlers.edit_subscription && handlers.edit_subscription(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/delete_subscription', ctx => {
    handlers.delete_subscription && handlers.delete_subscription(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/mention', ctx => {
    handlers.mention && handlers.mention(ctx.request.body || {}, context);
    ctx.body = { success: true };
  });

  router.post('/post', (() => {
    var _ref = _asyncToGenerator(function* (ctx) {
      if (!handlers.post) {
        ctx.body = { success: false, message: 'post endpoint not implemented' };
        return;
      }

      const params = ctx.request.body || {};

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

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

      try {
        const res = handlers.post(params, context);
        const payload = res && res.then ? yield res : res;
        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);
    };
  })());

  router.post('/preview', (() => {
    var _ref2 = _asyncToGenerator(function* (ctx) {
      if (!handlers.preview) {
        ctx.body = { success: false };
        return;
      }

      const params = ctx.request.body || {};

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

    return function (_x2) {
      return _ref2.apply(this, arguments);
    };
  })());

  return router;
};