move account parsing to its own module
This commit is contained in:
parent
668fd3a064
commit
c3c4518cb4
|
@ -0,0 +1,66 @@
|
||||||
|
const bip39 = require("bip39");
|
||||||
|
const hdkey = require('ethereumjs-wallet/hdkey');
|
||||||
|
const fs = require('../core/fs');
|
||||||
|
|
||||||
|
|
||||||
|
class AccountParser {
|
||||||
|
static parseAccountsConfig(accountsConfig, web3, logger) {
|
||||||
|
let accounts = [];
|
||||||
|
if (accountsConfig && accountsConfig.length) {
|
||||||
|
accountsConfig.forEach(accountConfig => {
|
||||||
|
const account = AccountParser.getAccount(accountConfig, web3, logger);
|
||||||
|
if (!account) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (Array.isArray(account)) {
|
||||||
|
accounts = accounts.concat(account);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
accounts.push(account);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return accounts;
|
||||||
|
}
|
||||||
|
|
||||||
|
static getAccount(accountConfig, web3, logger) {
|
||||||
|
if (!logger) {
|
||||||
|
logger = console;
|
||||||
|
}
|
||||||
|
if (accountConfig.privateKey) {
|
||||||
|
if (!accountConfig.privateKey.startsWith('0x')) {
|
||||||
|
accountConfig.privateKey = '0x' + accountConfig.privateKey;
|
||||||
|
}
|
||||||
|
return web3.eth.accounts.privateKeyToAccount(accountConfig.privateKey);
|
||||||
|
}
|
||||||
|
if (accountConfig.privateKeyFile) {
|
||||||
|
let fileContent = fs.readFileSync(fs.dappPath(accountConfig.privateKeyFile)).toString();
|
||||||
|
fileContent = fileContent.trim().split(/[,;]/);
|
||||||
|
return fileContent.map(key => {
|
||||||
|
if (!key.startsWith('0x')) {
|
||||||
|
key = '0x' + key;
|
||||||
|
}
|
||||||
|
return web3.eth.accounts.privateKeyToAccount(key);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (accountConfig.mnemonic) {
|
||||||
|
const hdwallet = hdkey.fromMasterSeed(bip39.mnemonicToSeed(accountConfig.mnemonic.trim()));
|
||||||
|
|
||||||
|
const addressIndex = accountConfig.addressIndex || 0;
|
||||||
|
const numAddresses = accountConfig.numAddresses || 1;
|
||||||
|
const wallet_hdpath = accountConfig.hdpath || "m/44'/60'/0'/0/";
|
||||||
|
|
||||||
|
const accounts = [];
|
||||||
|
for (let i = addressIndex; i < addressIndex + numAddresses; i++) {
|
||||||
|
const wallet = hdwallet.derivePath(wallet_hdpath + i).getWallet();
|
||||||
|
accounts.push(web3.eth.accounts.privateKeyToAccount('0x' + wallet.getPrivateKey().toString('hex')));
|
||||||
|
}
|
||||||
|
return accounts;
|
||||||
|
}
|
||||||
|
logger.warn('Unsupported account configuration: ' + JSON.stringify(accountConfig));
|
||||||
|
logger.warn('Try using one of those: ' +
|
||||||
|
'{ "privateKey": "your-private-key", "privateKeyFile": "path/to/file/containing/key", "mnemonic": "12 word mnemonic" }');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = AccountParser;
|
|
@ -1,8 +1,6 @@
|
||||||
const ProviderEngine = require('web3-provider-engine');
|
const ProviderEngine = require('web3-provider-engine');
|
||||||
const RpcSubprovider = require('web3-provider-engine/subproviders/rpc.js');
|
const RpcSubprovider = require('web3-provider-engine/subproviders/rpc.js');
|
||||||
const bip39 = require("bip39");
|
const AccountParser = require('./accountParser');
|
||||||
const hdkey = require('ethereumjs-wallet/hdkey');
|
|
||||||
const fs = require('../core/fs');
|
|
||||||
|
|
||||||
class Provider {
|
class Provider {
|
||||||
constructor(options) {
|
constructor(options) {
|
||||||
|
@ -18,35 +16,18 @@ class Provider {
|
||||||
rpcUrl: options.web3Endpoint
|
rpcUrl: options.web3Endpoint
|
||||||
}));
|
}));
|
||||||
|
|
||||||
if (this.accountsConfig && this.accountsConfig.length) {
|
this.accounts = AccountParser.parseAccountsConfig(this.accountsConfig, this.web3, this.logger);
|
||||||
this.accounts = [];
|
this.addresses = [];
|
||||||
this.addresses = [];
|
if (this.accounts.length) {
|
||||||
this.accountsConfig.forEach(accountConfig => {
|
this.accounts.forEach(account => {
|
||||||
const account = this.getAccount(accountConfig);
|
|
||||||
if (!account) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (Array.isArray(account)) {
|
|
||||||
this.accounts = this.accounts.concat(account);
|
|
||||||
account.forEach(acc => {
|
|
||||||
this.addresses.push(acc.address);
|
|
||||||
});
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
this.accounts.push(account);
|
|
||||||
this.addresses.push(account.address);
|
this.addresses.push(account.address);
|
||||||
|
this.web3.eth.accounts.wallet.add(account);
|
||||||
});
|
});
|
||||||
|
this.asyncMethods = {
|
||||||
if (this.accounts.length) {
|
eth_accounts: self.eth_accounts.bind(this)
|
||||||
this.accounts.forEach(account => {
|
};
|
||||||
this.web3.eth.accounts.wallet.add(account);
|
if (this.isDev) {
|
||||||
});
|
this.logger.warn('You are using your own account in the develop environment. It might not be funded.');
|
||||||
this.asyncMethods = {
|
|
||||||
eth_accounts: self.eth_accounts.bind(this)
|
|
||||||
};
|
|
||||||
if (this.isDev) {
|
|
||||||
this.logger.warn('You are using your own account in the develop environment. It might not be funded.');
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -58,43 +39,6 @@ class Provider {
|
||||||
this.engine.start();
|
this.engine.start();
|
||||||
}
|
}
|
||||||
|
|
||||||
getAccount(accountConfig) {
|
|
||||||
if (accountConfig.privateKey) {
|
|
||||||
if (!accountConfig.privateKey.startsWith('0x')) {
|
|
||||||
accountConfig.privateKey = '0x' + accountConfig.privateKey;
|
|
||||||
}
|
|
||||||
return this.web3.eth.accounts.privateKeyToAccount(accountConfig.privateKey);
|
|
||||||
}
|
|
||||||
if (accountConfig.privateKeyFile) {
|
|
||||||
let fileContent = fs.readFileSync(fs.dappPath(accountConfig.privateKeyFile)).toString();
|
|
||||||
fileContent = fileContent.trim().split(/[,;]/);
|
|
||||||
return fileContent.map(key => {
|
|
||||||
if (!key.startsWith('0x')) {
|
|
||||||
key = '0x' + key;
|
|
||||||
}
|
|
||||||
return this.web3.eth.accounts.privateKeyToAccount(key);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
if (accountConfig.mnemonic) {
|
|
||||||
const hdwallet = hdkey.fromMasterSeed(bip39.mnemonicToSeed(accountConfig.mnemonic.trim()));
|
|
||||||
|
|
||||||
const addressIndex = accountConfig.addressIndex || 0;
|
|
||||||
const numAddresses = accountConfig.numAddresses || 1;
|
|
||||||
const wallet_hdpath = accountConfig.hdpath || "m/44'/60'/0'/0/";
|
|
||||||
|
|
||||||
const accounts = [];
|
|
||||||
for (let i = addressIndex; i < addressIndex + numAddresses; i++) {
|
|
||||||
const wallet = hdwallet.derivePath(wallet_hdpath + i).getWallet();
|
|
||||||
accounts.push(this.web3.eth.accounts.privateKeyToAccount('0x' + wallet.getPrivateKey().toString('hex')));
|
|
||||||
}
|
|
||||||
return accounts;
|
|
||||||
}
|
|
||||||
this.logger.warn('Unsupported account configuration: ' + JSON.stringify(accountConfig));
|
|
||||||
this.logger.warn('Try using one of those: ' +
|
|
||||||
'{ "privateKey": "your-private-key", "privateKeyFile": "path/to/file/containing/key", "mnemonic": "12 word mnemonic" }');
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
eth_accounts(payload, cb) {
|
eth_accounts(payload, cb) {
|
||||||
return cb(null, this.addresses);
|
return cb(null, this.addresses);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue