pipework 0.7.13 → 0.7.17

package/dist/data/query/pg-core/query-builders/insert.js

@@ -0,0 +1,201 @@
+import { entityKind, is } from '../../entity.js';
+import { QueryPromise } from '../../query-promise.js';
+import { SelectionProxyHandler } from '../../selection-proxy.js';
+import { Param, SQL, sql } from '../../sql/sql.js';
+import { Columns, getTableName, Table } from '../../table.js';
+import { tracer } from '../../tracing.js';
+import { haveSameKeys, mapUpdateSet, orderSelectedFields } from '../../utils.js';
+import { extractUsedTable } from '../utils.js';
+import { QueryBuilder } from './query-builder.js';
+export class PgInsertBuilder {
+    table;
+    session;
+    dialect;
+    withList;
+    overridingSystemValue_;
+    static [entityKind] = 'PgInsertBuilder';
+    constructor(table, session, dialect, withList, overridingSystemValue_) {
+        this.table = table;
+        this.session = session;
+        this.dialect = dialect;
+        this.withList = withList;
+        this.overridingSystemValue_ = overridingSystemValue_;
+    }
+    authToken;
+    /** @internal */
+    setToken(token) {
+        this.authToken = token;
+        return this;
+    }
+    overridingSystemValue() {
+        this.overridingSystemValue_ = true;
+        return this;
+    }
+    values(values) {
+        values = Array.isArray(values) ? values : [values];
+        if (values.length === 0) {
+            throw new Error('values() must be called with at least one value');
+        }
+        const mappedValues = values.map((entry) => {
+            const result = {};
+            const cols = this.table[Table.Symbol.Columns];
+            for (const colKey of Object.keys(entry)) {
+                const colValue = entry[colKey];
+                result[colKey] = is(colValue, SQL) ? colValue : new Param(colValue, cols[colKey]);
+            }
+            return result;
+        });
+        return new PgInsertBase(this.table, mappedValues, this.session, this.dialect, this.withList, false, this.overridingSystemValue_).setToken(this.authToken);
+    }
+    select(selectQuery) {
+        const select = typeof selectQuery === 'function' ? selectQuery(new QueryBuilder()) : selectQuery;
+        if (!is(select, SQL)
+            && !haveSameKeys(this.table[Columns], select._.selectedFields)) {
+            throw new Error('Insert select error: selected fields are not the same or are in a different order compared to the table definition');
+        }
+        return new PgInsertBase(this.table, select, this.session, this.dialect, this.withList, true);
+    }
+}
+export class PgInsertBase extends QueryPromise {
+    session;
+    dialect;
+    static [entityKind] = 'PgInsert';
+    config;
+    cacheConfig;
+    constructor(table, values, session, dialect, withList, select, overridingSystemValue_) {
+        super();
+        this.session = session;
+        this.dialect = dialect;
+        this.config = { table, values: values, withList, select, overridingSystemValue_ };
+    }
+    returning(fields = this.config.table[Table.Symbol.Columns]) {
+        this.config.returningFields = fields;
+        this.config.returning = orderSelectedFields(fields);
+        return this;
+    }
+    /**
+     * Adds an `on conflict do nothing` clause to the query.
+     *
+     * Calling this method simply avoids inserting a row as its alternative action.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/insert#on-conflict-do-nothing}
+     *
+     * @param config The `target` and `where` clauses.
+     *
+     * @example
+     * ```ts
+     * // Insert one row and cancel the insert if there's a conflict
+     * await db.insert(cars)
+     *   .values({ id: 1, brand: 'BMW' })
+     *   .onConflictDoNothing();
+     *
+     * // Explicitly specify conflict target
+     * await db.insert(cars)
+     *   .values({ id: 1, brand: 'BMW' })
+     *   .onConflictDoNothing({ target: cars.id });
+     * ```
+     */
+    onConflictDoNothing(config = {}) {
+        if (config.target === undefined) {
+            this.config.onConflict = sql `do nothing`;
+        }
+        else {
+            let targetColumn = '';
+            targetColumn = Array.isArray(config.target)
+                ? config.target.map((it) => this.dialect.escapeName(this.dialect.casing.getColumnCasing(it))).join(',')
+                : this.dialect.escapeName(this.dialect.casing.getColumnCasing(config.target));
+            const whereSql = config.where ? sql ` where ${config.where}` : undefined;
+            this.config.onConflict = sql `(${sql.raw(targetColumn)})${whereSql} do nothing`;
+        }
+        return this;
+    }
+    /**
+     * Adds an `on conflict do update` clause to the query.
+     *
+     * Calling this method will update the existing row that conflicts with the row proposed for insertion as its alternative action.
+     *
+     * See docs: {@link https://orm.drizzle.team/docs/insert#upserts-and-conflicts}
+     *
+     * @param config The `target`, `set` and `where` clauses.
+     *
+     * @example
+     * ```ts
+     * // Update the row if there's a conflict
+     * await db.insert(cars)
+     *   .values({ id: 1, brand: 'BMW' })
+     *   .onConflictDoUpdate({
+     *     target: cars.id,
+     *     set: { brand: 'Porsche' }
+     *   });
+     *
+     * // Upsert with 'where' clause
+     * await db.insert(cars)
+     *   .values({ id: 1, brand: 'BMW' })
+     *   .onConflictDoUpdate({
+     *     target: cars.id,
+     *     set: { brand: 'newBMW' },
+     *     targetWhere: sql`${cars.createdAt} > '2023-01-01'::date`,
+     *   });
+     * ```
+     */
+    onConflictDoUpdate(config) {
+        if (config.where && (config.targetWhere || config.setWhere)) {
+            throw new Error('You cannot use both "where" and "targetWhere"/"setWhere" at the same time - "where" is deprecated, use "targetWhere" or "setWhere" instead.');
+        }
+        const whereSql = config.where ? sql ` where ${config.where}` : undefined;
+        const targetWhereSql = config.targetWhere ? sql ` where ${config.targetWhere}` : undefined;
+        const setWhereSql = config.setWhere ? sql ` where ${config.setWhere}` : undefined;
+        const setSql = this.dialect.buildUpdateSet(this.config.table, mapUpdateSet(this.config.table, config.set));
+        let targetColumn = '';
+        targetColumn = Array.isArray(config.target)
+            ? config.target.map((it) => this.dialect.escapeName(this.dialect.casing.getColumnCasing(it))).join(',')
+            : this.dialect.escapeName(this.dialect.casing.getColumnCasing(config.target));
+        this.config.onConflict = sql `(${sql.raw(targetColumn)})${targetWhereSql} do update set ${setSql}${whereSql}${setWhereSql}`;
+        return this;
+    }
+    /** @internal */
+    getSQL() {
+        return this.dialect.buildInsertQuery(this.config);
+    }
+    toSQL() {
+        const { typings: _typings, ...rest } = this.dialect.sqlToQuery(this.getSQL());
+        return rest;
+    }
+    /** @internal */
+    _prepare(name) {
+        return tracer.startActiveSpan('drizzle.prepareQuery', () => {
+            return this.session.prepareQuery(this.dialect.sqlToQuery(this.getSQL()), this.config.returning, name, true, undefined, {
+                type: 'insert',
+                tables: extractUsedTable(this.config.table),
+            }, this.cacheConfig);
+        });
+    }
+    prepare(name) {
+        return this._prepare(name);
+    }
+    authToken;
+    /** @internal */
+    setToken(token) {
+        this.authToken = token;
+        return this;
+    }
+    execute = (placeholderValues) => {
+        return tracer.startActiveSpan('drizzle.operation', () => {
+            return this._prepare().execute(placeholderValues, this.authToken);
+        });
+    };
+    /** @internal */
+    getSelectedFields() {
+        return (this.config.returningFields
+            ? new Proxy(this.config.returningFields, new SelectionProxyHandler({
+                alias: getTableName(this.config.table),
+                sqlAliasedBehavior: 'alias',
+                sqlBehavior: 'error',
+            }))
+            : undefined);
+    }
+    $dynamic() {
+        return this;
+    }
+}
+//# sourceMappingURL=insert.js.map

package/dist/data/query/pg-core/query-builders/insert.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"insert.js","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/insert.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,iBAAiB,CAAC;AAajD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,kBAAkB,CAAC;AAGnD,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,YAAY,EAAsB,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAErG,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AA4BlD,MAAM,OAAO,eAAe;IAQlB;IACA;IACA;IACA;IACA;IAPT,MAAM,CAAU,CAAC,UAAU,CAAC,GAAW,iBAAiB,CAAC;IAEzD,YACS,KAAa,EACb,OAAkB,EAClB,OAAkB,EAClB,QAAqB,EACrB,sBAAgC;QAJhC,UAAK,GAAL,KAAK,CAAQ;QACb,YAAO,GAAP,OAAO,CAAW;QAClB,YAAO,GAAP,OAAO,CAAW;QAClB,aAAQ,GAAR,QAAQ,CAAa;QACrB,2BAAsB,GAAtB,sBAAsB,CAAU;IACtC,CAAC;IAEI,SAAS,CAA6B;IAC9C,gBAAgB;IAChB,QAAQ,CAAC,KAAqB;QAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,qBAAqB;QACpB,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QACnC,OAAO,IAAW,CAAC;IACpB,CAAC;IAID,MAAM,CACL,MAA6E;QAE7E,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACpE,CAAC;QACD,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACzC,MAAM,MAAM,GAAgC,EAAE,CAAC;YAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC9C,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzC,MAAM,QAAQ,GAAG,KAAK,CAAC,MAA4B,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,CAAC;YACD,OAAO,MAAM,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,YAAY,CACtB,IAAI,CAAC,KAAK,EACV,YAAY,EACZ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,QAAQ,EACb,KAAK,EACL,IAAI,CAAC,sBAAsB,CAC3B,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAQ,CAAC;IACnC,CAAC;IAMD,MAAM,CACL,WAGmE;QAEnE,MAAM,MAAM,GAAG,OAAO,WAAW,KAAK,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;QAEjG,IACC,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC;eACb,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,EAC7D,CAAC;YACF,MAAM,IAAI,KAAK,CACd,oHAAoH,CACpH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC9F,CAAC;;AAmGF,MAAM,OAAO,YASX,SAAQ,YAAkG;IAiBlG;IACA;IATT,MAAM,CAAmB,CAAC,UAAU,CAAC,GAAW,UAAU,CAAC;IAEnD,MAAM,CAAyB;IAC7B,WAAW,CAAmB;IAExC,YACC,KAAa,EACb,MAAgC,EACxB,OAAkB,EAClB,OAAkB,EAC1B,QAAqB,EACrB,MAAgB,EAChB,sBAAgC;QAEhC,KAAK,EAAE,CAAC;QANA,YAAO,GAAP,OAAO,CAAW;QAClB,YAAO,GAAP,OAAO,CAAW;QAM1B,IAAI,CAAC,MAAM,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAa,EAAE,QAAQ,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC;IAC1F,CAAC;IA0BD,SAAS,CACR,SAA6B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC;QAEpE,IAAI,CAAC,MAAM,CAAC,eAAe,GAAG,MAAM,CAAC;QACrC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,mBAAmB,CAAW,MAAM,CAAC,CAAC;QAC9D,OAAO,IAAW,CAAC;IACpB,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;OAqBG;IACH,mBAAmB,CAClB,SAAgE,EAAE;QAElE,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAA,YAAY,CAAC;QAC1C,CAAC;aAAM,CAAC;YACP,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;gBAC1C,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACvG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/E,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAA,UAAU,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACxE,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAA,IAAI,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,QAAQ,aAAa,CAAC;QAChF,CAAC;QACD,OAAO,IAAW,CAAC;IACpB,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;IACH,kBAAkB,CACjB,MAA8C;QAE9C,IAAI,MAAM,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7D,MAAM,IAAI,KAAK,CACd,6IAA6I,CAC7I,CAAC;QACH,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAA,UAAU,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACxE,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,UAAU,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1F,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAA,UAAU,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3G,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;YAC1C,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;YACvG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAA,IAC3B,GAAG,CAAC,GAAG,CAAC,YAAY,CACrB,IAAI,cAAc,kBAAkB,MAAM,GAAG,QAAQ,GAAG,WAAW,EAAE,CAAC;QACtE,OAAO,IAAW,CAAC;IACpB,CAAC;IAED,gBAAgB;IAChB,MAAM;QACL,OAAO,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IAED,KAAK;QACJ,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,OAAO,IAAI,CAAC;IACb,CAAC;IAED,gBAAgB;IAChB,QAAQ,CAAC,IAAa;QACrB,OAAO,MAAM,CAAC,eAAe,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAI9B,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE;gBACvF,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;aAC3C,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,IAAY;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,SAAS,CAA6B;IAC9C,gBAAgB;IAChB,QAAQ,CAAC,KAAqB;QAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,OAAO,IAAI,CAAC;IACb,CAAC;IAEQ,OAAO,GAA2C,CAAC,iBAAiB,EAAE,EAAE;QAChF,OAAO,MAAM,CAAC,eAAe,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACvD,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC;IAEF,gBAAgB;IAChB,iBAAiB;QAChB,OAAO,CACN,IAAI,CAAC,MAAM,CAAC,eAAe;YAC1B,CAAC,CAAC,IAAI,KAAK,CACV,IAAI,CAAC,MAAM,CAAC,eAAe,EAC3B,IAAI,qBAAqB,CAAC;gBACzB,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACtC,kBAAkB,EAAE,OAAO;gBAC3B,WAAW,EAAE,OAAO;aACpB,CAAC,CACF;YACD,CAAC,CAAC,SAAS,CACmB,CAAC;IAClC,CAAC;IAED,QAAQ;QACP,OAAO,IAAW,CAAC;IACpB,CAAC"}

package/dist/data/query/pg-core/query-builders/query-builder.d.ts

@@ -0,0 +1,38 @@
+import { entityKind } from '../../entity.js';
+import type { PgDialectConfig } from '../dialect.js';
+import { PgDialect } from '../dialect.js';
+import type { SQLWrapper } from '../../sql/sql.js';
+import { WithSubquery } from '../../subquery.js';
+import type { PgColumn } from '../columns/index.js';
+import type { WithBuilder } from '../subquery.js';
+import { PgSelectBuilder } from './select.js';
+import type { SelectedFields } from './select.types.js';
+export declare class QueryBuilder {
+    static readonly [entityKind]: string;
+    private dialect;
+    private dialectConfig;
+    constructor(dialect?: PgDialect | PgDialectConfig);
+    $with: WithBuilder;
+    with(...queries: WithSubquery[]): {
+        select: {
+            (): PgSelectBuilder<undefined, "qb">;
+            <TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection, "qb">;
+        };
+        selectDistinct: {
+            (): PgSelectBuilder<undefined, "qb">;
+            <TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection, "qb">;
+        };
+        selectDistinctOn: {
+            (on: (PgColumn | SQLWrapper)[]): PgSelectBuilder<undefined, "qb">;
+            <TSelection extends SelectedFields>(on: (PgColumn | SQLWrapper)[], fields: TSelection): PgSelectBuilder<TSelection, "qb">;
+        };
+    };
+    select(): PgSelectBuilder<undefined, 'qb'>;
+    select<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection, 'qb'>;
+    selectDistinct(): PgSelectBuilder<undefined>;
+    selectDistinct<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
+    selectDistinctOn(on: (PgColumn | SQLWrapper)[]): PgSelectBuilder<undefined>;
+    selectDistinctOn<TSelection extends SelectedFields>(on: (PgColumn | SQLWrapper)[], fields: TSelection): PgSelectBuilder<TSelection>;
+    private getDialect;
+}
+//# sourceMappingURL=query-builder.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"query-builder.d.ts","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/query-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAM,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,OAAO,KAAK,EAAyB,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExD,qBAAa,YAAY;IACxB,MAAM,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,EAAE,MAAM,CAAoB;IAExD,OAAO,CAAC,OAAO,CAAwB;IACvC,OAAO,CAAC,aAAa,CAA8B;gBAEvC,OAAO,CAAC,EAAE,SAAS,GAAG,eAAe;IAKjD,KAAK,EAAE,WAAW,CAuBhB;IAEF,IAAI,CAAC,GAAG,OAAO,EAAE,YAAY,EAAE;;gBAGX,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC;aACnC,UAAU,SAAS,cAAc,UAAU,UAAU,GAAG,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC;;;gBAY9E,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC;aACnC,UAAU,SAAS,cAAc,UAAU,UAAU,GAAG,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC;;;iBAYnF,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC;aAChE,UAAU,SAAS,cAAc,MACtD,CAAC,QAAQ,GAAG,UAAU,CAAC,EAAE,UACrB,UAAU,GAChB,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC;;;IAgBrC,MAAM,IAAI,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC;IAC1C,MAAM,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,UAAU,GAAG,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC;IAShG,cAAc,IAAI,eAAe,CAAC,SAAS,CAAC;IAC5C,cAAc,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,UAAU,GAAG,eAAe,CAAC,UAAU,CAAC;IAUlG,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;IAc9B,OAAO,CAAC,UAAU;CAOlB"}

package/dist/data/query/pg-core/query-builders/query-builder.js

@@ -0,0 +1,83 @@
+import { entityKind, is } from '../../entity.js';
+import { PgDialect } from '../dialect.js';
+import { SelectionProxyHandler } from '../../selection-proxy.js';
+import { WithSubquery } from '../../subquery.js';
+import { PgSelectBuilder } from './select.js';
+export class QueryBuilder {
+    static [entityKind] = 'PgQueryBuilder';
+    dialect;
+    dialectConfig;
+    constructor(dialect) {
+        this.dialect = is(dialect, PgDialect) ? dialect : undefined;
+        this.dialectConfig = is(dialect, PgDialect) ? undefined : dialect;
+    }
+    $with = (alias, selection) => {
+        const queryBuilder = this;
+        const as = (qb) => {
+            if (typeof qb === 'function') {
+                qb = qb(queryBuilder);
+            }
+            return new Proxy(new WithSubquery(qb.getSQL(), selection ?? ('getSelectedFields' in qb ? qb.getSelectedFields() ?? {} : {}), alias, true), new SelectionProxyHandler({ alias, sqlAliasedBehavior: 'alias', sqlBehavior: 'error' }));
+        };
+        return { as };
+    };
+    with(...queries) {
+        const self = this;
+        function select(fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: undefined,
+                dialect: self.getDialect(),
+                withList: queries,
+            });
+        }
+        function selectDistinct(fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: undefined,
+                dialect: self.getDialect(),
+                distinct: true,
+            });
+        }
+        function selectDistinctOn(on, fields) {
+            return new PgSelectBuilder({
+                fields: fields ?? undefined,
+                session: undefined,
+                dialect: self.getDialect(),
+                distinct: { on },
+            });
+        }
+        return { select, selectDistinct, selectDistinctOn };
+    }
+    select(fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: undefined,
+            dialect: this.getDialect(),
+        });
+    }
+    selectDistinct(fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: undefined,
+            dialect: this.getDialect(),
+            distinct: true,
+        });
+    }
+    selectDistinctOn(on, fields) {
+        return new PgSelectBuilder({
+            fields: fields ?? undefined,
+            session: undefined,
+            dialect: this.getDialect(),
+            distinct: { on },
+        });
+    }
+    // Lazy load dialect to avoid circular dependency
+    getDialect() {
+        if (!this.dialect) {
+            this.dialect = new PgDialect(this.dialectConfig);
+        }
+        return this.dialect;
+    }
+}
+//# sourceMappingURL=query-builder.js.map

package/dist/data/query/pg-core/query-builders/query-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-builder.js","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/query-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAGjD,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAG9C,MAAM,OAAO,YAAY;IACxB,MAAM,CAAU,CAAC,UAAU,CAAC,GAAW,gBAAgB,CAAC;IAEhD,OAAO,CAAwB;IAC/B,aAAa,CAA8B;IAEnD,YAAY,OAAqC;QAChD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5D,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;IACnE,CAAC;IAED,KAAK,GAAgB,CAAC,KAAa,EAAE,SAA4B,EAAE,EAAE;QACpE,MAAM,YAAY,GAAG,IAAI,CAAC;QAC1B,MAAM,EAAE,GAAG,CACV,EAGgF,EAC/E,EAAE;YACH,IAAI,OAAO,EAAE,KAAK,UAAU,EAAE,CAAC;gBAC9B,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;YACvB,CAAC;YAED,OAAO,IAAI,KAAK,CACf,IAAI,YAAY,CACf,EAAE,CAAC,MAAM,EAAE,EACX,SAAS,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAmB,EAC9F,KAAK,EACL,IAAI,CACJ,EACD,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAChF,CAAC;QACV,CAAC,CAAC;QACF,OAAO,EAAE,EAAE,EAAE,CAAC;IACf,CAAC,CAAC;IAEF,IAAI,CAAC,GAAG,OAAuB;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC;QAIlB,SAAS,MAAM,CACd,MAAmB;YAEnB,OAAO,IAAI,eAAe,CAAC;gBAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;gBAC1B,QAAQ,EAAE,OAAO;aACjB,CAAC,CAAC;QACJ,CAAC;QAID,SAAS,cAAc,CACtB,MAAmB;YAEnB,OAAO,IAAI,eAAe,CAAC;gBAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;gBAC1B,QAAQ,EAAE,IAAI;aACd,CAAC,CAAC;QACJ,CAAC;QAOD,SAAS,gBAAgB,CACxB,EAA6B,EAC7B,MAAmB;YAEnB,OAAO,IAAI,eAAe,CAAC;gBAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;gBAC1B,QAAQ,EAAE,EAAE,EAAE,EAAE;aAChB,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,gBAAgB,EAAE,CAAC;IACrD,CAAC;IAID,MAAM,CAAoC,MAAmB;QAC5D,OAAO,IAAI,eAAe,CAAC;YAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;YAC3B,OAAO,EAAE,SAAS;YAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;SAC1B,CAAC,CAAC;IACJ,CAAC;IAID,cAAc,CAAoC,MAAmB;QACpE,OAAO,IAAI,eAAe,CAAC;YAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;YAC3B,OAAO,EAAE,SAAS;YAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;YAC1B,QAAQ,EAAE,IAAI;SACd,CAAC,CAAC;IACJ,CAAC;IAOD,gBAAgB,CACf,EAA6B,EAC7B,MAAmB;QAEnB,OAAO,IAAI,eAAe,CAAC;YAC1B,MAAM,EAAE,MAAM,IAAI,SAAS;YAC3B,OAAO,EAAE,SAAS;YAClB,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;YAC1B,QAAQ,EAAE,EAAE,EAAE,EAAE;SAChB,CAAC,CAAC;IACJ,CAAC;IAED,iDAAiD;IACzC,UAAU;QACjB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC,OAAO,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACrB,CAAC"}

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

@@ -0,0 +1,56 @@
+import { entityKind } from '../../entity.js';
+import { QueryPromise } from '../../query-promise.js';
+import { type BuildQueryResult, type DBQueryConfig, type TableRelationalConfig, type TablesRelationalConfig } from '../../relations.js';
+import type { RunnableQuery } from '../../runnable-query.js';
+import type { Query, SQL, SQLWrapper } from '../../sql/sql.js';
+import type { KnownKeysOnly, NeonAuthToken } from '../../utils.js';
+import type { PgDialect } from '../dialect.js';
+import type { PgPreparedQuery, PgSession, PreparedQueryConfig } from '../session.js';
+import type { PgTable } from '../table.js';
+export declare class RelationalQueryBuilder<TSchema extends TablesRelationalConfig, TFields extends TableRelationalConfig> {
+    private fullSchema;
+    private schema;
+    private tableNamesMap;
+    private table;
+    private tableConfig;
+    private dialect;
+    private session;
+    static readonly [entityKind]: string;
+    constructor(fullSchema: Record<string, unknown>, schema: TSchema, tableNamesMap: Record<string, string>, table: PgTable, tableConfig: TableRelationalConfig, dialect: PgDialect, session: PgSession);
+    findMany<TConfig extends DBQueryConfig<'many', true, TSchema, TFields>>(config?: KnownKeysOnly<TConfig, DBQueryConfig<'many', true, TSchema, TFields>>): PgRelationalQuery<BuildQueryResult<TSchema, TFields, TConfig>[]>;
+    findFirst<TSelection extends Omit<DBQueryConfig<'many', true, TSchema, TFields>, 'limit'>>(config?: KnownKeysOnly<TSelection, Omit<DBQueryConfig<'many', true, TSchema, TFields>, 'limit'>>): PgRelationalQuery<BuildQueryResult<TSchema, TFields, TSelection> | undefined>;
+}
+export declare class PgRelationalQuery<TResult> extends QueryPromise<TResult> implements RunnableQuery<TResult, 'pg'>, SQLWrapper {
+    private fullSchema;
+    private schema;
+    private tableNamesMap;
+    private table;
+    private tableConfig;
+    private dialect;
+    private session;
+    private config;
+    private mode;
+    static readonly [entityKind]: string;
+    readonly _: {
+        readonly dialect: 'pg';
+        readonly result: TResult;
+    };
+    constructor(fullSchema: Record<string, unknown>, schema: TablesRelationalConfig, tableNamesMap: Record<string, string>, table: PgTable, tableConfig: TableRelationalConfig, dialect: PgDialect, session: PgSession, config: DBQueryConfig<'many', true> | true, mode: 'many' | 'first');
+    /** @internal */
+    _prepare(name?: string): PgPreparedQuery<PreparedQueryConfig & {
+        execute: TResult;
+    }>;
+    prepare(name: string): PgPreparedQuery<PreparedQueryConfig & {
+        execute: TResult;
+    }>;
+    private _getQuery;
+    /** @internal */
+    getSQL(): SQL;
+    private _toSQL;
+    toSQL(): Query;
+    private authToken?;
+    /** @internal */
+    setToken(token?: NeonAuthToken): this;
+    execute(): Promise<TResult>;
+}
+//# sourceMappingURL=query.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/query.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EACN,KAAK,gBAAgB,EAErB,KAAK,aAAa,EAElB,KAAK,qBAAqB,EAC1B,KAAK,sBAAsB,EAC3B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,KAAK,EAAE,KAAK,EAAoB,GAAG,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEjF,OAAO,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AACnE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,eAAe,EAAE,SAAS,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACrF,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAE3C,qBAAa,sBAAsB,CAAC,OAAO,SAAS,sBAAsB,EAAE,OAAO,SAAS,qBAAqB;IAI/G,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,aAAa;IACrB,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,OAAO;IAThB,MAAM,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,EAAE,MAAM,CAA8B;gBAGzD,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,MAAM,EAAE,OAAO,EACf,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACrC,KAAK,EAAE,OAAO,EACd,WAAW,EAAE,qBAAqB,EAClC,OAAO,EAAE,SAAS,EAClB,OAAO,EAAE,SAAS;IAG3B,QAAQ,CAAC,OAAO,SAAS,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,EACrE,MAAM,CAAC,EAAE,aAAa,CAAC,OAAO,EAAE,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,GAC5E,iBAAiB,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC;IAcnE,SAAS,CAAC,UAAU,SAAS,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,EACxF,MAAM,CAAC,EAAE,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC,GAC9F,iBAAiB,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC;CAahF;AAED,qBAAa,iBAAiB,CAAC,OAAO,CAAE,SAAQ,YAAY,CAAC,OAAO,CACnE,YAAW,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,UAAU;IAUlD,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,aAAa;IACrB,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,IAAI;IAhBb,gBAAyB,CAAC,UAAU,CAAC,EAAE,MAAM,CAAuB;IAEpE,SAAiB,CAAC,EAAE;QACnB,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC;QACvB,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;KACzB,CAAC;gBAGO,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,MAAM,EAAE,sBAAsB,EAC9B,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACrC,KAAK,EAAE,OAAO,EACd,WAAW,EAAE,qBAAqB,EAClC,OAAO,EAAE,SAAS,EAClB,OAAO,EAAE,SAAS,EAClB,MAAM,EAAE,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,IAAI,EAC1C,IAAI,EAAE,MAAM,GAAG,OAAO;IAK/B,gBAAgB;IAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,eAAe,CAAC,mBAAmB,GAAG;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IAsBpF,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,CAAC,mBAAmB,GAAG;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IAIlF,OAAO,CAAC,SAAS;IAYjB,gBAAgB;IAChB,MAAM,IAAI,GAAG;IAIb,OAAO,CAAC,MAAM;IAQd,KAAK,IAAI,KAAK;IAId,OAAO,CAAC,SAAS,CAAC,CAA4B;IAC9C,gBAAgB;IAChB,QAAQ,CAAC,KAAK,CAAC,EAAE,aAAa;IAKrB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;CAKpC"}

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

@@ -0,0 +1,104 @@
+import { entityKind } from '../../entity.js';
+import { QueryPromise } from '../../query-promise.js';
+import { mapRelationalRow, } from '../../relations.js';
+import { tracer } from '../../tracing.js';
+export class RelationalQueryBuilder {
+    fullSchema;
+    schema;
+    tableNamesMap;
+    table;
+    tableConfig;
+    dialect;
+    session;
+    static [entityKind] = 'PgRelationalQueryBuilder';
+    constructor(fullSchema, schema, tableNamesMap, table, tableConfig, dialect, session) {
+        this.fullSchema = fullSchema;
+        this.schema = schema;
+        this.tableNamesMap = tableNamesMap;
+        this.table = table;
+        this.tableConfig = tableConfig;
+        this.dialect = dialect;
+        this.session = session;
+    }
+    findMany(config) {
+        return new PgRelationalQuery(this.fullSchema, this.schema, this.tableNamesMap, this.table, this.tableConfig, this.dialect, this.session, config ? config : {}, 'many');
+    }
+    findFirst(config) {
+        return new PgRelationalQuery(this.fullSchema, this.schema, this.tableNamesMap, this.table, this.tableConfig, this.dialect, this.session, config ? { ...config, limit: 1 } : { limit: 1 }, 'first');
+    }
+}
+export class PgRelationalQuery extends QueryPromise {
+    fullSchema;
+    schema;
+    tableNamesMap;
+    table;
+    tableConfig;
+    dialect;
+    session;
+    config;
+    mode;
+    static [entityKind] = 'PgRelationalQuery';
+    constructor(fullSchema, schema, tableNamesMap, table, tableConfig, dialect, session, config, mode) {
+        super();
+        this.fullSchema = fullSchema;
+        this.schema = schema;
+        this.tableNamesMap = tableNamesMap;
+        this.table = table;
+        this.tableConfig = tableConfig;
+        this.dialect = dialect;
+        this.session = session;
+        this.config = config;
+        this.mode = mode;
+    }
+    /** @internal */
+    _prepare(name) {
+        return tracer.startActiveSpan('drizzle.prepareQuery', () => {
+            const { query, builtQuery } = this._toSQL();
+            return this.session.prepareQuery(builtQuery, undefined, name, true, (rawRows, mapColumnValue) => {
+                const rows = rawRows.map((row) => mapRelationalRow(this.schema, this.tableConfig, row, query.selection, mapColumnValue));
+                if (this.mode === 'first') {
+                    return rows[0];
+                }
+                return rows;
+            });
+        });
+    }
+    prepare(name) {
+        return this._prepare(name);
+    }
+    _getQuery() {
+        return this.dialect.buildRelationalQueryWithoutPK({
+            fullSchema: this.fullSchema,
+            schema: this.schema,
+            tableNamesMap: this.tableNamesMap,
+            table: this.table,
+            tableConfig: this.tableConfig,
+            queryConfig: this.config,
+            tableAlias: this.tableConfig.tsName,
+        });
+    }
+    /** @internal */
+    getSQL() {
+        return this._getQuery().sql;
+    }
+    _toSQL() {
+        const query = this._getQuery();
+        const builtQuery = this.dialect.sqlToQuery(query.sql);
+        return { query, builtQuery };
+    }
+    toSQL() {
+        return this._toSQL().builtQuery;
+    }
+    authToken;
+    /** @internal */
+    setToken(token) {
+        this.authToken = token;
+        return this;
+    }
+    execute() {
+        return tracer.startActiveSpan('drizzle.operation', () => {
+            return this._prepare().execute(undefined, this.authToken);
+        });
+    }
+}
+//# sourceMappingURL=query.js.map

package/dist/data/query/pg-core/query-builders/query.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.js","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/query.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAIN,gBAAgB,GAGhB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAM1C,MAAM,OAAO,sBAAsB;IAIzB;IACA;IACA;IACA;IACA;IACA;IACA;IATT,MAAM,CAAU,CAAC,UAAU,CAAC,GAAW,0BAA0B,CAAC;IAElE,YACS,UAAmC,EACnC,MAAe,EACf,aAAqC,EACrC,KAAc,EACd,WAAkC,EAClC,OAAkB,EAClB,OAAkB;QANlB,eAAU,GAAV,UAAU,CAAyB;QACnC,WAAM,GAAN,MAAM,CAAS;QACf,kBAAa,GAAb,aAAa,CAAwB;QACrC,UAAK,GAAL,KAAK,CAAS;QACd,gBAAW,GAAX,WAAW,CAAuB;QAClC,YAAO,GAAP,OAAO,CAAW;QAClB,YAAO,GAAP,OAAO,CAAW;IACxB,CAAC;IAEJ,QAAQ,CACP,MAA8E;QAE9E,OAAO,IAAI,iBAAiB,CAC3B,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,MAAM,CAAC,CAAC,CAAE,MAAsC,CAAC,CAAC,CAAC,EAAE,EACrD,MAAM,CACN,CAAC;IACH,CAAC;IAED,SAAS,CACR,MAAgG;QAEhG,OAAO,IAAI,iBAAiB,CAC3B,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,MAAM,CAAC,CAAC,CAAC,EAAE,GAAI,MAAkD,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAC5F,OAAO,CACP,CAAC;IACH,CAAC;;AAGF,MAAM,OAAO,iBAA2B,SAAQ,YAAqB;IAW3D;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAhBT,MAAM,CAAmB,CAAC,UAAU,CAAC,GAAW,mBAAmB,CAAC;IAOpE,YACS,UAAmC,EACnC,MAA8B,EAC9B,aAAqC,EACrC,KAAc,EACd,WAAkC,EAClC,OAAkB,EAClB,OAAkB,EAClB,MAA0C,EAC1C,IAAsB;QAE9B,KAAK,EAAE,CAAC;QAVA,eAAU,GAAV,UAAU,CAAyB;QACnC,WAAM,GAAN,MAAM,CAAwB;QAC9B,kBAAa,GAAb,aAAa,CAAwB;QACrC,UAAK,GAAL,KAAK,CAAS;QACd,gBAAW,GAAX,WAAW,CAAuB;QAClC,YAAO,GAAP,OAAO,CAAW;QAClB,YAAO,GAAP,OAAO,CAAW;QAClB,WAAM,GAAN,MAAM,CAAoC;QAC1C,SAAI,GAAJ,IAAI,CAAkB;IAG/B,CAAC;IAED,gBAAgB;IAChB,QAAQ,CAAC,IAAa;QACrB,OAAO,MAAM,CAAC,eAAe,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAC1D,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YAE5C,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAC/B,UAAU,EACV,SAAS,EACT,IAAI,EACJ,IAAI,EACJ,CAAC,OAAO,EAAE,cAAc,EAAE,EAAE;gBAC3B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAChC,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,CAAC,CACrF,CAAC;gBACF,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,OAAO,IAAI,CAAC,CAAC,CAAY,CAAC;gBAC3B,CAAC;gBACD,OAAO,IAAe,CAAC;YACxB,CAAC,CACD,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,IAAY;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,SAAS;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,6BAA6B,CAAC;YACjD,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,MAAM;YACxB,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;SACnC,CAAC,CAAC;IACJ,CAAC;IAED,gBAAgB;IAChB,MAAM;QACL,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC,GAAU,CAAC;IACpC,CAAC;IAEO,MAAM;QACb,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,GAAU,CAAC,CAAC;QAE7D,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK;QACJ,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,UAAU,CAAC;IACjC,CAAC;IAEO,SAAS,CAA6B;IAC9C,gBAAgB;IAChB,QAAQ,CAAC,KAAqB;QAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,OAAO,IAAI,CAAC;IACb,CAAC;IAEQ,OAAO;QACf,OAAO,MAAM,CAAC,eAAe,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACvD,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACJ,CAAC"}

package/dist/data/query/pg-core/query-builders/raw.d.ts

@@ -0,0 +1,27 @@
+import { entityKind } from '../../entity.js';
+import { QueryPromise } from '../../query-promise.js';
+import type { RunnableQuery } from '../../runnable-query.js';
+import type { PreparedQuery } from '../../session.js';
+import type { Query, SQL, SQLWrapper } from '../../sql/sql.js';
+export interface PgRaw<TResult> extends QueryPromise<TResult>, RunnableQuery<TResult, 'pg'>, SQLWrapper {
+}
+export declare class PgRaw<TResult> extends QueryPromise<TResult> implements RunnableQuery<TResult, 'pg'>, SQLWrapper, PreparedQuery {
+    execute: () => Promise<TResult>;
+    private sql;
+    private query;
+    private mapBatchResult;
+    static readonly [entityKind]: string;
+    readonly _: {
+        readonly dialect: 'pg';
+        readonly result: TResult;
+    };
+    constructor(execute: () => Promise<TResult>, sql: SQL, query: Query, mapBatchResult: (result: unknown) => unknown);
+    /** @internal */
+    getSQL(): SQL<unknown>;
+    getQuery(): Query;
+    mapResult(result: unknown, isFromBatch?: boolean): unknown;
+    _prepare(): PreparedQuery;
+    /** @internal */
+    isResponseInArrayMode(): boolean;
+}
+//# sourceMappingURL=raw.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"raw.d.ts","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/raw.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE/D,MAAM,WAAW,KAAK,CAAC,OAAO,CAAE,SAAQ,YAAY,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,UAAU;CAAG;AAE1G,qBAAa,KAAK,CAAC,OAAO,CAAE,SAAQ,YAAY,CAAC,OAAO,CACvD,YAAW,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,UAAU,EAAE,aAAa;IAU1D,OAAO,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC;IACtC,OAAO,CAAC,GAAG;IACX,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,cAAc;IAXvB,gBAAyB,CAAC,UAAU,CAAC,EAAE,MAAM,CAAW;IAExD,SAAiB,CAAC,EAAE;QACnB,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC;QACvB,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;KACzB,CAAC;gBAGM,OAAO,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,EAC9B,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,KAAK,EACZ,cAAc,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO;IAKrD,gBAAgB;IAChB,MAAM;IAIN,QAAQ;IAIR,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,EAAE,OAAO;IAIhD,QAAQ,IAAI,aAAa;IAIzB,gBAAgB;IAChB,qBAAqB;CAGrB"}

package/dist/data/query/pg-core/query-builders/raw.js

@@ -0,0 +1,34 @@
+import { entityKind } from '../../entity.js';
+import { QueryPromise } from '../../query-promise.js';
+export class PgRaw extends QueryPromise {
+    execute;
+    sql;
+    query;
+    mapBatchResult;
+    static [entityKind] = 'PgRaw';
+    constructor(execute, sql, query, mapBatchResult) {
+        super();
+        this.execute = execute;
+        this.sql = sql;
+        this.query = query;
+        this.mapBatchResult = mapBatchResult;
+    }
+    /** @internal */
+    getSQL() {
+        return this.sql;
+    }
+    getQuery() {
+        return this.query;
+    }
+    mapResult(result, isFromBatch) {
+        return isFromBatch ? this.mapBatchResult(result) : result;
+    }
+    _prepare() {
+        return this;
+    }
+    /** @internal */
+    isResponseInArrayMode() {
+        return false;
+    }
+}
+//# sourceMappingURL=raw.js.map

package/dist/data/query/pg-core/query-builders/raw.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"raw.js","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/raw.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAOtD,MAAM,OAAO,KAAe,SAAQ,YAAqB;IAWhD;IACC;IACA;IACA;IAXT,MAAM,CAAmB,CAAC,UAAU,CAAC,GAAW,OAAO,CAAC;IAOxD,YACQ,OAA+B,EAC9B,GAAQ,EACR,KAAY,EACZ,cAA4C;QAEpD,KAAK,EAAE,CAAC;QALD,YAAO,GAAP,OAAO,CAAwB;QAC9B,QAAG,GAAH,GAAG,CAAK;QACR,UAAK,GAAL,KAAK,CAAO;QACZ,mBAAc,GAAd,cAAc,CAA8B;IAGrD,CAAC;IAED,gBAAgB;IAChB,MAAM;QACL,OAAO,IAAI,CAAC,GAAG,CAAC;IACjB,CAAC;IAED,QAAQ;QACP,OAAO,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAED,SAAS,CAAC,MAAe,EAAE,WAAqB;QAC/C,OAAO,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAC3D,CAAC;IAED,QAAQ;QACP,OAAO,IAAI,CAAC;IACb,CAAC;IAED,gBAAgB;IAChB,qBAAqB;QACpB,OAAO,KAAK,CAAC;IACd,CAAC"}

package/dist/data/query/pg-core/query-builders/refresh-materialized-view.d.ts

@@ -0,0 +1,38 @@
+import { entityKind } from '../../entity.js';
+import type { PgDialect } from '../dialect.js';
+import type { PgPreparedQuery, PgQueryResultHKT, PgQueryResultKind, PgSession, PreparedQueryConfig } from '../session.js';
+import type { PgMaterializedView } from '../view.js';
+import { QueryPromise } from '../../query-promise.js';
+import type { RunnableQuery } from '../../runnable-query.js';
+import type { Query, SQL, SQLWrapper } from '../../sql/sql.js';
+import type { NeonAuthToken } from '../../utils.js';
+export interface PgRefreshMaterializedView<TQueryResult extends PgQueryResultHKT> extends QueryPromise<PgQueryResultKind<TQueryResult, never>>, RunnableQuery<PgQueryResultKind<TQueryResult, never>, 'pg'>, SQLWrapper {
+    readonly _: {
+        readonly dialect: 'pg';
+        readonly result: PgQueryResultKind<TQueryResult, never>;
+    };
+}
+export declare class PgRefreshMaterializedView<TQueryResult extends PgQueryResultHKT> extends QueryPromise<PgQueryResultKind<TQueryResult, never>> implements RunnableQuery<PgQueryResultKind<TQueryResult, never>, 'pg'>, SQLWrapper {
+    private session;
+    private dialect;
+    static readonly [entityKind]: string;
+    private config;
+    constructor(view: PgMaterializedView, session: PgSession, dialect: PgDialect);
+    concurrently(): this;
+    withNoData(): this;
+    /** @internal */
+    getSQL(): SQL;
+    toSQL(): Query;
+    /** @internal */
+    _prepare(name?: string): PgPreparedQuery<PreparedQueryConfig & {
+        execute: PgQueryResultKind<TQueryResult, never>;
+    }>;
+    prepare(name: string): PgPreparedQuery<PreparedQueryConfig & {
+        execute: PgQueryResultKind<TQueryResult, never>;
+    }>;
+    private authToken?;
+    /** @internal */
+    setToken(token?: NeonAuthToken): this;
+    execute: ReturnType<this['prepare']>['execute'];
+}
+//# sourceMappingURL=refresh-materialized-view.d.ts.map

package/dist/data/query/pg-core/query-builders/refresh-materialized-view.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"refresh-materialized-view.d.ts","sourceRoot":"","sources":["../../../../../src/data/query/pg-core/query-builders/refresh-materialized-view.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EACX,eAAe,EACf,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,mBAAmB,EACnB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE/D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAGpD,MAAM,WAAW,yBAAyB,CAAC,YAAY,SAAS,gBAAgB,CAC/E,SACC,YAAY,CAAC,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,EACpD,aAAa,CAAC,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,EAC3D,UAAU;IAEX,QAAQ,CAAC,CAAC,EAAE;QACX,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC;QACvB,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;KACxD,CAAC;CACF;AAED,qBAAa,yBAAyB,CAAC,YAAY,SAAS,gBAAgB,CAC3E,SAAQ,YAAY,CAAC,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,CAC3D,YAAW,aAAa,CAAC,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,EAAE,UAAU;IAYjF,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,OAAO;IAXhB,gBAAyB,CAAC,UAAU,CAAC,EAAE,MAAM,CAA+B;IAE5E,OAAO,CAAC,MAAM,CAIZ;gBAGD,IAAI,EAAE,kBAAkB,EAChB,OAAO,EAAE,SAAS,EAClB,OAAO,EAAE,SAAS;IAM3B,YAAY,IAAI,IAAI;IAQpB,UAAU,IAAI,IAAI;IAQlB,gBAAgB;IAChB,MAAM,IAAI,GAAG;IAIb,KAAK,IAAI,KAAK;IAKd,gBAAgB;IAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,eAAe,CACvC,mBAAmB,GAAG;QACrB,OAAO,EAAE,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;KAChD,CACD;IAMD,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,CACrC,mBAAmB,GAAG;QACrB,OAAO,EAAE,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;KAChD,CACD;IAID,OAAO,CAAC,SAAS,CAAC,CAA4B;IAC9C,gBAAgB;IAChB,QAAQ,CAAC,KAAK,CAAC,EAAE,aAAa;IAK9B,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAI7C;CACF"}