app.js

"use strict";

var DockerWatcher = require('./lib/docker-watcher')
  , DockerContainer = require('./lib/docker-container')
  , ContainersHelper = require('./lib/containers-helper')
  , url = require('url')
  , Network = require('./lib/network')
  , HTTPProxy = require('./lib/http-proxy')
  , Logger = require('./lib/logger')
  , DNSServer = require('./lib/dns-server').DNSServer
  , exec = require('child_process').exec

require('colors')

/**
 * App
 *
 * ProxyDriver, domain, dns_ip, dns_port
 *
 * @constructor
 */
var App = function (ProxyDriver, HTTPDriver, DNSDriver, DockerDriver, domain, api_port, proxy_ip, dns_ip, dns_port, dns_ttl) {

  this.dockerInfos = url.parse(process.env.DOCKER_HOST)
  this.proxyDriver = ProxyDriver
  this.httpDriver = HTTPDriver
  this.dnsDriver = DNSDriver
  this.dockerDriver = DockerDriver
  this.domain = domain
  this.apiPort = api_port
  this.proxyIp = proxy_ip
  this.dnsIp = dns_ip
  this.dnsPort = dns_port
  this.dnsTtl = dns_ttl
  this.version = require('./package.json').version

  // DNS stuff
  this.dnsServer = new DNSServer(this, this.dnsDriver, domain);

  // proxy
  this.proxy = new HTTPProxy(this, this.proxyDriver, this.httpDriver, this.dnsServer)
}

/**
 * Return docker infos
 * @returns {Object}
 */
App.prototype.getDockerInfos = function () {
  return this.dockerInfos
}

/**
 * Run the app
 */
App.prototype.run = function () {

  Logger.info("Starting Muguet App".green)

  Network.setupResolver(this.domain, this.dnsPort, this.dnsIp)

  this.dnsServer.listen(this.dnsPort)
  var watcher = new DockerWatcher(this).run()

  watcher
    .on('setup', this._onContainerSetup.bind(this))
    .on('change', this._onContainerChange.bind(this))
    .on('error', this._onError.bind(this))
}

/**
 * Get proxy IP address
 *
 * @returns {String}
 */
App.prototype.getProxyIp = function () {
  return this.proxyIp
}

/**
 * Get domain
 *
 * @returns {String}
 */
App.prototype.getDomain = function () {
  return this.domain
}

/**
 * Callback on the first containers list received
 *
 * @param {Array} containers
 * @private
 */
App.prototype._onContainerSetup = function (containers) {
  Logger.info('Containers list received from Docker-watcher: %s containers found', String(containers.length).yellow)
  this.containers = containers
  this.dnsServer.setEntries(this.getDnsEntries())
  this.proxy.setRoutes(this.getProxyRoutes())
  this.proxy.listen(this.apiPort)
}

/**
 * Callback on containers list changes
 *
 * @param {Array} containers
 * @private
 */
App.prototype._onContainerChange = function (containers) {
  Logger.info("Containers have changed: %s running", String(containers.length).yellow)
  this.containers = containers
  this.dnsServer.setEntries(this.getDnsEntries())
  this.proxy.updateRoutes(this.getProxyRoutes())
}

/**
 * Error callback
 *
 * @param {String} err
 * @private
 */
App.prototype._onError = function (err) {
  exec('boot2docker status', function (error, stdout) {
    if (stdout.trim() === 'poweroff') {
      return Logger.error("Your boot2docker VM does not seem to be started. Please run `boot2docker up`.");
    }
    Logger.error("Error while watching containers. (" + err + ")")
  })
}


/**
 * Build proxy routes for all container
 *
 * @private
 */
App.prototype.getProxyRoutes = function () {

  var flags = DockerContainer.IS_RUNNING | DockerContainer.HAS_PROXY_ENABLED | DockerContainer.HAS_EXPOSED_PORTS;

  var routes = ContainersHelper.filterContainers(this.containers, flags).map(function (cnt) {
    return cnt.getProxyRoutes()
  })

  if (routes.length) {
    routes = routes.reduce(function (prev, next) {
      return prev.concat(next)
    })
  }
  return routes
}
/**
 * Build dns entries
 *
 * @private
 */
App.prototype.getDnsEntries = function () {

  var finalEntries = {}
    , entries = this.containers.map(function (cnt) {
      return cnt.getDnsEntries()
    })

  entries.forEach(function (entry) {
    for (var n in entry) {
      if (entry.hasOwnProperty(n)) {
        finalEntries[n] = (finalEntries[n] || []).concat(entry[n])
      }
    }
  })

  return finalEntries
}

module.exports = App