Added height check for rewind wallet

kryptokrona · Feb 4, 2024 · 37162ae · 37162ae
1 parent 0e72e25
commit 37162ae
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/backend/electron.cjs b/src/backend/electron.cjs
@@ -309,7 +309,7 @@ ipcMain.on("reset-wallet", (e, height) => {
 });
 
 ipcMain.on("rewind-wallet", async (e, height) => {
-
+  successMessage(`Rewind wallet from height ${height}`)
   walletBackend.rewind(parseInt(height));
 
 });

diff --git a/src/routes/wallet/settings/wallet/+page.svelte b/src/routes/wallet/settings/wallet/+page.svelte
@@ -7,16 +7,22 @@
   let height = 0;
 
   const resetWallet = () => {
-    if (height > $node.networkBlockCount) {
-      window.api.errorMessage('Cannot scan from height');
-      return;
-    }
+    if (!checkHeight(height)) return;
     window.api.resetWallet(height);
   };
 
   const rewindWallet = () => {
+    if (!checkHeight(height)) return;
     window.api.rewindWallet(height);
   };
+
+  const checkHeight = (h) => {
+    if (h > $node.networkBlockCount) {
+      window.api.errorMessage('Cannot scan from height');
+      return false;
+    }
+    return true;
+  };
 </script>
 
 <div class="wrapper">