@beaker/homebase/lib/server.js

An easy-to-administer hosting server for Hyperdrive.

const http = require('http')
const fs = require('fs')
const express = require('express')
const vhost = require('vhost')
const compression = require('compression')
const mkdirp = require('mkdirp')
const figures = require('figures')
const chalk = require('chalk')
const metrics = require('./metrics')
const vhostMethods = {
  hyperdrive: require('./vhosts/hyperdrive'),
  proxy: require('./vhosts/proxy'),
  redirect: require('./vhosts/redirect')
}
const packageJson = require('../package.json')

// constants
// =

const IS_DEBUG = (['debug', 'staging', 'test'].indexOf(process.env.NODE_ENV) !== -1)
const ENABLED = chalk.green(figures.tick + ' enabled')
const DISABLED = chalk.red(figures.cross + ' disabled')
const BULLET = chalk.dim(figures.play)

// globals
// =

var app
var activeRouter
var metricsServer

// exported api
// =

exports.start = async function (config, cb) {
  var server
  var plainServer

  mkdirp.sync(config.directory)

  console.log(`
${chalk.bold(`== Homebase ${packageJson.version} ==`)}

 ${BULLET} Directory:    ${config.directory}
 ${BULLET} Ports:        ${config.ports.http} ${chalk.dim('(HTTP)')}
 ${BULLET} HTTP mirror:  ${config.httpMirror ? ENABLED : DISABLED}
 ${BULLET} Dashboard:    ${config.dashboard ? ENABLED : DISABLED}
`)

  // create server app
  app = express()
  exports.configure(config)
  app.use(compression())
  app.use(function (req, res, next) {
    activeRouter(req, res, next)
  })
  app.use((err, req, res, next) => {
    console.log(err)
    res.status(500).end()
  })

  // start server
  server = http.createServer(app)
  server.listen(config.ports.http)
  if (plainServer) plainServer.on('error', onServerError)
  server.on('error', onServerError)
  function onServerError (err) {
    if (err.code === 'EACCES') {
      console.error(chalk.red(`ERROR: Failed to bind to ${chalk.bold(`port ${err.port}`)} (EACCES)

Make sure the ${chalk.bold(`port is not in use`)} and that you ${chalk.bold(`have permission`)} to use it.
See ${chalk.underline(`https://github.com/beakerbrowser/homebase/tree/master#port-setup-eacces-error`)}`))
      process.exit(1)
    }
    throw err
  }
  if (cb) {
    server.once('listening', cb)
  }

  // watch the config file for changes
  var watcher
  if (config.configPath) {
    var prevTime = 0
    watcher = fs.watch(config.configPath, function () {
      fs.lstat(config.configPath, function (_, st) {
        var now = Date.now()
        if (now - prevTime > 100) {
          console.log(`\nDetected change to ${config.configPath}, reloading...\n`)
          config.readFromFile()
          watcher.close()
          server.close(() => {
            exports.start(config)
          })
        }
        prevTime = now
      })
    })
  }

  return {
    config,
    close: cb => {
      if (watcher) watcher.close()
      server.close(cb)
    }
  }
}

exports.configure = function (config) {
  // create a new router
  activeRouter = express.Router()

  // add vhosts
  config.allVhosts.forEach(async vhostCfg => {
    var vhostServer = await vhostMethods[vhostCfg.vhostType].start(vhostCfg, config)
    vhostCfg.hostnames.forEach(hostname => activeRouter.use(vhost(hostname, vhostServer)))
  })

  // metrics server
  if (metricsServer) {
    metricsServer.close()
    metricsServer = null
  }
  if (config.dashboard) {
    metricsServer = http.createServer((req, res) => res.end(metrics.getMetrics()))
    metricsServer.listen(config.dashboard.port)
  }
}