Update: Handle error message, handle error blocks.

threefoldtech · Apr 21, 2024 · d097eea · d097eea
1 parent bd73ce3
commit d097eea
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 13 deletions.
diff --git a/packages/playground/src/utils/ssh.ts b/packages/playground/src/utils/ssh.ts
@@ -1,11 +1,10 @@
-import { InsufficientBalanceError } from "@threefold/types";
 import crypto from "crypto";
 
 import { useProfileManager } from "@/stores/profile_manager";
 import type { SSHKeyData } from "@/types";
 
 import { createCustomToast, ToastType } from "./custom_toast";
-import { getGrid, loadBalance, storeSSH } from "./grid";
+import { getGrid, storeSSH } from "./grid";
 import { downloadAsJson } from "./helpers";
 
 /**
@@ -83,18 +82,12 @@ class SSHKeysManagement {
   async update(keys: SSHKeyData[]): Promise<void> {
     const profileManager = useProfileManager();
     const grid = await getGrid(profileManager.profile!);
+
     if (!grid) {
       createCustomToast(`Error occurred because the grid has not initialized yet.`, ToastType.danger);
       return;
     }
 
-    const balance = await loadBalance(grid!);
-    if (balance.free === balance.locked || balance.free < this.updateCost) {
-      throw new Error(
-        "Your wallet balance is insufficient to save your SSH key. To avoid losing your SSH key, please recharge your wallet.",
-      );
-    }
-
     const copiedKeys = keys.map(
       ({ fingerPrint, activating, deleting, ...keyWithoutSensitiveProps }) => keyWithoutSensitiveProps,
     );

diff --git a/packages/playground/src/views/sshkey_view.vue b/packages/playground/src/views/sshkey_view.vue
@@ -101,6 +101,7 @@
 </template>
 
 <script lang="ts" setup>
+import { InsufficientBalanceError } from "@threefold/types";
 import { defineComponent, onMounted, ref } from "vue";
 import { generateKeyPair } from "web-ssh-keygen";
 
@@ -192,7 +193,15 @@ const updateActivation = async (key: SSHKeyData) => {
   try {
     await sshKeysManagement.update(allKeys.value);
   } catch (e: any) {
-    createCustomToast(e.message, ToastType.danger);
+    if (e instanceof InsufficientBalanceError) {
+      createCustomToast(
+        "Your wallet balance is insufficient to save your SSH key. To avoid losing your SSH key, please recharge your wallet.",
+        ToastType.danger,
+      );
+    } else {
+      createCustomToast(e.message, ToastType.danger);
+    }
+
     activating.value = key.activating = false;
     return;
   }
@@ -215,7 +224,15 @@ const deleteKey = async (selectedKeys: SSHKeyData[]) => {
   try {
     await sshKeysManagement.update(keysToNotDelete);
   } catch (e: any) {
-    createCustomToast(e.message, ToastType.danger);
+    if (e instanceof InsufficientBalanceError) {
+      createCustomToast(
+        "Your wallet balance is insufficient to save your SSH key. To avoid losing your SSH key, please recharge your wallet.",
+        ToastType.danger,
+      );
+    } else {
+      createCustomToast(e.message, ToastType.danger);
+    }
+
     deleting.value = false;
     return;
   }
@@ -236,13 +253,23 @@ const addKey = async (key: SSHKeyData) => {
   }
 
   const copiedAllkeys = [...allKeys.value, key];
+
   try {
     await sshKeysManagement.update(copiedAllkeys);
   } catch (e: any) {
-    createCustomToast(e.message, ToastType.danger);
+    if (e instanceof InsufficientBalanceError) {
+      createCustomToast(
+        "Your wallet balance is insufficient to save your SSH key. To avoid losing your SSH key, please recharge your wallet.",
+        ToastType.danger,
+      );
+    } else {
+      createCustomToast(e.message, ToastType.danger);
+    }
+
     savingKey.value = false;
     return;
   }
+
   loading.value = true;
   allKeys.value = copiedAllkeys;
   savingKey.value = false;
@@ -267,7 +294,15 @@ const generateSSHKeys = async (key: SSHKeyData) => {
   try {
     await sshKeysManagement.update(copiedAllkeys);
   } catch (e: any) {
-    createCustomToast(e.message, ToastType.danger);
+    if (e instanceof InsufficientBalanceError) {
+      createCustomToast(
+        "Your wallet balance is insufficient to save your SSH key. To avoid losing your SSH key, please recharge your wallet.",
+        ToastType.danger,
+      );
+    } else {
+      createCustomToast(e.message, ToastType.danger);
+    }
+
     generatingSSH.value = false;
     loading.value = false;
     return;