From 9dea0a3e21cd5a980d4b676e6ccbe0fa38c78730 Mon Sep 17 00:00:00 2001
From: Gabor Stefan <gabor.stefan@acodis.io>
Date: Fri, 10 Jun 2022 14:35:41 +0200
Subject: [PATCH] fix(store): allow a generic reducer (ngrx/platform #2996)

---
 src/store/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/store/index.ts b/src/store/index.ts
index 34ee60c..ac5ffb3 100644
--- a/src/store/index.ts
+++ b/src/store/index.ts
@@ -22,7 +22,7 @@ export interface ImmerOnReducer<State, AC extends ActionCreator[]> {
  * Immer wrapper around `on` to mutate state
  */
 export function immerOn<State, Creators extends ActionCreator[]>(
-	...args: [...creators: Creators, reducer: ImmerOnReducer<State, Creators>]
+	...args: [...creators: Creators, reducer: ImmerOnReducer<State extends infer S ? S : never, Creators>]
 ): ReducerTypes<State, Creators> {
 	const reducer = (args.pop() as Function) as ActionReducer<State>;
 	return (on as any)(...(args as ActionCreator[]), immerReducer(reducer));