api.js

const http = require('http');
const https = require('https');
const Koa = require('koa');
const compress = require('koa-compress');
const responseTime = require('koa-response-time');
const rateLimit = require('koa-simple-ratelimit');
const logger = require('koa-logger');
const bodyParser = require('koa-bodyparser');
const json = require('koa-json');
const errorHandler = require('koa-better-error-handler');
const removeTrailingSlashes = require('koa-no-trailing-slash');
const redis = require('redis');
const StoreIPAddress = require('@ladjs/store-ip-address');

const { Timeout } = require('./helpers');
const helpers = require('./helpers');
const config = require('./config');
const routes = require('./routes');

// initialize mongoose
const mongoose = new helpers.Mongoose();

// connect to redis
const redisClient = redis.createClient(config.redis);
redisClient.on('connect', () => helpers.logger.info('redis connected'));
redisClient.on('error', helpers.logger.error);

// initialize the app
const app = new Koa();

// store the server initialization
// so that we can gracefully exit
// later on with `server.close()`
let server;

app.on('error', helpers.logger.contextError);
app.on('log', helpers.logger.log);

// setup localization
app.use(helpers.i18n.middleware);

// trust proxy
app.proxy = true;

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

// override koa's undocumented error handler
app.context.onerror = errorHandler;

// specify that this is our api (used by error handler)
app.context.api = true;

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

// add the logger for development environment only
if (config.env === 'development') app.use(logger());

// rate limiting
app.use(
  rateLimit({
    ...config.rateLimit,
    db: redisClient
  })
);

// remove trailing slashes
app.use(removeTrailingSlashes);

// body parser
app.use(bodyParser());

// pretty-printed json responses
app.use(json());

// add context helpers
app.use(helpers.contextHelpers);

// auth
app.use(helpers.passport.initialize());

// configure timeout
app.use(async (ctx, next) => {
  try {
    const timeout = new Timeout({
      ms: config.apiRequestTimeoutMs,
      message: ctx.translate('REQUEST_TIMED_OUT')
    });
    await timeout.middleware(ctx, next);
  } catch (err) {
    ctx.throw(err);
  }
});

// store the user's last ip address in the background
app.use(new StoreIPAddress({ logger: helpers.logger }).middleware);

// mount the app's defined and nested routes
app.use(routes.api.routes());

// custom 404 handler since it's not already built in
app.use(helpers._404Handler);

// start server on either http or https
if (config.protocols.api === 'http') server = http.createServer(app.callback());
else server = https.createServer(config.ssl.api, app.callback());

if (!module.parent)
  server = server.listen(config.ports.api, () =>
    helpers.logger.info(`api server listening on ${config.ports.api}`)
  );

// handle process events and graceful restart
helpers.graceful(server, redisClient, mongoose);

module.exports = server;