simplify types

Author: devanshj

src/middleware/devtools.ts

@@ -58,18 +58,8 @@ type StoreDevtools<S> = S extends {
     ? {
         setState<R extends boolean | undefined>(
           partial:
-            | (R extends true
-                ? T
-                : R extends false | undefined
-                ? Partial<T>
-                : T | Partial<T>)
-            | ((
-                state: T
-              ) => R extends true
-                ? T
-                : R extends false | undefined
-                ? Partial<T>
-                : T | Partial<T>),
+            | (R extends true ? T : T | Partial<T>)
+            | ((state: T) => R extends true ? T : T | Partial<T>),
           replace?: R,
           action?: string | { type: unknown }
         ): Sr
@@ -81,11 +71,7 @@ type StoreDevtools<S> = S extends {
     ? {
         setState<R extends boolean | undefined>(
           nextStateOrUpdater:
-            | (R extends true
-                ? T
-                : R extends false | undefined
-                ? Partial<T>
-                : T | Partial<T>)
+            | (R extends true ? T : T | Partial<T>)
             | ((state: Draft<T>) => void),
           shouldReplace?: R,
           action?: string | { type: unknown }

src/middleware/immer.ts

@@ -41,11 +41,9 @@ type StoreImmer<S> = S extends {
 }
   ? {
       setState<R extends boolean | undefined = false>(
-        nextStateOrUpdater: R extends true
-          ? T
-          : R extends false | undefined
-          ? Partial<T>
-          : T | Partial<T> | ((state: Draft<T>) => void),
+        nextStateOrUpdater:
+          | (R extends true ? T : T | Partial<T>)
+          | ((state: Draft<T>) => void),
         shouldReplace?: R,
         ...a: SkipTwo<A>
       ): Sr

src/vanilla.ts

@@ -16,18 +16,8 @@ export type Subscribe<T extends State> = {
 export type SetState<T extends State> = {
   _<R extends boolean | undefined = false>(
     partial:
-      | (R extends true
-          ? T
-          : R extends false | undefined
-          ? Partial<T>
-          : T | Partial<T>)
-      | ((
-          state: T
-        ) => R extends true
-          ? T
-          : R extends false | undefined
-          ? Partial<T>
-          : T | Partial<T>),
+      | (R extends true ? T : T | Partial<T>)
+      | ((state: T) => R extends true ? T : T | Partial<T>),
     replace?: R
   ): void
 }['_']