pqb 0.62.1 → 0.64.0

package/dist/index.d.ts

@@ -1,7 +1,8 @@
 import { inspect } from 'node:util';
 import { AsyncLocalStorage } from 'node:async_hooks';
 import { MaybeArray as MaybeArray$1 } from 'rollup';
-import { RecordOptionalString as RecordOptionalString$1, DefaultPrivileges as DefaultPrivileges$1, Query as Query$1 } from 'pqb/internal';
+import { RecordOptionalString as RecordOptionalString$1, DefaultPrivileges as DefaultPrivileges$1 } from 'pqb/internal';
+import { Query as Query$1 } from 'pqb';
 
 type UnionToIntersection<U> = (U extends any ? (x: U) => void : never) extends (x: infer I) => void ? I : never;
 type MaybeArray<T> = T | T[];
@@ -252,6 +253,11 @@ declare abstract class QueryError<T extends PickQueryShape = PickQueryShape> ext
     get columns(): { [K in keyof T["shape"]]?: true | undefined; };
 }
 
+interface SqlSessionState {
+    role?: string;
+    setConfig?: Record<string, string | number | boolean>;
+}
+
 /**
  * Generic result returning from query methods.
  */
@@ -342,6 +348,7 @@ interface AdapterTransactionOptions {
     locals?: {
         [ConfigName: string]: string | number;
     };
+    sqlSessionState?: SqlSessionState;
 }
 type TransactionArgs<Result> = [
     cbOrOptions: undefined | AdapterTransactionOptions | ((adapter: TransactionAdapterBase) => Promise<Result>),
@@ -366,8 +373,8 @@ interface AdapterBase {
     getHost(): string;
     getSchema(): QuerySchema | undefined;
     connect?(): Promise<unknown>;
-    query<T extends QueryResultRow = QueryResultRow>(text: string, values?: unknown[], startingSavepoint?: string, releasingSavepoint?: string): Promise<QueryResult<T>>;
-    arrays<R extends any[] = any[]>(text: string, values?: unknown[], startingSavepoint?: string, releasingSavepoint?: string): Promise<QueryArraysResult<R>>;
+    query<T extends QueryResultRow = QueryResultRow>(text: string, values?: unknown[], startingSavepoint?: string, releasingSavepoint?: string, sqlSessionState?: SqlSessionState): Promise<QueryResult<T>>;
+    arrays<R extends any[] = any[]>(text: string, values?: unknown[], startingSavepoint?: string, releasingSavepoint?: string, sqlSessionState?: SqlSessionState): Promise<QueryArraysResult<R>>;
     /**
      * Run a transaction
      *
@@ -2499,18 +2506,21 @@ interface QueryDataSources {
     [K: string]: QuerySourceItem;
 }
 
-interface OrderArgSelf extends PickQuerySelectable, PickQueryResult, PickQueryTsQuery {
+declare namespace Order {
+    export interface ArgThis extends PickQuerySelectable, PickQueryResult, PickQueryTsQuery {
+    }
+    export type Arg<T extends ArgThis> = ArgKey<T> | ArgTsQuery<T> | {
+        [K in ArgKey<T> | ArgTsQuery<T>]?: K extends ArgTsQuery<T> ? OrderTsQueryConfig : SortDir;
+    } | Expression;
+    export type Args<T extends ArgThis> = Arg<T>[];
+    type ArgTsQuery<T extends ArgThis> = string | undefined extends T['__tsQuery'] ? never : Exclude<T['__tsQuery'], undefined>;
+    type ArgKey<T extends ArgThis> = {
+        [K in keyof T['__selectable']]: T['__selectable'][K]['column']['queryType'] extends undefined ? never : K;
+    }[keyof T['__selectable']] | {
+        [K in keyof T['result']]: T['result'][K]['dataType'] extends 'array' | 'object' | 'runtimeComputed' ? never : K;
+    }[keyof T['result']];
+    export {};
 }
-type OrderArg<T extends OrderArgSelf> = OrderArgKey<T> | OrderArgTsQuery<T> | {
-    [K in OrderArgKey<T> | OrderArgTsQuery<T>]?: K extends OrderArgTsQuery<T> ? OrderTsQueryConfig : SortDir;
-} | Expression;
-type OrderArgs<T extends OrderArgSelf> = OrderArg<T>[];
-type OrderArgTsQuery<T extends OrderArgSelf> = string | undefined extends T['__tsQuery'] ? never : Exclude<T['__tsQuery'], undefined>;
-type OrderArgKey<T extends OrderArgSelf> = {
-    [K in keyof T['__selectable']]: T['__selectable'][K]['column']['queryType'] extends undefined ? never : K;
-}[keyof T['__selectable']] | {
-    [K in keyof T['result']]: T['result'][K]['dataType'] extends 'array' | 'object' | 'runtimeComputed' ? never : K;
-}[keyof T['result']];
 declare class QueryOrder {
     /**
      * Adds an order by clause to the query.
@@ -2547,7 +2557,7 @@ declare class QueryOrder {
      *
      * @param args - column name(s) or an object with column names and sort directions.
      */
-    order<T extends OrderArgSelf>(this: T, ...args: OrderArgs<T>): T;
+    order<T extends Order.ArgThis>(this: T, ...args: Order.Args<T>): T;
     /**
      * Order by SQL expression
      *
@@ -2562,12 +2572,12 @@ declare class QueryOrder {
     orderSql<T>(this: T, ...args: SQLQueryArgs): T;
 }
 
-interface WindowArg<T extends OrderArgSelf> {
+interface WindowArg<T extends Order.ArgThis> {
     [K: string]: WindowArgDeclaration<T> | Expression;
 }
-interface WindowArgDeclaration<T extends OrderArgSelf = OrderArgSelf> {
+interface WindowArgDeclaration<T extends Order.ArgThis = Order.ArgThis> {
     partitionBy?: SelectableOrExpression<T> | SelectableOrExpressions<T>;
-    order?: OrderArg<T>;
+    order?: Order.Arg<T>;
 }
 type WindowResult<T, W extends RecordUnknown> = T & {
     windows: {
@@ -2601,12 +2611,12 @@ declare class QueryWindow {
      *
      * @param arg - window config
      */
-    window<T extends OrderArgSelf, W extends WindowArg<T>>(this: T, arg: W): WindowResult<T, W>;
+    window<T extends Order.ArgThis, W extends WindowArg<T>>(this: T, arg: W): WindowResult<T, W>;
 }
 
 interface AggregateOptions<T extends PickQuerySelectableResultRelationsWindows> {
     distinct?: boolean;
-    order?: OrderArg<T> | OrderArgs<T>;
+    order?: Order.Arg<T> | Order.Args<T>;
     filter?: WhereArg<T>;
     filterOr?: WhereArgs<T>;
     withinGroup?: boolean;
@@ -4149,7 +4159,7 @@ declare class QueryWithSchema {
     withSchema<T>(this: T, schema: QuerySchema | undefined): T;
 }
 
-interface AsyncState {
+interface AsyncState extends SqlSessionState {
     transactionAdapter?: TransactionAdapterBase;
     transactionId?: number;
     afterCommit?: TransactionAfterCommitHook[];
@@ -4158,7 +4168,7 @@ interface AsyncState {
     catchI?: number;
     schema?: QuerySchema;
 }
-interface StorageOptions {
+interface StorageOptions extends SqlSessionState {
     log?: boolean;
     schema?: QuerySchema;
 }
@@ -5161,7 +5171,9 @@ declare abstract class ExpressionTypeMethod {
 }
 
 type IsolationLevel = 'SERIALIZABLE' | 'REPEATABLE READ' | 'READ COMMITTED' | 'READ UNCOMMITTED';
-interface TransactionOptions extends StorageOptions {
+interface TransactionOptions {
+    log?: boolean;
+    schema?: QuerySchema;
     level?: IsolationLevel;
     readOnly?: boolean;
     deferrable?: boolean;
@@ -7110,22 +7122,27 @@ type CreateRelationsDataOmittingFKeys<T extends CreateSelf, Union> = (Union exte
 } & Partial<Union['nested']> : {
     [P in Union['columns'] & keyof T['inputType']]: CreateColumn<T, P>;
 } | Union['nested']) => void : never) extends (u: infer Obj) => void ? Obj : never;
-type CreateResult<T extends CreateSelf> = T extends {
+type CreateResult<T extends CreateSelf, Data> = T extends {
     isCount: true;
-} ? T : T['returnType'] extends undefined | 'all' ? SetQueryReturnsOneResult<T, NarrowCreateResult<T>> : T['returnType'] extends 'pluck' ? SetQueryReturnsColumnResult<T, NarrowCreateResult<T>> : SetQueryResult<T, NarrowCreateResult<T>>;
-type InsertResult<T extends CreateSelf> = T['__hasSelect'] extends true ? T['returnType'] extends undefined | 'all' ? SetQueryReturnsOneResult<T, NarrowCreateResult<T>> : T['returnType'] extends 'pluck' ? SetQueryReturnsColumnResult<T, NarrowCreateResult<T>> : SetQueryResult<T, NarrowCreateResult<T>> : SetQueryReturnsRowCount<T>;
+} ? T : T['returnType'] extends undefined | 'all' ? SetQueryReturnsOneResult<T, NarrowCreateResult<T, Data>> : T['returnType'] extends 'pluck' ? SetQueryReturnsColumnResult<T, NarrowCreateResult<T, Data>> : SetQueryResult<T, NarrowCreateResult<T, Data>>;
+type InsertResult<T extends CreateSelf, Data> = T['__hasSelect'] extends true ? T['returnType'] extends undefined | 'all' ? SetQueryReturnsOneResult<T, NarrowCreateResult<T, Data>> : T['returnType'] extends 'pluck' ? SetQueryReturnsColumnResult<T, NarrowCreateResult<T, Data>> : SetQueryResult<T, NarrowCreateResult<T, Data>> : SetQueryReturnsRowCount<T>;
 type CreateManyResult<T extends CreateSelf> = T extends {
     isCount: true;
-} ? SetQueryResult<T, NarrowCreateResult<T>> : T['returnType'] extends 'one' | 'oneOrThrow' ? SetQueryReturnsAllResult<T, NarrowCreateResult<T>> : T['returnType'] extends 'value' | 'valueOrThrow' ? SetQueryReturnsPluckColumnResult<T, NarrowCreateResult<T>> : SetQueryResult<T, NarrowCreateResult<T>>;
-type InsertManyResult<T extends CreateSelf> = T['__hasSelect'] extends true ? T['returnType'] extends 'one' | 'oneOrThrow' ? SetQueryReturnsAllResult<T, NarrowCreateResult<T>> : T['returnType'] extends 'value' | 'valueOrThrow' ? SetQueryReturnsPluckColumnResult<T, NarrowCreateResult<T>> : SetQueryResult<T, NarrowCreateResult<T>> : SetQueryReturnsRowCountMany<T>;
+} ? T : T['returnType'] extends 'one' | 'oneOrThrow' ? SetQueryReturnsAll<T> : T['returnType'] extends 'value' | 'valueOrThrow' ? SetValueQueryReturnsPluckColumn<T> : T;
+type InsertManyResult<T extends CreateSelf> = T['__hasSelect'] extends true ? T['returnType'] extends 'one' | 'oneOrThrow' ? SetQueryReturnsAll<T> : T['returnType'] extends 'value' | 'valueOrThrow' ? SetValueQueryReturnsPluckColumn<T> : T : SetQueryReturnsRowCountMany<T>;
 /**
  * When creating a record with a *belongs to* nested record,
  * un-nullify foreign key columns of the result.
  *
  * The same should work as well with any non-null columns passed to `create`, but it's to be implemented later.
  */
-type NarrowCreateResult<T extends CreateSelf> = EmptyObject extends T['relations'] ? T['result'] : {
-    [K in keyof T['result']]: K extends T['relations'][keyof T['relations']]['omitForeignKeyInCreate'] ? Column.Pick.QueryColumnOfTypeAndOps<string, Exclude<T['result'][K]['type'], null>, T['result'][K]['operators']> : T['result'][K];
+type NarrowCreateResult<T extends CreateSelf, Data> = EmptyObject extends T['relations'] ? T['result'] : {
+    [K in keyof T['result']]: true extends {
+        [R in keyof T['relations']]: K extends T['relations'][R]['omitForeignKeyInCreate'] ? T['relations'][R]['dataForCreate'] extends {
+            columns: unknown;
+            nested: unknown;
+        } ? keyof T['relations'][R]['dataForCreate']['nested'] extends keyof Data ? true : T['relations'][R]['dataForCreate']['columns'] extends keyof Data ? null | undefined extends Data[T['relations'][R]['dataForCreate']['columns']] ? never : true : never : never : never;
+    }[keyof T['relations']] ? Column.Pick.QueryColumnOfTypeAndOps<string, Exclude<T['result'][K]['outputType'], null>, T['result'][K]['operators']> : T['result'][K];
 };
 type IgnoreResult<T extends CreateSelf> = T['returnType'] extends 'oneOrThrow' ? QueryTakeOptional<T> : T['returnType'] extends 'valueOrThrow' ? SetQueryReturnsColumnOptional<T, T['result']['value']> : T;
 type OnConflictArg<T extends PickQueryUniqueProperties> = T['internal']['uniqueColumnNames'] | T['internal']['uniqueColumnTuples'] | Expression | {
@@ -7145,8 +7162,8 @@ interface CreateCtx {
     returnTypeAll?: true;
     resultAll: RecordUnknown[];
 }
-declare const _queryCreate: <T extends CreateSelf>(q: T, data: CreateData<T>) => CreateResult<T>;
-declare const _queryInsert: <T extends CreateSelf>(query: T, data: CreateData<T>) => InsertResult<T>;
+declare const _queryCreate: <T extends CreateSelf, Data extends CreateData<T>>(q: T, data: Data) => CreateResult<T, Data>;
+declare const _queryInsert: <T extends CreateSelf, Data extends CreateData<T>>(query: T, data: Data) => InsertResult<T, Data>;
 declare const _queryCreateMany: <T extends CreateSelf>(q: T, data: CreateData<T>[]) => CreateManyResult<T>;
 declare const _queryInsertMany: <T extends CreateSelf>(q: T, data: CreateData<T>[]) => InsertManyResult<T>;
 declare const _queryDefaults: <T extends CreateSelf, Data extends Partial<CreateData<T>>>(q: T, data: Data) => AddQueryDefaults<T, keyof Data>;
@@ -7157,6 +7174,7 @@ declare const _queryDefaults: <T extends CreateSelf, Data extends Partial<Create
  */
 type CreateMethodsNames = 'create' | 'insert' | 'createMany' | 'insertMany' | CreateFromMethodNames;
 type CreateManyMethodsNames = 'createMany' | 'insertMany' | CreateManyFromMethodNames;
+type ExtraPropertiesAreNotAllowed<T extends CreateSelf, Data> = keyof Data extends keyof T['inputType'] | keyof T['relations'] ? Data : `Extra properties are not allowed: ${Exclude<keyof Data, keyof T['inputType'] | keyof T['relations']> & string}`;
 declare class QueryCreate {
     /**
      * `create` and `insert` create a single record.
@@ -7207,16 +7225,12 @@ declare class QueryCreate {
      *   )
      *   .from('b');
      * ```
-     *
-     * @param data - data for the record, may have values, raw SQL, queries, relation operations.
      */
-    create<T extends CreateSelf>(this: T, data: CreateData<T>): CreateResult<T>;
+    create<T extends CreateSelf, Data extends CreateData<T>>(this: T, data: ExtraPropertiesAreNotAllowed<T, Data>): CreateResult<T, Data>;
     /**
      * Works exactly as {@link create}, except that it returns inserted row count by default.
-     *
-     * @param data - data for the record, may have values, raw SQL, queries, relation operations.
      */
-    insert<T extends CreateSelf>(this: T, data: CreateData<T>): InsertResult<T>;
+    insert<T extends CreateSelf, Data extends CreateData<T>>(this: T, data: ExtraPropertiesAreNotAllowed<T, Data>): InsertResult<T, Data>;
     /**
      * `createMany` and `insertMany` will create a batch of records.
      *
@@ -9028,7 +9042,7 @@ declare module "./index" {
          * @param search - name of the search to use the query from
          * @param options - `text` for a text source, `options` for `ts_headline` options
          */
-        headline<T extends OrderArgSelf>(this: T, search: HeadlineSearchArg<T>, options?: HeadlineParams<T>): SetQueryReturnsColumnOrThrow<T, Column.Pick.QueryColumnOfType<string>>;
+        headline<T extends Order.ArgThis>(this: T, search: HeadlineSearchArg<T>, options?: HeadlineParams<T>): SetQueryReturnsColumnOrThrow<T, Column.Pick.QueryColumnOfType<string>>;
     }
 }
 type SearchArg<T extends PickQuerySelectable, As extends string> = {
@@ -9768,6 +9782,7 @@ interface QueryHelper<T extends PickQueryTableMetaShapeTableAs, Args extends any
     args: Args;
     result: Result;
 }
+type QueryHelperResult<T extends QueryHelper<PickQueryTableMetaShapeTableAs, any[], MergeQueryArg>> = T['result'];
 interface NarrowTypeSelf extends PickQueryResultReturnType {
     returnType: undefined | 'all' | 'one' | 'oneOrThrow' | 'value' | 'valueOrThrow' | 'pluck';
 }
@@ -10344,6 +10359,12 @@ interface Query extends IsQuery, PickQueryTable, PickQueryShape, PickQuerySelect
     relationQueries: IsQueries;
     error: new (message: string, length: number, name: QueryErrorName) => QueryError;
 }
+declare namespace Query {
+    namespace Order {
+        type Arg<T extends Order.ArgThis> = Order.Arg<T>;
+        type Args<T extends Order.ArgThis> = Order.Args<T>;
+    }
+}
 type SelectableOfType<T extends PickQuerySelectable, Type> = {
     [K in keyof T['__selectable']]: T['__selectable'][K]['column']['type'] extends Type | null ? K : never;
 }[keyof T['__selectable']];
@@ -10899,5 +10920,5 @@ declare const testTransaction: {
     close(arg: Arg): Promise<void>;
 };
 
-export { ArrayColumn, BigIntColumn, BigSerialColumn, BitColumn, BitVaryingColumn, BooleanColumn, BoxColumn, ByteaColumn, CidrColumn, CircleColumn, CitextColumn, Column, ColumnsShape, CustomTypeColumn, DateBaseColumn, DateColumn, DateTimeBaseClass, DateTimeTzBaseClass, Db, DecimalColumn, DefaultPrivileges, DomainColumn, DoublePrecisionColumn, DynamicRawSQL, EnumColumn, Expression, InetColumn, IntegerBaseColumn, IntegerColumn, IntervalColumn, JSONColumn, JSONTextColumn, LimitedTextBaseColumn, LineColumn, LsegColumn, MacAddr8Column, MacAddrColumn, MoneyColumn, NotFoundError, NumberAsStringBaseColumn, NumberBaseColumn, Operators, OrchidOrmInternalError, PathColumn, PointColumn, PolygonColumn, PostgisGeographyPointColumn, QueryError, QueryHookUtils, QueryHooks, RawSql, RealColumn, SerialColumn, SmallIntColumn, SmallSerialColumn, StringColumn$1 as StringColumn, TableData, TextBaseColumn, TextColumn, TimeColumn, TimestampColumn, TimestampTZColumn, TsQueryColumn, TsVectorColumn, UUIDColumn, UnknownColumn, VarCharColumn, VirtualColumn, XMLColumn, _appendQuery, _clone, _createDbSqlMethod, _hookSelectColumns, _initQueryBuilder, _orCreate, _prependWith, _queryCreate, _queryCreateMany, _queryCreateManyFrom, _queryDefaults, _queryDelete, _queryFindBy, _queryFindByOptional, _queryHookAfterCreate, _queryHookAfterUpdate, _queryInsert, _queryInsertMany, _queryJoinOn, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpsert, _queryWhere, _queryWhereExists, _queryWhereIn, addCode, addTopCte, addTopCteSql, applyMixins, assignDbDataToColumn, backtickQuote, cloneQueryBaseUnscoped, codeToString, colors, columnsShapeToCode, constraintInnerToCode, consumeColumnName, copyTableData, createDbWithAdapter, deepCompare, defaultSchemaConfig, emptyArray, emptyObject, escapeForMigration, escapeString, excludeInnerToCode, exhaustive, getCallerFilePath, getClonedQueryData, getColumnBaseType, getColumnInfo, getColumnTypes, getFreeAlias, getFreeSetAlias, getImportPath, getPrimaryKeys, getQueryAs, getQuerySchema, getShapeFromSelect, getSqlText, getStackTrace, getSupportedDefaultPrivileges, indexInnerToCode, isExpression, isQueryReturnsAll, isRawSQL, logColors, logParamToLogObject, makeColumnNullable, makeColumnTypes, makeColumnsByType, noop, objectHasValues, omit, parseTableData, parseTableDataInput, pathToLog, pick, pluralize, prepareSubQueryForSql, primaryKeyInnerToCode, pushQueryOnForOuter, pushQueryValueImmutable, pushTableDataCode, quoteObjectKey, quoteTableWithSchema, raw, rawSqlToCode, referencesArgsToCode, returnArg, setColumnData, setColumnEncode, setColumnParse, setColumnParseNull, setConnectRetryConfig, setCurrentColumnName, setDataValue, setDefaultLanguage, setFreeAlias, setQueryObjectValueImmutable, singleQuote, snakeCaseKey, tableDataMethods, testTransaction, toArray, toCamelCase, toPascalCase, toSnakeCase, wrapAdapterFnWithConnectRetry };
-export type { AdapterBase, AdapterConfigBase, AfterCommitStandaloneHook, AfterHook, ArrayColumnValue, ArrayData, AsyncState, BaseNumberData, Code, Codes, ColumnFromDbParams, ColumnSchemaConfig, ColumnSchemaGetterColumns, ColumnSchemaGetterTableClass, ColumnToCodeCtx, ColumnTypeSchemaArg, ColumnsByType, ComputedColumnsFromOptions, ComputedOptionsConfig, ComputedOptionsFactory, CreateCtx, CreateData, CreateManyMethodsNames, CreateMethodsNames, CreateSelf, DateColumnData, DbDomainArg, DbExtension, DbOptions, DbResult, DbSharedOptions, DbSqlMethod, DbStructureDomainsMap, DbTableOptionScopes, DbTableOptions, DecimalColumnData, DefaultColumnTypes, DefaultSchemaConfig, DeleteMethodsNames, EmptyObject, EmptyTuple, FromArg, FromResult, GeneratorIgnore, HookSelectValue, IsQuery, IsolationLevel, JoinQueryMethod, JoinedShapes, MapTableScopesOption, MaybeArray, MaybePromise, MergeQuery, NoPrimaryKeyOption, NonUniqDataItem, NumberColumnData, OperatorsArray, OperatorsJson, OperatorsOrdinalText, Ord, PickQueryInputType, PickQueryInternal, PickQueryQ, PickQueryRelations, PickQuerySelectableRelations, PickQueryShape, Query, QueryAfterHook, QueryArraysResult, QueryBeforeActionHook, QueryBeforeHook, QueryData, QueryHasWhere, QueryInternal, QueryLogObject, QueryLogOptions, QueryLogger, QueryManyTake, QueryManyTakeOptional, QueryOrExpression, QueryResult, QueryResultRow, QueryReturnType, QuerySchema, QueryScopes, RawSqlBase, RecordKeyTrue, RecordOptionalString, RecordString, RecordStringOrNumber, RecordUnknown, RelationConfigBase, RelationJoinQuery, RelationsBase, SearchWeight, SelectableFromShape, SerialColumnData, ShallowSimplify, ShapeColumnPrimaryKeys, ShapeUniqueColumns, SingleSql, SingleSqlItem, Sql, SqlFn, StorageOptions, StringData, TableDataFn, TableDataInput, TableDataItem, TableDataItemsUniqueColumnTuples, TableDataItemsUniqueColumns, TableDataItemsUniqueConstraints, TableDataMethods, TemplateLiteralArgs, Timestamps, TransactionAdapterBase, TransactionArgs, TransactionOptions, UniqueConstraints, UniqueTableDataItem, UpdateData, UpsertData, UpsertThis, WhereArg };
+export { ArrayColumn, BigIntColumn, BigSerialColumn, BitColumn, BitVaryingColumn, BooleanColumn, BoxColumn, ByteaColumn, CidrColumn, CircleColumn, CitextColumn, Column, ColumnsShape, CustomTypeColumn, DateBaseColumn, DateColumn, DateTimeBaseClass, DateTimeTzBaseClass, Db, DecimalColumn, DefaultPrivileges, DomainColumn, DoublePrecisionColumn, DynamicRawSQL, EnumColumn, Expression, InetColumn, IntegerBaseColumn, IntegerColumn, IntervalColumn, JSONColumn, JSONTextColumn, LimitedTextBaseColumn, LineColumn, LsegColumn, MacAddr8Column, MacAddrColumn, MoneyColumn, NotFoundError, NumberAsStringBaseColumn, NumberBaseColumn, Operators, OrchidOrmInternalError, PathColumn, PointColumn, PolygonColumn, PostgisGeographyPointColumn, Query, QueryError, QueryHookUtils, QueryHooks, RawSql, RealColumn, SerialColumn, SmallIntColumn, SmallSerialColumn, StringColumn$1 as StringColumn, TableData, TextBaseColumn, TextColumn, TimeColumn, TimestampColumn, TimestampTZColumn, TsQueryColumn, TsVectorColumn, UUIDColumn, UnknownColumn, VarCharColumn, VirtualColumn, XMLColumn, _appendQuery, _clone, _createDbSqlMethod, _hookSelectColumns, _initQueryBuilder, _orCreate, _prependWith, _queryCreate, _queryCreateMany, _queryCreateManyFrom, _queryDefaults, _queryDelete, _queryFindBy, _queryFindByOptional, _queryHookAfterCreate, _queryHookAfterUpdate, _queryInsert, _queryInsertMany, _queryJoinOn, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpsert, _queryWhere, _queryWhereExists, _queryWhereIn, addCode, addTopCte, addTopCteSql, applyMixins, assignDbDataToColumn, backtickQuote, cloneQueryBaseUnscoped, codeToString, colors, columnsShapeToCode, constraintInnerToCode, consumeColumnName, copyTableData, createDbWithAdapter, deepCompare, defaultSchemaConfig, emptyArray, emptyObject, escapeForMigration, escapeString, excludeInnerToCode, exhaustive, getCallerFilePath, getClonedQueryData, getColumnBaseType, getColumnInfo, getColumnTypes, getFreeAlias, getFreeSetAlias, getImportPath, getPrimaryKeys, getQueryAs, getQuerySchema, getShapeFromSelect, getSqlText, getStackTrace, getSupportedDefaultPrivileges, indexInnerToCode, isExpression, isQueryReturnsAll, isRawSQL, logColors, logParamToLogObject, makeColumnNullable, makeColumnTypes, makeColumnsByType, noop, objectHasValues, omit, parseTableData, parseTableDataInput, pathToLog, pick, pluralize, prepareSubQueryForSql, primaryKeyInnerToCode, pushQueryOnForOuter, pushQueryValueImmutable, pushTableDataCode, quoteObjectKey, quoteTableWithSchema, raw, rawSqlToCode, referencesArgsToCode, returnArg, setColumnData, setColumnEncode, setColumnParse, setColumnParseNull, setConnectRetryConfig, setCurrentColumnName, setDataValue, setDefaultLanguage, setFreeAlias, setQueryObjectValueImmutable, singleQuote, snakeCaseKey, tableDataMethods, testTransaction, toArray, toCamelCase, toPascalCase, toSnakeCase, wrapAdapterFnWithConnectRetry };
+export type { AdapterBase, AdapterConfigBase, AfterCommitStandaloneHook, AfterHook, ArrayColumnValue, ArrayData, AsyncState, BaseNumberData, Code, Codes, ColumnFromDbParams, ColumnSchemaConfig, ColumnSchemaGetterColumns, ColumnSchemaGetterTableClass, ColumnToCodeCtx, ColumnTypeSchemaArg, ColumnsByType, ComputedColumnsFromOptions, ComputedOptionsConfig, ComputedOptionsFactory, CreateCtx, CreateData, CreateManyMethodsNames, CreateMethodsNames, CreateSelf, DateColumnData, DbDomainArg, DbExtension, DbOptions, DbResult, DbSharedOptions, DbSqlMethod, DbStructureDomainsMap, DbTableOptionScopes, DbTableOptions, DecimalColumnData, DefaultColumnTypes, DefaultSchemaConfig, DeleteMethodsNames, EmptyObject, EmptyTuple, FromArg, FromResult, GeneratorIgnore, HookSelectValue, IsQuery, IsolationLevel, JoinQueryMethod, JoinedShapes, MapTableScopesOption, MaybeArray, MaybePromise, MergeQuery, NoPrimaryKeyOption, NonUniqDataItem, NumberColumnData, OperatorsArray, OperatorsJson, OperatorsOrdinalText, Ord, PickQueryInputType, PickQueryInternal, PickQueryQ, PickQueryRelations, PickQuerySelectableRelations, PickQueryShape, QueryAfterHook, QueryArraysResult, QueryBeforeActionHook, QueryBeforeHook, QueryData, QueryHasWhere, QueryHelperResult, QueryInternal, QueryLogObject, QueryLogOptions, QueryLogger, QueryManyTake, QueryManyTakeOptional, QueryOrExpression, QueryResult, QueryResultRow, QueryReturnType, QuerySchema, QueryScopes, RawSqlBase, RecordKeyTrue, RecordOptionalString, RecordString, RecordStringOrNumber, RecordUnknown, RelationConfigBase, RelationJoinQuery, RelationsBase, SearchWeight, SelectableFromShape, SerialColumnData, ShallowSimplify, ShapeColumnPrimaryKeys, ShapeUniqueColumns, SingleSql, SingleSqlItem, Sql, SqlFn, SqlSessionState, StorageOptions, StringData, TableDataFn, TableDataInput, TableDataItem, TableDataItemsUniqueColumnTuples, TableDataItemsUniqueColumns, TableDataItemsUniqueConstraints, TableDataMethods, TemplateLiteralArgs, Timestamps, TransactionAdapterBase, TransactionArgs, TransactionOptions, UniqueConstraints, UniqueTableDataItem, UpdateData, UpsertData, UpsertThis, WhereArg };

package/dist/index.js

@@ -3975,6 +3975,14 @@ class UnhandledTypeError extends OrchidOrmInternalError {
     super(query, `Unhandled type: ${JSON.stringify(value)} received`);
   }
 }
+class NestedSqlSessionError extends OrchidOrmInternalError {
+  constructor(query) {
+    super(
+      query,
+      "Cannot nest SQL session scopes. Outer scope already has role or setConfig defined."
+    );
+  }
+}
 
 const escape = (value, migration, nested) => {
   const type = typeof value;
@@ -4071,15 +4079,53 @@ class QueryLog {
   }
 }
 
-const processStorageOptions = (query, options) => {
+const hasSqlSessionContextOptions = (options) => {
+  return options.role !== void 0 || options.setConfig !== void 0;
+};
+const hasActiveSqlSessionContext = (state) => {
+  if (!state) return false;
+  return state.role !== void 0 || state.setConfig !== void 0;
+};
+const sqlSessionContextNormalizeSetConfig = (setConfig) => {
+  return Object.fromEntries(
+    Object.entries(setConfig).map(([key, value]) => [key, String(value)])
+  );
+};
+const sqlSessionContextSetStorageOptions = (query, state, options, result) => {
+  if (hasSqlSessionContextOptions(options) && hasActiveSqlSessionContext(state)) {
+    throw new NestedSqlSessionError(query);
+  }
+  if (options.role !== void 0) {
+    result.role = options.role;
+  }
+  if (options.setConfig) {
+    result.setConfig = sqlSessionContextNormalizeSetConfig(options.setConfig);
+  }
+};
+const sqlSessionContextMergeStorageState = (state, options) => {
+  if (!options) return state;
+  return {
+    role: options.role ?? state?.role,
+    setConfig: options.setConfig ?? state?.setConfig
+  };
+};
+const sqlSessionContextGetStateFromAsyncState = (state) => {
+  return state?.role || state?.setConfig ? state : void 0;
+};
+
+const processStorageOptions = (query, state, options) => {
   let log;
   if (options.log !== void 0 && !query.q.log) {
     log = logParamToLogObject(query.q.logger, options.log);
   }
-  return log || "schema" in options ? {
-    log,
-    schema: options.schema
-  } : void 0;
+  const result = {};
+  if (log) result.log = log;
+  if ("schema" in options) result.schema = options.schema;
+  sqlSessionContextSetStorageOptions(query, state, options, result);
+  if (result.log === void 0 && result.schema === void 0 && result.role === void 0 && result.setConfig === void 0) {
+    return void 0;
+  }
+  return result;
 };
 let currentDefaultSchema;
 const setCurrentDefaultSchema = (schema) => {
@@ -4089,14 +4135,15 @@ const getQuerySchema$1 = (query) => query.q.schema || currentDefaultSchema;
 class QueryStorage {
   async withOptions(options, cb) {
     const state = this.internal.asyncStorage.getStore();
-    const opts = processStorageOptions(this, options);
-    return this.internal.asyncStorage.run(
-      {
-        ...state,
-        ...opts
-      },
-      cb
-    );
+    const opts = processStorageOptions(this, state, options);
+    const sqlSessionState = sqlSessionContextMergeStorageState(state, opts);
+    const newState = opts ? {
+      ...state,
+      log: opts.log ?? state?.log,
+      schema: opts.schema ?? state?.schema,
+      ...sqlSessionState
+    } : void 0;
+    return !opts ? state ? this.internal.asyncStorage.run(state, cb) : cb() : this.internal.asyncStorage.run(newState, cb);
   }
 }
 
@@ -4146,7 +4193,7 @@ class QueryTransaction {
     const sql = {
       values: emptyArray
     };
-    const opts = processStorageOptions(this, options);
+    const opts = processStorageOptions(this, void 0, options);
     const log = opts?.log || this.q.log;
     let logData;
     let state = this.internal.asyncStorage.getStore();
@@ -4948,7 +4995,8 @@ const then = async (q, adapter, state, beforeHooks, afterHooks, afterSaveHooks,
         method,
         sql,
         startingSavepoint,
-        releasingSavepoint
+        releasingSavepoint,
+        sqlSessionContextGetStateFromAsyncState(state)
       );
       const { runAfterQuery } = sql;
       if (log) {
@@ -4994,7 +5042,8 @@ const then = async (q, adapter, state, beforeHooks, afterHooks, afterSaveHooks,
             queryMethod,
             sql,
             i === 0 ? startingSavepoint : void 0,
-            i === last ? releasingSavepoint : void 0
+            i === last ? releasingSavepoint : void 0,
+            sqlSessionContextGetStateFromAsyncState(state)
           );
           if (queryResult) {
             queryResult.rowCount += result2.rowCount;
@@ -5286,12 +5335,13 @@ const then = async (q, adapter, state, beforeHooks, afterHooks, afterSaveHooks,
 const setCatchingSavepoint = (catchTrx) => {
   return catchTrx ? `s${catchTrx.catchI = (catchTrx.catchI || 0) + 1}` : void 0;
 };
-const execQuery = (adapter, method, sql, startingSavepoint, releasingSavepoint) => {
+const execQuery = (adapter, method, sql, startingSavepoint, releasingSavepoint, sqlSessionState) => {
   return adapter[method](
     sql.text,
     sql.values,
     startingSavepoint,
-    releasingSavepoint
+    releasingSavepoint,
+    sqlSessionState
   ).then((result) => {
     if (result.rowCount && !result.rows.length) {
       result.rows.length = result.rowCount;
@@ -12890,16 +12940,12 @@ class QueryCreate {
    *   )
    *   .from('b');
    * ```
-   *
-   * @param data - data for the record, may have values, raw SQL, queries, relation operations.
    */
   create(data) {
     return _queryCreate(_clone(this), data);
   }
   /**
    * Works exactly as {@link create}, except that it returns inserted row count by default.
-   *
-   * @param data - data for the record, may have values, raw SQL, queries, relation operations.
    */
   insert(data) {
     return _queryInsert(_clone(this), data);
@@ -15801,7 +15847,14 @@ const performQuery = async (q, args, method) => {
   let logData;
   if (log) logData = log.beforeQuery(sql);
   try {
-    const result = await (trx?.transactionAdapter || q.adapterNotInTransaction)[method](sql.text, sql.values);
+    const adapter = trx?.transactionAdapter || q.adapterNotInTransaction;
+    const result = await adapter[method](
+      sql.text,
+      sql.values,
+      void 0,
+      void 0,
+      sqlSessionContextGetStateFromAsyncState(trx)
+    );
     if (log) log.afterQuery(sql, logData);
     return result;
   } catch (err) {