@umituz/react-native-storage 2.6.5 → 2.6.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-storage",
-  "version": "2.6.5",
+  "version": "2.6.6",
   "description": "Unified storage solution with AsyncStorage persistence, Zustand state management, and in-memory caching for React Native",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/domain/factories/StoreFactory.ts

@@ -1,23 +1,10 @@
 /**
  * Store Factory
  * Create Zustand stores with AsyncStorage persistence and actions
- *
- * @example
- * ```typescript
- * const useCounterStore = createStore({
- *   name: 'counter',
- *   initialState: { count: 0 },
- *   actions: (set, get) => ({
- *     increment: () => set({ count: get().count + 1 }),
- *     reset: () => set({ count: 0 }),
- *   }),
- *   persist: true,
- * });
- * ```
  */
 
-import { create } from 'zustand';
-import { persist, createJSONStorage } from 'zustand/middleware';
+import { create, type StateCreator } from 'zustand';
+import { persist, createJSONStorage, type PersistOptions } from 'zustand/middleware';
 import type { StoreConfig } from '../types/Store';
 import { storageService } from '../../infrastructure/adapters/StorageService';
 
@@ -30,42 +17,43 @@ export function createStore<
 >(config: StoreConfig<TState, TActions>) {
   type Store = TState & TActions;
 
-  const stateCreator = (
-    set: (partial: Partial<Store>) => void,
-    get: () => Store
-  ): Store => {
-    const state = config.initialState as Store;
-    const actions = config.actions ? config.actions(set as any, get as any) : ({} as TActions);
-    return { ...state, ...actions };
+  const stateCreator: StateCreator<Store> = (set, get) => {
+    const state = config.initialState as TState;
+    const actions = config.actions
+      ? config.actions(
+          (partial: Partial<TState>) => set(partial as Partial<Store>),
+          get as () => TState
+        )
+      : ({} as TActions);
+    return { ...state, ...actions } as Store;
   };
 
   if (!config.persist) {
     return create<Store>(stateCreator);
   }
 
-  return create<Store>()(
-    persist(stateCreator, {
-      name: config.name,
-      storage: createJSONStorage(() => storageService),
-      version: config.version || 1,
-      partialize: config.partialize
-        ? (state) => config.partialize!(state) as Store
-        : (state) => {
-            // By default, exclude functions (actions) from persistence
-            const persisted: Record<string, unknown> = {};
-            for (const key of Object.keys(state)) {
-              if (typeof state[key as keyof Store] !== 'function') {
-                persisted[key] = state[key as keyof Store];
-              }
+  const persistOptions: PersistOptions<Store, Partial<Store>> = {
+    name: config.name,
+    storage: createJSONStorage(() => storageService),
+    version: config.version || 1,
+    partialize: config.partialize
+      ? (state) => config.partialize!(state as TState) as Partial<Store>
+      : (state) => {
+          const persisted: Record<string, unknown> = {};
+          for (const key of Object.keys(state)) {
+            if (typeof state[key as keyof Store] !== 'function') {
+              persisted[key] = state[key as keyof Store];
             }
-            return persisted as Store;
-          },
-      onRehydrateStorage: () => (state) => {
-        if (state && config.onRehydrate) {
-          config.onRehydrate(state);
-        }
-      },
-      migrate: config.migrate as any,
-    })
-  );
+          }
+          return persisted as Partial<Store>;
+        },
+    onRehydrateStorage: () => (state) => {
+      if (state && config.onRehydrate) {
+        config.onRehydrate(state as TState);
+      }
+    },
+    migrate: config.migrate as PersistOptions<Store, Partial<Store>>['migrate'],
+  };
+
+  return create<Store>()(persist(stateCreator, persistOptions));
 }