serverbase.js

var express = require('express');
const WebSocket = require('ws');
const http = require('http');
const url = require('url');
var app = express()
var onlist = {};
var parmlist = {};
var wssobj;

var _options = {
    doc: '/index.html',
    dir: __dirname,
    port: 80,
    url: null
}

//  todo!   must be implemented in class
function checkArgs(req, res) {
    for (rid in req.query) {
        for (id in parmlist) {
            if (id == rid) parmlist[id](req.params[0], req.query[rid]);
        }
    }
}

app.get(/^(.+)$/, function (req, res) {
    if (req.params[0] == '/') req.params[0] = _options.doc;
    checkArgs(req, res);
    res.sendFile(_options.dir + req.params[0]);
});

function handleStream(ar) {
    for (id in ar) {
        for (on in onlist) {
            if (id == on) onlist[on](ar[id])
        }
    }
}

class serverbase {
    constructor(options) {
        if (options != undefined) {
            console.log()
            for (var opt in options) {
                _options[opt] = options[opt];
            }
        }
        const server = http.createServer(app);
        var wss;
        if (!_options.url) wss = new WebSocket.Server({ server });
        else {
            var ar = _options.url.split(':');
            console.log(ar);
            wss = new WebSocket.Server({ server: ar[0], port: ar[1]});
        }
        wssobj = wss;
        wss.on('connection', function connection(ws, req) {
            handleStream({ 'connect': ws })
            ws.on('message', function incoming(data) {
                handleStream(JSON.parse(data));
            });
        });
        server.listen(_options.port, function () {
            handleStream({"listen": _options.port})
        })
    }

    on (id, func) {
        onlist[id] = func;
    }

    onparm(id, func) {
        parmlist[id] = func;
    }

    broadcast (data) {
        wssobj.clients.forEach(function each(client) {
            if (client.readyState === WebSocket.OPEN) {
                client.send(data);
            }
        });
    };
}

module.exports = serverbase;