Something went wrong. Try again.
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147// based on https://github.com/rocicorp/mono/tree/main/packages/zero-solid
import type { CustomMutatorDefs, DefaultContext, DefaultSchema, Falsy, HumanReadable, PullRow, QueryOrQueryRequest, ReadonlyJSONValue, Schema, TTL, Zero,} from '@rocicorp/zero'import type { ComputedRef, MaybeRefOrGetter } from 'vue'import type { QueryError, QueryStatus, VueView } from './view'
import { addContextToQuery, asQueryInternals, DEFAULT_TTL_MS } from '@rocicorp/zero/bindings'import { computed, getCurrentInstance, onUnmounted, shallowRef, toValue, watch,} from 'vue'import { vueViewFactory } from './view'
export interface UseQueryOptions { ttl?: TTL | undefined}
export interface QueryResult<TReturn> { data: ComputedRef<HumanReadable<TReturn>> status: ComputedRef<QueryStatus> error: ComputedRef<QueryError & { retry: () => void } | undefined>}
export interface MaybeQueryResult<TReturn> { data: ComputedRef<HumanReadable<TReturn> | undefined> status: ComputedRef<QueryStatus | 'disabled'> error: ComputedRef<QueryError & { retry: () => void } | undefined>}
// Overload 1: Queryexport function useQuery< TTable extends keyof TSchema['tables'] & string, TInput extends ReadonlyJSONValue | undefined, TOutput extends ReadonlyJSONValue | undefined, TSchema extends Schema = DefaultSchema, TReturn = PullRow<TTable, TSchema>, TContext = DefaultContext, MD extends CustomMutatorDefs | undefined = undefined,>( z: MaybeRefOrGetter<Zero<TSchema, MD, TContext>>, query: MaybeRefOrGetter<QueryOrQueryRequest<TTable, TInput, TOutput, TSchema, TReturn, TContext>>, options?: MaybeRefOrGetter<UseQueryOptions>,): QueryResult<TReturn>
// Overload 2: Maybe queryexport function useQuery< TTable extends keyof TSchema['tables'] & string, TInput extends ReadonlyJSONValue | undefined, TOutput extends ReadonlyJSONValue | undefined, TSchema extends Schema = DefaultSchema, TReturn = PullRow<TTable, TSchema>, TContext = DefaultContext, MD extends CustomMutatorDefs | undefined = undefined,>( z: MaybeRefOrGetter<Zero<TSchema, MD, TContext>>, query: MaybeRefOrGetter<QueryOrQueryRequest<TTable, TInput, TOutput, TSchema, TReturn, TContext> | Falsy>, options?: MaybeRefOrGetter<UseQueryOptions>,): MaybeQueryResult<TReturn>
// Implementationexport function useQuery< TTable extends keyof TSchema['tables'] & string, TInput extends ReadonlyJSONValue | undefined, TOutput extends ReadonlyJSONValue | undefined, TSchema extends Schema = DefaultSchema, TReturn = PullRow<TTable, TSchema>, TContext = DefaultContext, MD extends CustomMutatorDefs | undefined = undefined,>( z: MaybeRefOrGetter<Zero<TSchema, MD, TContext>>, query: MaybeRefOrGetter<QueryOrQueryRequest<TTable, TInput, TOutput, TSchema, TReturn, TContext> | Falsy>, options?: MaybeRefOrGetter<UseQueryOptions>,): QueryResult<TReturn> | MaybeQueryResult<TReturn> { const ttl = computed(() => toValue(options)?.ttl ?? DEFAULT_TTL_MS) const view = shallowRef<VueView | null>(null) const refetchKey = shallowRef(0)
const q = shallowRef() watch( [ () => toValue(query), () => toValue(z), ], ([query, z]) => { q.value = query ? addContextToQuery(toValue(query), toValue(z).context) : undefined }, { immediate: true }, )
const qi = computed(() => q.value ? asQueryInternals(q.value) : undefined) const hash = computed(() => qi.value?.hash())
watch( [ () => toValue(z), hash, refetchKey, ], ([z]) => { view.value?.destroy() if (!hash.value) { view.value = null } else { view.value = z.materialize(toValue(q), vueViewFactory, { ttl: toValue(ttl) }) } }, { immediate: true }, )
watch(ttl, (ttl) => { toValue(view)?.updateTTL(ttl) })
if (getCurrentInstance()) { onUnmounted(() => view.value?.destroy()) }
return { data: computed(() => view.value?.data as HumanReadable<TReturn>), status: computed(() => view.value?.status ?? 'disabled'), error: computed(() => view.value?.error ? { retry: () => { refetchKey.value++ }, ...view.value.error, } : undefined, ), }}