Allow typing for useFaustQuery

wpengine · Nov 3, 2023 · 3bd56f6 · 3bd56f6
1 parent d0ff52c
commit 3bd56f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/faustwp-core/src/hooks/useFaustContext.tsx b/packages/faustwp-core/src/hooks/useFaustContext.tsx
@@ -5,7 +5,7 @@ import { sha256 } from 'js-sha256';
 import { useContext } from 'react';
 import { FaustContext } from '../store/FaustContext.js';
 
-export function useFaustQuery(query: DocumentNode) {
+export function useFaustQuery<TData>(query: DocumentNode): TData {
   const context = useContext(FaustContext);
 
   if (context === undefined) {
@@ -15,5 +15,5 @@ export function useFaustQuery(query: DocumentNode) {
   const sha = sha256(print(query));
 
   // eslint-disable-next-line no-underscore-dangle
-  return context?.__FAUST_QUERIES__?.[sha];
+  return context?.__FAUST_QUERIES__?.[sha] as TData;
 }
diff --git a/packages/faustwp-core/src/store/FaustContext.tsx b/packages/faustwp-core/src/store/FaustContext.tsx
@@ -4,7 +4,7 @@ import { FaustPageProps } from '../components/FaustProvider.js';
 export const FaustContext = createContext<
   | ({
       __FAUST_QUERIES__?: {
-        [key: string]: string;
+        [key: string]: any;
       };
     } & FaustPageProps)
   | undefined