feat: update src-electron/libs/store

OneKeyHQ · Oct 18, 2023 · a34d8e6 · a34d8e6
1 parent 1871d01
commit a34d8e6
Showing 1 changed file with 24 additions and 5 deletions.
diff --git a/packages/desktop/src-electron/libs/store.ts b/packages/desktop/src-electron/libs/store.ts
@@ -27,18 +27,37 @@ export const clearUpdateSettings = () => {
 };
 
 export const getSecureItem = (key: string) => {
+  const available = safeStorage.isEncryptionAvailable();
+  if (!available) {
+    console.error('safeStorage is not available');
+    return undefined;
+  }
   const item = store.get(EncryptedData, {}) as Record<string, string>;
   const value = item[key];
   if (value) {
-    const result = safeStorage.decryptString(Buffer.from(value, 'hex'));
-    return result;
+    try {
+      const result = safeStorage.decryptString(Buffer.from(value, 'hex'));
+      return result;
+    } catch (e) {
+      console.error(`failed to decrypt ${key}`, e);
+      return undefined;
+    }
   }
 };
 
 export const setSecureItem = (key: string, value: string): void => {
-  const items = store.get(EncryptedData, {}) as Record<string, string>;
-  items[key] = safeStorage.encryptString(value).toString('hex');
-  store.set(EncryptedData, items);
+  const available = safeStorage.isEncryptionAvailable();
+  if (!available) {
+    console.error('safeStorage is not available');
+    return undefined;
+  }
+  try {
+    const items = store.get(EncryptedData, {}) as Record<string, string>;
+    items[key] = safeStorage.encryptString(value).toString('hex');
+    store.set(EncryptedData, items);
+  } catch (e) {
+    console.error(`failed to encrypt ${key} ${value}`, e);
+  }
 };
 
 export const clearSecureItem = (key: string) => {