embark-area-51/lib/modules/webserver/server.js

let finalhandler = require('finalhandler');
let http = require('http');
let serveStatic = require('serve-static');
const {canonicalHost, defaultHost, dockerHostSwap} = require('../../utils/host');
require('http-shutdown').extend();
var express = require('express');
let path = require('path');
var expressWebSocket = require('express-ws');

class Server {
  constructor(options) {
    this.events = options.events;
    this.dist = options.dist || 'dist/';
    this.port = options.port || 8000;
    this.hostname = dockerHostSwap(options.host) || defaultHost;
  }

  start(callback) {
    const self = this;
    if (this.server && this.server.listening) {
      let message = __("a webserver is already running at") + " " +
                      ("http://" + canonicalHost(this.hostname) +
                      ":" + this.port).bold.underline.green;
      return callback(null, message);
    }
    let serve = serveStatic(this.dist, {'index': ['index.html', 'index.htm']});

    var app = express();

    app.use(serve);
    app.use('/embark', serveStatic(path.join(__dirname, 'backend'), {'backend': ['index.html', 'index.htm']}));

    expressWebSocket(app);

    app.ws('/embark/logs', function(ws, req) {
      self.events.on("log", function(logLevel, logMsg) {
        ws.send(JSON.stringify({msg: logMsg, msg_clear: logMsg.stripColors, logLevel: logLevel}), () => {});
      });
    });

    app.ws('/embark/dashboard', function(ws, req) {
      let dashboardState = { contractsState: [], environment: "", status: "", availableServices: [] };

      // TODO: doesn't feel quite right, should be refactored into a shared
      // dashboard state
      self.events.request('setDashboardState');

      self.events.on('contractsState', (contracts) => {
        dashboardState.contractsState = [];

        contracts.forEach(function (row) {
          dashboardState.contractsState.push({contractName: row[0], address: row[1], status: row[2]});
        });
        ws.send(JSON.stringify(dashboardState));
      });
      self.events.on('status', (status) => {
        dashboardState.status = status;
        ws.send(JSON.stringify(dashboardState));
      });
      self.events.on('servicesState', (servicesState) => {
        dashboardState.availableServices = servicesState;
        ws.send(JSON.stringify(dashboardState));
      });
    });

    app.get('/embark', function (req, res) {
      res.send('Welcome to Embark')
    });

    app.get('/embark/console', function (req, res) {
      self.events.request('console:command', req.query.cmd, res.send.bind(res));
    });

    app.listen(this.port);

    let message = __("webserver available at") +
                    " " +
                    ("http://" + canonicalHost(this.hostname) +
                    ":" + this.port).bold.underline.green;
    this.server.listen(this.port, this.hostname);
    callback(null, message);
  }

  stop(callback) {
    if (!this.server || !this.server.listening) {
      return callback(null, __("no webserver is currently running"));
    }
    this.server.shutdown(function() {
      callback(null, __("Webserver stopped"));
    });
  }

}

module.exports = Server;