[models] Add UI model update when a wallet is created

fibercrypto · Feb 14, 2020 · f860804 · f860804
1 parent 44c84f6
commit f860804
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/models/walletsManager.go b/src/models/walletsManager.go
@@ -859,6 +859,12 @@ func (walletM *WalletManager) createEncryptedWallet(seed, label, wltType, passwo
 	logWalletManager.Info("Created encrypted wallet")
 	qWallet := fromWalletToQWallet(wlt, true, false)
 	walletM.wallets = append(walletM.wallets, qWallet)
+	wi := &updateWalletInfo{
+		isNew: true,
+		row: len(walletM.wallets)-1,
+		wallet: walletM.wallets[len(walletM.wallets)-1]
+	}
+	walletM.updaterChannel <- wi
 	return qWallet
 }
 
@@ -874,6 +880,11 @@ func (walletM *WalletManager) createUnencryptedWallet(seed, label, wltType strin
 
 	qWallet := fromWalletToQWallet(wlt, true, false)
 	walletM.wallets = append(walletM.wallets, qWallet)
+	wi := &updateWalletInfo{
+		isNew: true,
+		row: len(walletM.wallets)-1,
+		wallet: walletM.wallets[len(walletM.wallets)-1]
+	}
 	return qWallet
 
 }