mirror of https://github.com/embarklabs/embark.git
105 lines
3.9 KiB
JavaScript
105 lines
3.9 KiB
JavaScript
const bip39 = require("bip39");
|
|
const hdkey = require('ethereumjs-wallet/hdkey');
|
|
const ethereumjsWallet = require('ethereumjs-wallet');
|
|
const fs = require('../core/fs');
|
|
const {getHexBalanceFromString} = require('../utils/utils');
|
|
|
|
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;
|
|
}
|
|
let hexBalance = null;
|
|
if (accountConfig.balance) {
|
|
hexBalance = getHexBalanceFromString(accountConfig.balance, web3);
|
|
}
|
|
|
|
if (accountConfig.privateKey === 'random') {
|
|
let randomAccount = web3.eth.accounts.create();
|
|
accountConfig.privateKey = randomAccount.privateKey;
|
|
}
|
|
|
|
if (accountConfig.privateKey) {
|
|
if (!accountConfig.privateKey.startsWith('0x')) {
|
|
accountConfig.privateKey = '0x' + accountConfig.privateKey;
|
|
}
|
|
if (!web3.utils.isHexStrict(accountConfig.privateKey)) {
|
|
logger.warn(`Private key ending with ${accountConfig.privateKey.substr(accountConfig.privateKey.length - 5)} is not a HEX string`);
|
|
return null;
|
|
}
|
|
return Object.assign(web3.eth.accounts.privateKeyToAccount(accountConfig.privateKey), {hexBalance});
|
|
}
|
|
|
|
if (accountConfig.privateKeyFile) {
|
|
let fileContent = fs.readFileSync(fs.dappPath(accountConfig.privateKeyFile)).toString();
|
|
if (accountConfig.password) {
|
|
try {
|
|
fileContent = JSON.parse(fileContent);
|
|
if (!ethereumjsWallet['fromV' + fileContent.version]) {
|
|
logger.error(`Key file ${accountConfig.privateKeyFile} is not a valid keystore file`);
|
|
return null;
|
|
}
|
|
const wallet = ethereumjsWallet['fromV' + fileContent.version](fileContent, accountConfig.password);
|
|
|
|
return Object.assign(web3.eth.accounts.privateKeyToAccount('0x' + wallet.getPrivateKey().toString('hex')), {hexBalance});
|
|
} catch (e) {
|
|
logger.error('Private key file is not a keystore JSON file but a password was provided');
|
|
logger.error(e.message || e);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
fileContent = fileContent.trim().split(/[,;]/);
|
|
return fileContent.map((key, index) => {
|
|
if (!key.startsWith('0x')) {
|
|
key = '0x' + key;
|
|
}
|
|
if (!web3.utils.isHexStrict(key)) {
|
|
logger.warn(`Private key is not a HEX string in file ${accountConfig.privateKeyFile} at index ${index}`);
|
|
return null;
|
|
}
|
|
return Object.assign(web3.eth.accounts.privateKeyToAccount(key), {hexBalance});
|
|
});
|
|
}
|
|
|
|
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(Object.assign(web3.eth.accounts.privateKeyToAccount('0x' + wallet.getPrivateKey().toString('hex')), {hexBalance}));
|
|
}
|
|
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;
|