feat: add include support to persist plugin

leafsphp · Nov 30, 2023 · d303eb4 · d303eb4
1 parent 8b63427
commit d303eb4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/packages/store/src/@types/plugin.ts b/packages/store/src/@types/plugin.ts
@@ -7,6 +7,7 @@ export interface PersistPluginOptions {
   key?: string;
   env?: 'react' | 'react-native';
   exclude?: string[];
+  include?: string[];
 }
 
 export type Hook =

diff --git a/packages/store/src/plugins/persist.ts b/packages/store/src/plugins/persist.ts
@@ -11,6 +11,7 @@ export class PersistedState implements PluginClass {
     key: 'hana-store',
     env: 'react',
     exclude: [],
+    include: [],
   };
 
   public constructor(options?: PersistPluginOptions) {
@@ -68,15 +69,27 @@ export class PersistedState implements PluginClass {
       return;
     }
 
-    this._options.exclude.forEach((item) => {
-      if (state[item]) {
-        delete state[item];
-      }
-    });
+    let finalState: State = {};
+
+    if (this._options.include.length > 0) {
+      this._options.include.forEach((item) => {
+        if (state[item]) {
+          finalState[item] = state[item];
+        }
+      });
+    } else {
+      this._options.exclude.forEach((item) => {
+        if (state[item]) {
+          delete state[item];
+        }
+      });
+
+      finalState = state;
+    }
 
     return await this._options.storage?.setItem?.(
       this._options.key,
-      JSON.stringify(state)
+      JSON.stringify(finalState)
     );
   }