mirror of https://github.com/embarklabs/embark.git
112 lines
3.0 KiB
JavaScript
112 lines
3.0 KiB
JavaScript
/*global EmbarkJS, Web3, __MessageEvents, sendMessage, listenTo*/
|
|
|
|
// for the whisper v5 and web3.js 1.0
|
|
let __embarkWhisperNewWeb3 = {};
|
|
|
|
__embarkWhisperNewWeb3.setProvider = function(options) {
|
|
const self = this;
|
|
let provider;
|
|
if (options === undefined) {
|
|
provider = "localhost:8546";
|
|
} else {
|
|
provider = options.server + ':' + options.port;
|
|
}
|
|
// TODO: take into account type
|
|
self.web3 = new Web3(new Web3.providers.WebsocketProvider("ws://" + provider, options.providerOptions));
|
|
self.web3.currentProvider.on('connect', () => {
|
|
self.getWhisperVersion(function(err, version) {
|
|
if (err) {
|
|
console.log("whisper not available");
|
|
} else if (version >= 5) {
|
|
self.web3.shh.newSymKey().then((id) => {
|
|
self.symKeyID = id;
|
|
});
|
|
self.web3.shh.newKeyPair().then((id) => {
|
|
self.sig = id;
|
|
});
|
|
} else {
|
|
throw new Error("version of whisper not supported");
|
|
}
|
|
self.whisperVersion = self.web3.version.whisper;
|
|
});
|
|
});
|
|
self.web3.currentProvider.on('error', () => {
|
|
console.log("whisper not available");
|
|
});
|
|
};
|
|
|
|
__embarkWhisperNewWeb3.sendMessage = function(options) {
|
|
const data = options.data || options.payload;
|
|
if (!data) {
|
|
throw new Error("missing option: data");
|
|
}
|
|
Object.assign(options, {
|
|
sig: this.sig,
|
|
fromAscii: EmbarkJS.Utils.fromAscii,
|
|
toHex: this.web3.utils.toHex,
|
|
symKeyID: options.symKeyID || this.symKeyID,
|
|
post: this.web3.shh.post,
|
|
data
|
|
});
|
|
|
|
sendMessage(options, (err) => {
|
|
if (err) {
|
|
throw new Error(err);
|
|
}
|
|
});
|
|
};
|
|
|
|
__embarkWhisperNewWeb3.listenTo = function (options, callback) {
|
|
Object.assign(options, {
|
|
sig: this.sig,
|
|
toAscii: EmbarkJS.Utils.toAscii,
|
|
toHex: this.web3.utils.toHex,
|
|
symKeyID: options.symKeyID || this.symKeyID,
|
|
messageEvents: __MessageEvents,
|
|
subscribe: this.web3.shh.subscribe
|
|
});
|
|
listenTo(options, callback);
|
|
};
|
|
|
|
__embarkWhisperNewWeb3.getWhisperVersion = function(cb) {
|
|
// 1) Parity does not implement shh_version JSON-RPC method
|
|
// 2) web3 1.0 still does not implement web3_clientVersion
|
|
// so we must do all by our own
|
|
const self = this;
|
|
self.web3._requestManager.send({method: 'web3_clientVersion', params: []}, (err, clientVersion) => {
|
|
if (err) return cb(err);
|
|
if (clientVersion.indexOf("Parity-Ethereum//v2") === 0) {
|
|
// This is Parity
|
|
self.web3.shh.getInfo(function(err) {
|
|
if (err) {
|
|
return cb(err, 0);
|
|
}
|
|
// TOFIX Assume Whisper v6 until there's a way to understand it via JSON-RPC
|
|
return cb(err, 6);
|
|
});
|
|
} else {
|
|
// Assume it is a Geth compliant client
|
|
self.web3.shh.getVersion(function(err, version) {
|
|
cb(err, version);
|
|
});
|
|
}
|
|
});
|
|
};
|
|
|
|
__embarkWhisperNewWeb3.isAvailable = function() {
|
|
return new Promise((resolve, reject) => {
|
|
if (!this.web3.shh) {
|
|
return resolve(false);
|
|
}
|
|
try {
|
|
this.getWhisperVersion((err) => {
|
|
resolve(Boolean(!err));
|
|
});
|
|
}
|
|
catch (err) {
|
|
reject(err);
|
|
}
|
|
});
|
|
};
|
|
|