Function createLiveQueryCollection

packages/db/src/query/live-query-collection.ts:155–203

(
  configOrQuery:
    | (LiveQueryCollectionConfig<TContext, TResult> & { utils?: TUtils })
    | ((
        q: InitialQueryBuilder,
      ) => QueryBuilder<TContext> & RootObjectResultConstraint<TContext>),
)

// Implementation
export function createLiveQueryCollection<
  TContext extends Context,
  TResult extends object = RootQueryResult<TContext>,
  TUtils extends UtilsRecord = {},
>(
  configOrQuery:
    | (LiveQueryCollectionConfig<TContext, TResult> & { utils?: TUtils })
    | ((
        q: InitialQueryBuilder,
      ) => QueryBuilder<TContext> & RootObjectResultConstraint<TContext>),
): CollectionForContext<TContext, TResult> & {
  utils: LiveQueryCollectionUtils & TUtils
} {
  // Determine if the argument is a function (query) or a config object
  if (typeof configOrQuery === `function`) {
    // Simple query function case
    const config: LiveQueryCollectionConfig<TContext, TResult> = {
      query: configOrQuery as (
        q: InitialQueryBuilder,
      ) => QueryBuilder<TContext> & RootObjectResultConstraint<TContext>,
    }
    // The implementation accepts both overload shapes, but TypeScript cannot
    // preserve the overload-specific query-builder inference through this branch.
    const options = liveQueryCollectionOptions(config as any)
    return bridgeToCreateCollection(options) as CollectionForContext<
      TContext,
      TResult
    > & { utils: LiveQueryCollectionUtils & TUtils }
  } else {
    // Config object case
    const config = configOrQuery as LiveQueryCollectionConfig<
      TContext,
      TResult
    > & { utils?: TUtils }
    // Same overload implementation limitation as above: the config has already
    // been validated by the public signatures, but the branch loses that precision.
    const options = liveQueryCollectionOptions(config as any)

    // Merge custom utils if provided, preserving the getBuilder() method for dependency tracking
    if (config.utils) {
      options.utils = { ...options.utils, ...config.utils }
    }

    return bridgeToCreateCollection(options) as CollectionForContext<
      TContext,
      TResult
    > & { utils: LiveQueryCollectionUtils & TUtils }
  }
}

/**
 * Bridge function that handles the type compatibility between query2's TResult