jsOMS/Socket/Client/Client.js

(function (jsOMS) {
    'use strict';

    jsOMS.Client = function (ip, port, protocol) {
        this.port       = port;
        this.ip         = ip;
        this.protocol   = protocol;
        this.connection = null;
        this.messages   = [];
    };

    jsOMS.Client.prototype.setMessage = function (id, callback) {
        this.messages[id] = callback;
    };

    jsOMS.Client.prototype.setIp = function (ip) {
        this.ip = ip;
    };

    jsOMS.Client.prototype.setPort = function (port) {
        this.port = port;
    };

    jsOMS.Client.prototype.setProtocol = function (protocol) {
        this.protocol = protocol;
    };

    jsOMS.Client.prototype.connect = function () {
        const self      = this;
        this.connection = new WebSocket(this.ip, this.protocol);

        this.connection.onmessage = function (event) {
            const msg = JSON.parse(event.data);

            self.messages[msg.type](msg);
        };
    };

    jsOMS.Client.prototype.send = function (msg) {
        this.connection.send(JSON.stringify(msg));
    };

    jsOMS.Client.prototype.close = function () {
        this.connection.close();
    };
}(window.jsOMS = window.jsOMS || {}));