Frances Pangilinan
8 years ago
parent
9e3fa3cfba
commit
090935f90a
@ -0,0 +1,106 @@ |
||||
const async = require('async') |
||||
const EthQuery = require('eth-query') |
||||
const ethUtil = require('ethereumjs-util') |
||||
const BN = ethUtil.BN |
||||
const ethBinToOps = require('eth-bin-to-ops') |
||||
|
||||
module.exports = class txProviderUtils { |
||||
constructor (provider) { |
||||
this.provider = provider |
||||
this.query = new EthQuery(provider) |
||||
} |
||||
analyzeGasUsage (txData, cb) { |
||||
var self = this |
||||
this.query.getBlockByNumber('latest', true, (err, block) => { |
||||
if (err) return cb(err) |
||||
async.waterfall([ |
||||
self.estimateTxGas.bind(self, txData, block.gasLimit), |
||||
self.checkForTxGasError.bind(self, txData), |
||||
self.setTxGas.bind(self, txData, block.gasLimit), |
||||
], cb) |
||||
}) |
||||
} |
||||
|
||||
// perform static analyis on the target contract code
|
||||
analyzeForDelegateCall (txParams, cb) { |
||||
if (txParams.to) { |
||||
this.query.getCode(txParams.to, function (err, result) { |
||||
if (err) return cb(err) |
||||
|
||||
var code = ethUtil.toBuffer(result) |
||||
if (code !== '0x') { |
||||
var ops = ethBinToOps(code) |
||||
var containsDelegateCall = ops.some((op) => op.name === 'DELEGATECALL') |
||||
cb(containsDelegateCall) |
||||
} else { |
||||
cb() |
||||
} |
||||
}) |
||||
} else { |
||||
cb() |
||||
} |
||||
} |
||||
|
||||
estimateTxGas (txData, blockGasLimitHex, cb) { |
||||
const txParams = txData.txParams |
||||
// check if gasLimit is already specified
|
||||
txData.gasLimitSpecified = Boolean(txParams.gas) |
||||
// if not, fallback to block gasLimit
|
||||
if (!txData.gasLimitSpecified) { |
||||
txParams.gas = blockGasLimitHex |
||||
} |
||||
// run tx, see if it will OOG
|
||||
this.query.estimateGas(txParams, cb) |
||||
} |
||||
|
||||
checkForTxGasError (txData, estimatedGasHex, cb) { |
||||
txData.estimatedGas = estimatedGasHex |
||||
// all gas used - must be an error
|
||||
if (estimatedGasHex === txData.txParams.gas) { |
||||
txData.simulationFails = true |
||||
} |
||||
cb() |
||||
} |
||||
|
||||
handleFork (block) { |
||||
|
||||
} |
||||
|
||||
setTxGas (txData, blockGasLimitHex, cb) { |
||||
const txParams = txData.txParams |
||||
// if OOG, nothing more to do
|
||||
if (txData.simulationFails) { |
||||
cb() |
||||
return |
||||
} |
||||
// if gasLimit was specified and doesnt OOG,
|
||||
// use original specified amount
|
||||
if (txData.gasLimitSpecified) { |
||||
txData.estimatedGas = txParams.gas |
||||
cb() |
||||
return |
||||
} |
||||
// if gasLimit not originally specified,
|
||||
// try adding an additional gas buffer to our estimation for safety
|
||||
const estimatedGasBn = new BN(ethUtil.stripHexPrefix(txData.estimatedGas), 16) |
||||
const blockGasLimitBn = new BN(ethUtil.stripHexPrefix(blockGasLimitHex), 16) |
||||
const estimationWithBuffer = new BN(this.addGasBuffer(estimatedGasBn), 16) |
||||
// added gas buffer is too high
|
||||
if (estimationWithBuffer.gt(blockGasLimitBn)) { |
||||
txParams.gas = txData.estimatedGas |
||||
// added gas buffer is safe
|
||||
} else { |
||||
const gasWithBufferHex = ethUtil.intToHex(estimationWithBuffer) |
||||
txParams.gas = gasWithBufferHex |
||||
} |
||||
cb() |
||||
return |
||||
} |
||||
|
||||
addGasBuffer (gas) { |
||||
const gasBuffer = new BN('100000', 10) |
||||
const bnGas = new BN(ethUtil.stripHexPrefix(gas), 16) |
||||
const correct = bnGas.add(gasBuffer) |
||||
return ethUtil.addHexPrefix(correct.toString(16)) |
||||
} |
||||
} |
||||
@ -0,0 +1,179 @@ |
||||
const EventEmitter = require('events') |
||||
const extend = require('xtend') |
||||
const TxProviderUtil = require('./lib/provider-utils') |
||||
|
||||
module.exports = class TransactionManager extends EventEmitter { |
||||
constructor (opts) { |
||||
super() |
||||
this.txList = opts.TxListFromStore || [] |
||||
this._persistTxList = opts.setTxList |
||||
this._unconfTxCbs = {} |
||||
this.txLimit = opts.txLimit |
||||
this.provider = opts.provider |
||||
} |
||||
|
||||
// Returns the tx list
|
||||
getTxList () { |
||||
return this.txList |
||||
} |
||||
|
||||
// Saves the new/updated txList.
|
||||
// Function is intended only for internal use
|
||||
_saveTxList (txList) { |
||||
this.txList = txList |
||||
this._persistTxList(txList) |
||||
} |
||||
|
||||
// Adds a tx to the txlist
|
||||
addTx (txData, onTxDoneCb) { |
||||
var txList = this.getTxList() |
||||
var txLimit = this.txLimit |
||||
if (txList.length > txLimit - 1) { |
||||
txList.shift() |
||||
} |
||||
txList.push(txData) |
||||
this._saveTxList(txList) |
||||
this.addOnTxDoneCb(txData.id, onTxDoneCb) |
||||
this.emit('unapproved', txData) |
||||
this.emit('update') |
||||
} |
||||
|
||||
getTx (txId, cb) { |
||||
var txList = this.getTxList() |
||||
var tx = txList.find((tx) => tx.id === txId) |
||||
return cb ? cb(tx) : tx |
||||
} |
||||
|
||||
updateTx (txData) { |
||||
var txId = txData.id |
||||
var txList = this.getTxList() |
||||
|
||||
var updatedTxList = txList.map((tx) => { |
||||
if (tx.id === txId) { |
||||
tx = txData |
||||
} |
||||
return tx |
||||
}) |
||||
this._saveTxList(updatedTxList) |
||||
} |
||||
|
||||
get unConftxCount () { |
||||
return Object.keys(this.getUnapprovedTxList()).length |
||||
} |
||||
|
||||
get pendingTxCount () { |
||||
return this.getTxsByMetaData('status', 'signed').length |
||||
} |
||||
|
||||
getUnapprovedTxList () { |
||||
var txList = this.getTxList() |
||||
return txList.filter((tx) => { |
||||
return tx.status === 'unapproved' |
||||
}).reduce((result, tx) => { |
||||
result[tx.id] = tx |
||||
return result |
||||
}, {}) |
||||
} |
||||
|
||||
getFilterdTxList (opts) { |
||||
var filteredTxList |
||||
Object.keys(opts).forEach((key) => { |
||||
filteredTxList = this.getTxsByMetaData(key, opts[key], filteredTxList) |
||||
}) |
||||
return filteredTxList |
||||
} |
||||
|
||||
getTxsByMetaData (key, value, txList = this.getTxList()) { |
||||
return txList.filter((tx) => { |
||||
if (key in tx.txParams) { |
||||
return tx.txParams[key] === value |
||||
} else { |
||||
return tx[key] === value |
||||
} |
||||
}) |
||||
} |
||||
|
||||
addOnTxDoneCb (txId, cb) { |
||||
this._unconfTxCbs[txId] = cb || noop |
||||
} |
||||
|
||||
// should return the tx
|
||||
|
||||
// Should find the tx in the tx list and
|
||||
// update it.
|
||||
// should set the status in txData
|
||||
// // - `'unapproved'` the user has not responded
|
||||
// // - `'rejected'` the user has responded no!
|
||||
// // - `'signed'` the tx is signed
|
||||
// // - `'submitted'` the tx is sent to a server
|
||||
// // - `'confirmed'` the tx has been included in a block.
|
||||
setTxStatus (txId, status) { |
||||
var txData = this.getTx(txId) |
||||
txData.status = status |
||||
this.emit(status, txId) |
||||
this.updateTx(txData, status) |
||||
} |
||||
|
||||
|
||||
// should return the status of the tx.
|
||||
getTxStatus (txId, cb) { |
||||
const txData = this.getTx(txId) |
||||
return cb ? cb(txData.staus) : txData.status |
||||
} |
||||
|
||||
|
||||
// should update the status of the tx to 'signed'.
|
||||
setTxStatusSigned (txId) { |
||||
this.setTxStatus(txId, 'signed') |
||||
this.emit('update') |
||||
} |
||||
|
||||
// should update the status of the tx to 'rejected'.
|
||||
setTxStatusRejected (txId) { |
||||
this.setTxStatus(txId, 'rejected') |
||||
this.emit('update') |
||||
} |
||||
|
||||
setTxStatusConfirmed (txId) { |
||||
this.setTxStatus(txId, 'confirmed') |
||||
// this.removeListener(`check${txId}`, this.checkForTxInBlock)
|
||||
} |
||||
|
||||
// merges txParams obj onto txData.txParams
|
||||
// use extend to ensure that all fields are filled
|
||||
updateTxParams (txId, txParams) { |
||||
var txData = this.getTx(txId) |
||||
txData.txParams = extend(txData, txParams) |
||||
this.updateTx(txData) |
||||
} |
||||
|
||||
setProvider (provider) { |
||||
this.provider = provider |
||||
this.txProviderUtils = new TxProviderUtil(provider) |
||||
this.provider.on('block', this.checkForTxInBlock.bind(this)) |
||||
} |
||||
|
||||
checkForTxInBlock () { |
||||
var signedTxList = this.getFilterdTxList({status: 'signed'}) |
||||
if (!signedTxList.length) return |
||||
var self = this |
||||
signedTxList.forEach((tx) => { |
||||
var txHash = tx.hash |
||||
var txId = tx.id |
||||
if (!txHash) return |
||||
// var d
|
||||
this.txProviderUtils.query.getTransactionByHash(txHash, (err, txData) => { |
||||
if (err) { |
||||
tx |
||||
|
||||
return console.error(err) |
||||
} |
||||
if (txData.blockNumber !== null) { |
||||
self.setTxStatusConfirmed(txId) |
||||
} |
||||
}) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
function noop () {} |
||||
Loading…
Reference in new issue