lib - improve useHook types

CrimsonChi · Oct 4, 2024 · 2158ba6 · 2158ba6
1 parent 044f829
commit 2158ba6
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/packages/lib/src/hooks/utils.ts b/packages/lib/src/hooks/utils.ts
@@ -78,9 +78,22 @@ type HookCallback<T> = (state: HookCallbackState<T>) => any
 let currentHookName: string | null = null
 const nestedHookWarnings = new Set<string>()
 
+function useHook<
+  T extends () => Record<string, unknown>,
+  U extends HookCallback<ReturnType<T>>,
+>(hookName: string, hookInitializer: T, callback: U): ReturnType<U>
+
+function useHook<T extends Record<string, unknown>, U extends HookCallback<T>>(
+  hookName: string,
+  hookData: T,
+  callback: U
+): ReturnType<U>
+
 function useHook<
   T,
-  U extends T extends () => any ? HookCallback<ReturnType<T>> : HookCallback<T>,
+  U extends T extends () => Record<string, unknown>
+    ? HookCallback<ReturnType<T>>
+    : HookCallback<T>,
 >(
   hookName: string,
   hookDataOrInitializer: Kaioken.Hook<T> | (() => Kaioken.Hook<T>),