From 2158ba641b36795d2cb358610b1a4905d3944050 Mon Sep 17 00:00:00 2001
From: Robby6Strings <rausten93@gmail.com>
Date: Fri, 4 Oct 2024 18:34:52 +1300
Subject: [PATCH] lib - improve `useHook` types

---
 packages/lib/src/hooks/utils.ts | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/packages/lib/src/hooks/utils.ts b/packages/lib/src/hooks/utils.ts
index a204033..9f7fe7d 100644
--- 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>),