m/metamask-extension
1
0
Fork 0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 09:52:26 +01:00
Code

Refresh computed balances controller when restoring vault

Browse Source
This commit is contained in:
Dan Finlay 2017-10-18 14:22:04 -07:00
parent ea79eca8eb
commit 9cc1e8a6d8
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/scripts/controllers/computed-balances.js
View File

@ -25,6 +25,10 @@ class ComputedbalancesController {
}
}
forgetAllBalances () {
this.balances = {}
}
_initBalanceUpdating () {
const store = this.accountTracker.store.getState()
this.addAnyAccountsFromStore(store)

3
app/scripts/metamask-controller.js
View File

@ -473,7 +473,7 @@ module.exports = class MetamaskController extends EventEmitter {
}
forgetOldAccounts () {
const { accountTracker } = this
const { accountTracker, balancesController } = this
let oldAccounts = []
try {
oldAccounts = Object.keys(accountTracker.store.getState().accounts)
@ -481,6 +481,7 @@ module.exports = class MetamaskController extends EventEmitter {
log.warn('Could not load old accounts to forget', e)
}
oldAccounts.forEach(addr => accountTracker.removeAccount(addr))
balancesController.forgetAllBalances()
}
selectFirstIdentity (vault) {