feat: extension store use the async callback (#14)

Co-authored-by: alpers <[email protected]>
zCloak-Network · Jan 10, 2023 · d4978c5 · d4978c5
1 parent 80d9fee
commit d4978c5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/packages/ui-store/src/ExtensionSession.ts b/packages/ui-store/src/ExtensionSession.ts
@@ -33,12 +33,10 @@ export class ExtensionSession extends BaseStore {
   }
 
   public remove(key: string, fn?: () => void) {
-    session.remove(key);
-    fn && fn();
+    session.remove(key, fn);
   }
 
   public set(key: string, value: unknown, fn?: () => void) {
-    session.set({ [key]: value });
-    fn && fn();
+    session.set({ [key]: value }, fn);
   }
 }
diff --git a/packages/ui-store/src/ExtensionStore.ts b/packages/ui-store/src/ExtensionStore.ts
@@ -33,12 +33,10 @@ export class ExtensionStore extends BaseStore {
   }
 
   public remove(key: string, fn?: () => void) {
-    storage.remove(key);
-    fn && fn();
+    storage.remove(key, fn);
   }
 
   public set(key: string, value: unknown, fn?: () => void) {
-    storage.set({ [key]: value });
-    fn && fn();
+    storage.set({ [key]: value }, fn);
   }
 }