81 lines
2.1 KiB
JavaScript
81 lines
2.1 KiB
JavaScript
const ObservableStore = require('obs-store')
|
|
const PendingBalanceCalculator = require('../lib/pending-balance-calculator')
|
|
const BN = require('ethereumjs-util').BN
|
|
|
|
class BalanceController {
|
|
|
|
constructor (opts = {}) {
|
|
this._validateParams(opts)
|
|
const { address, accountTracker, txController, blockTracker } = opts
|
|
|
|
this.address = address
|
|
this.accountTracker = accountTracker
|
|
this.txController = txController
|
|
this.blockTracker = blockTracker
|
|
|
|
const initState = {
|
|
ethBalance: undefined,
|
|
}
|
|
this.store = new ObservableStore(initState)
|
|
|
|
this.balanceCalc = new PendingBalanceCalculator({
|
|
getBalance: () => this._getBalance(),
|
|
getPendingTransactions: this._getPendingTransactions.bind(this),
|
|
})
|
|
|
|
this._registerUpdates()
|
|
}
|
|
|
|
async updateBalance () {
|
|
const balance = await this.balanceCalc.getBalance()
|
|
this.store.updateState({
|
|
ethBalance: balance,
|
|
})
|
|
}
|
|
|
|
_registerUpdates () {
|
|
const update = this.updateBalance.bind(this)
|
|
|
|
this.txController.on('tx:status-update', (txId, status) => {
|
|
switch (status) {
|
|
case 'submitted':
|
|
case 'confirmed':
|
|
case 'failed':
|
|
update()
|
|
return
|
|
default:
|
|
return
|
|
}
|
|
})
|
|
this.accountTracker.store.subscribe(update)
|
|
this.blockTracker.on('block', update)
|
|
}
|
|
|
|
async _getBalance () {
|
|
const { accounts } = this.accountTracker.store.getState()
|
|
const entry = accounts[this.address]
|
|
const balance = entry.balance
|
|
return balance ? new BN(balance.substring(2), 16) : undefined
|
|
}
|
|
|
|
async _getPendingTransactions () {
|
|
const pending = this.txController.getFilteredTxList({
|
|
from: this.address,
|
|
status: 'submitted',
|
|
err: undefined,
|
|
})
|
|
return pending
|
|
}
|
|
|
|
_validateParams (opts) {
|
|
const { address, accountTracker, txController, blockTracker } = opts
|
|
if (!address || !accountTracker || !txController || !blockTracker) {
|
|
const error = 'Cannot construct a balance checker without address, accountTracker, txController, and blockTracker.'
|
|
throw new Error(error)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
module.exports = BalanceController
|