Fix setInterstate does not infer the type of the parameter in a sette…

…r function (schema interface)
turtleflyer · Apr 5, 2021 · 3eb27d7 · 3eb27d7
1 parent 9feb7e9
commit 3eb27d7
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,9 @@
+## 0.1.1-alpha.2 (Apr 4, 2021)
+
+### Fix
+
+- Fix setInterstate does not infer the type of the parameter in a setter function (schema interface)
+
 ## 0.1.1-alpha.1 (Mar 30, 2021)
 
 ### Change

diff --git a/src/UseInterstateTypes.ts b/src/UseInterstateTypes.ts
@@ -113,6 +113,8 @@ export type SetInterstate<M extends Interstate = never> = [M] extends [never]
       <K extends OnlyStringKeys<M>>(key: K, set: SetInterstateParam<M[K]>): void;
 
       <K extends ExtraStringKeys<M>>(set: SetInterstateSchemaParam<M, K> & EliminateNever<K>): void;
+
+      <K extends ExtraStringKeys<M>>(set: SetInterstateSchemaParamFn<M, K>): void;
     };
 
 export type SetInterstateParam<T> = Exclude<T, (...x: any) => any> | ((prevValue: T) => T);

diff --git a/src/test_suites/post/checkTypes.test.ts b/src/test_suites/post/checkTypes.test.ts
@@ -630,6 +630,10 @@ describe('Check types', () => {
         go: [state.a],
       }));
       setInterstateDefined({ a: (x: string) => `${x}new` });
+      setInterstateDefined((state) => ({
+        77: undefined,
+        go: [state.a],
+      }));
 
       // @ts-expect-error
       setInterstateDefined((state: { a: number }) => ({

diff --git a/src/test_suites/pre/checkTypes.test.ts b/src/test_suites/pre/checkTypes.test.ts
@@ -630,6 +630,10 @@ describe('Check types', () => {
         go: [state.a],
       }));
       setInterstateDefined({ a: (x: string) => `${x}new` });
+      setInterstateDefined((state) => ({
+        77: undefined,
+        go: [state.a],
+      }));
 
       // @ts-expect-error
       setInterstateDefined((state: { a: number }) => ({