fix(ts): improve IProduce

Fixes #205
immerjs · Dec 15, 2018 · b0af196 · b0af196
1 parent f50ae4e
commit b0af196
Showing 1 changed file with 13 additions and 71 deletions.
diff --git a/src/immer.d.ts b/src/immer.d.ts
@@ -59,84 +59,26 @@ export interface IProduce {
      * @param initialState - if a curried function is created and this argument was given, it will be used as fallback if the curried function is called with a state of undefined
      * @returns The next state: a new state, or the current state if nothing was modified
      */
-    <S = any>(
+    <S = any, R = never>(
         currentState: S,
-        recipe: (this: Draft<S>, draftState: Draft<S>) => void | S,
+        recipe: (this: Draft<S>, draftState: Draft<S>) => void | R,
         listener?: PatchListener
-    ): S
+    ): R
 
-    // curried invocations with default initial state
-    // 0 additional arguments
-    <S = any>(
-        recipe: (this: Draft<S>, draftState: Draft<S>) => void | S,
+    /** Curried producer with an initial state */
+    <S = any, R = never>(
+        recipe: (this: Draft<S>, draftState: Draft<S>) => void | R,
         initialState: S
-    ): (currentState: S | undefined) => S
-    // 1 additional argument of type A
-    <S = any, A = any>(
-        recipe: (this: Draft<S>, draftState: Draft<S>, a: A) => void | S,
-        initialState: S
-    ): (currentState: S | undefined, a: A) => S
-    // 2 additional arguments of types A and B
-    <S = any, A = any, B = any>(
-        recipe: (this: Draft<S>, draftState: Draft<S>, a: A, b: B) => void | S,
-        initialState: S
-    ): (currentState: S | undefined, a: A, b: B) => S
-    // 3 additional arguments of types A, B and C
-    <S = any, A = any, B = any, C = any>(
-        recipe: (
-            this: Draft<S>,
-            draftState: Draft<S>,
-            a: A,
-            b: B,
-            c: C
-        ) => void | S,
-        initialState: S
-    ): (currentState: S | undefined, a: A, b: B, c: C) => S
-    // any number of additional arguments, but with loss of type safety
-    // this may be alleviated if "variadic kinds" makes it into Typescript:
-    // https://github.com/Microsoft/TypeScript/issues/5453
-    <S = any>(
-        recipe: (
-            this: Draft<S>,
-            draftState: Draft<S>,
-            ...extraArgs: any[]
-        ) => void | S,
-        initialState: S
-    ): (currentState: S | undefined, ...extraArgs: any[]) => S
-
-    // curried invocations without default initial state
-    // 0 additional arguments
-    <S = any>(recipe: (this: Draft<S>, draftState: Draft<S>) => void | S): (
-        currentState: S
-    ) => S
-    // 1 additional argument of type A
-    <S = any, A = any>(
-        recipe: (this: Draft<S>, draftState: Draft<S>, a: A) => void | S
-    ): (currentState: S, a: A) => S
-    // 2 additional arguments of types A and B
-    <S = any, A = any, B = any>(
-        recipe: (this: Draft<S>, draftState: Draft<S>, a: A, b: B) => void | S
-    ): (currentState: S, a: A, b: B) => S
-    // 3 additional arguments of types A, B and C
-    <S = any, A = any, B = any, C = any>(
-        recipe: (
-            this: Draft<S>,
-            draftState: Draft<S>,
-            a: A,
-            b: B,
-            c: C
-        ) => void | S
-    ): (currentState: S, a: A, b: B, c: C) => S
-    // any number of additional arguments, but with loss of type safety
-    // this may be alleviated if "variadic kinds" makes it into Typescript:
-    // https://github.com/Microsoft/TypeScript/issues/5453
-    <S = any>(
+    ): (currentState: S | undefined) => R
+
+    /** Curried producer with no initial state */
+    <S = any, R = never, Args extends any[] = any[]>(
         recipe: (
             this: Draft<S>,
             draftState: Draft<S>,
-            ...extraArgs: any[]
-        ) => void | S
-    ): (currentState: S, ...extraArgs: any[]) => S
+            ...extraArgs: Args
+        ) => void | R
+    ): (currentState: S, ...extraArgs: Args) => R
 }
 
 export const produce: IProduce