Browse Source

retabed files

cl-refactor
Marek Kotewicz 10 years ago
parent
commit
84b36a7193
  1. 107
      lib/httprpc.js
  2. 732
      lib/main.js
  3. 33
      lib/qt.js
  4. 78
      lib/websocket.js

107
lib/httprpc.js

@ -21,74 +21,73 @@
* @date 2014 * @date 2014
*/ */
if(process.env.NODE_ENV !== "build") { if (process.env.NODE_ENV !== "build") {
var XMLHttpRequest = require('xmlhttprequest').XMLHttpRequest; // jshint ignore:line var XMLHttpRequest = require('xmlhttprequest').XMLHttpRequest; // jshint ignore:line
} }
var HttpRpcProvider = function (host) {
this.handlers = [];
this.host = host;
};
var HttpRpcProvider = function (host) { function formatJsonRpcObject(object) {
this.handlers = []; return {
this.host = host; jsonrpc: '2.0',
method: object.call,
params: object.args,
id: object._id
}; };
}
function formatJsonRpcObject(object) { function formatJsonRpcMessage(message) {
return { var object = JSON.parse(message);
jsonrpc: '2.0',
method: object.call,
params: object.args,
id: object._id
};
}
function formatJsonRpcMessage(message) {
var object = JSON.parse(message);
return {
_id: object.id,
data: object.result,
error: object.error
};
}
HttpRpcProvider.prototype.sendRequest = function (payload, cb) {
var data = formatJsonRpcObject(payload);
var request = new XMLHttpRequest(); return {
request.open("POST", this.host, true); _id: object.id,
request.send(JSON.stringify(data)); data: object.result,
request.onreadystatechange = function () { error: object.error
if (request.readyState === 4 && cb) {
cb(request);
}
};
}; };
}
HttpRpcProvider.prototype.send = function (payload) { HttpRpcProvider.prototype.sendRequest = function (payload, cb) {
var self = this; var data = formatJsonRpcObject(payload);
this.sendRequest(payload, function (request) {
self.handlers.forEach(function (handler) { var request = new XMLHttpRequest();
handler.call(self, formatJsonRpcMessage(request.responseText)); request.open("POST", this.host, true);
}); request.send(JSON.stringify(data));
}); request.onreadystatechange = function () {
if (request.readyState === 4 && cb) {
cb(request);
}
}; };
};
HttpRpcProvider.prototype.poll = function (payload, id) { HttpRpcProvider.prototype.send = function (payload) {
var self = this; var self = this;
this.sendRequest(payload, function (request) { this.sendRequest(payload, function (request) {
var parsed = JSON.parse(request.responseText); self.handlers.forEach(function (handler) {
if (parsed.error || (parsed.result instanceof Array ? parsed.result.length === 0 : !parsed.result)) { handler.call(self, formatJsonRpcMessage(request.responseText));
return;
}
self.handlers.forEach(function (handler) {
handler.call(self, {_event: payload.call, _id: id, data: parsed.result});
});
}); });
}; });
};
Object.defineProperty(HttpRpcProvider.prototype, "onmessage", { HttpRpcProvider.prototype.poll = function (payload, id) {
set: function (handler) { var self = this;
this.handlers.push(handler); this.sendRequest(payload, function (request) {
var parsed = JSON.parse(request.responseText);
if (parsed.error || (parsed.result instanceof Array ? parsed.result.length === 0 : !parsed.result)) {
return;
} }
self.handlers.forEach(function (handler) {
handler.call(self, {_event: payload.call, _id: id, data: parsed.result});
});
}); });
};
Object.defineProperty(HttpRpcProvider.prototype, "onmessage", {
set: function (handler) {
this.handlers.push(handler);
}
});
module.exports = HttpRpcProvider; module.exports = HttpRpcProvider;

732
lib/main.js

@ -22,136 +22,169 @@
* @date 2014 * @date 2014
*/ */
function flattenPromise (obj) { function flattenPromise (obj) {
if (obj instanceof Promise) { if (obj instanceof Promise) {
return Promise.resolve(obj); return Promise.resolve(obj);
} }
if (obj instanceof Array) {
return new Promise(function (resolve) {
var promises = obj.map(function (o) {
return flattenPromise(o);
});
return Promise.all(promises).then(function (res) { if (obj instanceof Array) {
for (var i = 0; i < obj.length; i++) { return new Promise(function (resolve) {
obj[i] = res[i]; var promises = obj.map(function (o) {
} return flattenPromise(o);
resolve(obj);
});
}); });
}
if (obj instanceof Object) {
return new Promise(function (resolve) {
var keys = Object.keys(obj);
var promises = keys.map(function (key) {
return flattenPromise(obj[key]);
});
return Promise.all(promises).then(function (res) { return Promise.all(promises).then(function (res) {
for (var i = 0; i < keys.length; i++) { for (var i = 0; i < obj.length; i++) {
obj[keys[i]] = res[i]; obj[i] = res[i];
} }
resolve(obj); resolve(obj);
});
}); });
} });
return Promise.resolve(obj);
} }
var ethMethods = function () { if (obj instanceof Object) {
var blockCall = function (args) { return new Promise(function (resolve) {
return typeof args[0] === "string" ? "eth_blockByHash" : "eth_blockByNumber"; var keys = Object.keys(obj);
}; var promises = keys.map(function (key) {
return flattenPromise(obj[key]);
});
var transactionCall = function (args) { return Promise.all(promises).then(function (res) {
return typeof args[0] === "string" ? 'eth_transactionByHash' : 'eth_transactionByNumber'; for (var i = 0; i < keys.length; i++) {
}; obj[keys[i]] = res[i];
}
resolve(obj);
});
});
}
var uncleCall = function (args) { return Promise.resolve(obj);
return typeof args[0] === "string" ? 'eth_uncleByHash' : 'eth_uncleByNumber'; }
};
var methods = [ var ethMethods = function () {
{ name: 'balanceAt', call: 'eth_balanceAt' }, var blockCall = function (args) {
{ name: 'stateAt', call: 'eth_stateAt' }, return typeof args[0] === "string" ? "eth_blockByHash" : "eth_blockByNumber";
{ name: 'countAt', call: 'eth_countAt'},
{ name: 'codeAt', call: 'eth_codeAt' },
{ name: 'transact', call: 'eth_transact' },
{ name: 'call', call: 'eth_call' },
{ name: 'block', call: blockCall },
{ name: 'transaction', call: transactionCall },
{ name: 'uncle', call: uncleCall },
{ name: 'compile', call: 'eth_compile' },
{ name: 'lll', call: 'eth_lll' }
];
return methods;
}; };
var ethProperties = function () { var transactionCall = function (args) {
return [ return typeof args[0] === "string" ? 'eth_transactionByHash' : 'eth_transactionByNumber';
{ name: 'coinbase', getter: 'eth_coinbase', setter: 'eth_setCoinbase' },
{ name: 'listening', getter: 'eth_listening', setter: 'eth_setListening' },
{ name: 'mining', getter: 'eth_mining', setter: 'eth_setMining' },
{ name: 'gasPrice', getter: 'eth_gasPrice' },
{ name: 'account', getter: 'eth_account' },
{ name: 'accounts', getter: 'eth_accounts' },
{ name: 'peerCount', getter: 'eth_peerCount' },
{ name: 'defaultBlock', getter: 'eth_defaultBlock', setter: 'eth_setDefaultBlock' },
{ name: 'number', getter: 'eth_number'}
];
}; };
var dbMethods = function () { var uncleCall = function (args) {
return [ return typeof args[0] === "string" ? 'eth_uncleByHash' : 'eth_uncleByNumber';
{ name: 'put', call: 'db_put' },
{ name: 'get', call: 'db_get' },
{ name: 'putString', call: 'db_putString' },
{ name: 'getString', call: 'db_getString' }
];
}; };
var shhMethods = function () { var methods = [
return [ { name: 'balanceAt', call: 'eth_balanceAt' },
{ name: 'post', call: 'shh_post' }, { name: 'stateAt', call: 'eth_stateAt' },
{ name: 'newIdentity', call: 'shh_newIdentity' }, { name: 'countAt', call: 'eth_countAt'},
{ name: 'haveIdentity', call: 'shh_haveIdentity' }, { name: 'codeAt', call: 'eth_codeAt' },
{ name: 'newGroup', call: 'shh_newGroup' }, { name: 'transact', call: 'eth_transact' },
{ name: 'addToGroup', call: 'shh_addToGroup' } { name: 'call', call: 'eth_call' },
]; { name: 'block', call: blockCall },
{ name: 'transaction', call: transactionCall },
{ name: 'uncle', call: uncleCall },
{ name: 'compile', call: 'eth_compile' },
{ name: 'lll', call: 'eth_lll' }
];
return methods;
};
var ethProperties = function () {
return [
{ name: 'coinbase', getter: 'eth_coinbase', setter: 'eth_setCoinbase' },
{ name: 'listening', getter: 'eth_listening', setter: 'eth_setListening' },
{ name: 'mining', getter: 'eth_mining', setter: 'eth_setMining' },
{ name: 'gasPrice', getter: 'eth_gasPrice' },
{ name: 'account', getter: 'eth_account' },
{ name: 'accounts', getter: 'eth_accounts' },
{ name: 'peerCount', getter: 'eth_peerCount' },
{ name: 'defaultBlock', getter: 'eth_defaultBlock', setter: 'eth_setDefaultBlock' },
{ name: 'number', getter: 'eth_number'}
];
};
var dbMethods = function () {
return [
{ name: 'put', call: 'db_put' },
{ name: 'get', call: 'db_get' },
{ name: 'putString', call: 'db_putString' },
{ name: 'getString', call: 'db_getString' }
];
};
var shhMethods = function () {
return [
{ name: 'post', call: 'shh_post' },
{ name: 'newIdentity', call: 'shh_newIdentity' },
{ name: 'haveIdentity', call: 'shh_haveIdentity' },
{ name: 'newGroup', call: 'shh_newGroup' },
{ name: 'addToGroup', call: 'shh_addToGroup' }
];
};
var ethWatchMethods = function () {
var newFilter = function (args) {
return typeof args[0] === 'string' ? 'eth_newFilterString' : 'eth_newFilter';
}; };
var ethWatchMethods = function () { return [
var newFilter = function (args) { { name: 'newFilter', call: newFilter },
return typeof args[0] === 'string' ? 'eth_newFilterString' : 'eth_newFilter'; { name: 'uninstallFilter', call: 'eth_uninstallFilter' },
{ name: 'getMessages', call: 'eth_getMessages' }
];
};
var shhWatchMethods = function () {
return [
{ name: 'newFilter', call: 'shh_newFilter' },
{ name: 'uninstallFilter', call: 'shh_uninstallFilter' },
{ name: 'getMessage', call: 'shh_getMessages' }
];
};
var setupMethods = function (obj, methods) {
methods.forEach(function (method) {
obj[method.name] = function () {
return flattenPromise(Array.prototype.slice.call(arguments)).then(function (args) {
var call = typeof method.call === "function" ? method.call(args) : method.call;
return {call: call, args: args};
}).then(function (request) {
return new Promise(function (resolve, reject) {
web3.provider.send(request, function (err, result) {
if (!err) {
resolve(result);
return;
}
reject(err);
});
});
}).catch(function(err) {
console.error(err);
});
}; };
});
return [ };
{ name: 'newFilter', call: newFilter },
{ name: 'uninstallFilter', call: 'eth_uninstallFilter' }, var setupProperties = function (obj, properties) {
{ name: 'getMessages', call: 'eth_getMessages' } properties.forEach(function (property) {
]; var proto = {};
}; proto.get = function () {
return new Promise(function(resolve, reject) {
var shhWatchMethods = function () { web3.provider.send({call: property.getter}, function(err, result) {
return [ if (!err) {
{ name: 'newFilter', call: 'shh_newFilter' }, resolve(result);
{ name: 'uninstallFilter', call: 'shh_uninstallFilter' }, return;
{ name: 'getMessage', call: 'shh_getMessages' } }
]; reject(err);
}; });
});
var setupMethods = function (obj, methods) { };
methods.forEach(function (method) { if (property.setter) {
obj[method.name] = function () { proto.set = function (val) {
return flattenPromise(Array.prototype.slice.call(arguments)).then(function (args) { return flattenPromise([val]).then(function (args) {
var call = typeof method.call === "function" ? method.call(args) : method.call; return new Promise(function (resolve) {
return {call: call, args: args}; web3.provider.send({call: property.setter, args: args}, function (err, result) {
}).then(function (request) {
return new Promise(function (resolve, reject) {
web3.provider.send(request, function (err, result) {
if (!err) { if (!err) {
resolve(result); resolve(result);
return; return;
@ -159,294 +192,261 @@
reject(err); reject(err);
}); });
}); });
}).catch(function(err) { }).catch(function (err) {
console.error(err); console.error(err);
}); });
}; };
}); }
}; Object.defineProperty(obj, property.name, proto);
});
var setupProperties = function (obj, properties) { };
properties.forEach(function (property) {
var proto = {}; var web3 = {
proto.get = function () { _callbacks: {},
return new Promise(function(resolve, reject) { _events: {},
web3.provider.send({call: property.getter}, function(err, result) { providers: {},
if (!err) { toHex: function(str) {
resolve(result); var hex = "";
return; for(var i = 0; i < str.length; i++) {
} var n = str.charCodeAt(i).toString(16);
reject(err); hex += n.length < 2 ? '0' + n : n;
}); }
});
};
if (property.setter) {
proto.set = function (val) {
return flattenPromise([val]).then(function (args) {
return new Promise(function (resolve) {
web3.provider.send({call: property.setter, args: args}, function (err, result) {
if (!err) {
resolve(result);
return;
}
reject(err);
});
});
}).catch(function (err) {
console.error(err);
});
};
}
Object.defineProperty(obj, property.name, proto);
});
};
var web3 = {
_callbacks: {},
_events: {},
providers: {},
toHex: function(str) {
var hex = "";
for(var i = 0; i < str.length; i++) {
var n = str.charCodeAt(i).toString(16);
hex += n.length < 2 ? '0' + n : n;
}
return hex;
},
toAscii: function(hex) {
// Find termination
var str = "";
var i = 0, l = hex.length;
if (hex.substring(0, 2) === '0x')
i = 2;
for(; i < l; i+=2) {
var code = hex.charCodeAt(i);
if(code === 0) {
break;
}
str += String.fromCharCode(parseInt(hex.substr(i, 2), 16));
}
return str;
},
toDecimal: function (val) {
return parseInt(val, 16);
},
fromAscii: function(str, pad) {
pad = pad === undefined ? 32 : pad;
var hex = this.toHex(str);
while(hex.length < pad*2)
hex += "00";
return "0x" + hex;
},
eth: {
prototype: Object(), // jshint ignore:line
watch: function (params) {
return new Filter(params, ethWatch);
}
},
db: {
prototype: Object() // jshint ignore:line
},
shh: {
prototype: Object(), // jshint ignore:line
watch: function (params) {
return new Filter(params, shhWatch);
}
},
on: function(event, id, cb) {
if(web3._events[event] === undefined) {
web3._events[event] = {};
}
web3._events[event][id] = cb;
return this;
},
off: function(event, id) { return hex;
if(web3._events[event] !== undefined) { },
delete web3._events[event][id];
toAscii: function(hex) {
// Find termination
var str = "";
var i = 0, l = hex.length;
if (hex.substring(0, 2) === '0x')
i = 2;
for(; i < l; i+=2) {
var code = hex.charCodeAt(i);
if(code === 0) {
break;
} }
return this; str += String.fromCharCode(parseInt(hex.substr(i, 2), 16));
},
trigger: function(event, id, data) {
var callbacks = web3._events[event];
if (!callbacks || !callbacks[id]) {
return;
}
var cb = callbacks[id];
cb(data);
} }
};
var eth = web3.eth;
setupMethods(eth, ethMethods());
setupProperties(eth, ethProperties());
setupMethods(web3.db, dbMethods());
setupMethods(web3.shh, shhMethods());
var ethWatch = { return str;
changed: 'eth_changed' },
};
setupMethods(ethWatch, ethWatchMethods()); toDecimal: function (val) {
var shhWatch = { return parseInt(val, 16);
changed: 'shh_changed' },
};
setupMethods(shhWatch, shhWatchMethods()); fromAscii: function(str, pad) {
pad = pad === undefined ? 32 : pad;
var ProviderManager = function() { var hex = this.toHex(str);
this.queued = []; while(hex.length < pad*2)
this.polls = []; hex += "00";
this.ready = false; return "0x" + hex;
this.provider = undefined; },
this.id = 1;
eth: {
var self = this; prototype: Object(), // jshint ignore:line
var poll = function () { watch: function (params) {
if (self.provider && self.provider.poll) { return new Filter(params, ethWatch);
self.polls.forEach(function (data) {
data.data._id = self.id;
self.id++;
self.provider.poll(data.data, data.id);
});
}
setTimeout(poll, 12000);
};
poll();
};
ProviderManager.prototype.send = function(data, cb) {
data._id = this.id;
if (cb) {
web3._callbacks[data._id] = cb;
} }
},
data.args = data.args || []; db: {
this.id++; prototype: Object() // jshint ignore:line
},
if(this.provider !== undefined) { shh: {
this.provider.send(data); prototype: Object(), // jshint ignore:line
} else { watch: function (params) {
console.warn("provider is not set"); return new Filter(params, shhWatch);
this.queued.push(data);
} }
}; },
ProviderManager.prototype.set = function(provider) { on: function(event, id, cb) {
if(this.provider !== undefined && this.provider.unload !== undefined) { if(web3._events[event] === undefined) {
this.provider.unload(); web3._events[event] = {};
} }
this.provider = provider; web3._events[event][id] = cb;
this.ready = true; return this;
}; },
ProviderManager.prototype.sendQueued = function() { off: function(event, id) {
for(var i = 0; this.queued.length; i++) { if(web3._events[event] !== undefined) {
// Resend delete web3._events[event][id];
this.send(this.queued[i]);
} }
};
ProviderManager.prototype.installed = function() { return this;
return this.provider !== undefined; },
};
ProviderManager.prototype.startPolling = function (data, pollId) { trigger: function(event, id, data) {
if (!this.provider || !this.provider.poll) { var callbacks = web3._events[event];
if (!callbacks || !callbacks[id]) {
return; return;
} }
this.polls.push({data: data, id: pollId}); var cb = callbacks[id];
}; cb(data);
}
ProviderManager.prototype.stopPolling = function (pollId) { };
for (var i = this.polls.length; i--;) {
var poll = this.polls[i]; var eth = web3.eth;
if (poll.id === pollId) { setupMethods(eth, ethMethods());
this.polls.splice(i, 1); setupProperties(eth, ethProperties());
} setupMethods(web3.db, dbMethods());
setupMethods(web3.shh, shhMethods());
var ethWatch = {
changed: 'eth_changed'
};
setupMethods(ethWatch, ethWatchMethods());
var shhWatch = {
changed: 'shh_changed'
};
setupMethods(shhWatch, shhWatchMethods());
var ProviderManager = function() {
this.queued = [];
this.polls = [];
this.ready = false;
this.provider = undefined;
this.id = 1;
var self = this;
var poll = function () {
if (self.provider && self.provider.poll) {
self.polls.forEach(function (data) {
data.data._id = self.id;
self.id++;
self.provider.poll(data.data, data.id);
});
} }
setTimeout(poll, 12000);
}; };
poll();
};
web3.provider = new ProviderManager(); ProviderManager.prototype.send = function(data, cb) {
data._id = this.id;
web3.setProvider = function(provider) { if (cb) {
provider.onmessage = messageHandler; web3._callbacks[data._id] = cb;
web3.provider.set(provider); }
web3.provider.sendQueued();
};
var Filter = function(options, impl) { data.args = data.args || [];
this.impl = impl; this.id++;
this.callbacks = [];
var self = this; if(this.provider !== undefined) {
this.promise = impl.newFilter(options); this.provider.send(data);
this.promise.then(function (id) { } else {
self.id = id; console.warn("provider is not set");
web3.on(impl.changed, id, self.trigger.bind(self)); this.queued.push(data);
web3.provider.startPolling({call: impl.changed, args: [id]}, id); }
}); };
};
Filter.prototype.arrived = function(callback) { ProviderManager.prototype.set = function(provider) {
this.changed(callback); if(this.provider !== undefined && this.provider.unload !== undefined) {
}; this.provider.unload();
}
Filter.prototype.changed = function(callback) { this.provider = provider;
var self = this; this.ready = true;
this.promise.then(function(id) { };
self.callbacks.push(callback);
});
};
Filter.prototype.trigger = function(messages) { ProviderManager.prototype.sendQueued = function() {
for(var i = 0; i < this.callbacks.length; i++) { for(var i = 0; this.queued.length; i++) {
this.callbacks[i].call(this, messages); // Resend
} this.send(this.queued[i]);
}; }
};
Filter.prototype.uninstall = function() {
var self = this;
this.promise.then(function (id) {
self.impl.uninstallFilter(id);
web3.provider.stopPolling(id);
web3.off(impl.changed, id);
});
};
Filter.prototype.messages = function() { ProviderManager.prototype.installed = function() {
var self = this; return this.provider !== undefined;
return this.promise.then(function (id) { };
return self.impl.getMessages(id);
});
};
function messageHandler(data) { ProviderManager.prototype.startPolling = function (data, pollId) {
if(data._event !== undefined) { if (!this.provider || !this.provider.poll) {
web3.trigger(data._event, data._id, data.data); return;
return; }
this.polls.push({data: data, id: pollId});
};
ProviderManager.prototype.stopPolling = function (pollId) {
for (var i = this.polls.length; i--;) {
var poll = this.polls[i];
if (poll.id === pollId) {
this.polls.splice(i, 1);
} }
}
};
web3.provider = new ProviderManager();
web3.setProvider = function(provider) {
provider.onmessage = messageHandler;
web3.provider.set(provider);
web3.provider.sendQueued();
};
var Filter = function(options, impl) {
this.impl = impl;
this.callbacks = [];
var self = this;
this.promise = impl.newFilter(options);
this.promise.then(function (id) {
self.id = id;
web3.on(impl.changed, id, self.trigger.bind(self));
web3.provider.startPolling({call: impl.changed, args: [id]}, id);
});
};
Filter.prototype.arrived = function(callback) {
this.changed(callback);
};
Filter.prototype.changed = function(callback) {
var self = this;
this.promise.then(function(id) {
self.callbacks.push(callback);
});
};
Filter.prototype.trigger = function(messages) {
for(var i = 0; i < this.callbacks.length; i++) {
this.callbacks[i].call(this, messages);
}
};
Filter.prototype.uninstall = function() {
var self = this;
this.promise.then(function (id) {
self.impl.uninstallFilter(id);
web3.provider.stopPolling(id);
web3.off(impl.changed, id);
});
};
Filter.prototype.messages = function() {
var self = this;
return this.promise.then(function (id) {
return self.impl.getMessages(id);
});
};
function messageHandler(data) {
if(data._event !== undefined) {
web3.trigger(data._event, data._id, data.data);
return;
}
if(data._id) { if(data._id) {
var cb = web3._callbacks[data._id]; var cb = web3._callbacks[data._id];
if (cb) { if (cb) {
cb.call(this, data.error, data.data); cb.call(this, data.error, data.data);
delete web3._callbacks[data._id]; delete web3._callbacks[data._id];
}
} }
} }
}
module.exports = web3; module.exports = web3;

33
lib/qt.js

@ -16,29 +16,30 @@
*/ */
/** @file qt.js /** @file qt.js
* @authors: * @authors:
* Jeffrey Wilcke <jeff@ethdev.com>
* Marek Kotewicz <marek@ethdev.com> * Marek Kotewicz <marek@ethdev.com>
* @date 2014 * @date 2014
*/ */
var QtProvider = function() { var QtProvider = function() {
this.handlers = []; this.handlers = [];
var self = this; var self = this;
navigator.qt.onmessage = function (message) { navigator.qt.onmessage = function (message) {
self.handlers.forEach(function (handler) { self.handlers.forEach(function (handler) {
handler.call(self, JSON.parse(message.data)); handler.call(self, JSON.parse(message.data));
}); });
};
}; };
};
QtProvider.prototype.send = function(payload) { QtProvider.prototype.send = function(payload) {
navigator.qt.postMessage(JSON.stringify(payload)); navigator.qt.postMessage(JSON.stringify(payload));
}; };
Object.defineProperty(QtProvider.prototype, "onmessage", { Object.defineProperty(QtProvider.prototype, "onmessage", {
set: function(handler) { set: function(handler) {
this.handlers.push(handler); this.handlers.push(handler);
} }
}); });
module.exports = QtProvider; module.exports = QtProvider;

78
lib/websocket.js

@ -16,59 +16,61 @@
*/ */
/** @file websocket.js /** @file websocket.js
* @authors: * @authors:
* Jeffrey Wilcke <jeff@ethdev.com>
* Marek Kotewicz <marek@ethdev.com> * Marek Kotewicz <marek@ethdev.com>
* Marian Oancea <marian@ethdev.com> * Marian Oancea <marian@ethdev.com>
* @date 2014 * @date 2014
*/ */
if(process.env.NODE_ENV !== "build") { if (process.env.NODE_ENV !== "build") {
var WebSocket = require('ws'); // jshint ignore:line var WebSocket = require('ws'); // jshint ignore:line
} }
var WebSocketProvider = function(host) { var WebSocketProvider = function(host) {
// onmessage handlers // onmessage handlers
this.handlers = []; this.handlers = [];
// queue will be filled with messages if send is invoked before the ws is ready // queue will be filled with messages if send is invoked before the ws is ready
this.queued = []; this.queued = [];
this.ready = false; this.ready = false;
this.ws = new WebSocket(host); this.ws = new WebSocket(host);
var self = this; var self = this;
this.ws.onmessage = function(event) { this.ws.onmessage = function(event) {
for(var i = 0; i < self.handlers.length; i++) { for(var i = 0; i < self.handlers.length; i++) {
self.handlers[i].call(self, JSON.parse(event.data), event); self.handlers[i].call(self, JSON.parse(event.data), event);
} }
};
this.ws.onopen = function() {
self.ready = true;
for(var i = 0; i < self.queued.length; i++) {
// Resend
self.send(self.queued[i]);
}
};
}; };
WebSocketProvider.prototype.send = function(payload) {
if(this.ready) {
var data = JSON.stringify(payload);
this.ws.send(data); this.ws.onopen = function() {
} else { self.ready = true;
this.queued.push(payload);
for(var i = 0; i < self.queued.length; i++) {
// Resend
self.send(self.queued[i]);
} }
}; };
};
WebSocketProvider.prototype.onMessage = function(handler) { WebSocketProvider.prototype.send = function(payload) {
this.handlers.push(handler); if(this.ready) {
}; var data = JSON.stringify(payload);
WebSocketProvider.prototype.unload = function() { this.ws.send(data);
this.ws.close(); } else {
}; this.queued.push(payload);
Object.defineProperty(WebSocketProvider.prototype, "onmessage", { }
set: function(provider) { this.onMessage(provider); } };
});
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); }
});
module.exports = WebSocketProvider; module.exports = WebSocketProvider;

Loading…
Cancel
Save