This repository has been archived by the owner on Jan 9, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1
/
private-provider.js
83 lines (65 loc) · 2.5 KB
/
private-provider.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
const ProviderEngine = require("web3-provider-engine");
const WalletSubprovider = require('web3-provider-engine/subproviders/wallet');
const RpcSubprovider = require('web3-provider-engine/subproviders/rpc');
const EthereumjsWallet = require('ethereumjs-wallet');
function ChainIdSubProvider(chainId) {
this.chainId = chainId;
}
ChainIdSubProvider.prototype.setEngine = function(engine) {
const self = this
if (self.engine) return
self.engine = engine
}
ChainIdSubProvider.prototype.handleRequest = function(payload, next, end) {
if (payload.method == "eth_sendTransaction" && payload.params.length > 0 && typeof payload.params[0].chainId == "undefined") {
payload.params[0].chainId = this.chainId;
}
next()
}
function NonceSubProvider() {
}
NonceSubProvider.prototype.setEngine = function(engine) {
const self = this
if (self.engine) return
self.engine = engine
}
NonceSubProvider.prototype.handleRequest = function(payload, next, end) {
if (payload.method == "eth_sendTransaction") {
this.engine.sendAsync({
jsonrpc:"2.0",
id: Math.ceil(Math.random() * 4415011859092441),
method: "eth_getTransactionCount",
params: [payload.params[0].from, "latest"]
}, (err, result) => {
const nonce = typeof result.result == "string" ?
result.result == "0x" ? 0 : parseInt(result.result.substring(2), 16) : 0;
payload.params[0].nonce = '0x' + (nonce || 0);
next();
})
} else {
next()
}
}
function PrivateKeyProvider(privateKey, providerUrl, chainId) {
if (!privateKey) {
throw new Error(`Private Key missing, non-empty string expected, got "${privateKey}"`);
}
if (!providerUrl) {
throw new Error(`Provider URL missing, non-empty string expected, got "${providerUrl}"`);
}
this.wallet = EthereumjsWallet.default.fromPrivateKey(new Buffer(privateKey, "hex"));
this.address = "0x" + this.wallet.getAddress().toString("hex");
this.engine = new ProviderEngine({useSkipCache: false});
this.engine.addProvider(new ChainIdSubProvider(chainId));
this.engine.addProvider(new NonceSubProvider());
this.engine.addProvider(new WalletSubprovider(this.wallet, {}));
this.engine.addProvider(new RpcSubprovider({ rpcUrl: providerUrl }));
this.engine.start();
}
PrivateKeyProvider.prototype.sendAsync = function(payload, callback) {
return this.engine.sendAsync.apply(this.engine, arguments);
};
PrivateKeyProvider.prototype.send = function() {
return this.engine.send.apply(this.engine, arguments);
};
module.exports = PrivateKeyProvider;