From f97dcd92acd916484c26c671b504bbe0995dbfc1 Mon Sep 17 00:00:00 2001
From: Ahmed Ibrahim <ib153507@gmail.com>
Date: Tue, 18 Jan 2022 11:54:37 +0200
Subject: [PATCH] Fix Flow types

---
 packages/yoroi-extension/app/containers/wallet/Wallet.js      | 3 ++-
 packages/yoroi-extension/app/containers/wallet/Wallet.mock.js | 3 +--
 packages/yoroi-extension/app/stores/toplevel/WalletStore.js   | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/packages/yoroi-extension/app/containers/wallet/Wallet.js b/packages/yoroi-extension/app/containers/wallet/Wallet.js
index 15649de40e..18f608d446 100644
--- a/packages/yoroi-extension/app/containers/wallet/Wallet.js
+++ b/packages/yoroi-extension/app/containers/wallet/Wallet.js
@@ -97,6 +97,7 @@ class Wallet extends Component<AllProps> {
 
   renderOverlay(): null | React$Element<typeof WalletSyncingOverlay> {
     const publicDeriver = this.generated.stores.wallets.selected;
+    if (publicDeriver == null) throw new Error(`${nameof(this.renderOverlay)} no public deriver`);
 
     if (this.generated.stores.wallets.firstSync === publicDeriver.getPublicDeriverId()) {
       return (
@@ -254,7 +255,7 @@ class Wallet extends Component<AllProps> {
       |},
       wallets: {|
         selected: null | PublicDeriver<>,
-        firstSync: boolean,
+        firstSync: ?number,
       |},
       router: {| location: any |},
       transactions: {|
diff --git a/packages/yoroi-extension/app/containers/wallet/Wallet.mock.js b/packages/yoroi-extension/app/containers/wallet/Wallet.mock.js
index b5beaabd50..59bdf8925f 100644
--- a/packages/yoroi-extension/app/containers/wallet/Wallet.mock.js
+++ b/packages/yoroi-extension/app/containers/wallet/Wallet.mock.js
@@ -34,14 +34,13 @@ export const mockWalletProps: {
     stores: {
       app: {
         currentRoute: request.location,
-      
       },
       profile: {
         isClassicTheme: true
       },
       wallets: {
         selected: request.selected,
-        firstSync: false,
+        firstSync: null,
       },
       transactions: {
         getBalanceRequest: {
diff --git a/packages/yoroi-extension/app/stores/toplevel/WalletStore.js b/packages/yoroi-extension/app/stores/toplevel/WalletStore.js
index c35b9541d1..b9237a6d38 100644
--- a/packages/yoroi-extension/app/stores/toplevel/WalletStore.js
+++ b/packages/yoroi-extension/app/stores/toplevel/WalletStore.js
@@ -93,7 +93,7 @@ export default class WalletStore extends Store<StoresMap, ActionsMap> {
   WALLET_REFRESH_INTERVAL: number = environment.getWalletRefreshInterval();
   ON_VISIBLE_DEBOUNCE_WAIT: number = 1000;
 
-  @observable firstSync: null | number;
+  @observable firstSync: ?number;
   @observable publicDerivers: Array<PublicDeriver<>>;
   @observable selected: null | PublicDeriver<>;
   @observable getInitialWallets: Request<GetWalletsFunc> = new Request<GetWalletsFunc>(getWallets);
@@ -239,7 +239,7 @@ export default class WalletStore extends Store<StoresMap, ActionsMap> {
       await this.stores.addresses.refreshAddressesFromDb(publicDeriver);
 
       await updateSyncedWallets(publicDeriver)
-       if (this.firstSync) {
+       if (typeof this.firstSync === 'number') {
         runInAction(() => {
           this.firstSync = null
         })