tacmap_pli.js

/* 
 * Copyright (C) 2015 jdn
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
(function () {
    "use strict";
    var express = require('express');
    var compression = require('compression');
    var url = require('url');
    var request = require('axios');
    var bodyParser = require('body-parser');
    var fs = require('fs');
    var cesium = require('./geoserver/cesiumserver');
    //var http = require('http');
    var cors = require('cors');

    var server_port = 8181;
    var yargs = require('yargs').options({
        'port': {
            'default': server_port,
            'description': 'Port to listen on.'
        },
        'public': {
            'type': 'boolean',
            'description': 'Run a public server that listens on all interfaces.'
        },
        'publicssl': {
            'type': 'boolean',
            'description': 'Run a public https server that listens on all interfaces.'
        },
        'upstream-proxy': {
            'description': 'A standard proxy server that will be used to retrieve data.  Specify a URL including port, e.g. "http://proxy:8000".'
        },
        'bypass-upstream-proxy-hosts': {
            'description': 'A comma separated list of hosts that will bypass the specified upstream_proxy, e.g. "lanhost1,lanhost2"'
        },
        'help': {
            'alias': 'h',
            'type': 'boolean',
            'description': 'Show this help.'
        }
    });
    var argv = yargs.argv;
    if (argv.help) {
        return yargs.showHelp();
    }
    var app = express();
    app.use(bodyParser.json());
    app.use(compression());
    app.use(express.static(__dirname + '/public'));
    app.use(cors());

    function getRemoteUrlFromParam(req) {
        var remoteUrl = req.params[0];
        if (remoteUrl) {
            // add http:// to the URL if no protocol is present
            if (!/^https?:\/\//.test(remoteUrl)) {
                remoteUrl = 'http://' + remoteUrl;
            }
            remoteUrl = url.parse(remoteUrl);
            // copy query string
            remoteUrl.search = url.parse(req.url).search;
        }
        return remoteUrl;
    }

    var dontProxyHeaderRegex = /^(?:Host|Proxy-Connection|Connection|Keep-Alive|Transfer-Encoding|TE|Trailer|Proxy-Authorization|Proxy-Authenticate|Upgrade)$/i;

    function filterHeaders(req, headers) {
        var result = {};
        // filter out headers that are listed in the regex above
        Object.keys(headers).forEach(function (name) {
            if (!dontProxyHeaderRegex.test(name)) {
                result[name] = headers[name];
            }
        });
        return result;
    }

    var upstreamProxy = argv['upstream-proxy'];
    var bypassUpstreamProxyHosts = {};
    if (argv['bypass-upstream-proxy-hosts']) {
        argv['bypass-upstream-proxy-hosts'].split(',').forEach(function (host) {
            bypassUpstreamProxyHosts[host.toLowerCase()] = true;
        });
    }
    app.get('/proxy/*', function (req, res, next) {
        // look for request like http://localhost:8080/proxy/http://example.com/file?query=1
        var remoteUrl = getRemoteUrlFromParam(req);
        if (!remoteUrl) {
            // look for request like http://localhost:8080/proxy/?http%3A%2F%2Fexample.com%2Ffile%3Fquery%3D1
            remoteUrl = Object.keys(req.query)[0];
            if (remoteUrl) {
                remoteUrl = new URL(remoteUrl);
            }
        }

        if (!remoteUrl) {
            return res.send(400, 'No url specified.');
        }

        if (!remoteUrl.protocol) {
            remoteUrl.protocol = 'http:';
        }

        var proxy;
        if (upstreamProxy && !(remoteUrl.host in bypassUpstreamProxyHosts)) {
            proxy = upstreamProxy;
        }

        // encoding : null means "body" passed to the callback will be raw bytes

        request.get({
            url: url.format(remoteUrl),
            headers: filterHeaders(req, req.headers),
            encoding: null,
            proxy: proxy
        }, function (error, response, body) {
            var code = 500;
            if (response) {
                code = response.statusCode;
                res.header(filterHeaders(req, response.headers));
            }

            res.send(code, body);
        });
    });
    var server_ip_address = '127.0.0.1';
    var server = app.listen(server_port, argv.public ? undefined : server_ip_address, function () {
        if (argv.public) {
            console.log('TacMap development server running publicly.  Connect to http://localhost:%d/', server.address().port);
        } else if (argv.publicssl) {
            server.key = fs.readFileSync('key.pem');
            server.cert = fs.readFileSync('cert.pem');
            console.log('TacMap development server running publicly.  Connect to https://localhost:%d/', server.address().port);
        }
        else {
            console.log('TacMap development server running locally.  Connect to http://localhost:%d/', server.address().port);
        }
    });
    server.on('error', function (e) {
        if (e.code === 'EADDRINUSE') {
            console.log('Error: Port %d is already in use, select a different port.', argv.port);
            console.log('Example: node server.js --port %d', argv.port + 1);
        }
        else if (e.code === 'EACCES') {
            console.log('Error: This process does not have permission to listen on port %d.', argv.port);
            if (argv.port < 1024) {
                console.log('Try a port number higher than 1024.');
            }
        }
        console.log(e);
        process.exit(1);
    });
    server.on('close', function () {
        console.log('Cesium development server stopped.');
    });
    process.on('SIGINT', function () {
        server.close(function () {
            process.exit(0);
        });
    });
    //SOCKET IO
    /**
     * @param req Request
     * @param res Result
     * **/
    app.get('/', function (req, res) {
        res.sendFile(__dirname + '/public/geoview.html');
    });
    app.get('/node_modules/*', function (req, res) {
        res.sendFile(__dirname + '/' + req.url);
    });
    app.get('/json/*', function (req, res) {
        res.sendFile(__dirname + '/' + req.url);
    });
    app.post('/json/*', function (req, res) {
        //console.log(request.body);
        fs.writeFile(__dirname + '/public' + req.url, JSON.stringify(req.body), function () {
            res.end();
        });
    });
    app.put('/json/*', function (req, res) {
        //console.log(req.body);
        fs.writeFile(__dirname + '/public' + req.url, JSON.stringify(req.body), function () {
            res.end();
        });
    });
    app.get('/xml/*', function (req, res) {
        res.sendFile(__dirname + '/' + req.url);
    });
    app.put('/xml/*', function (req, res) {
        console.log("Put " + req.url);
        console.log(req.body);
        fs.writeFile(__dirname + '/' + req.url, req.body, function () {
            res.end();
        });
    });
    app.put('/entity/*', function (req, res) {
        console.log("Put entity " + req.url);
        console.log(req.body);
        fs.writeFile(__dirname + '/public' + req.url, req.body, function () {
            res.end();
        });
    });

    app.post('/msg/*', cors(), function (req, res) {
        var jsonmsg = req.body;
        //console.log(jsonmsg);
        sio.emit(jsonmsg.scktmsg, { scktid: jsonmsg.sctkid, payload: jsonmsg.payload });
        res.send(req.body);
    });


    /** SocketIO Services **/
    var sio = require('socket.io')(server);
    sio.serveClient(true);


    /** @param topsocket Top level socket connection **/

    sio.of('').on('connection', function (topsocket) {
        topsocket.emit('connection', {
            message: 'Msg Socket Ready',
            socketid: topsocket.id
        });
        /** 
         * Namespaces correspond to Map Views ..
         * @param mapdta {id,netname,mapviewid}
         * @param callback  
         * **/
        topsocket.on('join namespace', function (ep, callback) {
            console.log("join namespace: " + ep.user_id);
            sio.of('/' + ep.user_id)
                .once('connection', function (map_socket) {
                    console.log('user connected to ' + ep.user_id);
                    socketOps(map_socket);
                });
            callback(ep);
        });
        /** @param endpoint {id,netname,mapviewid}**/
        topsocket.on('initial connection', function (endpoint) {
            sio.to(endpoint.socketid).emit('load map', endpoint);
        });


        // When a Map View is created - another namespace is set up with
        // SocketOps functions.  Each new user has a namespace auomatically assigned
        // and will be avalalabe to others to select.
        // @todo Implement security to restrict/permit access to Map Views / Namespaces

        // Create a mapview
        /** @param mapdata {user_id,name,data} **/
        topsocket.on('create map', function (mapdata) {
            console.log('create map');
            sio.of('/' + mapdata.id)
                .once('connection', function (map_socket) {
                    console.log('user connected to ' + mapdata.id);
                    socketOps(map_socket);
                });
        });
        // Remove a mapview
        /** @param mapdata {id,name,url,data} **/
        topsocket.on('remove map', function (mapdata) {
            sio.emit('remove map', mapdata);
        });
        // Update a mapview
        /** @param mapdata {id,name,url,data} **/
        topsocket.on('update map', function (mapdata) {
            sio.emit('update map', mapdata);
        });
    });
    var socketOps = function (socket) {
        // Disconnect endpoint.  Remove from all lists
        socket.once('socketDisconnect', function () {
            console.log('Socket Disconnect ' + socket.id);
            sio.emit('socketDisconnect', {
                socketid: socket.id
            });
        });
        // Join a network.
        socket.on('join network', function (netdata) {
            socket.join(netdata.network_id);
            sio.emit('join net', netdata);
        });
        // Leave a network.
        socket.on('leave network', function (netdata) {
            socket.leave(netdata.network_id);
            sio.emit('leave net', netdata);
        });
        // Rename a network.
        socket.on('update network', function (netdata) {
            sio.emit('update net', netdata);
        });
        // Create a network as a socketIO room
        socket.on('create network', function (netdata) {
            sio.emit('create net', netdata);
        });
        // Remove a network 
        socket.on('remove network', function (netdata) {
            sio.emit('remove net', netdata);
        });
        // Synch Tracks 
        socket.on('Sync Tracks', function (trackdata) {
            console.log('Sync Tracks');
            sio.emit('update tracks', { scktid: trackdata.scktid, tracks: trackdata.tracks });
        });
        //
        //Publish message to single socketid, or to one or all network
        socket.on('publish msg', function (msg) {
            if (typeof msg.data.destination !== 'undefined') {
                sio.to(msg.data.destination.socketid).emit(msg.scktmsg, msg.data);
            }
            else if (typeof msg.data.network !== 'undefined') {
                sio.to(msg.data.network.id).emit(msg.scktmsg, msg.data);
            }
            else {
                sio.emit(msg.scktmsg, msg.data);
            }
        });
    };
})();