pipework 0.7.13 → 0.7.18

package/dist/data/query/pg-core/db.d.ts

@@ -0,0 +1,292 @@
+import type { Cache } from '../cache/core/cache.js';
+import { entityKind } from '../entity.js';
+import type { PgDialect } from './dialect.js';
+import { PgDeleteBase, PgInsertBuilder, PgSelectBuilder, PgUpdateBuilder } from './query-builders/index.js';
+import type { PgQueryResultHKT, PgQueryResultKind, PgSession, PgTransaction, PgTransactionConfig } from './session.js';
+import type { PgTable } from './table.js';
+import type { ExtractTablesWithRelations, RelationalSchemaConfig, TablesRelationalConfig } from '../relations.js';
+import { type SQL, type SQLWrapper } from '../sql/sql.js';
+import { WithSubquery } from '../subquery.js';
+import type { DrizzleTypeError, NeonAuthToken } from '../utils.js';
+import type { PgColumn } from './columns/index.js';
+import { PgCountBuilder } from './query-builders/count.js';
+import { RelationalQueryBuilder } from './query-builders/query.js';
+import { PgRaw } from './query-builders/raw.js';
+import { PgRefreshMaterializedView } from './query-builders/refresh-materialized-view.js';
+import type { SelectedFields } from './query-builders/select.types.js';
+import type { WithBuilder } from './subquery.js';
+import type { PgViewBase } from './view-base.js';
+import type { PgMaterializedView } from './view.js';
+export declare class PgDatabase<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
+    /** @internal */
+    readonly dialect: PgDialect;
+    /** @internal */
+    readonly session: PgSession<any, any, any>;
+    static readonly [entityKind]: string;
+    readonly _: {
+        readonly schema: TSchema | undefined;
+        readonly fullSchema: TFullSchema;
+        readonly tableNamesMap: Record<string, string>;
+        readonly session: PgSession<TQueryResult, TFullSchema, TSchema>;
+    };
+    query: TFullSchema extends Record<string, never> ? DrizzleTypeError<'Seems like the schema generic is missing - did you forget to add it to your DB type?'> : {
+        [K in keyof TSchema]: RelationalQueryBuilder<TSchema, TSchema[K]>;
+    };
+    constructor(
+    /** @internal */
+    dialect: PgDialect, 
+    /** @internal */
+    session: PgSession<any, any, any>, schema: RelationalSchemaConfig<TSchema> | undefined);
+    /**
+     * Creates a subquery that defines a temporary named result set as a CTE.
+     *
+     * It is useful for breaking down complex queries into simpler parts and for reusing the result set in subsequent parts of the query.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+     *
+     * @param alias The alias for the subquery.
+     *
+     * Failure to provide an alias will result in a DrizzleTypeError, preventing the subquery from being referenced in other queries.
+     *
+     * @example
+     *
+     * ```ts
+     * // Create a subquery with alias 'sq' and use it in the select query
+     * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+     *
+     * const result = await db.with(sq).select().from(sq);
+     * ```
+     *
+     * To select arbitrary SQL values as fields in a CTE and reference them in other CTEs or in the main query, you need to add aliases to them:
+     *
+     * ```ts
+     * // Select an arbitrary SQL value as a field in a CTE and reference it in the main query
+     * const sq = db.$with('sq').as(db.select({
+     *   name: sql<string>`upper(${users.name})`.as('name'),
+     * })
+     * .from(users));
+     *
+     * const result = await db.with(sq).select({ name: sq.name }).from(sq);
+     * ```
+     */
+    $with: WithBuilder;
+    $count(source: PgTable | PgViewBase | SQL | SQLWrapper, filters?: SQL<unknown>): PgCountBuilder<PgSession<any, any, any>>;
+    $cache: {
+        invalidate: Cache['onMutate'];
+    };
+    /**
+     * Incorporates a previously defined CTE (using `$with`) into the main query.
+     *
+     * This method allows the main query to reference a temporary named result set.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+     *
+     * @param queries The CTEs to incorporate into the main query.
+     *
+     * @example
+     *
+     * ```ts
+     * // Define a subquery 'sq' as a CTE using $with
+     * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+     *
+     * // Incorporate the CTE 'sq' into the main query and select from it
+     * const result = await db.with(sq).select().from(sq);
+     * ```
+     */
+    with(...queries: WithSubquery[]): {
+        select: {
+            (): PgSelectBuilder<undefined>;
+            <TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
+        };
+        selectDistinct: {
+            (): PgSelectBuilder<undefined>;
+            <TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
+        };
+        selectDistinctOn: {
+            (on: (PgColumn | SQLWrapper)[]): PgSelectBuilder<undefined>;
+            <TSelection extends SelectedFields>(on: (PgColumn | SQLWrapper)[], fields: TSelection): PgSelectBuilder<TSelection>;
+        };
+        update: <TTable extends PgTable>(table: TTable) => PgUpdateBuilder<TTable, TQueryResult>;
+        insert: <TTable extends PgTable>(table: TTable) => PgInsertBuilder<TTable, TQueryResult>;
+        delete: <TTable extends PgTable>(table: TTable) => PgDeleteBase<TTable, TQueryResult>;
+    };
+    /**
+     * Creates a select query.
+     *
+     * Calling this method with no arguments will select all columns from the table. Pass a selection object to specify the columns you want to select.
+     *
+     * Use `.from()` method to specify which table to select from.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select}
+     *
+     * @param fields The selection object.
+     *
+     * @example
+     *
+     * ```ts
+     * // Select all columns and all rows from the 'cars' table
+     * const allCars: Car[] = await db.select().from(cars);
+     *
+     * // Select specific columns and all rows from the 'cars' table
+     * const carsIdsAndBrands: { id: number; brand: string }[] = await db.select({
+     *   id: cars.id,
+     *   brand: cars.brand
+     * })
+     *   .from(cars);
+     * ```
+     *
+     * Like in SQL, you can use arbitrary expressions as selection fields, not just table columns:
+     *
+     * ```ts
+     * // Select specific columns along with expression and all rows from the 'cars' table
+     * const carsIdsAndLowerNames: { id: number; lowerBrand: string }[] = await db.select({
+     *   id: cars.id,
+     *   lowerBrand: sql<string>`lower(${cars.brand})`,
+     * })
+     *   .from(cars);
+     * ```
+     */
+    select(): PgSelectBuilder<undefined>;
+    select<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
+    /**
+     * Adds `distinct` expression to the select query.
+     *
+     * Calling this method will return only unique values. When multiple columns are selected, it returns rows with unique combinations of values in these columns.
+     *
+     * Use `.from()` method to specify which table to select from.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#distinct}
+     *
+     * @param fields The selection object.
+     *
+     * @example
+     * ```ts
+     * // Select all unique rows from the 'cars' table
+     * await db.selectDistinct()
+     *   .from(cars)
+     *   .orderBy(cars.id, cars.brand, cars.color);
+     *
+     * // Select all unique brands from the 'cars' table
+     * await db.selectDistinct({ brand: cars.brand })
+     *   .from(cars)
+     *   .orderBy(cars.brand);
+     * ```
+     */
+    selectDistinct(): PgSelectBuilder<undefined>;
+    selectDistinct<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
+    /**
+     * Adds `distinct on` expression to the select query.
+     *
+     * Calling this method will specify how the unique rows are determined.
+     *
+     * Use `.from()` method to specify which table to select from.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#distinct}
+     *
+     * @param on The expression defining uniqueness.
+     * @param fields The selection object.
+     *
+     * @example
+     * ```ts
+     * // Select the first row for each unique brand from the 'cars' table
+     * await db.selectDistinctOn([cars.brand])
+     *   .from(cars)
+     *   .orderBy(cars.brand);
+     *
+     * // Selects the first occurrence of each unique car brand along with its color from the 'cars' table
+     * await db.selectDistinctOn([cars.brand], { brand: cars.brand, color: cars.color })
+     *   .from(cars)
+     *   .orderBy(cars.brand, cars.color);
+     * ```
+     */
+    selectDistinctOn(on: (PgColumn | SQLWrapper)[]): PgSelectBuilder<undefined>;
+    selectDistinctOn<TSelection extends SelectedFields>(on: (PgColumn | SQLWrapper)[], fields: TSelection): PgSelectBuilder<TSelection>;
+    /**
+     * Creates an update query.
+     *
+     * Calling this method without `.where()` clause will update all rows in a table. The `.where()` clause specifies which rows should be updated.
+     *
+     * Use `.set()` method to specify which values to update.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/update}
+     *
+     * @param table The table to update.
+     *
+     * @example
+     *
+     * ```ts
+     * // Update all rows in the 'cars' table
+     * await db.update(cars).set({ color: 'red' });
+     *
+     * // Update rows with filters and conditions
+     * await db.update(cars).set({ color: 'red' }).where(eq(cars.brand, 'BMW'));
+     *
+     * // Update with returning clause
+     * const updatedCar: Car[] = await db.update(cars)
+     *   .set({ color: 'red' })
+     *   .where(eq(cars.id, 1))
+     *   .returning();
+     * ```
+     */
+    update<TTable extends PgTable>(table: TTable): PgUpdateBuilder<TTable, TQueryResult>;
+    /**
+     * Creates an insert query.
+     *
+     * Calling this method will create new rows in a table. Use `.values()` method to specify which values to insert.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/insert}
+     *
+     * @param table The table to insert into.
+     *
+     * @example
+     *
+     * ```ts
+     * // Insert one row
+     * await db.insert(cars).values({ brand: 'BMW' });
+     *
+     * // Insert multiple rows
+     * await db.insert(cars).values([{ brand: 'BMW' }, { brand: 'Porsche' }]);
+     *
+     * // Insert with returning clause
+     * const insertedCar: Car[] = await db.insert(cars)
+     *   .values({ brand: 'BMW' })
+     *   .returning();
+     * ```
+     */
+    insert<TTable extends PgTable>(table: TTable): PgInsertBuilder<TTable, TQueryResult>;
+    /**
+     * Creates a delete query.
+     *
+     * Calling this method without `.where()` clause will delete all rows in a table. The `.where()` clause specifies which rows should be deleted.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/delete}
+     *
+     * @param table The table to delete from.
+     *
+     * @example
+     *
+     * ```ts
+     * // Delete all rows in the 'cars' table
+     * await db.delete(cars);
+     *
+     * // Delete rows with filters and conditions
+     * await db.delete(cars).where(eq(cars.color, 'green'));
+     *
+     * // Delete with returning clause
+     * const deletedCar: Car[] = await db.delete(cars)
+     *   .where(eq(cars.id, 1))
+     *   .returning();
+     * ```
+     */
+    delete<TTable extends PgTable>(table: TTable): PgDeleteBase<TTable, TQueryResult>;
+    refreshMaterializedView<TView extends PgMaterializedView>(view: TView): PgRefreshMaterializedView<TQueryResult>;
+    protected authToken?: NeonAuthToken;
+    execute<TRow extends Record<string, unknown> = Record<string, unknown>>(query: SQLWrapper | string): PgRaw<PgQueryResultKind<TQueryResult, TRow>>;
+    transaction<T>(transaction: (tx: PgTransaction<TQueryResult, TFullSchema, TSchema>) => Promise<T>, config?: PgTransactionConfig): Promise<T>;
+}
+export type PgWithReplicas<Q> = Q & {
+    $primary: Q;
+    $replicas: Q[];
+};
+export declare const withReplicas: <HKT extends PgQueryResultHKT, TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig, Q extends PgDatabase<HKT, TFullSchema, TSchema extends Record<string, unknown> ? ExtractTablesWithRelations<TFullSchema> : TSchema>>(primary: Q, replicas: [Q, ...Q[]], getReplica?: (replicas: Q[]) => Q) => PgWithReplicas<Q>;
+//# sourceMappingURL=db.d.ts.map

package/dist/data/query/pg-core/db.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.d.ts","sourceRoot":"","sources":["../../../../src/data/query/pg-core/db.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EACN,YAAY,EACZ,eAAe,EACf,eAAe,EACf,eAAe,EAEf,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EACX,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,aAAa,EACb,mBAAmB,EAEnB,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,KAAK,EAAE,0BAA0B,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAElH,OAAO,EAAyB,KAAK,GAAG,EAAO,KAAK,UAAU,EAAE,MAAM,eAAe,CAAC;AACtF,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAChD,OAAO,EAAE,yBAAyB,EAAE,MAAM,+CAA+C,CAAC;AAC1F,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAEpD,qBAAa,UAAU,CACtB,YAAY,SAAS,gBAAgB,EACrC,WAAW,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,EACnE,OAAO,SAAS,sBAAsB,GAAG,0BAA0B,CAAC,WAAW,CAAC;IAkB/E,gBAAgB;IAChB,QAAQ,CAAC,OAAO,EAAE,SAAS;IAC3B,gBAAgB;IAChB,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;IAnB3C,MAAM,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,EAAE,MAAM,CAAgB;IAEpD,SAAiB,CAAC,EAAE;QACnB,QAAQ,CAAC,MAAM,EAAE,OAAO,GAAG,SAAS,CAAC;QACrC,QAAQ,CAAC,UAAU,EAAE,WAAW,CAAC;QACjC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/C,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,YAAY,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;KAChE,CAAC;IAEF,KAAK,EAAE,WAAW,SAAS,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GAC7C,gBAAgB,CAAC,sFAAsF,CAAC,GACxG;SACA,CAAC,IAAI,MAAM,OAAO,GAAG,sBAAsB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;KACjE,CAAC;;IAGF,gBAAgB;IACP,OAAO,EAAE,SAAS;IAC3B,gBAAgB;IACP,OAAO,EAAE,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAC1C,MAAM,EAAE,sBAAsB,CAAC,OAAO,CAAC,GAAG,SAAS;IAgCpD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+BG;IACH,KAAK,EAAE,WAAW,CAuBhB;IAEF,MAAM,CACL,MAAM,EAAE,OAAO,GAAG,UAAU,GAAG,GAAG,GAAG,UAAU,EAC/C,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC;IAKvB,MAAM,EAAE;QAAE,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAA;KAAE,CAAC;IAE1C;;;;;;;;;;;;;;;;;;OAkBG;IACH,IAAI,CAAC,GAAG,OAAO,EAAE,YAAY,EAAE;;gBAuCX,eAAe,CAAC,SAAS,CAAC;aAC7B,UAAU,SAAS,cAAc,UAAU,UAAU,GAAG,eAAe,CAAC,UAAU,CAAC;;;gBAkCxE,eAAe,CAAC,SAAS,CAAC;aAC7B,UAAU,SAAS,cAAc,UAAU,UAAU,GAAG,eAAe,CAAC,UAAU,CAAC;;;iBAsC7E,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,GAAG,eAAe,CAAC,SAAS,CAAC;aAC1D,UAAU,SAAS,cAAc,MACtD,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,UACrB,UAAU,GAChB,eAAe,CAAC,UAAU,CAAC;;iBAyCd,MAAM,SAAS,OAAO,SAAS,MAAM,KAAG,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC;iBA4B7E,MAAM,SAAS,OAAO,SAAS,MAAM,KAAG,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC;iBA4B5E,MAAM,SAAS,OAAO,SAAS,MAAM,KAAG,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC;;IAO5F;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmCG;IACH,MAAM,IAAI,eAAe,CAAC,SAAS,CAAC;IACpC,MAAM,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,UAAU,GAAG,eAAe,CAAC,UAAU,CAAC;IAS1F;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,cAAc,IAAI,eAAe,CAAC,SAAS,CAAC;IAC5C,cAAc,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,UAAU,GAAG,eAAe,CAAC,UAAU,CAAC;IAUlG;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACH,gBAAgB,CAAC,EAAE,EAAE,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,GAAG,eAAe,CAAC,SAAS,CAAC;IAC3E,gBAAgB,CAAC,UAAU,SAAS,cAAc,EACjD,EAAE,EAAE,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,EAC7B,MAAM,EAAE,UAAU,GAChB,eAAe,CAAC,UAAU,CAAC;IAa9B;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,MAAM,CAAC,MAAM,SAAS,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC;IAIpF;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,MAAM,CAAC,MAAM,SAAS,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC;IAIpF;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,MAAM,CAAC,MAAM,SAAS,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC;IAIjF,uBAAuB,CAAC,KAAK,SAAS,kBAAkB,EAAE,IAAI,EAAE,KAAK,GAAG,yBAAyB,CAAC,YAAY,CAAC;IAI/G,SAAS,CAAC,SAAS,CAAC,EAAE,aAAa,CAAC;IAEpC,OAAO,CAAC,IAAI,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrE,KAAK,EAAE,UAAU,GAAG,MAAM,GACxB,KAAK,CAAC,iBAAiB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAmB/C,WAAW,CAAC,CAAC,EACZ,WAAW,EAAE,CAAC,EAAE,EAAE,aAAa,CAAC,YAAY,EAAE,WAAW,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,EAClF,MAAM,CAAC,EAAE,mBAAmB,GAC1B,OAAO,CAAC,CAAC,CAAC;CAGb;AAED,MAAM,MAAM,cAAc,CAAC,CAAC,IAAI,CAAC,GAAG;IAAE,QAAQ,EAAE,CAAC,CAAC;IAAC,SAAS,EAAE,CAAC,EAAE,CAAA;CAAE,CAAC;AAEpE,eAAO,MAAM,YAAY,GACxB,GAAG,SAAS,gBAAgB,EAC5B,WAAW,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC3C,OAAO,SAAS,sBAAsB,EACtC,CAAC,SAAS,UAAU,CACnB,GAAG,EACH,WAAW,EACX,OAAO,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,0BAA0B,CAAC,WAAW,CAAC,GAAG,OAAO,CAC3F,EAED,SAAS,CAAC,EACV,UAAU,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,EACrB,aAAY,CAAC,QAAQ,EAAE,CAAC,EAAE,KAAK,CAAgE,KAC7F,cAAc,CAAC,CAAC,CAoClB,CAAC"}

package/dist/data/query/pg-core/db.js

@@ -0,0 +1,377 @@
+import { entityKind } from '../entity.js';
+import { PgDeleteBase, PgInsertBuilder, PgSelectBuilder, PgUpdateBuilder, QueryBuilder, } from './query-builders/index.js';
+import { SelectionProxyHandler } from '../selection-proxy.js';
+import { sql } from '../sql/sql.js';
+import { WithSubquery } from '../subquery.js';
+import { PgCountBuilder } from './query-builders/count.js';
+import { RelationalQueryBuilder } from './query-builders/query.js';
+import { PgRaw } from './query-builders/raw.js';
+import { PgRefreshMaterializedView } from './query-builders/refresh-materialized-view.js';
+export class PgDatabase {
+    dialect;
+    session;
+    static [entityKind] = 'PgDatabase';
+    query;
+    constructor(
+    /** @internal */
+    dialect, 
+    /** @internal */
+    session, schema) {
+        this.dialect = dialect;
+        this.session = session;
+        this._ = schema
+            ? {
+                schema: schema.schema,
+                fullSchema: schema.fullSchema,
+                tableNamesMap: schema.tableNamesMap,
+                session,
+            }
+            : {
+                schema: undefined,
+                fullSchema: {},
+                tableNamesMap: {},
+                session,
+            };
+        this.query = {};
+        if (this._.schema) {
+            for (const [tableName, columns] of Object.entries(this._.schema)) {
+                this.query[tableName] = new RelationalQueryBuilder(schema.fullSchema, this._.schema, this._.tableNamesMap, schema.fullSchema[tableName], columns, dialect, session);
+            }
+        }
+        this.$cache = { invalidate: async (_params) => { } };
+    }
+    /**
+     * Creates a subquery that defines a temporary named result set as a CTE.
+     *
+     * It is useful for breaking down complex queries into simpler parts and for reusing the result set in subsequent parts of the query.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+     *
+     * @param alias The alias for the subquery.
+     *
+     * Failure to provide an alias will result in a DrizzleTypeError, preventing the subquery from being referenced in other queries.
+     *
+     * @example
+     *
+     * ```ts
+     * // Create a subquery with alias 'sq' and use it in the select query
+     * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+     *
+     * const result = await db.with(sq).select().from(sq);
+     * ```
+     *
+     * To select arbitrary SQL values as fields in a CTE and reference them in other CTEs or in the main query, you need to add aliases to them:
+     *
+     * ```ts
+     * // Select an arbitrary SQL value as a field in a CTE and reference it in the main query
+     * const sq = db.$with('sq').as(db.select({
+     *   name: sql<string>`upper(${users.name})`.as('name'),
+     * })
+     * .from(users));
+     *
+     * const result = await db.with(sq).select({ name: sq.name }).from(sq);
+     * ```
+     */
+    $with = (alias, selection) => {
+        const self = this;
+        const as = (qb) => {
+            if (typeof qb === 'function') {
+                qb = qb(new QueryBuilder(self.dialect));
+            }
+            return new Proxy(new WithSubquery(qb.getSQL(), selection ?? ('getSelectedFields' in qb ? qb.getSelectedFields() ?? {} : {}), alias, true), new SelectionProxyHandler({ alias, sqlAliasedBehavior: 'alias', sqlBehavior: 'error' }));
+        };
+        return { as };
+    };
+    $count(source, filters) {
+        return new PgCountBuilder({ source, filters, session: this.session });
+    }
+    $cache;
+    /**
+     * Incorporates a previously defined CTE (using `$with`) into the main query.
+     *
+     * This method allows the main query to reference a temporary named result set.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+     *
+     * @param queries The CTEs to incorporate into the main query.
+     *
+     * @example
+     *
+     * ```ts
+     * // Define a subquery 'sq' as a CTE using $with
+     * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+     *
+     * // Incorporate the CTE 'sq' into the main query and select from it
+     * const result = await db.with(sq).select().from(sq);
+     * ```
+     */
+    with(...queries) {
+        const self = this;
+        function select(fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: self.session,
+                dialect: self.dialect,
+                withList: queries,
+            });
+        }
+        function selectDistinct(fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: self.session,
+                dialect: self.dialect,
+                withList: queries,
+                distinct: true,
+            });
+        }
+        function selectDistinctOn(on, fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: self.session,
+                dialect: self.dialect,
+                withList: queries,
+                distinct: { on },
+            });
+        }
+        /**
+         * Creates an update query.
+         *
+         * Calling this method without `.where()` clause will update all rows in a table. The `.where()` clause specifies which rows should be updated.
+         *
+         * Use `.set()` method to specify which values to update.
+         *
+         * See docs: {@link https://orm.drizzle.team/docs/update}
+         *
+         * @param table The table to update.
+         *
+         * @example
+         *
+         * ```ts
+         * // Update all rows in the 'cars' table
+         * await db.update(cars).set({ color: 'red' });
+         *
+         * // Update rows with filters and conditions
+         * await db.update(cars).set({ color: 'red' }).where(eq(cars.brand, 'BMW'));
+         *
+         * // Update with returning clause
+         * const updatedCar: Car[] = await db.update(cars)
+         *   .set({ color: 'red' })
+         *   .where(eq(cars.id, 1))
+         *   .returning();
+         * ```
+         */
+        function update(table) {
+            return new PgUpdateBuilder(table, self.session, self.dialect, queries);
+        }
+        /**
+         * Creates an insert query.
+         *
+         * Calling this method will create new rows in a table. Use `.values()` method to specify which values to insert.
+         *
+         * See docs: {@link https://orm.drizzle.team/docs/insert}
+         *
+         * @param table The table to insert into.
+         *
+         * @example
+         *
+         * ```ts
+         * // Insert one row
+         * await db.insert(cars).values({ brand: 'BMW' });
+         *
+         * // Insert multiple rows
+         * await db.insert(cars).values([{ brand: 'BMW' }, { brand: 'Porsche' }]);
+         *
+         * // Insert with returning clause
+         * const insertedCar: Car[] = await db.insert(cars)
+         *   .values({ brand: 'BMW' })
+         *   .returning();
+         * ```
+         */
+        function insert(table) {
+            return new PgInsertBuilder(table, self.session, self.dialect, queries);
+        }
+        /**
+         * Creates a delete query.
+         *
+         * Calling this method without `.where()` clause will delete all rows in a table. The `.where()` clause specifies which rows should be deleted.
+         *
+         * See docs: {@link https://orm.drizzle.team/docs/delete}
+         *
+         * @param table The table to delete from.
+         *
+         * @example
+         *
+         * ```ts
+         * // Delete all rows in the 'cars' table
+         * await db.delete(cars);
+         *
+         * // Delete rows with filters and conditions
+         * await db.delete(cars).where(eq(cars.color, 'green'));
+         *
+         * // Delete with returning clause
+         * const deletedCar: Car[] = await db.delete(cars)
+         *   .where(eq(cars.id, 1))
+         *   .returning();
+         * ```
+         */
+        function delete_(table) {
+            return new PgDeleteBase(table, self.session, self.dialect, queries);
+        }
+        return { select, selectDistinct, selectDistinctOn, update, insert, delete: delete_ };
+    }
+    select(fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: this.session,
+            dialect: this.dialect,
+        });
+    }
+    selectDistinct(fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: this.session,
+            dialect: this.dialect,
+            distinct: true,
+        });
+    }
+    selectDistinctOn(on, fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: this.session,
+            dialect: this.dialect,
+            distinct: { on },
+        });
+    }
+    /**
+     * Creates an update query.
+     *
+     * Calling this method without `.where()` clause will update all rows in a table. The `.where()` clause specifies which rows should be updated.
+     *
+     * Use `.set()` method to specify which values to update.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/update}
+     *
+     * @param table The table to update.
+     *
+     * @example
+     *
+     * ```ts
+     * // Update all rows in the 'cars' table
+     * await db.update(cars).set({ color: 'red' });
+     *
+     * // Update rows with filters and conditions
+     * await db.update(cars).set({ color: 'red' }).where(eq(cars.brand, 'BMW'));
+     *
+     * // Update with returning clause
+     * const updatedCar: Car[] = await db.update(cars)
+     *   .set({ color: 'red' })
+     *   .where(eq(cars.id, 1))
+     *   .returning();
+     * ```
+     */
+    update(table) {
+        return new PgUpdateBuilder(table, this.session, this.dialect);
+    }
+    /**
+     * Creates an insert query.
+     *
+     * Calling this method will create new rows in a table. Use `.values()` method to specify which values to insert.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/insert}
+     *
+     * @param table The table to insert into.
+     *
+     * @example
+     *
+     * ```ts
+     * // Insert one row
+     * await db.insert(cars).values({ brand: 'BMW' });
+     *
+     * // Insert multiple rows
+     * await db.insert(cars).values([{ brand: 'BMW' }, { brand: 'Porsche' }]);
+     *
+     * // Insert with returning clause
+     * const insertedCar: Car[] = await db.insert(cars)
+     *   .values({ brand: 'BMW' })
+     *   .returning();
+     * ```
+     */
+    insert(table) {
+        return new PgInsertBuilder(table, this.session, this.dialect);
+    }
+    /**
+     * Creates a delete query.
+     *
+     * Calling this method without `.where()` clause will delete all rows in a table. The `.where()` clause specifies which rows should be deleted.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/delete}
+     *
+     * @param table The table to delete from.
+     *
+     * @example
+     *
+     * ```ts
+     * // Delete all rows in the 'cars' table
+     * await db.delete(cars);
+     *
+     * // Delete rows with filters and conditions
+     * await db.delete(cars).where(eq(cars.color, 'green'));
+     *
+     * // Delete with returning clause
+     * const deletedCar: Car[] = await db.delete(cars)
+     *   .where(eq(cars.id, 1))
+     *   .returning();
+     * ```
+     */
+    delete(table) {
+        return new PgDeleteBase(table, this.session, this.dialect);
+    }
+    refreshMaterializedView(view) {
+        return new PgRefreshMaterializedView(view, this.session, this.dialect);
+    }
+    authToken;
+    execute(query) {
+        const sequel = typeof query === 'string' ? sql.raw(query) : query.getSQL();
+        const builtQuery = this.dialect.sqlToQuery(sequel);
+        const prepared = this.session.prepareQuery(builtQuery, undefined, undefined, false);
+        return new PgRaw(() => prepared.execute(undefined, this.authToken), sequel, builtQuery, (result) => prepared.mapResult(result, true));
+    }
+    transaction(transaction, config) {
+        return this.session.transaction(transaction, config);
+    }
+}
+export const withReplicas = (primary, replicas, getReplica = () => replicas[Math.floor(Math.random() * replicas.length)]) => {
+    const select = (...args) => getReplica(replicas).select(...args);
+    const selectDistinct = (...args) => getReplica(replicas).selectDistinct(...args);
+    const selectDistinctOn = (...args) => getReplica(replicas).selectDistinctOn(...args);
+    const $count = (...args) => getReplica(replicas).$count(...args);
+    const _with = (...args) => getReplica(replicas).with(...args);
+    const $with = (arg) => getReplica(replicas).$with(arg);
+    const update = (...args) => primary.update(...args);
+    const insert = (...args) => primary.insert(...args);
+    const $delete = (...args) => primary.delete(...args);
+    const execute = (...args) => primary.execute(...args);
+    const transaction = (...args) => primary.transaction(...args);
+    const refreshMaterializedView = (...args) => primary.refreshMaterializedView(...args);
+    return {
+        ...primary,
+        update,
+        insert,
+        delete: $delete,
+        execute,
+        transaction,
+        refreshMaterializedView,
+        $primary: primary,
+        $replicas: replicas,
+        select,
+        selectDistinct,
+        selectDistinctOn,
+        $count,
+        $with,
+        with: _with,
+        get query() {
+            return getReplica(replicas).query;
+        },
+    };
+};
+//# sourceMappingURL=db.js.map