const async = require('async'); let serveStatic = require('serve-static'); const {canonicalHost, defaultHost, dockerHostSwap} = require('../../utils/host'); const expressWebSocket = require('express-ws'); const express = require('express'); const fs = require('../../core/fs'); require('http-shutdown').extend(); var express = require('express'); let path = require('path'); var expressWebSocket = require('express-ws'); class Server { constructor(options) { this.logger = options.logger; this.buildDir = options.buildDir; this.events = options.events; this.port = options.port || 8000; this.hostname = dockerHostSwap(options.host) || defaultHost; this.isFirstStart = true; this.opened = false; this.openBrowser = options.openBrowser; this.logging = false; this.events.once('outputDone', () => { this.logger.info(this._getMessage()); }); } enableLogging(callback) { this.logging = true; return callback(null, __("Enabled Webserver Logs")); } disableLogging(callback) { this.logging = false; return callback(null, __("Disabled Webserver Logs")); } 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); } const coverage = serveStatic(fs.dappPath('coverage/__root__/'), {'index': ['index.html', 'index.htm']}); const coverageStyle = serveStatic(fs.dappPath('coverage/')); const main = serveStatic(this.buildDir, {'index': ['index.html', 'index.htm']}); this.app = express(); const expressWs = expressWebSocket(this.app); // Assign Logging Function this.app.use(function(req, res, next) { if (self.logging) { if (!req.headers.upgrade) { console.log('Webserver> ' + req.method + " " + req.originalUrl); } } next(); }); this.app.use(main); this.app.use('/coverage', coverage); this.app.use(coverageStyle); expressWebSocket(this.app); this.app.ws('/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)); }); }); this.app.ws('/', function(ws, _req) { self.events.on('outputDone', () => { if (ws.readyState === WEB_SOCKET_STATE_OPEN) { return ws.send('outputDone'); } // if the socket wasn't yet opened, listen for the 'open' event, // then send the 'outputDone' data ws.addEventListener('open', _event => { ws.send('outputDone'); }); }); }); async.waterfall([ function createPlaceholderPage(next) { if (!self.isFirstStart) { return next(); } self.isFirstStart = false; self.events.request('build-placeholder', next); }, function listen(next) { self.server = self.app.listen(self.port, self.hostname, () => { self.port = self.server.address().port; next(); }); }, function openBrowser(next) { if (!self.openBrowser || self.opened) { return next(); } self.opened = true; self.events.request('open-browser', next); } ], function (err) { if (err) { return callback(err); } callback(null, self._getMessage(), self.port); }); } _getMessage() { return __('webserver available at') + ' ' + ('http://' + canonicalHost(this.hostname) + ':' + this.port).bold.underline.green; } stop(callback) { if (!this.server || !this.server.listening) { return callback(null, __("no webserver is currently running")); } this.server.close(function() { callback(null, __("Webserver stopped")); }); } } module.exports = Server;