@tanstack/db/dist/cjs/query/query-builder.cjs.map

A reactive client store for building super fast apps on sync

{"version":3,"file":"query-builder.cjs","sources":["../../../src/query/query-builder.ts"],"sourcesContent":["import type { Collection } from \"../collection\"\nimport type {\n  Comparator,\n  ComparatorValue,\n  Condition,\n  From,\n  JoinClause,\n  Limit,\n  LiteralValue,\n  Offset,\n  OrderBy,\n  Query,\n  Select,\n  WhereCallback,\n  WithQuery,\n} from \"./schema.js\"\nimport type {\n  Context,\n  Flatten,\n  InferResultTypeFromSelectTuple,\n  Input,\n  InputReference,\n  PropertyReference,\n  PropertyReferenceString,\n  RemoveIndexSignature,\n  Schema,\n} from \"./types.js\"\n\ntype CollectionRef = { [K: string]: Collection<any> }\n\nexport class BaseQueryBuilder<TContext extends Context<Schema>> {\n  private readonly query: Partial<Query<TContext>> = {}\n\n  /**\n   * Create a new QueryBuilder instance.\n   */\n  constructor(query: Partial<Query<TContext>> = {}) {\n    this.query = query\n  }\n\n  from<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ): QueryBuilder<{\n    baseSchema: Flatten<\n      TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n          (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n            ? T\n            : never) &\n            Input\n        >\n      }\n    >\n    schema: Flatten<{\n      [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n        (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n          ? T\n          : never) &\n          Input\n      >\n    }>\n    default: keyof TCollectionRef & string\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(\n    collection: T\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: T\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(\n    collection: T,\n    as: TAs\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: TAs\n  }>\n\n  /**\n   * Specify the collection to query from.\n   * This is the first method that must be called in the chain.\n   *\n   * @param collection The collection name to query from\n   * @param as Optional alias for the collection\n   * @returns A new QueryBuilder with the from clause set\n   */\n  from<\n    T extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof collection === `object` && collection !== null) {\n      return this.fromCollectionRef(collection)\n    } else if (typeof collection === `string`) {\n      return this.fromInputReference(\n        collection as InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>,\n        as\n      )\n    } else {\n      throw new Error(`Invalid collection type`)\n    }\n  }\n\n  private fromCollectionRef<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ) {\n    const keys = Object.keys(collectionRef)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key`)\n    }\n\n    const key = keys[0]!\n    const collection = collectionRef[key]!\n\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = key as From<TContext>\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      schema: {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      default: keyof TCollectionRef & string\n    }>\n  }\n\n  private fromInputReference<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = collection as From<TContext>\n    if (as) {\n      newBuilder.query.as = as\n    }\n\n    // Calculate the result type without deep nesting\n    type ResultSchema = TAs extends undefined\n      ? { [K in T]: TContext[`baseSchema`][T] }\n      : { [K in string & TAs]: TContext[`baseSchema`][T] }\n\n    type ResultDefault = TAs extends undefined ? T : string & TAs\n\n    // Use simpler type assertion to avoid excessive depth\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: ResultSchema\n      default: ResultDefault\n    }>\n  }\n\n  /**\n   * Specify what columns to select.\n   * Overwrites any previous select clause.\n   * Also supports callback functions that receive the row context and return selected data.\n   *\n   * @param selects The columns to select (can include callbacks)\n   * @returns A new QueryBuilder with the select clause set\n   */\n  select<TSelects extends Array<Select<TContext>>>(\n    this: QueryBuilder<TContext>,\n    ...selects: TSelects\n  ) {\n    // Validate function calls in the selects\n    // Need to use a type assertion to bypass deep recursive type checking\n    const validatedSelects = selects.map((select) => {\n      // If the select is an object with aliases, validate each value\n      if (\n        typeof select === `object` &&\n        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n        select !== null &&\n        !Array.isArray(select)\n      ) {\n        const result: Record<string, any> = {}\n\n        for (const [key, value] of Object.entries(select)) {\n          // If it's a function call (object with a single key that is an allowed function name)\n          if (\n            typeof value === `object` &&\n            value !== null &&\n            !Array.isArray(value)\n          ) {\n            const keys = Object.keys(value)\n            if (keys.length === 1) {\n              const funcName = keys[0]!\n              // List of allowed function names from AllowedFunctionName\n              const allowedFunctions = [\n                `SUM`,\n                `COUNT`,\n                `AVG`,\n                `MIN`,\n                `MAX`,\n                `DATE`,\n                `JSON_EXTRACT`,\n                `JSON_EXTRACT_PATH`,\n                `UPPER`,\n                `LOWER`,\n                `COALESCE`,\n                `CONCAT`,\n                `LENGTH`,\n                `ORDER_INDEX`,\n              ]\n\n              if (!allowedFunctions.includes(funcName)) {\n                console.warn(\n                  `Unsupported function: ${funcName}. Expected one of: ${allowedFunctions.join(`, `)}`\n                )\n              }\n            }\n          }\n\n          result[key] = value\n        }\n\n        return result\n      }\n\n      return select\n    })\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called after orderBy\n    if (this._query.orderBy) {\n      validatedSelects.push({ _orderByIndex: { ORDER_INDEX: `numeric` } })\n    }\n\n    const newBuilder = new BaseQueryBuilder<TContext>(\n      (this as BaseQueryBuilder<TContext>).query\n    )\n    newBuilder.query.select = validatedSelects as Array<Select<TContext>>\n\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `result`> & {\n          result: InferResultTypeFromSelectTuple<TContext, TSelects>\n        }\n      >\n    >\n  }\n\n  /**\n   * Add a where clause comparing two values.\n   */\n  where<T extends Comparator>(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: T,\n    right: ComparatorValue<T, TContext>\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a complete condition object.\n   */\n  where(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a callback function.\n   */\n  where(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause to filter the results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the where clause added\n   */\n  where(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    // Use simplistic approach to avoid deep type errors\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Where is always an array, so initialize or append\n    if (!newBuilder.query.where) {\n      newBuilder.query.where = [condition]\n    } else {\n      newBuilder.query.where = [...newBuilder.query.where, condition]\n    }\n\n    return newBuilder as unknown as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a having clause comparing two values.\n   * For filtering results after they have been grouped.\n   */\n  having(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: Comparator,\n    right: PropertyReferenceString<TContext> | LiteralValue\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a complete condition object.\n   * For filtering results after they have been grouped.\n   */\n  having(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a callback function.\n   * For filtering results after they have been grouped.\n   */\n  having(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause to filter the grouped results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the having clause added\n   */\n  having(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Having is always an array, so initialize or append\n    if (!newBuilder.query.having) {\n      newBuilder.query.having = [condition]\n    } else {\n      newBuilder.query.having = [...newBuilder.query.having, condition]\n    }\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a join clause to the query using a CollectionRef.\n   */\n  join<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query without specifying an alias.\n   * The collection name will be used as the default alias.\n   */\n  join<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: T\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: { [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]> }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query with a specified alias.\n   */\n  join<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  join<\n    TFrom extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n  }): QueryBuilder<any> {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof joinClause.from === `object` && joinClause.from !== null) {\n      return this.joinCollectionRef(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: CollectionRef\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    } else {\n      return this.joinInputReference(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: InputReference<{\n            baseSchema: TContext[`baseSchema`]\n            schema: TContext[`baseSchema`]\n          }>\n          as?: TAs\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    }\n  }\n\n  private joinCollectionRef<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Get the collection key\n    const keys = Object.keys(joinClause.from)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key in CollectionRef`)\n    }\n    const key = keys[0]!\n    const collection = joinClause.from[key]\n    if (!collection) {\n      throw new Error(`Collection not found for key: ${key}`)\n    }\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = {\n      type: joinClause.type,\n      from: key,\n      on: joinClause.on,\n      where: joinClause.where,\n    } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Add the collection to the collections map\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n              (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n                ? T\n                : never) &\n                Input\n            >\n          }\n        }\n      >\n    >\n  }\n\n  private joinInputReference<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = { ...joinClause } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Determine the alias or use the collection name as default\n    const _effectiveAlias = joinClause.as ?? joinClause.from\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in typeof _effectiveAlias]: TContext[`baseSchema`][TFrom]\n          }\n        }\n      >\n    >\n  }\n\n  /**\n   * Add an orderBy clause to sort the results.\n   * Overwrites any previous orderBy clause.\n   *\n   * @param orderBy The order specification\n   * @returns A new QueryBuilder with the orderBy clause set\n   */\n  orderBy(orderBy: OrderBy<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the orderBy clause\n    newBuilder.query.orderBy = orderBy\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called before orderBy\n    newBuilder.query.select = [\n      ...(newBuilder.query.select ?? []),\n      { _orderByIndex: { ORDER_INDEX: `numeric` } },\n    ]\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set a limit on the number of results returned.\n   *\n   * @param limit Maximum number of results to return\n   * @returns A new QueryBuilder with the limit set\n   */\n  limit(limit: Limit<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the limit\n    newBuilder.query.limit = limit\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set an offset to skip a number of results.\n   *\n   * @param offset Number of results to skip\n   * @returns A new QueryBuilder with the offset set\n   */\n  offset(offset: Offset<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the offset\n    newBuilder.query.offset = offset\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a groupBy clause to group the results by one or more columns.\n   *\n   * @param groupBy The column(s) to group by\n   * @returns A new QueryBuilder with the groupBy clause set\n   */\n  groupBy(\n    groupBy: PropertyReference<TContext> | Array<PropertyReference<TContext>>\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the groupBy clause\n    newBuilder.query.groupBy = groupBy\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Define a Common Table Expression (CTE) that can be referenced in the main query.\n   * This allows referencing the CTE by name in subsequent from/join clauses.\n   *\n   * @param name The name of the CTE\n   * @param queryBuilderCallback A function that builds the CTE query\n   * @returns A new QueryBuilder with the CTE added\n   */\n  with<TName extends string, TResult = Record<string, unknown>>(\n    name: TName,\n    queryBuilderCallback: (\n      builder: InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    ) => QueryBuilder<any>\n  ): InitialQueryBuilder<{\n    baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n    schema: TContext[`schema`]\n  }> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a new builder for the CTE\n    const cteBuilder = new BaseQueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: {}\n    }>()\n\n    // Get the CTE query from the callback\n    const cteQueryBuilder = queryBuilderCallback(\n      cteBuilder as InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    )\n\n    // Get the query from the builder\n    const cteQuery = cteQueryBuilder._query\n\n    // Add an 'as' property to the CTE\n    const withQuery: WithQuery<any> = {\n      ...cteQuery,\n      as: name,\n    }\n\n    // Add the CTE to the with array\n    if (!newBuilder.query.with) {\n      newBuilder.query.with = [withQuery]\n    } else {\n      newBuilder.query.with = [...newBuilder.query.with, withQuery]\n    }\n\n    // Use a type cast that simplifies the type structure to avoid recursion\n    return newBuilder as unknown as InitialQueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n      schema: TContext[`schema`]\n    }>\n  }\n\n  get _query(): Query<TContext> {\n    return this.query as Query<TContext>\n  }\n}\n\nexport type InitialQueryBuilder<TContext extends Context<Schema>> = Pick<\n  BaseQueryBuilder<TContext>,\n  `from` | `with`\n>\n\nexport type QueryBuilder<TContext extends Context<Schema>> = Omit<\n  BaseQueryBuilder<TContext>,\n  `from`\n>\n\n/**\n * Create a new query builder with the given schema\n */\nexport function queryBuilder<TBaseSchema extends Schema = {}>() {\n  return new BaseQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>() as InitialQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>\n}\n\nexport type ResultsFromContext<TContext extends Context<Schema>> = Flatten<\n  TContext[`result`] extends object\n    ? TContext[`result`] // If there is a select we will have a result type\n    : TContext[`hasJoin`] extends true\n      ? TContext[`schema`] // If there is a join, the query returns the namespaced schema\n      : TContext[`default`] extends keyof TContext[`schema`]\n        ? TContext[`schema`][TContext[`default`]] // If there is no join we return the flat default schema\n        : never // Should never happen\n>\n\nexport type ResultFromQueryBuilder<TQueryBuilder> = Flatten<\n  TQueryBuilder extends QueryBuilder<infer C>\n    ? C extends { result: infer R }\n      ? R\n      : never\n    : never\n>\n"],"names":[],"mappings":";;AA8BO,MAAM,iBAAmD;AAAA;AAAA;AAAA;AAAA,EAM9D,YAAY,QAAkC,IAAI;AALlD,SAAiB,QAAkC,CAAC;AAMlD,SAAK,QAAQ;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmEf,KAQE,YAAe,IAAU;AAEzB,QAAI,OAAO,eAAe,YAAY,eAAe,MAAM;AAClD,aAAA,KAAK,kBAAkB,UAAU;AAAA,IAC1C,WAAW,OAAO,eAAe,UAAU;AACzC,aAAO,KAAK;AAAA,QACV;AAAA,QAIA;AAAA,MACF;AAAA,IAAA,OACK;AACC,YAAA,IAAI,MAAM,yBAAyB;AAAA,IAAA;AAAA,EAC3C;AAAA,EAGM,kBACN,eACA;;AACM,UAAA,OAAO,OAAO,KAAK,aAAa;AAClC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,0BAA0B;AAAA,IAAA;AAGtC,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,cAAc,GAAG;AAE9B,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACb,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAE7B,WAAA;AAAA,EAAA;AAAA,EAuBD,mBAMN,YAAe,IAAU;AACnB,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACxB,QAAI,IAAI;AACN,iBAAW,MAAM,KAAK;AAAA,IAAA;AAWjB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeT,UAEK,SACH;AAGA,UAAM,mBAAmB,QAAQ,IAAI,CAAC,WAAW;AAE/C,UACE,OAAO,WAAW;AAAA,MAElB,WAAW,QACX,CAAC,MAAM,QAAQ,MAAM,GACrB;AACA,cAAM,SAA8B,CAAC;AAErC,mBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,MAAM,GAAG;AAG/C,cAAA,OAAO,UAAU,YACjB,UAAU,QACV,CAAC,MAAM,QAAQ,KAAK,GACpB;AACM,kBAAA,OAAO,OAAO,KAAK,KAAK;AAC1B,gBAAA,KAAK,WAAW,GAAG;AACf,oBAAA,WAAW,KAAK,CAAC;AAEvB,oBAAM,mBAAmB;AAAA,gBACvB;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACF;AAEA,kBAAI,CAAC,iBAAiB,SAAS,QAAQ,GAAG;AAChC,wBAAA;AAAA,kBACN,yBAAyB,QAAQ,sBAAsB,iBAAiB,KAAK,IAAI,CAAC;AAAA,gBACpF;AAAA,cAAA;AAAA,YACF;AAAA,UACF;AAGF,iBAAO,GAAG,IAAI;AAAA,QAAA;AAGT,eAAA;AAAA,MAAA;AAGF,aAAA;AAAA,IAAA,CACR;AAIG,QAAA,KAAK,OAAO,SAAS;AACvB,uBAAiB,KAAK,EAAE,eAAe,EAAE,aAAa,UAAA,GAAa;AAAA,IAAA;AAGrE,UAAM,aAAa,IAAI;AAAA,MACpB,KAAoC;AAAA,IACvC;AACA,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAsCT,MACE,2BACA,UACA,OACwB;AAGlB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,OAAO;AAChB,iBAAA,MAAM,QAAQ,CAAC,SAAS;AAAA,IAAA,OAC9B;AACL,iBAAW,MAAM,QAAQ,CAAC,GAAG,WAAW,MAAM,OAAO,SAAS;AAAA,IAAA;AAGzD,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmCT,OACE,2BACA,UACA,OACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,QAAQ;AACjB,iBAAA,MAAM,SAAS,CAAC,SAAS;AAAA,IAAA,OAC/B;AACL,iBAAW,MAAM,SAAS,CAAC,GAAG,WAAW,MAAM,QAAQ,SAAS;AAAA,IAAA;AAG3D,WAAA;AAAA,EAAA;AAAA,EAgIT,KAQE,YA4CoB;AAEpB,QAAI,OAAO,WAAW,SAAS,YAAY,WAAW,SAAS,MAAM;AACnE,aAAO,KAAK;AAAA,QACV;AAAA,MAMF;AAAA,IAAA,OACK;AACL,aAAO,KAAK;AAAA,QACV;AAAA,MAUF;AAAA,IAAA;AAAA,EACF;AAAA,EAGM,kBAAwD,YAK1C;;AAEd,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,UAAM,OAAO,OAAO,KAAK,WAAW,IAAI;AACpC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,2CAA2C;AAAA,IAAA;AAEvD,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,WAAW,KAAK,GAAG;AACtC,QAAI,CAAC,YAAY;AACf,YAAM,IAAI,MAAM,iCAAiC,GAAG,EAAE;AAAA,IAAA;AAIxD,UAAM,iBAAiB;AAAA,MACrB,MAAM,WAAW;AAAA,MACjB,MAAM;AAAA,MACN,IAAI,WAAW;AAAA,MACf,OAAO,WAAW;AAAA,IACpB;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAIxD,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAG7B,WAAA;AAAA,EAAA;AAAA,EAgBD,mBAMN,YAMoB;AAEd,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,iBAAiB,EAAE,GAAG,WAAW;AAGnC,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAI3C,eAAW,MAAM,WAAW;AAG7C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBT,QAAQ,SAAoD;AAEpD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAI3B,eAAW,MAAM,SAAS;AAAA,MACxB,GAAI,WAAW,MAAM,UAAU,CAAC;AAAA,MAChC,EAAE,eAAe,EAAE,aAAa,UAAY,EAAA;AAAA,IAC9C;AAEO,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,MAAM,OAAgD;AAE9C,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,QAAQ;AAElB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,OAAO,QAAkD;AAEjD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,QACE,SACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAEpB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWT,KACE,MACA,sBASC;AAEK,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,aAAa,IAAI,iBAGpB;AAGH,UAAM,kBAAkB;AAAA,MACtB;AAAA,IAIF;AAGA,UAAM,WAAW,gBAAgB;AAGjC,UAAM,YAA4B;AAAA,MAChC,GAAG;AAAA,MACH,IAAI;AAAA,IACN;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,SAAS;AAAA,IAAA,OAC7B;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,SAAS;AAAA,IAAA;AAIvD,WAAA;AAAA,EAAA;AAAA,EAMT,IAAI,SAA0B;AAC5B,WAAO,KAAK;AAAA,EAAA;AAEhB;AAeO,SAAS,eAAgD;AAC9D,SAAO,IAAI,iBAGR;AAIL;;;"}