index.js

/**
 * Module dependencies.
 */

var responseTime = require('koa-response-time'),
    bodyParser   = require('koa-bodyparser'),
    ratelimit    = require('koa-ratelimit'),
    compress     = require('koa-compress'),
    winston      = require('winston'),
    logger       = require('koa-logger'),
    router       = require('koa-router'),
    load         = require('./lib/load'),
    redis        = require('redis'),
    koa          = require('koa');

/**
 * Environment.
 */

const env = process.env.NODE_ENV || 'development';

/**
 * Initialize an app with the given `opts`.
 *
 * @param {Object} opts
 * @return {Application}
 * @api public
 */

module.exports = function api (opts) {
  require('./lib/load/mongoose')();

  let _opts = opts || {},
      app = koa();

  app.use(require('koa-simple-qs')({
    depth: 10
  }));

  // logging
  if ( env !== 'test' ) {
    app.use(logger());
  }

  // x-response-time
  app.use(responseTime());

  // compression
  app.use(compress());

  // request body parsing
  app.use(bodyParser());

  // rate limiting
  if ( process.env.ENABLE_RATELIMITING ) {
    app.use(ratelimit({
      max: _opts.ratelimit,
      duration: _opts.duration,
      db: redis.createClient()
    }));
  }

  // routing
  app.use(router(app));

  if ( process.env.NODE_ENV === 'test' ) {
    app.use(function *(next) {
      try {
        yield next;
      } catch (err) {
        this.status = err.status || 500;
        if ( this.status > 499 ) {
          winston.error(err);
        }
        this.body = err.message;
        this.app.emit('error', err, this);
      }
    });
  }

  // boot
  load(app, __dirname + '/api');

  return app;
};