refactor some unnecessary if/else conditions

pull/1/head
Iuri Matias 5 years ago
parent 8daf16f0d5
commit 9cf3f70ca2
  1. 57
      src/blockchain/blockchain.js

@ -268,7 +268,9 @@ class Blockchain {
runOrCallContractMethod (contractName, contractAbi, funABI, value, address, callType, lookupOnly, logMsg, logCallback, outputCb, confirmationCb, continueCb, promptCb) { runOrCallContractMethod (contractName, contractAbi, funABI, value, address, callType, lookupOnly, logMsg, logCallback, outputCb, confirmationCb, continueCb, promptCb) {
// contractsDetails is used to resolve libraries // contractsDetails is used to resolve libraries
txFormat.buildData(contractName, contractAbi, {}, false, funABI, callType, (error, data) => { txFormat.buildData(contractName, contractAbi, {}, false, funABI, callType, (error, data) => {
if (!error) { if (error) {
return logCallback(`${logMsg} errored: ${error} `)
}
if (!lookupOnly) { if (!lookupOnly) {
logCallback(`${logMsg} pending ... `) logCallback(`${logMsg} pending ... `)
} else { } else {
@ -276,26 +278,21 @@ class Blockchain {
} }
if (funABI.type === 'fallback') data.dataHex = value if (funABI.type === 'fallback') data.dataHex = value
this.callFunction(address, data, funABI, confirmationCb, continueCb, promptCb, (error, txResult) => { this.callFunction(address, data, funABI, confirmationCb, continueCb, promptCb, (error, txResult) => {
if (!error) { if (error) {
return logCallback(`${logMsg} errored: ${error} `)
}
const isVM = this.executionContext.isVM() const isVM = this.executionContext.isVM()
if (isVM) { if (isVM) {
const vmError = txExecution.checkVMError(txResult) const vmError = txExecution.checkVMError(txResult)
if (vmError.error) { if (vmError.error) {
logCallback(`${logMsg} errored: ${vmError.message} `) return logCallback(`${logMsg} errored: ${vmError.message} `)
return
} }
} }
if (lookupOnly) { if (lookupOnly) {
const returnValue = (this.executionContext.isVM() ? txResult.result.execResult.returnValue : ethJSUtil.toBuffer(txResult.result)) const returnValue = (this.executionContext.isVM() ? txResult.result.execResult.returnValue : ethJSUtil.toBuffer(txResult.result))
outputCb(returnValue) outputCb(returnValue)
} }
} else {
logCallback(`${logMsg} errored: ${error} `)
}
}) })
} else {
logCallback(`${logMsg} errored: ${error} `)
}
}, },
(msg) => { (msg) => {
logCallback(msg) logCallback(msg)
@ -380,21 +377,20 @@ class Blockchain {
} }
newAccount (_password, passwordPromptCb, cb) { newAccount (_password, passwordPromptCb, cb) {
if (!this.executionContext.isVM()) { if (this.executionContext.isVM()) {
if (!this.config.get('settings/personal-mode')) {
return cb('Not running in personal mode')
}
passwordPromptCb((passphrase) => {
this.executionContext.web3().personal.newAccount(passphrase, cb)
})
} else {
let privateKey let privateKey
do { do {
privateKey = crypto.randomBytes(32) privateKey = crypto.randomBytes(32)
} while (!isValidPrivate(privateKey)) } while (!isValidPrivate(privateKey))
this._addAccount(privateKey, '0x56BC75E2D63100000') this._addAccount(privateKey, '0x56BC75E2D63100000')
cb(null, '0x' + privateToAddress(privateKey).toString('hex')) return cb(null, '0x' + privateToAddress(privateKey).toString('hex'))
}
if (!this.config.get('settings/personal-mode')) {
return cb('Not running in personal mode')
} }
passwordPromptCb((passphrase) => {
this.executionContext.web3().personal.newAccount(passphrase, cb)
})
} }
/** Add an account to the list of account (only for Javascript VM) */ /** Add an account to the list of account (only for Javascript VM) */
@ -429,8 +425,7 @@ class Blockchain {
case 'vm': { case 'vm': {
if (!this.accounts) { if (!this.accounts) {
if (cb) cb('No accounts?') if (cb) cb('No accounts?')
reject('No accounts?') return reject('No accounts?')
return
} }
if (cb) cb(null, Object.keys(this.accounts)) if (cb) cb(null, Object.keys(this.accounts))
resolve(Object.keys(this.accounts)) resolve(Object.keys(this.accounts))
@ -468,37 +463,33 @@ class Blockchain {
address = stripHexPrefix(address) address = stripHexPrefix(address)
if (!this.executionContext.isVM()) { if (!this.executionContext.isVM()) {
this.executionContext.web3().eth.getBalance(address, (err, res) => { return this.executionContext.web3().eth.getBalance(address, (err, res) => {
if (err) { if (err) {
cb(err) return cb(err)
} else {
cb(null, res.toString(10))
} }
cb(null, res.toString(10))
}) })
} else { }
if (!this.accounts) { if (!this.accounts) {
return cb('No accounts?') return cb('No accounts?')
} }
this.executionContext.vm().stateManager.getAccount(Buffer.from(address, 'hex'), (err, res) => { this.executionContext.vm().stateManager.getAccount(Buffer.from(address, 'hex'), (err, res) => {
if (err) { if (err) {
cb('Account not found') return cb('Account not found')
} else {
cb(null, new BN(res.balance).toString(10))
} }
cb(null, new BN(res.balance).toString(10))
}) })
} }
}
/** Get the balance of an address, and convert wei to ether */ /** Get the balance of an address, and convert wei to ether */
getBalanceInEther (address, callback) { getBalanceInEther (address, callback) {
this.getBalance(address, (error, balance) => { this.getBalance(address, (error, balance) => {
if (error) { if (error) {
callback(error) return callback(error)
} else { }
// callback(null, this.executionContext.web3().fromWei(balance, 'ether')) // callback(null, this.executionContext.web3().fromWei(balance, 'ether'))
callback(null, Web3.utils.fromWei(balance.toString(10), 'ether')) callback(null, Web3.utils.fromWei(balance.toString(10), 'ether'))
}
}) })
} }

Loading…
Cancel
Save