app.js

'use strict'
/**
 * app.js
 * Entry point for the Badged App
 * (C) Habib Rehman
 ******************************/

// Init logger to override console
const logger = require('./script/logger')

const controllersPath = `${__dirname}/controllers`,
  badge = require(`${controllersPath}/badge`),
  home = require(`${controllersPath}/home`),
  {resolve} = require('path'),
  koa = require('koa'),
  responseTime = require('koa-response-time'),
  serve = require('koa-static'),
  compress = require('koa-compress'),
  klogger = require('koa-logger'),
  render = require('koa-ejs'),
  Router = require('koa-router'),
  _ = require('lodash'),
  MongoClient = require('mongodb').MongoClient

// Constants
const VIEW_PATH = `${__dirname}/views`,
  ENV = process.env.NODE_ENV || 'development',
  PORT = process.env.PORT || '4000',
  MONGODB_URI = process.env.MONGODB_URI,
  DEFAULT_DB_COLLECTION = 'downloads'

// Init
const app = new koa()
const router = new Router()
var _db

/**
 * Config
 */

// Normalize
function normalize (path) {
  return resolve(path.toString().toLowerCase())
}

// Ejs setup
render(app, {
  root: VIEW_PATH,
  viewExt: 'html',
  layout: false,
  cache: true,
  debug: true
})

// Middleware to protect against HTTP Parameter Pollution attacks
app.use(async (ctx, next) => {
  for (var param in ctx.query) {
    if (_.has(ctx.query, param)) {
      ctx.query[param] = _.isArray(ctx.query[param]) ? ctx.query[param][0] : ctx.query[param]
    }
  }
  await next()
})

// Middleware for normalizing request path
app.use(async (ctx, next) => {
  ctx.path = normalize(ctx.path)
  await next()
})

// logging
if ('test' != ENV) app.use(klogger())

// Mongodb logging
require('mongodb').Logger.setLevel('info')

// serve static files
app.use(serve(`${__dirname}/public`))

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

// Compress
app.use(compress())

// Add database to the context
app.use(async (ctx, next) => {
  ctx.db = _db
  ctx.downloads = _db.collection(DEFAULT_DB_COLLECTION)
  await next()
})

/**
 * Routes
 */
router.get('/', home.index)
router.get('/status', badge.status)
router.get('/:owner/:repo', badge.release)
router.get('/:owner/:repo/total', badge.total)
router.get('/:owner/:repo/:id', badge.releaseById)
router.get('/:owner/:repo/tags/:tag', badge.releaseByTag)

// catch all
router.get('/*', home.index)

app
  .use(router.routes())
  .use(router.allowedMethods())

MongoClient.connect(MONGODB_URI)
  .then((db) => {
    _db = db
    logger.info(`Connected to db!`)
    // Start server after connected to db
    app.listen(PORT, () => {
      logger.info(`listening on port ${PORT}`)
    })
  })
  .catch(function (err) {
    logger.error(err)
  })