index.js

var http = require('http')

var HttpHashRouter = require('http-hash-router')
var catNames = require('cat-names')
var concat = require('concat-stream')
var pumpify = require('pumpify')
var corsify = require('corsify')
var debug = require('debug')('cat-lobby')

var limitStream = require('./limit-stream.js')

module.exports = function create (lobbyOpts) {
  if (!lobbyOpts) lobbyOpts = {}

  var router = HttpHashRouter()

  var state = {
    pings: {},
    pongs: {},
    timeouts: []
  }

  var utils = {
    makeName: makeName,
    rnd: rnd,
    uploadStream: uploadStream
  }

  // old alpha API
  router.set('/ping', deprecated)
  router.set('/pong/:name', deprecated)
  router.set('/ping/:name', deprecated)
  router.set('/pongs/:name', deprecated)

  var v1API = require('./versions/v1.js')(state, utils)

  router.set('/v1', v1API.create)
  router.set('/v1/:name/ping', v1API.ping)
  router.set('/v1/:name/pong', v1API.pong)
  router.set('/v1/:name/pings', v1API.pings)
  router.set('/v1/:name/pongs', v1API.pongs)

  return createServer(router)

  function deprecated (req, res, opts, cb) {
    cb(new Error('This version of ScreenCat is unsupported, please upgrade.'))
  }

  function uploadStream (cb) {
    var limiter = limitStream(1024 * 5) // 5kb max

    var concatter = concat(function concatted (buff) {
      cb(buff)
    })

    return pumpify(limiter, concatter)
  }

  function makeName () {
    var n = [utils.rnd(), utils.rnd(), utils.rnd()].join('-')
    if (state.pings[n]) return utils.makeName()
    return n
  }

  function rnd () {
    return catNames.random().toLowerCase().replace(/\s/g, '-')
  }

  function createServer (router) {
    var cors = corsify({
      'Access-Control-Allow-Methods': 'POST, GET'
    })

    var server = http.createServer(handler)

    function handler (req, res) {
      debug(req.url, 'request/response start')

      // redirect https
      if (req.headers['x-forwarded-proto'] === 'http') {
        var httpsURL = 'https://' + req.headers.host + req.url
        debug('https redirect', httpsURL)
        res.writeHead(302, {'Location': httpsURL })
        res.end()
        return
      }

      req.on('end', function logReqEnd () {
        debug(req.url, 'request end')
      })

      res.on('end', function logResEnd () {
        debug(req.url, 'response end')
      })

      cors(route)(req, res)

      function route (req, res) {
        router(req, res, {}, onError)
      }

      function onError (err) {
        if (err) {
          debug('error', {path: req.url, message: err.message})
          res.statusCode = err.statusCode || 500
          res.end(JSON.stringify({name: err.message}))
        }
      }
    }

    server.on('close', function closed () {
      // to prevent process from hanging open
      state.timeouts.forEach(function each (t) {
        clearTimeout(t)
      })
    })

    return server
  }
}