@livestore/react 0.4.0-dev.2 → 0.4.0-dev.20

package/src/useStore.ts

@@ -1,3 +1,4 @@
+import type { LiveStoreSchema } from '@livestore/common/schema'
 import type { Store } from '@livestore/livestore'
 import React from 'react'
 
@@ -6,14 +7,14 @@ import { LiveStoreContext } from './LiveStoreContext.ts'
 import { useClientDocument } from './useClientDocument.ts'
 import { useQuery } from './useQuery.ts'
 
-export const withReactApi = (store: Store): Store & ReactApi => {
+export const withReactApi = <TSchema extends LiveStoreSchema>(store: Store<TSchema>): Store<TSchema> & ReactApi => {
   // @ts-expect-error TODO properly implement this
 
-  store.useQuery = (queryDef) => useQuery(queryDef, { store })
+  store.useQuery = (queryable) => useQuery(queryable, { store })
   // @ts-expect-error TODO properly implement this
 
   store.useClientDocument = (table, idOrOptions, options) => useClientDocument(table, idOrOptions, options, { store })
-  return store as Store & ReactApi
+  return store as Store<TSchema> & ReactApi
 }
 
 export const useStore = (options?: { store?: Store }): { store: Store & ReactApi } => {