@erfanium/drizzle-orm 0.1.0 → 1.0.0

package/mssql-core/db.d.ts

@@ -0,0 +1,235 @@
+import { MsSqlDeleteBase } from "./query-builders/delete.js";
+import { MsSqlInsertBuilder } from "./query-builders/insert.js";
+import { MsSqlUpdateBuilder } from "./query-builders/update.js";
+import { MsSqlTable } from "./table.js";
+import { MsSqlDialect } from "./dialect.js";
+import { MsSqlSelectBuilder } from "./query-builders/select.js";
+import { QueryBuilder } from "./query-builders/query-builder.js";
+import { RelationalQueryBuilder } from "./query-builders/query.js";
+import { WithSubqueryWithSelection } from "./subquery.js";
+import { MsSqlSession, MsSqlTransaction, MsSqlTransactionConfig, PreparedQueryHKTBase, QueryResultHKT, QueryResultKind } from "./session.js";
+import { SelectedFields } from "./query-builders/select.types.js";
+import { entityKind } from "../entity.js";
+import { DrizzleTypeError } from "../utils.js";
+import { ColumnsSelection, SQLWrapper } from "../sql/sql.js";
+import { WithSubquery } from "../subquery.js";
+import { TypedQueryBuilder } from "../query-builders/query-builder.js";
+import * as V1 from "../_relations.js";
+
+//#region src/mssql-core/db.d.ts
+declare class MsSqlDatabase<TQueryResult extends QueryResultHKT, TPreparedQueryHKT extends PreparedQueryHKTBase, TFullSchema extends Record<string, unknown> = {}, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> {
+  static readonly [entityKind]: string;
+  readonly _: {
+    readonly schema: TSchema | undefined;
+    readonly tableNamesMap: Record<string, string>;
+  };
+  _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<TPreparedQueryHKT, TSchema, TSchema[K]> };
+  constructor(/** @internal */
+
+  dialect: MsSqlDialect, /** @internal */
+
+  session: MsSqlSession<any, any, any, any>, schema: V1.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<TAlias extends string>(alias: TAlias): {
+    as<TSelection extends ColumnsSelection>(qb: TypedQueryBuilder<TSelection> | ((qb: QueryBuilder) => TypedQueryBuilder<TSelection>)): WithSubqueryWithSelection<TSelection, TAlias>;
+  };
+  /**
+   * 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: {
+      (): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;
+      <TSelection extends SelectedFields>(fields: TSelection): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;
+    };
+    selectDistinct: {
+      (): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;
+      <TSelection extends SelectedFields>(fields: TSelection): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;
+    };
+  };
+  /**
+   * 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(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;
+  select<TSelection extends SelectedFields>(fields: TSelection): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;
+  /**
+   * 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(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;
+  selectDistinct<TSelection extends SelectedFields>(fields: TSelection): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;
+  /**
+   * 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<TTable extends MsSqlTable>(table: TTable): MsSqlUpdateBuilder<TTable, TQueryResult, TPreparedQueryHKT>;
+  /**
+   * 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<TTable extends MsSqlTable>(table: TTable): MsSqlInsertBuilder<TTable, TQueryResult, TPreparedQueryHKT>;
+  /**
+   * 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<TTable extends MsSqlTable>(table: TTable): MsSqlDeleteBase<TTable, TQueryResult, TPreparedQueryHKT>;
+  execute<T extends {
+    [column: string]: any;
+  } | {
+    [column: string]: any;
+  }[]>(query: SQLWrapper | string): Promise<QueryResultKind<TQueryResult, T>>;
+  transaction<T>(transaction: (tx: MsSqlTransaction<TQueryResult, TPreparedQueryHKT, TFullSchema, TSchema>, config?: MsSqlTransactionConfig) => Promise<T>, config?: MsSqlTransactionConfig): Promise<T>;
+}
+type MySQLWithReplicas<Q> = Q & {
+  $primary: Q;
+};
+declare const withReplicas: <HKT extends QueryResultHKT, TPreparedQueryHKT extends PreparedQueryHKTBase, TFullSchema extends Record<string, unknown>, TSchema extends V1.TablesRelationalConfig, Q extends MsSqlDatabase<HKT, TPreparedQueryHKT, TFullSchema, TSchema extends Record<string, unknown> ? V1.ExtractTablesWithRelations<TFullSchema> : TSchema>>(primary: Q, replicas: [Q, ...Q[]], getReplica?: (replicas: Q[]) => Q) => MySQLWithReplicas<Q>;
+//#endregion
+export { MsSqlDatabase, MySQLWithReplicas, withReplicas };
+//# sourceMappingURL=db.d.ts.map

package/mssql-core/db.js

@@ -0,0 +1,234 @@
+import { MsSqlSelectBuilder } from "./query-builders/select.js";
+import { QueryBuilder } from "./query-builders/query-builder.js";
+import { MsSqlDeleteBase } from "./query-builders/delete.js";
+import { MsSqlInsertBuilder } from "./query-builders/insert.js";
+import { MsSqlUpdateBuilder } from "./query-builders/update.js";
+import { RelationalQueryBuilder } from "./query-builders/query.js";
+import { entityKind } from "../entity.js";
+import { sql } from "../sql/sql.js";
+import { WithSubquery } from "../subquery.js";
+import { SelectionProxyHandler } from "../selection-proxy.js";
+
+//#region src/mssql-core/db.ts
+var MsSqlDatabase = class {
+	static [entityKind] = "MsSqlDatabase";
+	_query;
+	constructor(dialect, session, schema) {
+		this.dialect = dialect;
+		this.session = session;
+		this._ = schema ? {
+			schema: schema.schema,
+			tableNamesMap: schema.tableNamesMap
+		} : {
+			schema: void 0,
+			tableNamesMap: {}
+		};
+		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);
+	}
+	/**
+	* 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) {
+		const self = this;
+		return { as(qb) {
+			if (typeof qb === "function") qb = qb(new QueryBuilder(self.dialect));
+			return new Proxy(new WithSubquery(qb.getSQL(), qb.getSelectedFields(), alias, true), new SelectionProxyHandler({
+				alias,
+				sqlAliasedBehavior: "alias",
+				sqlBehavior: "error"
+			}));
+		} };
+	}
+	/**
+	* 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 MsSqlSelectBuilder({
+				fields: fields ?? void 0,
+				session: self.session,
+				dialect: self.dialect,
+				withList: queries
+			});
+		}
+		function selectDistinct(fields) {
+			return new MsSqlSelectBuilder({
+				fields: fields ?? void 0,
+				session: self.session,
+				dialect: self.dialect,
+				withList: queries,
+				distinct: true
+			});
+		}
+		return {
+			select,
+			selectDistinct
+		};
+	}
+	select(fields) {
+		return new MsSqlSelectBuilder({
+			fields: fields ?? void 0,
+			session: this.session,
+			dialect: this.dialect
+		});
+	}
+	selectDistinct(fields) {
+		return new MsSqlSelectBuilder({
+			fields: fields ?? void 0,
+			session: this.session,
+			dialect: this.dialect,
+			distinct: true
+		});
+	}
+	/**
+	* 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(table) {
+		return new MsSqlUpdateBuilder(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(table) {
+		return new MsSqlInsertBuilder(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(table) {
+		return new MsSqlDeleteBase(table, this.session, this.dialect);
+	}
+	execute(query) {
+		return this.session.execute((typeof query === "string" ? sql.raw(query) : query).getSQL());
+	}
+	transaction(transaction, config) {
+		return this.session.transaction(transaction, config);
+	}
+};
+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 $with = (...args) => getReplica(replicas).with(...args);
+	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);
+	return {
+		...primary,
+		update,
+		insert,
+		delete: $delete,
+		execute,
+		transaction,
+		$primary: primary,
+		$replicas: replicas,
+		select,
+		selectDistinct,
+		with: $with,
+		get _query() {
+			return getReplica(replicas)._query;
+		}
+	};
+};
+
+//#endregion
+export { MsSqlDatabase, withReplicas };
+//# sourceMappingURL=db.js.map

package/mssql-core/db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.js","names":[],"sources":["../../src/mssql-core/db.ts"],"sourcesContent":["import type * as V1 from '~/_relations.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { TypedQueryBuilder } from '~/query-builders/query-builder.ts';\nimport { SelectionProxyHandler } from '~/selection-proxy.ts';\nimport { type ColumnsSelection, sql, type SQLWrapper } from '~/sql/sql.ts';\nimport { WithSubquery } from '~/subquery.ts';\nimport type { DrizzleTypeError } from '~/utils.ts';\nimport type { MsSqlDialect } from './dialect.ts';\nimport {\n\tMsSqlDeleteBase,\n\tMsSqlInsertBuilder,\n\tMsSqlSelectBuilder,\n\tMsSqlUpdateBuilder,\n\tQueryBuilder,\n} from './query-builders/index.ts';\nimport { RelationalQueryBuilder } from './query-builders/query.ts';\nimport type { SelectedFields } from './query-builders/select.types.ts';\nimport type {\n\tMsSqlSession,\n\tMsSqlTransaction,\n\tMsSqlTransactionConfig,\n\tPreparedQueryHKTBase,\n\tQueryResultHKT,\n\tQueryResultKind,\n} from './session.ts';\nimport type { WithSubqueryWithSelection } from './subquery.ts';\nimport type { MsSqlTable } from './table.ts';\n\nexport class MsSqlDatabase<\n\tTQueryResult extends QueryResultHKT,\n\tTPreparedQueryHKT extends PreparedQueryHKTBase,\n\tTFullSchema extends Record<string, unknown> = {},\n\tTSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> {\n\tstatic readonly [entityKind]: string = 'MsSqlDatabase';\n\n\tdeclare readonly _: {\n\t\treadonly schema: TSchema | undefined;\n\t\treadonly tableNamesMap: Record<string, string>;\n\t};\n\n\t_query: TFullSchema extends Record<string, never>\n\t\t? DrizzleTypeError<'Seems like the schema generic is missing - did you forget to add it to your DB type?'>\n\t\t: {\n\t\t\t[K in keyof TSchema]: RelationalQueryBuilder<TPreparedQueryHKT, TSchema, TSchema[K]>;\n\t\t};\n\n\tconstructor(\n\t\t/** @internal */\n\t\treadonly dialect: MsSqlDialect,\n\t\t/** @internal */\n\t\treadonly session: MsSqlSession<any, any, any, any>,\n\t\tschema: V1.RelationalSchemaConfig<TSchema> | undefined,\n\t) {\n\t\tthis._ = schema\n\t\t\t? { schema: schema.schema, tableNamesMap: schema.tableNamesMap }\n\t\t\t: { schema: undefined, tableNamesMap: {} };\n\t\tthis._query = {} as typeof this['_query'];\n\t\tif (this._.schema) {\n\t\t\tfor (const [tableName, columns] of Object.entries(this._.schema)) {\n\t\t\t\t(this._query as MsSqlDatabase<TQueryResult, TPreparedQueryHKT, Record<string, any>>['_query'])[tableName] =\n\t\t\t\t\tnew RelationalQueryBuilder(\n\t\t\t\t\t\tschema!.fullSchema,\n\t\t\t\t\t\tthis._.schema,\n\t\t\t\t\t\tthis._.tableNamesMap,\n\t\t\t\t\t\tschema!.fullSchema[tableName] as MsSqlTable,\n\t\t\t\t\t\tcolumns,\n\t\t\t\t\t\tdialect,\n\t\t\t\t\t\tsession,\n\t\t\t\t\t);\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Creates a subquery that defines a temporary named result set as a CTE.\n\t *\n\t * It is useful for breaking down complex queries into simpler parts and for reusing the result set in subsequent parts of the query.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}\n\t *\n\t * @param alias The alias for the subquery.\n\t *\n\t * Failure to provide an alias will result in a DrizzleTypeError, preventing the subquery from being referenced in other queries.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Create a subquery with alias 'sq' and use it in the select query\n\t * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));\n\t *\n\t * const result = await db.with(sq).select().from(sq);\n\t * ```\n\t *\n\t * 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:\n\t *\n\t * ```ts\n\t * // Select an arbitrary SQL value as a field in a CTE and reference it in the main query\n\t * const sq = db.$with('sq').as(db.select({\n\t *   name: sql<string>`upper(${users.name})`.as('name'),\n\t * })\n\t * .from(users));\n\t *\n\t * const result = await db.with(sq).select({ name: sq.name }).from(sq);\n\t * ```\n\t */\n\t$with<TAlias extends string>(alias: TAlias) {\n\t\tconst self = this;\n\n\t\treturn {\n\t\t\tas<TSelection extends ColumnsSelection>(\n\t\t\t\tqb: TypedQueryBuilder<TSelection> | ((qb: QueryBuilder) => TypedQueryBuilder<TSelection>),\n\t\t\t): WithSubqueryWithSelection<TSelection, TAlias> {\n\t\t\t\tif (typeof qb === 'function') {\n\t\t\t\t\tqb = qb(new QueryBuilder(self.dialect));\n\t\t\t\t}\n\n\t\t\t\treturn new Proxy(\n\t\t\t\t\tnew WithSubquery(qb.getSQL(), qb.getSelectedFields() as SelectedFields, alias, true),\n\t\t\t\t\tnew SelectionProxyHandler({ alias, sqlAliasedBehavior: 'alias', sqlBehavior: 'error' }),\n\t\t\t\t) as WithSubqueryWithSelection<TSelection, TAlias>;\n\t\t\t},\n\t\t};\n\t}\n\n\t/**\n\t * Incorporates a previously defined CTE (using `$with`) into the main query.\n\t *\n\t * This method allows the main query to reference a temporary named result set.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}\n\t *\n\t * @param queries The CTEs to incorporate into the main query.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Define a subquery 'sq' as a CTE using $with\n\t * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));\n\t *\n\t * // Incorporate the CTE 'sq' into the main query and select from it\n\t * const result = await db.with(sq).select().from(sq);\n\t * ```\n\t */\n\twith(...queries: WithSubquery[]) {\n\t\tconst self = this;\n\n\t\tfunction select(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;\n\t\tfunction select<TSelection extends SelectedFields>(\n\t\t\tfields: TSelection,\n\t\t): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;\n\t\tfunction select(fields?: SelectedFields): MsSqlSelectBuilder<SelectedFields | undefined, TPreparedQueryHKT> {\n\t\t\treturn new MsSqlSelectBuilder({\n\t\t\t\tfields: fields ?? undefined,\n\t\t\t\tsession: self.session,\n\t\t\t\tdialect: self.dialect,\n\t\t\t\twithList: queries,\n\t\t\t});\n\t\t}\n\n\t\tfunction selectDistinct(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;\n\t\tfunction selectDistinct<TSelection extends SelectedFields>(\n\t\t\tfields: TSelection,\n\t\t): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;\n\t\tfunction selectDistinct(\n\t\t\tfields?: SelectedFields,\n\t\t): MsSqlSelectBuilder<SelectedFields | undefined, TPreparedQueryHKT> {\n\t\t\treturn new MsSqlSelectBuilder({\n\t\t\t\tfields: fields ?? undefined,\n\t\t\t\tsession: self.session,\n\t\t\t\tdialect: self.dialect,\n\t\t\t\twithList: queries,\n\t\t\t\tdistinct: true,\n\t\t\t});\n\t\t}\n\n\t\treturn { select, selectDistinct };\n\t}\n\n\t/**\n\t * Creates a select query.\n\t *\n\t * 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.\n\t *\n\t * Use `.from()` method to specify which table to select from.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/select}\n\t *\n\t * @param fields The selection object.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Select all columns and all rows from the 'cars' table\n\t * const allCars: Car[] = await db.select().from(cars);\n\t *\n\t * // Select specific columns and all rows from the 'cars' table\n\t * const carsIdsAndBrands: { id: number; brand: string }[] = await db.select({\n\t *   id: cars.id,\n\t *   brand: cars.brand\n\t * })\n\t *   .from(cars);\n\t * ```\n\t *\n\t * Like in SQL, you can use arbitrary expressions as selection fields, not just table columns:\n\t *\n\t * ```ts\n\t * // Select specific columns along with expression and all rows from the 'cars' table\n\t * const carsIdsAndLowerNames: { id: number; lowerBrand: string }[] = await db.select({\n\t *   id: cars.id,\n\t *   lowerBrand: sql<string>`lower(${cars.brand})`,\n\t * })\n\t *   .from(cars);\n\t * ```\n\t */\n\tselect(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;\n\tselect<TSelection extends SelectedFields>(fields: TSelection): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;\n\tselect(fields?: SelectedFields): MsSqlSelectBuilder<SelectedFields | undefined, TPreparedQueryHKT> {\n\t\treturn new MsSqlSelectBuilder({ fields: fields ?? undefined, session: this.session, dialect: this.dialect });\n\t}\n\n\t/**\n\t * Adds `distinct` expression to the select query.\n\t *\n\t * Calling this method will return only unique values. When multiple columns are selected, it returns rows with unique combinations of values in these columns.\n\t *\n\t * Use `.from()` method to specify which table to select from.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/select#distinct}\n\t *\n\t * @param fields The selection object.\n\t *\n\t * @example\n\t * ```ts\n\t * // Select all unique rows from the 'cars' table\n\t * await db.selectDistinct()\n\t *   .from(cars)\n\t *   .orderBy(cars.id, cars.brand, cars.color);\n\t *\n\t * // Select all unique brands from the 'cars' table\n\t * await db.selectDistinct({ brand: cars.brand })\n\t *   .from(cars)\n\t *   .orderBy(cars.brand);\n\t * ```\n\t */\n\tselectDistinct(): MsSqlSelectBuilder<undefined, TPreparedQueryHKT>;\n\tselectDistinct<TSelection extends SelectedFields>(\n\t\tfields: TSelection,\n\t): MsSqlSelectBuilder<TSelection, TPreparedQueryHKT>;\n\tselectDistinct(fields?: SelectedFields): MsSqlSelectBuilder<SelectedFields | undefined, TPreparedQueryHKT> {\n\t\treturn new MsSqlSelectBuilder({\n\t\t\tfields: fields ?? undefined,\n\t\t\tsession: this.session,\n\t\t\tdialect: this.dialect,\n\t\t\tdistinct: true,\n\t\t});\n\t}\n\n\t/**\n\t * Creates an update query.\n\t *\n\t * Calling this method without `.where()` clause will update all rows in a table. The `.where()` clause specifies which rows should be updated.\n\t *\n\t * Use `.set()` method to specify which values to update.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/update}\n\t *\n\t * @param table The table to update.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Update all rows in the 'cars' table\n\t * await db.update(cars).set({ color: 'red' });\n\t *\n\t * // Update rows with filters and conditions\n\t * await db.update(cars).set({ color: 'red' }).where(eq(cars.brand, 'BMW'));\n\t * ```\n\t */\n\tupdate<TTable extends MsSqlTable>(table: TTable): MsSqlUpdateBuilder<TTable, TQueryResult, TPreparedQueryHKT> {\n\t\treturn new MsSqlUpdateBuilder(table, this.session, this.dialect);\n\t}\n\n\t/**\n\t * Creates an insert query.\n\t *\n\t * Calling this method will create new rows in a table. Use `.values()` method to specify which values to insert.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/insert}\n\t *\n\t * @param table The table to insert into.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Insert one row\n\t * await db.insert(cars).values({ brand: 'BMW' });\n\t *\n\t * // Insert multiple rows\n\t * await db.insert(cars).values([{ brand: 'BMW' }, { brand: 'Porsche' }]);\n\t * ```\n\t */\n\tinsert<TTable extends MsSqlTable>(table: TTable): MsSqlInsertBuilder<TTable, TQueryResult, TPreparedQueryHKT> {\n\t\treturn new MsSqlInsertBuilder(table, this.session, this.dialect);\n\t}\n\n\t/**\n\t * Creates a delete query.\n\t *\n\t * Calling this method without `.where()` clause will delete all rows in a table. The `.where()` clause specifies which rows should be deleted.\n\t *\n\t * See docs: {@link https://orm.drizzle.team/docs/delete}\n\t *\n\t * @param table The table to delete from.\n\t *\n\t * @example\n\t *\n\t * ```ts\n\t * // Delete all rows in the 'cars' table\n\t * await db.delete(cars);\n\t *\n\t * // Delete rows with filters and conditions\n\t * await db.delete(cars).where(eq(cars.color, 'green'));\n\t * ```\n\t */\n\tdelete<TTable extends MsSqlTable>(table: TTable): MsSqlDeleteBase<TTable, TQueryResult, TPreparedQueryHKT> {\n\t\treturn new MsSqlDeleteBase(table, this.session, this.dialect);\n\t}\n\n\texecute<T extends { [column: string]: any } | { [column: string]: any }[]>(\n\t\tquery: SQLWrapper | string,\n\t): Promise<QueryResultKind<TQueryResult, T>> {\n\t\treturn this.session.execute((typeof query === 'string' ? sql.raw(query) : query).getSQL());\n\t}\n\n\ttransaction<T>(\n\t\ttransaction: (\n\t\t\ttx: MsSqlTransaction<TQueryResult, TPreparedQueryHKT, TFullSchema, TSchema>,\n\t\t\tconfig?: MsSqlTransactionConfig,\n\t\t) => Promise<T>,\n\t\tconfig?: MsSqlTransactionConfig,\n\t): Promise<T> {\n\t\treturn this.session.transaction(transaction, config);\n\t}\n}\n\nexport type MySQLWithReplicas<Q> = Q & { $primary: Q };\n\nexport const withReplicas = <\n\tHKT extends QueryResultHKT,\n\tTPreparedQueryHKT extends PreparedQueryHKTBase,\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n\tQ extends MsSqlDatabase<\n\t\tHKT,\n\t\tTPreparedQueryHKT,\n\t\tTFullSchema,\n\t\tTSchema extends Record<string, unknown> ? V1.ExtractTablesWithRelations<TFullSchema> : TSchema\n\t>,\n>(\n\tprimary: Q,\n\treplicas: [Q, ...Q[]],\n\tgetReplica: (replicas: Q[]) => Q = () => replicas[Math.floor(Math.random() * replicas.length)]!,\n): MySQLWithReplicas<Q> => {\n\tconst select: Q['select'] = (...args: []) => getReplica(replicas).select(...args);\n\tconst selectDistinct: Q['selectDistinct'] = (...args: []) => getReplica(replicas).selectDistinct(...args);\n\tconst $with: Q['with'] = (...args: []) => getReplica(replicas).with(...args);\n\n\tconst update: Q['update'] = (...args: [any]) => primary.update(...args);\n\tconst insert: Q['insert'] = (...args: [any]) => primary.insert(...args);\n\tconst $delete: Q['delete'] = (...args: [any]) => primary.delete(...args);\n\tconst execute: Q['execute'] = (...args: [any]) => primary.execute(...args);\n\tconst transaction: Q['transaction'] = (...args: [any, any]) => primary.transaction(...args);\n\n\treturn {\n\t\t...primary,\n\t\tupdate,\n\t\tinsert,\n\t\tdelete: $delete,\n\t\texecute,\n\t\ttransaction,\n\t\t$primary: primary,\n\t\t$replicas: replicas,\n\t\tselect,\n\t\tselectDistinct,\n\t\twith: $with,\n\t\tget _query() {\n\t\t\treturn getReplica(replicas)._query;\n\t\t},\n\t};\n};\n"],"mappings":";;;;;;;;;;;;AA4BA,IAAa,gBAAb,MAKE;CACD,QAAiB,cAAsB;CAOvC;CAMA,YAEC,AAAS,SAET,AAAS,SACT,QACC;EAJQ;EAEA;AAGT,OAAK,IAAI,SACN;GAAE,QAAQ,OAAO;GAAQ,eAAe,OAAO;GAAe,GAC9D;GAAE,QAAQ;GAAW,eAAe,EAAE;GAAE;AAC3C,OAAK,SAAS,EAAE;AAChB,MAAI,KAAK,EAAE,OACV,MAAK,MAAM,CAAC,WAAW,YAAY,OAAO,QAAQ,KAAK,EAAE,OAAO,CAC/D,CAAC,KAAK,OAAyF,aAC9F,IAAI,uBACH,OAAQ,YACR,KAAK,EAAE,QACP,KAAK,EAAE,eACP,OAAQ,WAAW,YACnB,SACA,SACA,QACA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAqCL,MAA6B,OAAe;EAC3C,MAAM,OAAO;AAEb,SAAO,EACN,GACC,IACgD;AAChD,OAAI,OAAO,OAAO,WACjB,MAAK,GAAG,IAAI,aAAa,KAAK,QAAQ,CAAC;AAGxC,UAAO,IAAI,MACV,IAAI,aAAa,GAAG,QAAQ,EAAE,GAAG,mBAAmB,EAAoB,OAAO,KAAK,EACpF,IAAI,sBAAsB;IAAE;IAAO,oBAAoB;IAAS,aAAa;IAAS,CAAC,CACvF;KAEF;;;;;;;;;;;;;;;;;;;;;CAsBF,KAAK,GAAG,SAAyB;EAChC,MAAM,OAAO;EAMb,SAAS,OAAO,QAA4F;AAC3G,UAAO,IAAI,mBAAmB;IAC7B,QAAQ,UAAU;IAClB,SAAS,KAAK;IACd,SAAS,KAAK;IACd,UAAU;IACV,CAAC;;EAOH,SAAS,eACR,QACoE;AACpE,UAAO,IAAI,mBAAmB;IAC7B,QAAQ,UAAU;IAClB,SAAS,KAAK;IACd,SAAS,KAAK;IACd,UAAU;IACV,UAAU;IACV,CAAC;;AAGH,SAAO;GAAE;GAAQ;GAAgB;;CAyClC,OAAO,QAA4F;AAClG,SAAO,IAAI,mBAAmB;GAAE,QAAQ,UAAU;GAAW,SAAS,KAAK;GAAS,SAAS,KAAK;GAAS,CAAC;;CA+B7G,eAAe,QAA4F;AAC1G,SAAO,IAAI,mBAAmB;GAC7B,QAAQ,UAAU;GAClB,SAAS,KAAK;GACd,SAAS,KAAK;GACd,UAAU;GACV,CAAC;;;;;;;;;;;;;;;;;;;;;;;CAwBH,OAAkC,OAA4E;AAC7G,SAAO,IAAI,mBAAmB,OAAO,KAAK,SAAS,KAAK,QAAQ;;;;;;;;;;;;;;;;;;;;;CAsBjE,OAAkC,OAA4E;AAC7G,SAAO,IAAI,mBAAmB,OAAO,KAAK,SAAS,KAAK,QAAQ;;;;;;;;;;;;;;;;;;;;;CAsBjE,OAAkC,OAAyE;AAC1G,SAAO,IAAI,gBAAgB,OAAO,KAAK,SAAS,KAAK,QAAQ;;CAG9D,QACC,OAC4C;AAC5C,SAAO,KAAK,QAAQ,SAAS,OAAO,UAAU,WAAW,IAAI,IAAI,MAAM,GAAG,OAAO,QAAQ,CAAC;;CAG3F,YACC,aAIA,QACa;AACb,SAAO,KAAK,QAAQ,YAAY,aAAa,OAAO;;;AAMtD,MAAa,gBAYZ,SACA,UACA,mBAAyC,SAAS,KAAK,MAAM,KAAK,QAAQ,GAAG,SAAS,OAAO,MACnE;CAC1B,MAAM,UAAuB,GAAG,SAAa,WAAW,SAAS,CAAC,OAAO,GAAG,KAAK;CACjF,MAAM,kBAAuC,GAAG,SAAa,WAAW,SAAS,CAAC,eAAe,GAAG,KAAK;CACzG,MAAM,SAAoB,GAAG,SAAa,WAAW,SAAS,CAAC,KAAK,GAAG,KAAK;CAE5E,MAAM,UAAuB,GAAG,SAAgB,QAAQ,OAAO,GAAG,KAAK;CACvE,MAAM,UAAuB,GAAG,SAAgB,QAAQ,OAAO,GAAG,KAAK;CACvE,MAAM,WAAwB,GAAG,SAAgB,QAAQ,OAAO,GAAG,KAAK;CACxE,MAAM,WAAyB,GAAG,SAAgB,QAAQ,QAAQ,GAAG,KAAK;CAC1E,MAAM,eAAiC,GAAG,SAAqB,QAAQ,YAAY,GAAG,KAAK;AAE3F,QAAO;EACN,GAAG;EACH;EACA;EACA,QAAQ;EACR;EACA;EACA,UAAU;EACV,WAAW;EACX;EACA;EACA,MAAM;EACN,IAAI,SAAS;AACZ,UAAO,WAAW,SAAS,CAAC;;EAE7B"}