fix: allow deleting last wallet (#122)

getAlby · Sep 17, 2024 · 4f05700 · 4f05700
1 parent 8f9a136
commit 4f05700
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 6 deletions.
diff --git a/lib/state/appStore.ts b/lib/state/appStore.ts
@@ -99,7 +99,15 @@ export const useAppStore = create<AppState>()((set, get) => {
   const removeCurrentWallet = () => {
     const wallets = [...get().wallets];
     if (wallets.length <= 1) {
-      // cannot delete last wallet
+      // set to initial wallet status
+      secureStorage.removeItem(hasOnboardedKey);
+      secureStorage.setItem(selectedWalletIdKey, "0");
+      secureStorage.setItem(getWalletKey(0), JSON.stringify({}));
+      set({
+        nwcClient: undefined,
+        selectedWalletId: 0,
+        wallets: [{}],
+      });
       return;
     }
     const selectedWalletId = get().selectedWalletId;
@@ -191,17 +199,22 @@ export const useAppStore = create<AppState>()((set, get) => {
       for (let i = 0; i < get().addressBookEntries.length; i++) {
         secureStorage.removeItem(getAddressBookEntryKey(i));
       }
-      // clear selected wallet ID
-      secureStorage.removeItem(selectedWalletIdKey);
+
+      // clear fiat currency
+      secureStorage.removeItem(fiatCurrencyKey);
 
       // clear onboarding status
       secureStorage.removeItem(hasOnboardedKey);
 
+      // set to initial wallet status
+      secureStorage.setItem(selectedWalletIdKey, "0");
+      secureStorage.setItem(getWalletKey(0), JSON.stringify({}));
+
       set({
         nwcClient: undefined,
         fiatCurrency: undefined,
-        selectedWalletId: undefined,
-        wallets: [],
+        selectedWalletId: 0,
+        wallets: [{}],
         addressBookEntries: [],
       });
     },

diff --git a/pages/settings/wallets/EditWallet.tsx b/pages/settings/wallets/EditWallet.tsx
@@ -117,7 +117,12 @@ export function EditWallet() {
               {
                 text: "Confirm",
                 onPress: () => {
-                  router.back();
+                  if (useAppStore.getState().wallets.length == 1) {
+                    router.dismissAll();
+                    router.replace("/onboarding");
+                  } else {
+                    router.back();
+                  }
                   useAppStore.getState().removeCurrentWallet();
                 },
               },