Parameterize the type

thefrontside · Jan 20, 2024 · 193acaf · 193acaf
1 parent 9a25484
commit 193acaf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/hooks/useLoader.ts b/src/hooks/useLoader.ts
@@ -27,7 +27,7 @@ export function useLoader<T>(
     retryingMessageInterval = 1000,
   } = options;
 
-  const [state, setState] = useState<LoaderState<unknown>>({ type: "initial" });
+  const [state, setState] = useState<LoaderState<T>>({ type: "initial" });
   const [key, setKey] = useState<number>(0);
 
   const restart = useCallback(() => {
@@ -58,7 +58,7 @@ export function useLoader<T>(
 
   useEffect(() => {
     const task = run(function* () {
-      yield* initLoaderStateContext();
+      yield* initLoaderStateContext<T>();
 
       // eslint-disable-next-line react-hooks/rules-of-hooks
       const state = yield* useLoaderState();

diff --git a/src/operations/LoaderStateContext.ts b/src/operations/LoaderStateContext.ts
@@ -4,9 +4,9 @@ import { LoaderState } from "../types";
 export const LoaderStateContext =
   createContext<Channel<LoaderState<unknown>, void>>("loader-state");
 
-export function* initLoaderStateContext() {
-  const state = yield* resource<Channel<LoaderState<unknown>, void>>(function* (provide) {
-    const channel = createChannel<LoaderState<unknown>>();
+export function* initLoaderStateContext<T>() {
+  const state = yield* resource<Channel<LoaderState<T>, void>>(function* (provide) {
+    const channel = createChannel<LoaderState<T>>();
 
     try {
       yield* provide(channel);