websocket.js 2.14 KB
Newer Older
Marian Oancea's avatar
Marian Oancea committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
    This file is part of ethereum.js.

    ethereum.js is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    ethereum.js 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with ethereum.js.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file websocket.js
 * @authors:
Marek Kotewicz's avatar
Marek Kotewicz committed
19
 *   Jeffrey Wilcke <jeff@ethdev.com>
Marian Oancea's avatar
Marian Oancea committed
20 21 22 23 24
 *   Marek Kotewicz <marek@ethdev.com>
 *   Marian Oancea <marian@ethdev.com>
 * @date 2014
 */

Marek Kotewicz's avatar
Marek Kotewicz committed
25
if (process.env.NODE_ENV !== "build") {
Marian Oancea's avatar
Marian Oancea committed
26 27
    var WebSocket = require('ws'); // jshint ignore:line
}
Marian Oancea's avatar
Marian Oancea committed
28

Marek Kotewicz's avatar
Marek Kotewicz committed
29 30 31 32 33 34
var WebSocketProvider = function(host) {
    // onmessage handlers
    this.handlers = [];
    // queue will be filled with messages if send is invoked before the ws is ready
    this.queued = [];
    this.ready = false;
Marian Oancea's avatar
Marian Oancea committed
35

Marek Kotewicz's avatar
Marek Kotewicz committed
36
    this.ws = new WebSocket(host);
Marian Oancea's avatar
Marian Oancea committed
37

Marek Kotewicz's avatar
Marek Kotewicz committed
38 39 40 41 42
    var self = this;
    this.ws.onmessage = function(event) {
        for(var i = 0; i < self.handlers.length; i++) {
            self.handlers[i].call(self, JSON.parse(event.data), event);
        }
Marian Oancea's avatar
Marian Oancea committed
43 44
    };

Marek Kotewicz's avatar
Marek Kotewicz committed
45 46 47 48 49 50
    this.ws.onopen = function() {
        self.ready = true;

        for(var i = 0; i < self.queued.length; i++) {
            // Resend
            self.send(self.queued[i]);
Marian Oancea's avatar
Marian Oancea committed
51 52
        }
    };
Marek Kotewicz's avatar
Marek Kotewicz committed
53
};
Marian Oancea's avatar
Marian Oancea committed
54

Marek Kotewicz's avatar
Marek Kotewicz committed
55 56 57
WebSocketProvider.prototype.send = function(payload) {
    if(this.ready) {
        var data = JSON.stringify(payload);
Marian Oancea's avatar
Marian Oancea committed
58

Marek Kotewicz's avatar
Marek Kotewicz committed
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
        this.ws.send(data);
    } else {
        this.queued.push(payload);
    }
};

WebSocketProvider.prototype.onMessage = function(handler) {
    this.handlers.push(handler);
};

WebSocketProvider.prototype.unload = function() {
    this.ws.close();
};
Object.defineProperty(WebSocketProvider.prototype, "onmessage", {
    set: function(provider) { this.onMessage(provider); }
});
Marian Oancea's avatar
Marian Oancea committed
75 76

module.exports = WebSocketProvider;