Improve generics - types are now passed to the returned reducer.

index.d.ts

@@ -1,4 +1,4 @@
-import { Action, PreloadedState, Reducer, StoreCreator, StoreEnhancer } from 'redux';
+import { Action, AnyAction, PreloadedState, Reducer, StoreCreator, StoreEnhancer } from 'redux';
 declare const REMEMBER_REHYDRATED = "@@REMEMBER_REHYDRATED";
 declare const REMEMBER_PERSISTED = "@@REMEMBER_PERSISTED";
 type SerializeFunction = (data: any) => any;
@@ -14,6 +14,6 @@ type Options = {
     persistThrottle: number;
     persistWholeStore: boolean;
 };
-declare const rememberReducer: (reducers: Reducer) => Reducer;
+declare const rememberReducer: <S = any, A extends Action<any> = AnyAction>(reducers: Reducer<S, A>) => Reducer<S, A>;
 declare const rememberEnhancer: <S, A extends Action<any>, Ext>(driver: Driver, rememberedKeys: string[], { prefix, serialize, unserialize, persistThrottle, persistWholeStore }?: Partial<Options>) => (createStore: StoreCreator) => (rootReducer: Reducer<S, A>, initialState?: PreloadedState<S> | undefined, enhancer?: StoreEnhancer<Ext, {}> | undefined) => import("redux").Store<S, A> & Ext;
 export { rememberReducer, rememberEnhancer, REMEMBER_REHYDRATED, REMEMBER_PERSISTED };

package.json

@@ -1,6 +1,6 @@
 {
   "name": "redux-remember",
-  "version": "2.1.1",
+  "version": "2.1.2",
   "description": "Saves and loads your redux state from a key-value store of your choice",
   "main": "lib/index.js",
   "module": "es/index.js",

src/index.ts

@@ -10,12 +10,14 @@ import {
   StoreEnhancer
 } from 'redux';
 
-const rememberReducer = (reducers: Reducer): Reducer => {
-  const data = {
+const rememberReducer = <S = any, A extends Action = AnyAction>(
+  reducers: Reducer<S, A>
+): Reducer<S, A> => {
+  const data: any = {
     state: {}
   };
 
-  return (state = data.state, action: AnyAction) => {
+  return (state: S = data.state, action: any) => {
     if (action.type && (
       action.type === '@@INIT'
       || action.type.startsWith('@@redux/INIT')
@@ -30,7 +32,7 @@ const rememberReducer = (reducers: Reducer): Reducer => {
             ...data.state,
             ...(action.payload || {})
           },
-          { type: REMEMBER_REHYDRATED }
+          { type: REMEMBER_REHYDRATED } as any
         );
 
         return data.state;