Update the useSharedFacet typing (#145)

viccon · pirelenito · web-flow · commit c0693d7dd54c · 2025-03-12T11:38:07.000+01:00
Co-authored-by: Paulo Ragonha &lt;paulo@ragonha.me&gt;
diff --git a/packages/@react-facet/shared-facet/src/context.ts b/packages/@react-facet/shared-facet/src/context.ts
@@ -8,6 +8,8 @@ export const sharedFacetDriverContext = createContext<SharedFacetDriver>(dummyCo
 
 export const SharedFacetDriverProvider = sharedFacetDriverContext.Provider
 
-export const useSharedFacet = <T>(sharedFacet: SharedFacet<T>): Facet<T> => {
-  return sharedFacet(useContext(sharedFacetDriverContext))
+type InferFacet<T> = T extends SharedFacet<infer U> ? Facet<U> : never
+
+export const useSharedFacet = <T extends SharedFacet<unknown>>(sharedFacet: T): InferFacet<T> => {
+  return sharedFacet(useContext(sharedFacetDriverContext)) as InferFacet<T>
 }

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,8 @@ export const sharedFacetDriverContext = createContext<SharedFacetDriver>(dummyCo`
`8`	`8`
`9`	`9`	`export const SharedFacetDriverProvider = sharedFacetDriverContext.Provider`
`10`	`10`
`11`		`-export const useSharedFacet = <T>(sharedFacet: SharedFacet<T>): Facet<T> => {`
`12`		`- return sharedFacet(useContext(sharedFacetDriverContext))`
	`11`	`+type InferFacet<T> = T extends SharedFacet<infer U> ? Facet<U> : never`
	`12`	`+`
	`13`	`+export const useSharedFacet = <T extends SharedFacet<unknown>>(sharedFacet: T): InferFacet<T> => {`
	`14`	`+ return sharedFacet(useContext(sharedFacetDriverContext)) as InferFacet<T>`
`13`	`15`	`}`