fix(resolveHookState): by accident removed needed type.

streamich · Nov 6, 2019 · 59aa41c · 59aa41c
1 parent 1840b57
commit 59aa41c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/util/resolveHookState.ts b/src/util/resolveHookState.ts
@@ -5,10 +5,10 @@ export type InitialHookState<S> = S | InitialStateSetter<S>;
 export type HookState<S> = S | StateSetter<S>;
 export type ResolvableHookState<S> = S | StateSetter<S> | InitialStateSetter<S>;
 
-export function resolveHookState<S>(newState: InitialStateSetter<S>): S;
-export function resolveHookState<S>(newState: StateSetter<S>, currentState: S): S;
-export function resolveHookState<S>(newState: ResolvableHookState<S>, currentState?: S): S;
-export function resolveHookState<S>(newState: ResolvableHookState<S>, currentState?: S): S {
+export function resolveHookState<S, C extends S>(newState: InitialStateSetter<S>): S;
+export function resolveHookState<S, C extends S>(newState: StateSetter<S>, currentState: C): S;
+export function resolveHookState<S, C extends S>(newState: ResolvableHookState<S>, currentState?: C): S;
+export function resolveHookState<S, C extends S>(newState: ResolvableHookState<S>, currentState?: C): S {
   if (typeof newState === 'function') {
     return (newState as Function)(currentState);
   }