hihat/lib/watchify-server.js

local Node/Browser development with Chrome DevTools

var ecstatic = require('ecstatic')
var Router = require('routes-router')
var http = require('http')
var Emitter = require('events/')
var defaultIndex = require('simple-html-index')
var fs = require('fs')
var path = require('path')
var isAbsolute = require('path-is-absolute')

module.exports = function (opts) {
  var handler = createHandler(opts)
  var server = http.createServer(handler.router)

  server.update = function (contents) {
    handler.pending = false
    handler.contents = contents
    handler.emit('update')
  }

  server.pending = function () {
    handler.pending = true
  }

  return server
}

function createHandler (opts) {
  var cwd = process.cwd()
  var basedir = opts.dir || cwd // where to serve
  var entries = opts.entries
  var staticHandler = ecstatic(basedir)
  var router = Router()

  // JS/HTML files are relative to CWD
  var htmlIndex = opts.index
  if (htmlIndex) {
    htmlIndex = isAbsolute(htmlIndex)
      ? htmlIndex
      : path.resolve(cwd, htmlIndex)
  }

  var emitter = new Emitter()
  emitter.router = router
  emitter.pending = false
  emitter.contents = ''

  if (entries.length > 0) {
    router.addRoute('/' + opts.serve, function (req, res) {
      if (emitter.pending) {
        emitter.once('update', function () {
          submit(req, res)
        })
      } else {
        submit(req, res)
      }
    })
  }

  router.addRoute('/index.html', generateIndex)
  router.addRoute('/', generateIndex)
  router.addRoute('*', staticHandler)

  return emitter

  function submit (req, res) {
    res.setHeader('content-type', 'application/javascript; charset=utf-8')
    res.setHeader('content-length', emitter.contents.length)
    res.statusCode = req.statusCode || 200

    res.end(emitter.contents)
  }

  function generateIndex (req, res) {
    res.setHeader('content-type', 'text/html; charset=utf-8')

    var stream
    if (htmlIndex) {
      stream = fs.createReadStream(htmlIndex)
    } else {
      stream = defaultIndex({
        title: 'hihat',
        entry: opts.serve
      })
    }
    stream.pipe(res)
  }
}