occaecatidicta/tools/omelox-robot/lib/console/http.ts

// ------------------------------------
// HTTP Server
// ------------------------------------
//
// This file defines HttpServer and the singleton HTTP_SERVER.
//
// This file defines a generic HTTP server that serves static files and that can be configured
// with new routes. It also starts the nodeload HTTP server unless require('nodeload/config')
// .disableServer() was called.
//
import * as http from 'http';
import * as fs from 'fs';
import * as net from 'net';
import * as util from '../common/util';
let qputs = console.log;
import { EventEmitter } from 'events';

import * as __ from 'underscore';
import { getData } from '../monitor/monitor';
/** By default, HttpServer knows how to return static files from the current directory. Add new route
regexs using HttpServer.on(). */

export class HttpServer extends EventEmitter {
    server: http.Server;
    hostname: string;
    port: number;
    connections: net.Socket[];
    routes: { regex: any, handler: Function }[] = [];
    running = false;
    robotMasterUrl: string;
    /** Start the server listening on the given port */
    start(robotMasterUrl: string , port?: number, hostname?: string) {
        if (this.running) { return; }
        this.running = true;
        this.robotMasterUrl = robotMasterUrl;

        let self = this;
        port = port || 8000;
        self.hostname = hostname || 'localhost';
        self.port = port;
        self.connections = [];

        self.server = http.createServer(function (req, res) { self.route_(req, res); });
        self.server.on('connection', function (c) {
            // We need to track incoming connections, beause Server.close() won't terminate active
            // connections by default.
            c.on('close', function () {
                let idx = self.connections.indexOf(c);
                if (idx !== -1) {
                    self.connections.splice(idx, 1);
                }
            });
            self.connections.push(c);
        });
        self.server.listen(this.port, this.hostname);

        self.emit('start', self.hostname, self.port);
        return self;
    }
    /** Terminate the server */
    stop() {
        if (!this.running) { return; }
        this.running = false;
        this.connections.forEach(function (c) { c.destroy(); });
        this.server.close();
        this.server = null;
        this.emit('end');
    }
    /** When an incoming request matches a given regex, route it to the provided handler:
    function(url, ServerRequest, ServerResponse) */
    addRoute(regex: any, handler: Function) {
        this.routes.unshift({ regex: regex, handler: handler });
        return this;
    }
    removeRoute(regex: any, handler: Function) {
        this.routes = this.routes.filter(function (r) {
            return !((regex === r.regex) && (!handler || handler === r.handler));
        });
        return this;
    }
    route_(req: any, res: any) {
        for (let i = 0; i < this.routes.length; i++) {
            if (req.url.match(this.routes[i].regex)) {
                this.routes[i].handler(req.url, req, res);
                return;
            }
        }
        if (req.method === 'GET') {
            this.serveFile_('.' + req.url, res);
        } else {
            res.writeHead(405, { 'Content-Length': '0' });
            res.end();
        }
    }

    doReport(response: any) {
        let pdata = getData();
        // console.log('pdata %j',pdata);
        let mdata: any[] = [];
        let _show = false;
        __.each(pdata, function (val: any, key) {
            let single: any = {};
            _show = true;
            single['name'] = key;
            single['uid'] = key;
            let keycolumns: any = [];
            let maxId = 0;
            __.each(val, function (kval, akey) {
                let _length = __.size(kval);
                if (_length > maxId) maxId = _length;
                if (_length > 0) keycolumns.push(akey);
            });
            let gcolumns: any = [];
            gcolumns.push('users');
            let glastkeyData = {};
            __.each(keycolumns, function (dkey) { gcolumns.push(dkey); });
            let grows: any = [];
            for (let i = 0; i < maxId; i++) {
                let rows: any = [];
                rows.push(i + 1);
                __.each(keycolumns, function (dkey: any) {
                    // console.log('dkey' + dkey + ' ' +i + JSON.stringify(val[dkey]))
                    rows.push(val[dkey][i] || 0);
                    // _vaild = true;
                });
                grows.push(rows);
            }
            let gsummary: any = {};
            __.each(keycolumns, function (dkey: any) {
                let summary: any = {};
                let kdata = val[dkey];
                let min = Number.MAX_VALUE, max = 0;
                let sindex = 0, sum = 0;
                __.each(kdata, function (time: number) {
                    if (time > max) max = time;
                    if (time < min) min = time;
                    sum += time;
                    ++sindex;
                });
                let avg = Math.round(sum / sindex);
                summary = { 'max': max, 'min': min, 'avg': avg, 'qs': Math.round(sindex * 1000 / avg) };
                gsummary[dkey] = (summary);
            });
            single['summary'] = gsummary;
            single['charts'] = { 'latency': { 'name': 'robot', 'uid': single['uid'], 'columns': gcolumns, 'rows': grows } };
            if (grows.length > 0) mdata.push(single);
        });
        if (_show) {
            let data = JSON.stringify(mdata);
            // response.writeHead(200, { 'Content-Length': pdata.length });
            response.write(data, 'binary');
        }
        response.end();
    }


    serveFile_(basefile: string, response: any) {
        if (basefile.lastIndexOf('report') !== -1) {
            this.doReport(response);
            return;
        }
        if (basefile === './')
        basefile = 'index.html';
        let file = __dirname + '/' + basefile;
        fs.stat(file,  (err, stat) => {
            if (err) {
                response.writeHead(404, { 'Content-Type': 'text/plain' });
                response.write('Cannot find file: ' + file);
                response.end();
                return;
            }

            fs.readFile(file, 'binary',  (err, data) => {
                if (err) {
                    response.writeHead(500, { 'Content-Type': 'text/plain' });
                    response.write('Error opening file ' + file + ': ' + err);
                } else {
                    if (file.lastIndexOf('.html') === -1) {
                        if(basefile === './js/webclient.js') {
                            data = data.replace('${robotMasterUrl}' , this.robotMasterUrl);
                        }
                        response.writeHead(200, { 'Content-Length': data.length });
                        response.write(data, 'binary');
                    } else {
                        response.writeHead(200, { 'Content-Length': data.length, 'Content-Type': 'text/html; charset=utf-8' });
                        response.write(data, 'binary');
                    }
                }
                response.end();
            });
        });
    }
}
// =================
// Singletons
// =================
/** The global HTTP server used by nodeload */
export let HTTP_SERVER = new HttpServer();
HTTP_SERVER.on('start', function (hostname, port) {
    console.log(`Started Robot Master Http server on : http://${hostname}:${port}`);
});

HTTP_SERVER.on('end', function () {
    qputs('Shutdown Robot Master Http server.');
});