Function extractCollectionsFromQuery

packages/db/src/query/live/utils.ts:19–102

(
  query: any,
)

 * Maps collections by their ID (not alias) as expected by the compiler.
 */
export function extractCollectionsFromQuery(
  query: any,
): Record<string, Collection<any, any, any>> {
  const collections: Record<string, any> = {}

  // Helper function to recursively extract collections from a query or source
  function extractFromSource(source: any) {
    if (source.type === `collectionRef`) {
      collections[source.collection.id] = source.collection
    } else if (source.type === `queryRef`) {
      // Recursively extract from subquery
      extractFromQuery(source.query)
    } else if (source.type === `unionFrom`) {
      for (const childSource of source.sources) {
        extractFromSource(childSource)
      }
    } else if (source.type === `unionAll`) {
      for (const branch of source.queries) {
        extractFromQuery(branch)
      }
    }
  }

  // Helper function to recursively extract collections from a query
  function extractFromQuery(q: any) {
    // Extract from FROM clause
    if (q.from) {
      extractFromSource(q.from)
    }

    // Extract from JOIN clauses
    if (q.join && Array.isArray(q.join)) {
      for (const joinClause of q.join) {
        if (joinClause.from) {
          extractFromSource(joinClause.from)
        }
      }
    }

    // Extract from SELECT (for IncludesSubquery)
    if (q.select) {
      extractFromSelect(q.select)
    }
  }

  function extractFromSelect(select: any) {
    for (const [key, value] of Object.entries(select)) {
      if (typeof key === `string` && key.startsWith(`__SPREAD_SENTINEL__`)) {
        continue
      }
      if (value instanceof IncludesSubquery) {
        extractFromQuery(value.query)
      } else if (value instanceof ConditionalSelect) {
        extractFromConditionalSelect(value)
      } else if (isNestedSelectObject(value)) {
        extractFromSelect(value)
      }
    }