You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
88 lines
2.5 KiB
import ObservableStore from 'obs-store'
|
|
|
|
/**
|
|
* @typedef {Object} CachedBalancesOptions
|
|
* @property {Object} accountTracker An {@code AccountTracker} reference
|
|
* @property {Function} getNetwork A function to get the current network
|
|
* @property {Object} initState The initial controller state
|
|
*/
|
|
|
|
/**
|
|
* Background controller responsible for maintaining
|
|
* a cache of account balances in local storage
|
|
*/
|
|
export default class CachedBalancesController {
|
|
/**
|
|
* Creates a new controller instance
|
|
*
|
|
* @param {CachedBalancesOptions} [opts] Controller configuration parameters
|
|
*/
|
|
constructor (opts = {}) {
|
|
const { accountTracker, getNetwork } = opts
|
|
|
|
this.accountTracker = accountTracker
|
|
this.getNetwork = getNetwork
|
|
|
|
const initState = Object.assign({
|
|
cachedBalances: {},
|
|
}, opts.initState)
|
|
this.store = new ObservableStore(initState)
|
|
|
|
this._registerUpdates()
|
|
}
|
|
|
|
/**
|
|
* Updates the cachedBalances property for the current network. Cached balances will be updated to those in the passed accounts
|
|
* if balances in the passed accounts are truthy.
|
|
*
|
|
* @param {Object} obj - The the recently updated accounts object for the current network
|
|
* @returns {Promise<void>}
|
|
*/
|
|
async updateCachedBalances ({ accounts }) {
|
|
const network = await this.getNetwork()
|
|
const balancesToCache = await this._generateBalancesToCache(accounts, network)
|
|
this.store.updateState({
|
|
cachedBalances: balancesToCache,
|
|
})
|
|
}
|
|
|
|
_generateBalancesToCache (newAccounts, currentNetwork) {
|
|
const { cachedBalances } = this.store.getState()
|
|
const currentNetworkBalancesToCache = { ...cachedBalances[currentNetwork] }
|
|
|
|
Object.keys(newAccounts).forEach((accountID) => {
|
|
const account = newAccounts[accountID]
|
|
|
|
if (account.balance) {
|
|
currentNetworkBalancesToCache[accountID] = account.balance
|
|
}
|
|
})
|
|
const balancesToCache = {
|
|
...cachedBalances,
|
|
[currentNetwork]: currentNetworkBalancesToCache,
|
|
}
|
|
|
|
return balancesToCache
|
|
}
|
|
|
|
/**
|
|
* Removes cachedBalances
|
|
*/
|
|
|
|
clearCachedBalances () {
|
|
this.store.updateState({ cachedBalances: {} })
|
|
}
|
|
|
|
/**
|
|
* Sets up listeners and subscriptions which should trigger an update of cached balances. These updates will
|
|
* happen when the current account changes. Which happens on block updates, as well as on network and account
|
|
* selections.
|
|
*
|
|
* @private
|
|
*
|
|
*/
|
|
_registerUpdates () {
|
|
const update = this.updateCachedBalances.bind(this)
|
|
this.accountTracker.store.subscribe(update)
|
|
}
|
|
}
|
|
|