@zenstackhq/runtime 3.0.0-alpha.1 → 3.0.0-alpha.11

package/dist/{contract-DguafRNB.d.cts → contract-BQGRBbOU.d.cts}

@@ -1,12 +1,16 @@
-import { SchemaDef, GetModels, ScalarFields, ForeignKeyFields, GetFields, FieldHasDefault, GetFieldType, FieldIsOptional, RelationFields, RelationFieldType, FieldIsArray, GetEnums, GetEnum, BuiltinType, NonRelationFields, GetModel, FieldDef, GetField, FieldIsRelationArray, RelationInfo, FieldType, FieldIsRelation, DataSourceProviderType, ProcedureDef, DataSourceProvider } from '@zenstackhq/sdk/schema';
 import Decimal, { Decimal as Decimal$1 } from 'decimal.js';
-import { Kysely, Generated, ExpressionBuilder, OperandExpression, SqlBool, SelectQueryBuilder, Expression, ExpressionWrapper, RootOperationNode, QueryResult, UnknownRow, OperationNode, KyselyConfig, SqliteDialectConfig, PostgresDialectConfig } from 'kysely';
-import { Optional } from 'utility-types';
-import { Model } from '@zenstackhq/language/ast';
+import { SchemaDef, GetModels, ScalarFields, ForeignKeyFields, GetFields, FieldHasDefault, GetFieldType, FieldIsOptional, NonRelationFields, RelationFields, FieldIsArray, RelationFieldType, GetField, FieldIsRelation, GetEnums, GetEnum, BuiltinType, GetModel, FieldDef, FieldIsRelationArray, FieldType, RelationInfo, DataSourceProviderType, DataSourceProvider, ProcedureDef } from '@zenstackhq/sdk/schema';
+import { Generated, Kysely, ExpressionBuilder, OperandExpression, SqlBool, SelectQueryBuilder, Expression, ExpressionWrapper, RootOperationNode, QueryResult, UnknownRow, OperationNode, SqliteDialectConfig, PostgresDialectConfig, KyselyConfig } from 'kysely';
 
+type Optional<T extends object, K extends keyof T = keyof T> = Omit<T, K> & Partial<Pick<T, K>>;
 type NullableIf<T, Condition extends boolean> = Condition extends true ? T | null : T;
+type _Preserve = Date | Function | Decimal | Uint8Array | JsonObject | JsonValue;
+type _Depth = [never, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9];
+type Simplify<T, D extends number = 6> = D extends 0 ? T : T extends object ? T extends _Preserve ? T : {
+    [K in keyof T]: Simplify<T[K], _Depth[D]>;
+} & {} : T;
 type WrapType<T, Optional = false, Array = false> = Optional extends true ? T | null : Array extends true ? T[] : T;
-type MapBaseType$1<T> = T extends 'String' ? string : T extends 'Boolean' ? boolean : T extends 'Int' | 'Float' ? number : T extends 'BigInt' ? bigint : T extends 'Decimal' ? Decimal : T extends 'DateTime' ? Date : T extends 'Json' ? JsonValue : unknown;
+type MapBaseType$1<T> = T extends 'String' ? string : T extends 'Boolean' ? boolean : T extends 'Int' | 'Float' ? number : T extends 'BigInt' ? bigint : T extends 'Decimal' ? Decimal : T extends 'DateTime' ? Date : T extends 'Bytes' ? Uint8Array : T extends 'Json' ? JsonValue : unknown;
 type JsonValue = string | number | boolean | null | JsonObject | JsonArray;
 type JsonObject = {
     [key: string]: JsonValue;
@@ -24,10 +28,13 @@ type AtLeast<O extends object, K extends string> = NoExpand<O extends unknown ?
 type Without<T, U> = {
     [P in Exclude<keyof T, keyof U>]?: never;
 };
-type XOR<T, U> = T extends object ? U extends object ? (Without<T, U> & U) | (Without<U, T> & T) : U : T;
+type XOR<T, U> = T extends object ? (U extends object ? (Without<T, U> & U) | (Without<U, T> & T) : U) : T;
 type MaybePromise<T> = T | Promise<T>;
 type PrependParameter<Param, Func> = Func extends (...args: any[]) => infer R ? (p: Param, ...args: Parameters<Func>) => R : never;
 type OrUndefinedIf<T, Condition extends boolean> = Condition extends true ? T | undefined : T;
+type UnwrapTuplePromises<T extends readonly unknown[]> = {
+    [K in keyof T]: Awaited<T[K]>;
+};
 
 type ToKyselySchema<Schema extends SchemaDef> = {
     [Model in GetModels<Schema>]: ToKyselyTable<Schema, Model>;
@@ -45,7 +52,7 @@ type DefaultModelResult<Schema extends SchemaDef, Model extends GetModels<Schema
     [Key in NonRelationFields<Schema, Model> as Key extends keyof Omit ? Omit[Key] extends true ? never : Key : Key]: MapFieldType<Schema, Model, Key>;
 }, Optional, Array>;
 type ModelSelectResult<Schema extends SchemaDef, Model extends GetModels<Schema>, Select, Omit> = {
-    [Key in keyof Select as Select[Key] extends false | undefined ? never : Key extends keyof Omit ? Omit[Key] extends true ? never : Key : Key extends '_count' ? Select[Key] extends SelectCount<Schema, Model> ? Key : never : Key]: Key extends '_count' ? SelectCountResult<Schema, Model, Select[Key]> : Key extends NonRelationFields<Schema, Model> ? MapFieldType<Schema, Model, Key> : Key extends RelationFields<Schema, Model> ? Select[Key] extends FindArgs<Schema, RelationFieldType<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> ? ModelResult<Schema, RelationFieldType<Schema, Model, Key>, Select[Key], FieldIsOptional<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> : DefaultModelResult<Schema, RelationFieldType<Schema, Model, Key>, Omit, FieldIsOptional<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> : never;
+    [Key in keyof Select as Select[Key] extends false | undefined ? never : Key extends keyof Omit ? Omit[Key] extends true ? never : Key : Key extends '_count' ? Select[Key] extends SelectCount<Schema, Model> ? Key : never : Key]: Key extends '_count' ? SelectCountResult<Schema, Model, Select[Key]> : Key extends NonRelationFields<Schema, Model> ? MapFieldType<Schema, Model, Key> : Key extends RelationFields<Schema, Model> ? Select[Key] extends FindArgs<Schema, RelationFieldType<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> ? 'select' extends keyof Select[Key] ? ModelResult<Schema, RelationFieldType<Schema, Model, Key>, Pick<Select[Key], 'select'>, FieldIsOptional<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> : ModelResult<Schema, RelationFieldType<Schema, Model, Key>, Pick<Select[Key], 'include' | 'omit'>, FieldIsOptional<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> : DefaultModelResult<Schema, RelationFieldType<Schema, Model, Key>, Omit, FieldIsOptional<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>> : never;
 };
 type SelectCountResult<Schema extends SchemaDef, Model extends GetModels<Schema>, C> = C extends true ? {
     [Key in RelationFields<Schema, Model> as FieldIsArray<Schema, Model, Key> extends true ? Key : never]: number;
@@ -65,6 +72,7 @@ type ModelResult<Schema extends SchemaDef, Model extends GetModels<Schema>, Args
 } : Args extends {
     omit: infer O;
 } ? DefaultModelResult<Schema, Model, O> : DefaultModelResult<Schema, Model>, Optional, Array>;
+type SimplifiedModelResult<Schema extends SchemaDef, Model extends GetModels<Schema>, Args extends SelectIncludeOmit<Schema, Model, boolean>, Optional = false, Array = false> = Simplify<ModelResult<Schema, Model, Args, Optional, Array>>;
 type BatchResult = {
     count: number;
 };
@@ -90,7 +98,7 @@ type ArrayFilter<T extends string> = {
     hasSome?: MapBaseType$1<T>[];
     isEmpty?: boolean;
 };
-type PrimitiveFilter<T extends string, Nullable extends boolean> = T extends 'String' ? StringFilter<Nullable> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable> : T extends 'Boolean' ? BooleanFilter<Nullable> : T extends 'DateTime' ? DateTimeFilter<Nullable> : T extends 'Json' ? 'Not implemented yet' : never;
+type PrimitiveFilter<T extends string, Nullable extends boolean> = T extends 'String' ? StringFilter<Nullable> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable> : T extends 'Boolean' ? BooleanFilter<Nullable> : T extends 'DateTime' ? DateTimeFilter<Nullable> : T extends 'Bytes' ? BytesFilter<Nullable> : T extends 'Json' ? 'Not implemented yet' : never;
 type CommonPrimitiveFilter<DataType, T extends BuiltinType, Nullable extends boolean> = {
     equals?: NullableIf<DataType, Nullable>;
     in?: DataType[];
@@ -143,13 +151,13 @@ type WhereUniqueInput<Schema extends SchemaDef, Model extends GetModels<Schema>>
         [Key1 in keyof GetModel<Schema, Model>['uniqueFields'][Key]]: GetModel<Schema, Model>['uniqueFields'][Key][Key1] extends Pick<FieldDef, 'type'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key][Key1]> : never;
     };
 } & WhereInput<Schema, Model>, Extract<keyof GetModel<Schema, Model>['uniqueFields'], string>>;
-type OmitFields<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
+type OmitInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     [Key in NonRelationFields<Schema, Model>]?: true;
 };
 type SelectIncludeOmit<Schema extends SchemaDef, Model extends GetModels<Schema>, AllowCount extends boolean> = {
-    select?: Select<Schema, Model, AllowCount, boolean>;
-    include?: Include<Schema, Model>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, AllowCount, boolean>;
+    include?: IncludeInput<Schema, Model>;
+    omit?: OmitInput<Schema, Model>;
 };
 type Distinct<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     distinct?: OrArray<NonRelationFields<Schema, Model>>;
@@ -157,9 +165,9 @@ type Distinct<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
 type Cursor<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     cursor?: WhereUniqueInput<Schema, Model>;
 };
-type Select<Schema extends SchemaDef, Model extends GetModels<Schema>, AllowCount extends Boolean, AllowRelation extends boolean = true> = {
+type SelectInput<Schema extends SchemaDef, Model extends GetModels<Schema>, AllowCount extends boolean = true, AllowRelation extends boolean = true> = {
     [Key in NonRelationFields<Schema, Model>]?: true;
-} & (AllowRelation extends true ? Include<Schema, Model> : {}) & // relation fields
+} & (AllowRelation extends true ? IncludeInput<Schema, Model> : {}) & // relation fields
 (AllowCount extends true ? {
     _count?: SelectCount<Schema, Model>;
 } : {});
@@ -170,7 +178,7 @@ type SelectCount<Schema extends SchemaDef, Model extends GetModels<Schema>> = tr
         };
     };
 };
-type Include<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
+type IncludeInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     [Key in RelationFields<Schema, Model>]?: boolean | FindArgs<Schema, RelationFieldType<Schema, Model, Key>, FieldIsArray<Schema, Model, Key>, FieldIsArray<Schema, Model, Key> extends true ? true : FieldIsOptional<Schema, Model, Key> extends true ? true : false>;
 };
 type Subset<T, U> = {
@@ -202,7 +210,7 @@ type OptionalFieldsForCreate<Schema extends SchemaDef, Model extends GetModels<S
 };
 type GetRelation<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetFields<Schema, Model>> = GetField<Schema, Model, Field>['relation'];
 type OppositeRelation<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetFields<Schema, Model>, FT = FieldType<Schema, Model, Field>> = FT extends GetModels<Schema> ? GetRelation<Schema, Model, Field> extends RelationInfo ? GetRelation<Schema, Model, Field>['opposite'] extends GetFields<Schema, FT> ? Schema['models'][FT]['fields'][GetRelation<Schema, Model, Field>['opposite']]['relation'] : never : never : never;
-type OppositeRelationFields<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetFields<Schema, Model>, Opposite = OppositeRelation<Schema, Model, Field>> = Opposite extends RelationInfo ? Opposite['fields'] extends string[] ? Opposite['fields'] : [] : [];
+type OppositeRelationFields<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetFields<Schema, Model>, Opposite = OppositeRelation<Schema, Model, Field>> = Opposite extends RelationInfo ? (Opposite['fields'] extends string[] ? Opposite['fields'] : []) : [];
 type OppositeRelationAndFK<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetFields<Schema, Model>, FT = FieldType<Schema, Model, Field>, Relation = GetField<Schema, Model, Field>['relation'], Opposite = Relation extends RelationInfo ? Relation['opposite'] : never> = FT extends GetModels<Schema> ? Opposite extends GetFields<Schema, FT> ? Opposite | OppositeRelationFields<Schema, Model, Field>[number] : never : never;
 type FindArgs<Schema extends SchemaDef, Model extends GetModels<Schema>, Collection extends boolean, AllowFilter extends boolean = true> = (Collection extends true ? {
     skip?: number;
@@ -211,19 +219,21 @@ type FindArgs<Schema extends SchemaDef, Model extends GetModels<Schema>, Collect
 } : {}) & (AllowFilter extends true ? {
     where?: WhereInput<Schema, Model>;
 } : {}) & SelectIncludeOmit<Schema, Model, Collection> & Distinct<Schema, Model> & Cursor<Schema, Model>;
+type FindManyArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = FindArgs<Schema, Model, true>;
+type FindFirstArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = FindArgs<Schema, Model, false>;
 type FindUniqueArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     where?: WhereUniqueInput<Schema, Model>;
 } & SelectIncludeOmit<Schema, Model, true>;
 type CreateArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     data: CreateInput<Schema, Model>;
-    select?: Select<Schema, Model, true>;
-    include?: Include<Schema, Model>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, true>;
+    include?: IncludeInput<Schema, Model>;
+    omit?: OmitInput<Schema, Model>;
 };
-type CreateManyArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateManyPayload<Schema, Model>;
-type CreateManyAndReturnArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateManyPayload<Schema, Model> & {
-    select?: Select<Schema, Model, false, false>;
-    omit?: OmitFields<Schema, Model>;
+type CreateManyArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateManyInput<Schema, Model>;
+type CreateManyAndReturnArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateManyInput<Schema, Model> & {
+    select?: SelectInput<Schema, Model, false, false>;
+    omit?: OmitInput<Schema, Model>;
 };
 type OptionalWrap<Schema extends SchemaDef, Model extends GetModels<Schema>, T extends object> = Optional<T, keyof T & OptionalFieldsForCreate<Schema, Model>>;
 type CreateScalarPayload<Schema extends SchemaDef, Model extends GetModels<Schema>> = OptionalWrap<Schema, Model, {
@@ -244,30 +254,33 @@ type CreateRelationFieldPayload<Schema extends SchemaDef, Model extends GetModel
 type CreateRelationPayload<Schema extends SchemaDef, Model extends GetModels<Schema>> = OptionalWrap<Schema, Model, {
     [Key in RelationFields<Schema, Model>]: CreateRelationFieldPayload<Schema, Model, Key>;
 }>;
-type CreateWithFKInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateScalarPayload<Schema, Model> & CreateFKPayload<Schema, Model>;
+type CreateWithFKInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateScalarPayload<Schema, Model> & CreateFKPayload<Schema, Model> & CreateWithNonOwnedRelationPayload<Schema, Model>;
 type CreateWithRelationInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = CreateScalarPayload<Schema, Model> & CreateRelationPayload<Schema, Model>;
+type CreateWithNonOwnedRelationPayload<Schema extends SchemaDef, Model extends GetModels<Schema>> = OptionalWrap<Schema, Model, {
+    [Key in NonOwnedRelationFields<Schema, Model>]: CreateRelationFieldPayload<Schema, Model, Key>;
+}>;
 type ConnectOrCreatePayload<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = {
     where: WhereUniqueInput<Schema, Model>;
     create: CreateInput<Schema, Model, Without>;
 };
-type CreateManyPayload<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = {
+type CreateManyInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = {
     data: OrArray<Omit<CreateScalarPayload<Schema, Model>, Without> & Omit<CreateFKPayload<Schema, Model>, Without>>;
     skipDuplicates?: boolean;
 };
 type CreateInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = XOR<Omit<CreateWithFKInput<Schema, Model>, Without>, Omit<CreateWithRelationInput<Schema, Model>, Without>>;
 type NestedCreateInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = OrArray<CreateInput<Schema, RelationFieldType<Schema, Model, Field>, OppositeRelationAndFK<Schema, Model, Field>>, FieldIsArray<Schema, Model, Field>>;
-type NestedCreateManyInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = CreateManyPayload<Schema, RelationFieldType<Schema, Model, Field>, OppositeRelationAndFK<Schema, Model, Field>>;
+type NestedCreateManyInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = CreateManyInput<Schema, RelationFieldType<Schema, Model, Field>, OppositeRelationAndFK<Schema, Model, Field>>;
 type UpdateArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     data: UpdateInput<Schema, Model>;
     where: WhereUniqueInput<Schema, Model>;
-    select?: Select<Schema, Model, true>;
-    include?: Include<Schema, Model>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, true>;
+    include?: IncludeInput<Schema, Model>;
+    omit?: OmitInput<Schema, Model>;
 };
 type UpdateManyArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = UpdateManyPayload<Schema, Model>;
 type UpdateManyAndReturnArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = UpdateManyPayload<Schema, Model> & {
-    select?: Select<Schema, Model, false, false>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, false, false>;
+    omit?: OmitInput<Schema, Model>;
 };
 type UpdateManyPayload<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = {
     data: OrArray<UpdateScalarInput<Schema, Model, Without>>;
@@ -278,9 +291,9 @@ type UpsertArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     create: CreateInput<Schema, Model>;
     update: UpdateInput<Schema, Model>;
     where: WhereUniqueInput<Schema, Model>;
-    select?: Select<Schema, Model, true>;
-    include?: Include<Schema, Model>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, true>;
+    include?: IncludeInput<Schema, Model>;
+    omit?: OmitInput<Schema, Model>;
 };
 type UpdateScalarInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Without extends string = never> = Omit<{
     [Key in NonRelationFields<Schema, Model>]?: ScalarUpdatePayload<Schema, Model, Key>;
@@ -325,9 +338,9 @@ type ToOneRelationUpdateInput<Schema extends SchemaDef, Model extends GetModels<
 } : {});
 type DeleteArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     where: WhereUniqueInput<Schema, Model>;
-    select?: Select<Schema, Model, true>;
-    include?: Include<Schema, Model>;
-    omit?: OmitFields<Schema, Model>;
+    select?: SelectInput<Schema, Model, true>;
+    include?: IncludeInput<Schema, Model>;
+    omit?: OmitInput<Schema, Model>;
 };
 type DeleteManyArgs<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
     where?: WhereInput<Schema, Model>;
@@ -448,607 +461,92 @@ type NestedUpsertInput<Schema extends SchemaDef, Model extends GetModels<Schema>
 type NestedUpdateManyInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = OrArray<UpdateManyPayload<Schema, RelationFieldType<Schema, Model, Field>, OppositeRelationAndFK<Schema, Model, Field>>>;
 type NestedDeleteInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = FieldIsArray<Schema, Model, Field> extends true ? OrArray<WhereUniqueInput<Schema, RelationFieldType<Schema, Model, Field>>, true> : boolean | WhereInput<Schema, RelationFieldType<Schema, Model, Field>>;
 type NestedDeleteManyInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>> = OrArray<WhereInput<Schema, RelationFieldType<Schema, Model, Field>, true>>;
-interface ModelOperations<Schema extends SchemaDef, Model extends GetModels<Schema>> {
+type NonOwnedRelationFields<Schema extends SchemaDef, Model extends GetModels<Schema>> = keyof {
+    [Key in RelationFields<Schema, Model> as GetField<Schema, Model, Key>['relation'] extends {
+        references: unknown[];
+    } ? never : Key]: Key;
+};
+
+type AuthType<Schema extends SchemaDef> = string extends GetModels<Schema> ? Record<string, unknown> : Schema['authType'] extends GetModels<Schema> ? Partial<ModelResult<Schema, Schema['authType']>> : never;
+
+/**
+ * Client API methods that are not supported in transactions.
+ */
+declare const TRANSACTION_UNSUPPORTED_METHODS: readonly ["$transaction", "$disconnect", "$use"];
+
+declare abstract class BaseCrudDialect<Schema extends SchemaDef> {
+    protected readonly schema: Schema;
+    protected readonly options: ClientOptions<Schema>;
+    constructor(schema: Schema, options: ClientOptions<Schema>);
+    abstract get provider(): DataSourceProviderType;
+    transformPrimitive(value: unknown, _type: BuiltinType, _forArrayField: boolean): unknown;
+    abstract buildRelationSelection(query: SelectQueryBuilder<any, any, any>, model: string, relationField: string, parentAlias: string, payload: true | FindArgs<Schema, GetModels<Schema>, true>): SelectQueryBuilder<any, any, any>;
+    abstract buildSkipTake(query: SelectQueryBuilder<any, any, any>, skip: number | undefined, take: number | undefined): SelectQueryBuilder<any, any, any>;
+    buildFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, where: boolean | object | undefined): Expression<SqlBool>;
+    protected buildCompositeFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, key: 'AND' | 'OR' | 'NOT', payload: any): Expression<SqlBool>;
+    private buildRelationFilter;
+    private buildToOneRelationFilter;
+    private buildToManyRelationFilter;
+    private buildArrayFilter;
+    buildPrimitiveFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, field: string, fieldDef: FieldDef, payload: any): Expression<SqlBool>;
+    private buildLiteralFilter;
+    private buildStandardFilter;
+    private buildStringFilter;
+    private prepStringCasing;
+    private buildNumberFilter;
+    private buildBooleanFilter;
+    private buildDateTimeFilter;
+    private buildBytesFilter;
+    private buildEnumFilter;
+    buildOrderBy(query: SelectQueryBuilder<any, any, any>, model: string, modelAlias: string, orderBy: OrArray<OrderBy<Schema, GetModels<Schema>, boolean, boolean>> | undefined, useDefaultIfEmpty: boolean, negated: boolean): SelectQueryBuilder<any, any, any>;
+    private negateSort;
+    true(eb: ExpressionBuilder<any, any>): Expression<SqlBool>;
+    false(eb: ExpressionBuilder<any, any>): Expression<SqlBool>;
+    isTrue(expression: Expression<SqlBool>): boolean;
+    isFalse(expression: Expression<SqlBool>): boolean;
+    protected and(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): Expression<SqlBool>;
+    protected or(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): Expression<SqlBool>;
+    protected not(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): ExpressionWrapper<any, any, SqlBool>;
     /**
-     * Returns a list of entities.
-     * @param args - query args
-     * @returns a list of entities
-     *
-     * @example
-     * ```ts
-     * // find all users and return all scalar fields
-     * await client.user.findMany();
-     *
-     * // find all users with name 'Alex'
-     * await client.user.findMany({
-     *     where: {
-     *         name: 'Alex'
-     *     }
-     * });
-     *
-     * // select fields
-     * await client.user.findMany({
-     *     select: {
-     *         name: true,
-     *         email: true,
-     *     }
-     * }); // result: `Array<{ name: string, email: string }>`
-     *
-     * // omit fields
-     * await client.user.findMany({
-     *     omit: {
-     *         name: true,
-     *     }
-     * }); // result: `Array<{ id: number; email: string; ... }>`
-     *
-     * // include relations (and all scalar fields)
-     * await client.user.findMany({
-     *     include: {
-     *         posts: true,
-     *     }
-     * }); // result: `Array<{ ...; posts: Post[] }>`
-     *
-     * // include relations with filter
-     * await client.user.findMany({
-     *     include: {
-     *         posts: {
-     *             where: {
-     *                 published: true
-     *             }
-     *         }
-     *     }
-     * });
-     *
-     * // pagination and sorting
-     * await client.user.findMany({
-     *     skip: 10,
-     *     take: 10,
-     *     orderBy: [{ name: 'asc' }, { email: 'desc' }],
-     * });
-     *
-     * // pagination with cursor (https://www.prisma.io/docs/orm/prisma-client/queries/pagination#cursor-based-pagination)
-     * await client.user.findMany({
-     *     cursor: { id: 10 },
-     *     skip: 1,
-     *     take: 10,
-     *     orderBy: { id: 'asc' },
-     * });
-     *
-     * // distinct
-     * await client.user.findMany({
-     *     distinct: ['name']
-     * });
-     *
-     * // count all relations
-     * await client.user.findMany({
-     *     _count: true,
-     * }); // result: `{ _count: { posts: number; ... } }`
-     *
-     * // count selected relations
-     * await client.user.findMany({
-     *     _count: { select: { posts: true } },
-     * }); // result: `{ _count: { posts: number } }`
-     * ```
+     * Builds an Kysely expression that returns a JSON object for the given key-value pairs.
      */
-    findMany<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): Promise<ModelResult<Schema, Model, T>[]>;
+    abstract buildJsonObject(eb: ExpressionBuilder<any, any>, value: Record<string, Expression<unknown>>): ExpressionWrapper<any, any, unknown>;
     /**
-     * Returns a uniquely identified entity.
-     * @param args - query args
-     * @returns a single entity or null if not found
-     * @see {@link findMany}
+     * Builds an Kysely expression that returns the length of an array.
      */
-    findUnique<T extends FindUniqueArgs<Schema, Model>>(args?: SelectSubset<T, FindUniqueArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T> | null>;
+    abstract buildArrayLength(eb: ExpressionBuilder<any, any>, array: Expression<unknown>): ExpressionWrapper<any, any, number>;
     /**
-     * Returns a uniquely identified entity or throws `NotFoundError` if not found.
-     * @param args - query args
-     * @returns a single entity
-     * @see {@link findMany}
+     * Builds an array literal SQL string for the given values.
      */
-    findUniqueOrThrow<T extends FindUniqueArgs<Schema, Model>>(args?: SelectSubset<T, FindUniqueArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>>;
+    abstract buildArrayLiteralSQL(values: unknown[]): string;
     /**
-     * Returns the first entity.
-     * @param args - query args
-     * @returns a single entity or null if not found
-     * @see {@link findMany}
+     * Whether the dialect supports updating with a limit on the number of updated rows.
      */
-    findFirst<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): Promise<ModelResult<Schema, Model, T> | null>;
+    abstract get supportsUpdateWithLimit(): boolean;
     /**
-     * Returns the first entity or throws `NotFoundError` if not found.
-     * @param args - query args
-     * @returns a single entity
-     * @see {@link findMany}
+     * Whether the dialect supports deleting with a limit on the number of deleted rows.
      */
-    findFirstOrThrow<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): Promise<ModelResult<Schema, Model, T>>;
+    abstract get supportsDeleteWithLimit(): boolean;
     /**
-     * Creates a new entity.
-     * @param args - create args
-     * @returns the created entity
-     *
-     * @example
-     * ```ts
-     * // simple create
-     * await client.user.create({
-     *    data: { name: 'Alex', email: 'alex@zenstack.dev' }
-     * });
-     *
-     * // nested create with relation
-     * await client.user.create({
-     *    data: {
-     *        email: 'alex@zenstack.dev',
-     *        posts: { create: { title: 'Hello World' } }
-     *    }
-     * });
-     *
-     * // you can use `select`, `omit`, and `include` to control
-     * // the fields returned by the query, as with `findMany`
-     * await client.user.create({
-     *    data: {
-     *        email: 'alex@zenstack.dev',
-     *        posts: { create: { title: 'Hello World' } }
-     *    },
-     *    include: { posts: true }
-     * }); // result: `{ id: number; posts: Post[] }`
-     *
-     * // connect relations
-     * await client.user.create({
-     *    data: {
-     *        email: 'alex@zenstack.dev',
-     *        posts: { connect: { id: 1 } }
-     *    }
-     * });
-     *
-     * // connect relations, and create if not found
-     * await client.user.create({
-     *    data: {
-     *        email: 'alex@zenstack.dev',
-     *        posts: {
-     *            connectOrCreate: {
-     *                where: { id: 1 },
-     *                create: { title: 'Hello World' }
-     *            }
-     *        }
-     *    }
-     * });
-     * ```
+     * Whether the dialect supports DISTINCT ON.
      */
-    create<T extends CreateArgs<Schema, Model>>(args: SelectSubset<T, CreateArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>>;
+    abstract get supportsDistinctOn(): boolean;
+}
+
+type CrudOperation = 'findMany' | 'findUnique' | 'findFirst' | 'create' | 'createMany' | 'createManyAndReturn' | 'update' | 'updateMany' | 'updateManyAndReturn' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
+
+/**
+ * The result of the hooks interception filter.
+ */
+type MutationInterceptionFilterResult = {
     /**
-     * Creates multiple entities. Only scalar fields are allowed.
-     * @param args - create args
-     * @returns count of created entities: `{ count: number }`
-     *
-     * @example
-     * ```ts
-     * // create multiple entities
-     * await client.user.createMany({
-     *     data: [
-     *         { name: 'Alex', email: 'alex@zenstack.dev' },
-     *         { name: 'John', email: 'john@zenstack.dev' }
-     *     ]
-     * });
-     *
-     * // skip items that cause unique constraint violation
-     * await client.user.createMany({
-     *     data: [
-     *         { name: 'Alex', email: 'alex@zenstack.dev' },
-     *         { name: 'John', email: 'john@zenstack.dev' }
-     *     ],
-     *     skipDuplicates: true
-     * });
-     * ```
+     * Whether to intercept the mutation or not.
      */
-    createMany(args?: CreateManyPayload<Schema, Model>): Promise<BatchResult>;
+    intercept: boolean;
     /**
-     * Creates multiple entities and returns them.
-     * @param args - create args. See {@link createMany} for input. Use
-     * `select` and `omit` to control the fields returned.
-     * @returns the created entities
-     *
-     * @example
-     * ```ts
-     * // create multiple entities and return selected fields
-     * await client.user.createManyAndReturn({
-     *     data: [
-     *         { name: 'Alex', email: 'alex@zenstack.dev' },
-     *         { name: 'John', email: 'john@zenstack.dev' }
-     *     ],
-     *     select: { id: true, email: true }
-     * });
-     * ```
+     * Whether entities should be loaded before the mutation.
      */
-    createManyAndReturn<T extends CreateManyAndReturnArgs<Schema, Model>>(args?: SelectSubset<T, CreateManyAndReturnArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>[]>;
-    /**
-     * Updates a uniquely identified entity.
-     * @param args - update args. See {@link findMany} for how to control
-     * fields and relations returned.
-     * @returns the updated entity. Throws `NotFoundError` if the entity is not found.
-     *
-     * @example
-     * ```ts
-     * // update fields
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { name: 'Alex' }
-     * });
-     *
-     * // connect a relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { posts: { connect: { id: 1 } } }
-     * });
-     *
-     * // connect relation, and create if not found
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *            connectOrCreate: {
-     *                where: { id: 1 },
-     *                create: { title: 'Hello World' }
-     *            }
-     *         }
-     *     }
-     * });
-     *
-     * // create many related entities (only available for one-to-many relations)
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *             createMany: {
-     *                 data: [{ title: 'Hello World' }, { title: 'Hello World 2' }],
-     *             }
-     *         }
-     *     }
-     * });
-     *
-     * // disconnect a one-to-many relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { posts: { disconnect: { id: 1 } } }
-     * });
-     *
-     * // disconnect a one-to-one relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { profile: { disconnect: true } }
-     * });
-     *
-     * // replace a relation (only available for one-to-many relations)
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *             set: [{ id: 1 }, { id: 2 }]
-     *         }
-     *     }
-     * });
-     *
-     * // update a relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *             update: { where: { id: 1 }, data: { title: 'Hello World' } }
-     *         }
-     *     }
-     * });
-     *
-     * // upsert a relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *             upsert: {
-     *                 where: { id: 1 },
-     *                 create: { title: 'Hello World' },
-     *                 update: { title: 'Hello World' }
-     *             }
-     *         }
-     *     }
-     * });
-     *
-     * // update many related entities (only available for one-to-many relations)
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: {
-     *         posts: {
-     *             updateMany: {
-     *                 where: { published: true },
-     *                 data: { title: 'Hello World' }
-     *             }
-     *         }
-     *     }
-     * });
-     *
-     * // delete a one-to-many relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { posts: { delete: { id: 1 } } }
-     * });
-     *
-     * // delete a one-to-one relation
-     * await client.user.update({
-     *     where: { id: 1 },
-     *     data: { profile: { delete: true } }
-     * });
-     * ```
-     */
-    update<T extends UpdateArgs<Schema, Model>>(args: SelectSubset<T, UpdateArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>>;
-    /**
-     * Updates multiple entities.
-     * @param args - update args. Only scalar fields are allowed for data.
-     * @returns count of updated entities: `{ count: number }`
-     *
-     * @example
-     * ```ts
-     * // update many entities
-     * await client.user.updateMany({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     data: { role: 'ADMIN' }
-     * });
-     *
-     * // limit the number of updated entities
-     * await client.user.updateMany({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     data: { role: 'ADMIN' },
-     *     limit: 10
-     * });
-     */
-    updateMany<T extends UpdateManyArgs<Schema, Model>>(args: Subset<T, UpdateManyArgs<Schema, Model>>): Promise<BatchResult>;
-    /**
-     * Updates multiple entities and returns them.
-     * @param args - update args. Only scalar fields are allowed for data.
-     * @returns the updated entities
-     *
-     * @example
-     * ```ts
-     * // update many entities and return selected fields
-     * await client.user.updateManyAndReturn({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     data: { role: 'ADMIN' },
-     *     select: { id: true, email: true }
-     * }); // result: `Array<{ id: string; email: string }>`
-     *
-     * // limit the number of updated entities
-     * await client.user.updateManyAndReturn({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     data: { role: 'ADMIN' },
-     *     limit: 10
-     * });
-     * ```
-     */
-    updateManyAndReturn<T extends UpdateManyAndReturnArgs<Schema, Model>>(args: Subset<T, UpdateManyAndReturnArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>[]>;
-    /**
-     * Creates or updates an entity.
-     * @param args - upsert args
-     * @returns the upserted entity
-     *
-     * @example
-     * ```ts
-     * // upsert an entity
-     * await client.user.upsert({
-     *     // `where` clause is used to find the entity
-     *     where: { id: 1 },
-     *     // `create` clause is used if the entity is not found
-     *     create: { email: 'alex@zenstack.dev', name: 'Alex' },
-     *     // `update` clause is used if the entity is found
-     *     update: { name: 'Alex-new' },
-     *     // `select` and `omit` can be used to control the returned fields
-     *     ...
-     * });
-     * ```
-     */
-    upsert<T extends UpsertArgs<Schema, Model>>(args: SelectSubset<T, UpsertArgs<Schema, Model>>): Promise<ModelResult<Schema, Model, T>>;
-    /**
-     * Deletes a uniquely identifiable entity.
-     * @param args - delete args
-     * @returns the deleted entity. Throws `NotFoundError` if the entity is not found.
-     *
-     * @example
-     * ```ts
-     * // delete an entity
-     * await client.user.delete({
-     *     where: { id: 1 }
-     * });
-     *
-     * // delete an entity and return selected fields
-     * await client.user.delete({
-     *     where: { id: 1 },
-     *     select: { id: true, email: true }
-     * }); // result: `{ id: string; email: string }`
-     * ```
-     */
-    delete<T extends DeleteArgs<Schema, Model>>(args: SelectSubset<T, DeleteArgs<Schema, Model>>): Promise<ModelResult<Schema, Model>>;
-    /**
-     * Deletes multiple entities.
-     * @param args - delete args
-     * @returns count of deleted entities: `{ count: number }`
-     *
-     * @example
-     * ```ts
-     * // delete many entities
-     * await client.user.deleteMany({
-     *     where: { email: { endsWith: '@zenstack.dev' } }
-     * });
-     *
-     * // limit the number of deleted entities
-     * await client.user.deleteMany({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     limit: 10
-     * });
-     * ```
-     */
-    deleteMany<T extends DeleteManyArgs<Schema, Model>>(args?: Subset<T, DeleteManyArgs<Schema, Model>>): Promise<BatchResult>;
-    /**
-     * Counts rows or field values.
-     * @param args - count args
-     * @returns `number`, or an object containing count of selected relations
-     *
-     * @example
-     * ```ts
-     * // count all
-     * await client.user.count();
-     *
-     * // count with a filter
-     * await client.user.count({ where: { email: { endsWith: '@zenstack.dev' } } });
-     *
-     * // count rows and field values
-     * await client.user.count({
-     *     select: { _all: true, email: true }
-     * }); // result: `{ _all: number, email: number }`
-     */
-    count<T extends CountArgs<Schema, Model>>(args?: Subset<T, CountArgs<Schema, Model>>): Promise<CountResult<Schema, Model, T>>;
-    /**
-     * Aggregates rows.
-     * @param args - aggregation args
-     * @returns an object containing aggregated values
-     *
-     * @example
-     * ```ts
-     * // aggregate rows
-     * await client.profile.aggregate({
-     *     where: { email: { endsWith: '@zenstack.dev' } },
-     *     _count: true,
-     *     _avg: { age: true },
-     *     _sum: { age: true },
-     *     _min: { age: true },
-     *     _max: { age: true }
-     * }); // result: `{ _count: number, _avg: { age: number }, ... }`
-     */
-    aggregate<T extends AggregateArgs<Schema, Model>>(args: Subset<T, AggregateArgs<Schema, Model>>): Promise<AggregateResult<Schema, Model, T>>;
-    /**
-     * Groups rows by columns.
-     * @param args - groupBy args
-     * @returns an object containing grouped values
-     *
-     * @example
-     * ```ts
-     * // group by a field
-     * await client.profile.groupBy({
-     *     by: 'country',
-     *     _count: true
-     * }); // result: `Array<{ country: string, _count: number }>`
-     *
-     * // group by multiple fields
-     * await client.profile.groupBy({
-     *     by: ['country', 'city'],
-     *     _count: true
-     * }); // result: `Array<{ country: string, city: string, _count: number }>`
-     *
-     * // group by with sorting, the `orderBy` fields must be in the `by` list
-     * await client.profile.groupBy({
-     *     by: 'country',
-     *     orderBy: { country: 'desc' }
-     * });
-     *
-     * // group by with having (post-aggregation filter), the `having` fields must
-     * // be in the `by` list
-     * await client.profile.groupBy({
-     *     by: 'country',
-     *     having: { country: 'US' }
-     * });
-     */
-    groupBy<T extends GroupByArgs<Schema, Model>>(args: Subset<T, GroupByArgs<Schema, Model>>): Promise<GroupByResult<Schema, Model, T>>;
-}
-
-type AuthType<Schema extends SchemaDef> = string extends GetModels<Schema> ? Record<string, unknown> : Schema['authType'] extends GetModels<Schema> ? Partial<ModelResult<Schema, Schema['authType']>> : never;
-
-declare abstract class BaseCrudDialect<Schema extends SchemaDef> {
-    protected readonly schema: Schema;
-    protected readonly options: ClientOptions<Schema>;
-    constructor(schema: Schema, options: ClientOptions<Schema>);
-    abstract get provider(): DataSourceProviderType;
-    transformPrimitive(value: unknown, _type: BuiltinType): unknown;
-    abstract buildRelationSelection(query: SelectQueryBuilder<any, any, any>, model: string, relationField: string, parentAlias: string, payload: true | FindArgs<Schema, GetModels<Schema>, true>): SelectQueryBuilder<any, any, any>;
-    abstract buildSkipTake(query: SelectQueryBuilder<any, any, any>, skip: number | undefined, take: number | undefined): SelectQueryBuilder<any, any, any>;
-    buildFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, where: boolean | object | undefined): Expression<SqlBool>;
-    protected buildCompositeFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, key: 'AND' | 'OR' | 'NOT', payload: any): Expression<SqlBool>;
-    private buildRelationFilter;
-    private buildToOneRelationFilter;
-    private buildToManyRelationFilter;
-    private buildArrayFilter;
-    buildPrimitiveFilter(eb: ExpressionBuilder<any, any>, model: string, modelAlias: string, field: string, fieldDef: FieldDef, payload: any): Expression<SqlBool>;
-    private buildLiteralFilter;
-    private buildStandardFilter;
-    private buildStringFilter;
-    private prepStringCasing;
-    private buildNumberFilter;
-    private buildBooleanFilter;
-    private buildDateTimeFilter;
-    private buildBytesFilter;
-    private buildEnumFilter;
-    buildOrderBy(query: SelectQueryBuilder<any, any, any>, model: string, modelAlias: string, orderBy: OrArray<OrderBy<Schema, GetModels<Schema>, boolean, boolean>> | undefined, useDefaultIfEmpty: boolean, negated: boolean): SelectQueryBuilder<any, any, any>;
-    private negateSort;
-    true(eb: ExpressionBuilder<any, any>): Expression<SqlBool>;
-    false(eb: ExpressionBuilder<any, any>): Expression<SqlBool>;
-    isTrue(expression: Expression<SqlBool>): boolean;
-    isFalse(expression: Expression<SqlBool>): boolean;
-    protected and(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): Expression<SqlBool>;
-    protected or(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): Expression<SqlBool>;
-    protected not(eb: ExpressionBuilder<any, any>, ...args: Expression<SqlBool>[]): ExpressionWrapper<any, any, SqlBool>;
-    /**
-     * Builds an Kysely expression that returns a JSON object for the given key-value pairs.
-     */
-    abstract buildJsonObject(eb: ExpressionBuilder<any, any>, value: Record<string, Expression<unknown>>): ExpressionWrapper<any, any, unknown>;
-    /**
-     * Builds an Kysely expression that returns the length of an array.
-     */
-    abstract buildArrayLength(eb: ExpressionBuilder<any, any>, array: Expression<unknown>): ExpressionWrapper<any, any, number>;
-    /**
-     * Builds an array literal SQL string for the given values.
-     */
-    abstract buildArrayLiteralSQL(values: unknown[]): string;
-    /**
-     * Whether the dialect supports updating with a limit on the number of updated rows.
-     */
-    abstract get supportsUpdateWithLimit(): boolean;
-    /**
-     * Whether the dialect supports deleting with a limit on the number of deleted rows.
-     */
-    abstract get supportsDeleteWithLimit(): boolean;
-    /**
-     * Whether the dialect supports DISTINCT ON.
-     */
-    abstract get supportsDistinctOn(): boolean;
-}
-
-type CrudOperation = 'findMany' | 'findUnique' | 'findFirst' | 'create' | 'createMany' | 'createManyAndReturn' | 'update' | 'updateMany' | 'updateManyAndReturn' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
-
-type QueryContext<Schema extends SchemaDef> = {
-    /**
-     * The ZenStack client that's invoking the plugin.
-     */
-    client: ClientContract<Schema>;
-    /**
-     * The model that is being queried.
-     */
-    model: GetModels<Schema>;
-    /**
-     * The query operation that is being performed.
-     */
-    operation: CrudOperation;
-    /**
-     * The query arguments.
-     */
-    queryArgs: unknown;
-};
-/**
- * The result of the hooks interception filter.
- */
-type MutationInterceptionFilterResult = {
-    /**
-     * Whether to intercept the mutation or not.
-     */
-    intercept: boolean;
-    /**
-     * Whether to use a transaction for the mutation.
-     */
-    useTransactionForMutation?: boolean;
-    /**
-     * Whether entities should be loaded before the mutation.
-     */
-    loadBeforeMutationEntity?: boolean;
+    loadBeforeMutationEntity?: boolean;
     /**
      * Whether entities should be loaded after the mutation.
      */
@@ -1068,9 +566,6 @@ type MutationHooksArgs<Schema extends SchemaDef> = {
      */
     queryNode: OperationNode;
 };
-type OnQueryArgs<Schema extends SchemaDef> = QueryContext<Schema> & {
-    proceed: ProceedQueryFunction<Schema>;
-};
 type PluginBeforeEntityMutationArgs<Schema extends SchemaDef> = MutationHooksArgs<Schema> & {
     entities?: Record<string, unknown>[];
 };
@@ -1078,16 +573,12 @@ type PluginAfterEntityMutationArgs<Schema extends SchemaDef> = MutationHooksArgs
     beforeMutationEntities?: Record<string, unknown>[];
     afterMutationEntities?: Record<string, unknown>[];
 };
-type ProceedQueryFunction<Schema extends SchemaDef> = (queryArgs: unknown, tx?: ClientContract<Schema>) => Promise<unknown>;
-type OnKyselyQueryTransactionCallback = (proceed: ProceedKyselyQueryFunction) => Promise<QueryResult<any>>;
-type OnKyselyQueryTransaction = (callback: OnKyselyQueryTransactionCallback) => Promise<QueryResult<any>>;
 type OnKyselyQueryArgs<Schema extends SchemaDef> = {
     kysely: ToKysely<Schema>;
     schema: SchemaDef;
     client: ClientContract<Schema>;
     query: RootOperationNode;
     proceed: ProceedKyselyQueryFunction;
-    transaction: OnKyselyQueryTransaction;
 };
 type ProceedKyselyQueryFunction = (query: RootOperationNode) => Promise<QueryResult<any>>;
 /**
@@ -1109,7 +600,7 @@ interface RuntimePlugin<Schema extends SchemaDef = SchemaDef> {
     /**
      * Intercepts an ORM query.
      */
-    onQuery?: (args: OnQueryArgs<Schema>) => Promise<unknown>;
+    onQuery?: OnQueryHooks<Schema>;
     /**
      * Intercepts a Kysely query.
      */
@@ -1132,141 +623,762 @@ interface RuntimePlugin<Schema extends SchemaDef = SchemaDef> {
      * @param args.afterMutationEntity Only available if `loadAfterMutationEntity` is set to true in the
      * return value of {@link RuntimePlugin.mutationInterceptionFilter}.
      */
-    afterEntityMutation?: (args: PluginAfterEntityMutationArgs<Schema>) => MaybePromise<void>;
-}
-type CliGeneratorContext = {
-    model: Model;
-    outputPath: string;
-    tsSchemaFile: string;
-};
-type CliGenerator = (context: CliGeneratorContext) => MaybePromise<void>;
-
-type DialectConfig<Provider extends DataSourceProvider> = Provider['type'] extends 'sqlite' ? Optional<SqliteDialectConfig, 'database'> : Provider extends 'postgresql' ? Optional<PostgresDialectConfig, 'pool'> : never;
-type ZModelFunctionContext<Schema extends SchemaDef> = {
-    dialect: BaseCrudDialect<Schema>;
-    model: GetModels<Schema>;
-    operation: CRUD;
-};
-type ZModelFunction<Schema extends SchemaDef> = (eb: ExpressionBuilder<ToKyselySchema<Schema>, keyof ToKyselySchema<Schema>>, args: Expression<any>[], context: ZModelFunctionContext<Schema>) => Expression<unknown>;
-/**
- * ZenStack client options.
- */
-type ClientOptions<Schema extends SchemaDef> = {
+    afterEntityMutation?: (args: PluginAfterEntityMutationArgs<Schema>) => MaybePromise<void>;
+}
+type OnQueryHooks<Schema extends SchemaDef = SchemaDef> = {
+    [Model in GetModels<Schema> as Uncapitalize<Model>]?: OnQueryOperationHooks<Schema, Model>;
+} & {
+    $allModels?: OnQueryOperationHooks<Schema, GetModels<Schema>>;
+};
+type OnQueryOperationHooks<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
+    [Operation in keyof ModelOperations<Schema, Model>]?: (ctx: OnQueryHookContext<Schema, Model, Operation>) => Promise<Awaited<ReturnType<ModelOperations<Schema, Model>[Operation]>>>;
+} & {
+    $allOperations?: (ctx: {
+        model: Model;
+        operation: CrudOperation;
+        args: unknown;
+        query: (args: unknown) => Promise<unknown>;
+    }) => MaybePromise<unknown>;
+};
+type OnQueryHookContext<Schema extends SchemaDef, Model extends GetModels<Schema>, Operation extends keyof ModelOperations<Schema, Model>> = {
+    /**
+     * The model that is being queried.
+     */
+    model: Model;
+    /**
+     * The operation that is being performed.
+     */
+    operation: Operation;
+    /**
+     * The query arguments.
+     */
+    args: Parameters<ModelOperations<Schema, Model>[Operation]>[0];
+    /**
+     * The query function to proceed with the original query.
+     * It takes the same arguments as the operation method.
+     *
+     * @param args The query arguments.
+     */
+    query: (args: Parameters<ModelOperations<Schema, Model>[Operation]>[0]) => ReturnType<ModelOperations<Schema, Model>[Operation]>;
+    /**
+     * The ZenStack client that is performing the operation.
+     */
+    client: ClientContract<Schema>;
+};
+/**
+ * Defines a ZenStack runtime plugin.
+ */
+declare function definePlugin<Schema extends SchemaDef>(plugin: RuntimePlugin<Schema>): RuntimePlugin<Schema>;
+
+type DialectConfig<Provider extends DataSourceProvider> = Provider['type'] extends 'sqlite' ? Optional<SqliteDialectConfig, 'database'> : Provider['type'] extends 'postgresql' ? Optional<PostgresDialectConfig, 'pool'> : never;
+type ZModelFunctionContext<Schema extends SchemaDef> = {
+    dialect: BaseCrudDialect<Schema>;
+    model: GetModels<Schema>;
+    operation: CRUD;
+};
+type ZModelFunction<Schema extends SchemaDef> = (eb: ExpressionBuilder<ToKyselySchema<Schema>, keyof ToKyselySchema<Schema>>, args: Expression<any>[], context: ZModelFunctionContext<Schema>) => Expression<unknown>;
+/**
+ * ZenStack client options.
+ */
+type ClientOptions<Schema extends SchemaDef> = {
+    /**
+     * Database dialect configuration.
+     */
+    dialectConfig: DialectConfig<Schema['provider']>;
+    /**
+     * Custom function definitions.
+     */
+    functions?: Record<string, ZModelFunction<Schema>>;
+    /**
+     * Plugins.
+     */
+    plugins?: RuntimePlugin<Schema>[];
+    /**
+     * Logging configuration.
+     */
+    log?: KyselyConfig['log'];
+    /**
+     * Debug mode.
+     */
+    debug?: boolean;
+} & (HasComputedFields<Schema> extends true ? {
+    /**
+     * Computed field definitions.
+     */
+    computedFields: ComputedFieldsOptions<Schema>;
+} : {}) & (HasProcedures<Schema> extends true ? {
+    /**
+     * Custom procedure definitions.
+     */
+    procedures: ProceduresOptions<Schema>;
+} : {});
+type ComputedFieldsOptions<Schema extends SchemaDef> = {
+    [Model in GetModels<Schema> as 'computedFields' extends keyof GetModel<Schema, Model> ? Model : never]: {
+        [Field in keyof Schema['models'][Model]['computedFields']]: PrependParameter<ExpressionBuilder<ToKyselySchema<Schema>, Model>, Schema['models'][Model]['computedFields'][Field]>;
+    };
+};
+type HasComputedFields<Schema extends SchemaDef> = string extends GetModels<Schema> ? false : keyof ComputedFieldsOptions<Schema> extends never ? false : true;
+type ProceduresOptions<Schema extends SchemaDef> = Schema extends {
+    procedures: Record<string, ProcedureDef>;
+} ? {
+    [Key in keyof Schema['procedures']]: PrependParameter<ClientContract<Schema>, ProcedureFunc<Schema, Schema['procedures'][Key]>>;
+} : {};
+type HasProcedures<Schema extends SchemaDef> = Schema extends {
+    procedures: Record<string, ProcedureDef>;
+} ? true : false;
+
+/**
+ * A promise that only executes when it's awaited or .then() is called.
+ */
+type ZenStackPromise<Schema extends SchemaDef, T> = Promise<T> & {
+    /**
+     * @private
+     * Callable to get a plain promise.
+     */
+    cb: (txClient?: ClientContract<Schema>) => Promise<T>;
+};
+
+type TransactionUnsupportedMethods = (typeof TRANSACTION_UNSUPPORTED_METHODS)[number];
+/**
+ * Transaction isolation levels.
+ */
+declare enum TransactionIsolationLevel {
+    ReadUncommitted = "read uncommitted",
+    ReadCommitted = "read committed",
+    RepeatableRead = "repeatable read",
+    Serializable = "serializable",
+    Snapshot = "snapshot"
+}
+/**
+ * ZenStack client interface.
+ */
+type ClientContract<Schema extends SchemaDef> = {
+    readonly $schema: Schema;
+    /**
+     * The client options.
+     */
+    readonly $options: ClientOptions<Schema>;
+    /**
+     * Executes a prepared raw query and returns the number of affected rows.
+     * @example
+     * ```
+     * const result = await client.$executeRaw`UPDATE User SET cool = ${true} WHERE email = ${'user@email.com'};`
+     * ```
+     */
+    $executeRaw(query: TemplateStringsArray, ...values: any[]): ZenStackPromise<Schema, number>;
+    /**
+     * Executes a raw query and returns the number of affected rows.
+     * This method is susceptible to SQL injections.
+     * @example
+     * ```
+     * const result = await client.$executeRawUnsafe('UPDATE User SET cool = $1 WHERE email = $2 ;', true, 'user@email.com')
+     * ```
+     */
+    $executeRawUnsafe(query: string, ...values: any[]): ZenStackPromise<Schema, number>;
+    /**
+     * Performs a prepared raw query and returns the `SELECT` data.
+     * @example
+     * ```
+     * const result = await client.$queryRaw`SELECT * FROM User WHERE id = ${1} OR email = ${'user@email.com'};`
+     * ```
+     */
+    $queryRaw<T = unknown>(query: TemplateStringsArray, ...values: any[]): ZenStackPromise<Schema, T>;
+    /**
+     * Performs a raw query and returns the `SELECT` data.
+     * This method is susceptible to SQL injections.
+     * @example
+     * ```
+     * const result = await client.$queryRawUnsafe('SELECT * FROM User WHERE id = $1 OR email = $2;', 1, 'user@email.com')
+     * ```
+     */
+    $queryRawUnsafe<T = unknown>(query: string, ...values: any[]): ZenStackPromise<Schema, T>;
+    /**
+     * The current user identity.
+     */
+    get $auth(): AuthType<Schema> | undefined;
+    /**
+     * Sets the current user identity.
+     */
+    $setAuth(auth: AuthType<Schema> | undefined): ClientContract<Schema>;
+    /**
+     * The Kysely query builder instance.
+     */
+    readonly $qb: ToKysely<Schema>;
+    /**
+     * The raw Kysely query builder without any ZenStack enhancements.
+     */
+    readonly $qbRaw: ToKysely<any>;
+    /**
+     * Starts an interactive transaction.
+     */
+    $transaction<T>(callback: (tx: Omit<ClientContract<Schema>, TransactionUnsupportedMethods>) => Promise<T>, options?: {
+        isolationLevel?: TransactionIsolationLevel;
+    }): Promise<T>;
+    /**
+     * Starts a sequential transaction.
+     */
+    $transaction<P extends ZenStackPromise<Schema, any>[]>(arg: [...P], options?: {
+        isolationLevel?: TransactionIsolationLevel;
+    }): Promise<UnwrapTuplePromises<P>>;
+    /**
+     * Returns a new client with the specified plugin installed.
+     */
+    $use(plugin: RuntimePlugin<Schema>): ClientContract<Schema>;
+    /**
+     * Returns a new client with the specified plugin removed.
+     */
+    $unuse(pluginId: string): ClientContract<Schema>;
+    /**
+     * Returns a new client with all plugins removed.
+     */
+    $unuseAll(): ClientContract<Schema>;
+    /**
+     * Disconnects the underlying Kysely instance from the database.
+     */
+    $disconnect(): Promise<void>;
+    /**
+     * Pushes the schema to the database. For testing purposes only.
+     * @private
+     */
+    $pushSchema(): Promise<void>;
+} & {
+    [Key in GetModels<Schema> as Uncapitalize<Key>]: ModelOperations<Schema, Key>;
+} & Procedures<Schema>;
+type _TypeMap = {
+    String: string;
+    Int: number;
+    Float: number;
+    BigInt: bigint;
+    Decimal: Decimal$1;
+    Boolean: boolean;
+    DateTime: Date;
+};
+type MapType<Schema extends SchemaDef, T extends string> = T extends keyof _TypeMap ? _TypeMap[T] : T extends GetModels<Schema> ? ModelResult<Schema, T> : unknown;
+type Procedures<Schema extends SchemaDef> = Schema['procedures'] extends Record<string, ProcedureDef> ? {
+    $procedures: {
+        [Key in keyof Schema['procedures']]: ProcedureFunc<Schema, Schema['procedures'][Key]>;
+    };
+} : {};
+type ProcedureFunc<Schema extends SchemaDef, Proc extends ProcedureDef> = (...args: MapProcedureParams<Schema, Proc['params']>) => Promise<MapType<Schema, Proc['returnType']>>;
+type MapProcedureParams<Schema extends SchemaDef, Params> = {
+    [P in keyof Params]: Params[P] extends {
+        type: infer U;
+    } ? OrUndefinedIf<MapType<Schema, U & string>, Params[P] extends {
+        optional: true;
+    } ? true : false> : never;
+};
+/**
+ * Creates a new ZenStack client instance.
+ */
+interface ClientConstructor {
+    new <Schema extends SchemaDef>(schema: Schema, options: ClientOptions<Schema>): ClientContract<Schema>;
+}
+/**
+ * CRUD operations.
+ */
+type CRUD = 'create' | 'read' | 'update' | 'delete';
+interface ModelOperations<Schema extends SchemaDef, Model extends GetModels<Schema>> {
+    /**
+     * Returns a list of entities.
+     * @param args - query args
+     * @returns a list of entities
+     *
+     * @example
+     * ```ts
+     * // find all users and return all scalar fields
+     * await client.user.findMany();
+     *
+     * // find all users with name 'Alex'
+     * await client.user.findMany({
+     *     where: {
+     *         name: 'Alex'
+     *     }
+     * });
+     *
+     * // select fields
+     * await client.user.findMany({
+     *     select: {
+     *         name: true,
+     *         email: true,
+     *     }
+     * }); // result: `Array<{ name: string, email: string }>`
+     *
+     * // omit fields
+     * await client.user.findMany({
+     *     omit: {
+     *         name: true,
+     *     }
+     * }); // result: `Array<{ id: number; email: string; ... }>`
+     *
+     * // include relations (and all scalar fields)
+     * await client.user.findMany({
+     *     include: {
+     *         posts: true,
+     *     }
+     * }); // result: `Array<{ ...; posts: Post[] }>`
+     *
+     * // include relations with filter
+     * await client.user.findMany({
+     *     include: {
+     *         posts: {
+     *             where: {
+     *                 published: true
+     *             }
+     *         }
+     *     }
+     * });
+     *
+     * // pagination and sorting
+     * await client.user.findMany({
+     *     skip: 10,
+     *     take: 10,
+     *     orderBy: [{ name: 'asc' }, { email: 'desc' }],
+     * });
+     *
+     * // pagination with cursor (https://www.prisma.io/docs/orm/prisma-client/queries/pagination#cursor-based-pagination)
+     * await client.user.findMany({
+     *     cursor: { id: 10 },
+     *     skip: 1,
+     *     take: 10,
+     *     orderBy: { id: 'asc' },
+     * });
+     *
+     * // distinct
+     * await client.user.findMany({
+     *     distinct: ['name']
+     * });
+     *
+     * // count all relations
+     * await client.user.findMany({
+     *     _count: true,
+     * }); // result: `{ _count: { posts: number; ... } }`
+     *
+     * // count selected relations
+     * await client.user.findMany({
+     *     _count: { select: { posts: true } },
+     * }); // result: `{ _count: { posts: number } }`
+     * ```
+     */
+    findMany<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>[]>;
     /**
-     * Database dialect configuration.
+     * Returns a uniquely identified entity.
+     * @param args - query args
+     * @returns a single entity or null if not found
+     * @see {@link findMany}
      */
-    dialectConfig?: DialectConfig<Schema['provider']>;
+    findUnique<T extends FindUniqueArgs<Schema, Model>>(args?: SelectSubset<T, FindUniqueArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>> | null>;
     /**
-     * Custom function definitions.
+     * Returns a uniquely identified entity or throws `NotFoundError` if not found.
+     * @param args - query args
+     * @returns a single entity
+     * @see {@link findMany}
      */
-    functions?: Record<string, ZModelFunction<Schema>>;
+    findUniqueOrThrow<T extends FindUniqueArgs<Schema, Model>>(args?: SelectSubset<T, FindUniqueArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
     /**
-     * Plugins.
+     * Returns the first entity.
+     * @param args - query args
+     * @returns a single entity or null if not found
+     * @see {@link findMany}
      */
-    plugins?: RuntimePlugin<Schema>[];
+    findFirst<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>> | null>;
     /**
-     * Logging configuration.
+     * Returns the first entity or throws `NotFoundError` if not found.
+     * @param args - query args
+     * @returns a single entity
+     * @see {@link findMany}
      */
-    log?: KyselyConfig['log'];
-} & (HasComputedFields<Schema> extends true ? {
+    findFirstOrThrow<T extends FindArgs<Schema, Model, true>>(args?: SelectSubset<T, FindArgs<Schema, Model, true>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
     /**
-     * Computed field definitions.
+     * Creates a new entity.
+     * @param args - create args
+     * @returns the created entity
+     *
+     * @example
+     * ```ts
+     * // simple create
+     * await client.user.create({
+     *    data: { name: 'Alex', email: 'alex@zenstack.dev' }
+     * });
+     *
+     * // nested create with relation
+     * await client.user.create({
+     *    data: {
+     *        email: 'alex@zenstack.dev',
+     *        posts: { create: { title: 'Hello World' } }
+     *    }
+     * });
+     *
+     * // you can use `select`, `omit`, and `include` to control
+     * // the fields returned by the query, as with `findMany`
+     * await client.user.create({
+     *    data: {
+     *        email: 'alex@zenstack.dev',
+     *        posts: { create: { title: 'Hello World' } }
+     *    },
+     *    include: { posts: true }
+     * }); // result: `{ id: number; posts: Post[] }`
+     *
+     * // connect relations
+     * await client.user.create({
+     *    data: {
+     *        email: 'alex@zenstack.dev',
+     *        posts: { connect: { id: 1 } }
+     *    }
+     * });
+     *
+     * // connect relations, and create if not found
+     * await client.user.create({
+     *    data: {
+     *        email: 'alex@zenstack.dev',
+     *        posts: {
+     *            connectOrCreate: {
+     *                where: { id: 1 },
+     *                create: { title: 'Hello World' }
+     *            }
+     *        }
+     *    }
+     * });
+     * ```
      */
-    computedFields: ComputedFieldsOptions<Schema>;
-} : {}) & (HasProcedures<Schema> extends true ? {
+    create<T extends CreateArgs<Schema, Model>>(args: SelectSubset<T, CreateArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
     /**
-     * Custom procedure definitions.
+     * Creates multiple entities. Only scalar fields are allowed.
+     * @param args - create args
+     * @returns count of created entities: `{ count: number }`
+     *
+     * @example
+     * ```ts
+     * // create multiple entities
+     * await client.user.createMany({
+     *     data: [
+     *         { name: 'Alex', email: 'alex@zenstack.dev' },
+     *         { name: 'John', email: 'john@zenstack.dev' }
+     *     ]
+     * });
+     *
+     * // skip items that cause unique constraint violation
+     * await client.user.createMany({
+     *     data: [
+     *         { name: 'Alex', email: 'alex@zenstack.dev' },
+     *         { name: 'John', email: 'john@zenstack.dev' }
+     *     ],
+     *     skipDuplicates: true
+     * });
+     * ```
      */
-    procedures: ProceduresOptions<Schema>;
-} : {});
-type ComputedFieldsOptions<Schema extends SchemaDef> = {
-    [Model in GetModels<Schema> as 'computedFields' extends keyof GetModel<Schema, Model> ? Model : never]: {
-        [Field in keyof Schema['models'][Model]['computedFields']]: PrependParameter<ExpressionBuilder<ToKyselySchema<Schema>, Model>, Schema['models'][Model]['computedFields'][Field]>;
-    };
-};
-type HasComputedFields<Schema extends SchemaDef> = string extends GetModels<Schema> ? false : keyof ComputedFieldsOptions<Schema> extends never ? false : true;
-type ProceduresOptions<Schema extends SchemaDef> = Schema extends {
-    procedures: Record<string, ProcedureDef>;
-} ? {
-    [Key in keyof Schema['procedures']]: PrependParameter<ClientContract<Schema>, ProcedureFunc<Schema, Schema['procedures'][Key]>>;
-} : {};
-type HasProcedures<Schema extends SchemaDef> = Schema extends {
-    procedures: Record<string, ProcedureDef>;
-} ? true : false;
-
-/**
- * ZenStack client interface.
- */
-type ClientContract<Schema extends SchemaDef> = {
-    readonly $schema: Schema;
+    createMany<T extends CreateManyArgs<Schema, Model>>(args?: SelectSubset<T, CreateManyArgs<Schema, Model>>): ZenStackPromise<Schema, BatchResult>;
     /**
-     * The client options.
+     * Creates multiple entities and returns them.
+     * @param args - create args. See {@link createMany} for input. Use
+     * `select` and `omit` to control the fields returned.
+     * @returns the created entities
+     *
+     * @example
+     * ```ts
+     * // create multiple entities and return selected fields
+     * await client.user.createManyAndReturn({
+     *     data: [
+     *         { name: 'Alex', email: 'alex@zenstack.dev' },
+     *         { name: 'John', email: 'john@zenstack.dev' }
+     *     ],
+     *     select: { id: true, email: true }
+     * });
+     * ```
      */
-    readonly $options: ClientOptions<Schema>;
+    createManyAndReturn<T extends CreateManyAndReturnArgs<Schema, Model>>(args?: SelectSubset<T, CreateManyAndReturnArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>[]>;
     /**
-     * The current user identity.
+     * Updates a uniquely identified entity.
+     * @param args - update args. See {@link findMany} for how to control
+     * fields and relations returned.
+     * @returns the updated entity. Throws `NotFoundError` if the entity is not found.
+     *
+     * @example
+     * ```ts
+     * // update fields
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { name: 'Alex' }
+     * });
+     *
+     * // connect a relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { posts: { connect: { id: 1 } } }
+     * });
+     *
+     * // connect relation, and create if not found
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *            connectOrCreate: {
+     *                where: { id: 1 },
+     *                create: { title: 'Hello World' }
+     *            }
+     *         }
+     *     }
+     * });
+     *
+     * // create many related entities (only available for one-to-many relations)
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *             createMany: {
+     *                 data: [{ title: 'Hello World' }, { title: 'Hello World 2' }],
+     *             }
+     *         }
+     *     }
+     * });
+     *
+     * // disconnect a one-to-many relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { posts: { disconnect: { id: 1 } } }
+     * });
+     *
+     * // disconnect a one-to-one relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { profile: { disconnect: true } }
+     * });
+     *
+     * // replace a relation (only available for one-to-many relations)
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *             set: [{ id: 1 }, { id: 2 }]
+     *         }
+     *     }
+     * });
+     *
+     * // update a relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *             update: { where: { id: 1 }, data: { title: 'Hello World' } }
+     *         }
+     *     }
+     * });
+     *
+     * // upsert a relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *             upsert: {
+     *                 where: { id: 1 },
+     *                 create: { title: 'Hello World' },
+     *                 update: { title: 'Hello World' }
+     *             }
+     *         }
+     *     }
+     * });
+     *
+     * // update many related entities (only available for one-to-many relations)
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: {
+     *         posts: {
+     *             updateMany: {
+     *                 where: { published: true },
+     *                 data: { title: 'Hello World' }
+     *             }
+     *         }
+     *     }
+     * });
+     *
+     * // delete a one-to-many relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { posts: { delete: { id: 1 } } }
+     * });
+     *
+     * // delete a one-to-one relation
+     * await client.user.update({
+     *     where: { id: 1 },
+     *     data: { profile: { delete: true } }
+     * });
+     * ```
      */
-    get $auth(): AuthType<Schema> | undefined;
+    update<T extends UpdateArgs<Schema, Model>>(args: SelectSubset<T, UpdateArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
     /**
-     * Sets the current user identity.
+     * Updates multiple entities.
+     * @param args - update args. Only scalar fields are allowed for data.
+     * @returns count of updated entities: `{ count: number }`
+     *
+     * @example
+     * ```ts
+     * // update many entities
+     * await client.user.updateMany({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     data: { role: 'ADMIN' }
+     * });
+     *
+     * // limit the number of updated entities
+     * await client.user.updateMany({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     data: { role: 'ADMIN' },
+     *     limit: 10
+     * });
      */
-    $setAuth(auth: AuthType<Schema> | undefined): ClientContract<Schema>;
+    updateMany<T extends UpdateManyArgs<Schema, Model>>(args: Subset<T, UpdateManyArgs<Schema, Model>>): ZenStackPromise<Schema, BatchResult>;
     /**
-     * The Kysely query builder instance.
+     * Updates multiple entities and returns them.
+     * @param args - update args. Only scalar fields are allowed for data.
+     * @returns the updated entities
+     *
+     * @example
+     * ```ts
+     * // update many entities and return selected fields
+     * await client.user.updateManyAndReturn({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     data: { role: 'ADMIN' },
+     *     select: { id: true, email: true }
+     * }); // result: `Array<{ id: string; email: string }>`
+     *
+     * // limit the number of updated entities
+     * await client.user.updateManyAndReturn({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     data: { role: 'ADMIN' },
+     *     limit: 10
+     * });
+     * ```
      */
-    readonly $qb: ToKysely<Schema>;
+    updateManyAndReturn<T extends UpdateManyAndReturnArgs<Schema, Model>>(args: Subset<T, UpdateManyAndReturnArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>[]>;
     /**
-     * The raw Kysely query builder without any ZenStack enhancements.
+     * Creates or updates an entity.
+     * @param args - upsert args
+     * @returns the upserted entity
+     *
+     * @example
+     * ```ts
+     * // upsert an entity
+     * await client.user.upsert({
+     *     // `where` clause is used to find the entity
+     *     where: { id: 1 },
+     *     // `create` clause is used if the entity is not found
+     *     create: { email: 'alex@zenstack.dev', name: 'Alex' },
+     *     // `update` clause is used if the entity is found
+     *     update: { name: 'Alex-new' },
+     *     // `select` and `omit` can be used to control the returned fields
+     *     ...
+     * });
+     * ```
      */
-    readonly $qbRaw: ToKysely<any>;
+    upsert<T extends UpsertArgs<Schema, Model>>(args: SelectSubset<T, UpsertArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
     /**
-     * Starts a transaction.
+     * Deletes a uniquely identifiable entity.
+     * @param args - delete args
+     * @returns the deleted entity. Throws `NotFoundError` if the entity is not found.
+     *
+     * @example
+     * ```ts
+     * // delete an entity
+     * await client.user.delete({
+     *     where: { id: 1 }
+     * });
+     *
+     * // delete an entity and return selected fields
+     * await client.user.delete({
+     *     where: { id: 1 },
+     *     select: { id: true, email: true }
+     * }); // result: `{ id: string; email: string }`
+     * ```
      */
-    $transaction<T>(callback: (tx: ClientContract<Schema>) => Promise<T>): Promise<T>;
+    delete<T extends DeleteArgs<Schema, Model>>(args: SelectSubset<T, DeleteArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model>>>;
     /**
-     * Returns a new client with the specified plugin installed.
+     * Deletes multiple entities.
+     * @param args - delete args
+     * @returns count of deleted entities: `{ count: number }`
+     *
+     * @example
+     * ```ts
+     * // delete many entities
+     * await client.user.deleteMany({
+     *     where: { email: { endsWith: '@zenstack.dev' } }
+     * });
+     *
+     * // limit the number of deleted entities
+     * await client.user.deleteMany({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     limit: 10
+     * });
+     * ```
      */
-    $use(plugin: RuntimePlugin<Schema>): ClientContract<Schema>;
+    deleteMany<T extends DeleteManyArgs<Schema, Model>>(args?: Subset<T, DeleteManyArgs<Schema, Model>>): ZenStackPromise<Schema, BatchResult>;
     /**
-     * Returns a new client with all plugins removed.
+     * Counts rows or field values.
+     * @param args - count args
+     * @returns `number`, or an object containing count of selected relations
+     *
+     * @example
+     * ```ts
+     * // count all
+     * await client.user.count();
+     *
+     * // count with a filter
+     * await client.user.count({ where: { email: { endsWith: '@zenstack.dev' } } });
+     *
+     * // count rows and field values
+     * await client.user.count({
+     *     select: { _all: true, email: true }
+     * }); // result: `{ _all: number, email: number }`
      */
-    $unuseAll(): ClientContract<Schema>;
+    count<T extends CountArgs<Schema, Model>>(args?: Subset<T, CountArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<CountResult<Schema, Model, T>>>;
     /**
-     * Disconnects the underlying Kysely instance from the database.
+     * Aggregates rows.
+     * @param args - aggregation args
+     * @returns an object containing aggregated values
+     *
+     * @example
+     * ```ts
+     * // aggregate rows
+     * await client.profile.aggregate({
+     *     where: { email: { endsWith: '@zenstack.dev' } },
+     *     _count: true,
+     *     _avg: { age: true },
+     *     _sum: { age: true },
+     *     _min: { age: true },
+     *     _max: { age: true }
+     * }); // result: `{ _count: number, _avg: { age: number }, ... }`
      */
-    $disconnect(): Promise<void>;
+    aggregate<T extends AggregateArgs<Schema, Model>>(args: Subset<T, AggregateArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<AggregateResult<Schema, Model, T>>>;
     /**
-     * Pushes the schema to the database. For testing purposes only.
-     * @private
+     * Groups rows by columns.
+     * @param args - groupBy args
+     * @returns an object containing grouped values
+     *
+     * @example
+     * ```ts
+     * // group by a field
+     * await client.profile.groupBy({
+     *     by: 'country',
+     *     _count: true
+     * }); // result: `Array<{ country: string, _count: number }>`
+     *
+     * // group by multiple fields
+     * await client.profile.groupBy({
+     *     by: ['country', 'city'],
+     *     _count: true
+     * }); // result: `Array<{ country: string, city: string, _count: number }>`
+     *
+     * // group by with sorting, the `orderBy` fields must be in the `by` list
+     * await client.profile.groupBy({
+     *     by: 'country',
+     *     orderBy: { country: 'desc' }
+     * });
+     *
+     * // group by with having (post-aggregation filter), the `having` fields must
+     * // be in the `by` list
+     * await client.profile.groupBy({
+     *     by: 'country',
+     *     having: { country: 'US' }
+     * });
      */
-    $pushSchema(): Promise<void>;
-} & {
-    [Key in GetModels<Schema> as Uncapitalize<Key>]: ModelOperations<Schema, Key>;
-} & Procedures<Schema>;
-type MapType<Schema extends SchemaDef, T extends string> = T extends 'String' ? string : T extends 'Int' ? number : T extends 'Float' ? number : T extends 'BigInt' ? bigint : T extends 'Decimal' ? Decimal$1 : T extends 'Boolean' ? boolean : T extends 'DateTime' ? Date : T extends GetModels<Schema> ? ModelResult<Schema, T> : unknown;
-type Procedures<Schema extends SchemaDef> = Schema['procedures'] extends Record<string, ProcedureDef> ? {
-    $procedures: {
-        [Key in keyof Schema['procedures']]: ProcedureFunc<Schema, Schema['procedures'][Key]>;
-    };
-} : {};
-type ProcedureFunc<Schema extends SchemaDef, Proc extends ProcedureDef> = (...args: MapProcedureParams<Schema, Proc['params']>) => Promise<MapType<Schema, Proc['returnType']>>;
-type MapProcedureParams<Schema extends SchemaDef, Params> = {
-    [P in keyof Params]: Params[P] extends {
-        type: infer U;
-    } ? OrUndefinedIf<MapType<Schema, U & string>, Params[P] extends {
-        optional: true;
-    } ? true : false> : never;
-};
-/**
- * Creates a new ZenStack client instance.
- */
-interface ClientConstructor {
-    new <Schema extends SchemaDef>(schema: HasComputedFields<Schema> extends false ? Schema : never): ClientContract<Schema>;
-    new <Schema extends SchemaDef>(schema: Schema, options: ClientOptions<Schema>): ClientContract<Schema>;
+    groupBy<T extends GroupByArgs<Schema, Model>>(args: Subset<T, GroupByArgs<Schema, Model>>): ZenStackPromise<Schema, Simplify<GroupByResult<Schema, Model, T>>>;
 }
-/**
- * CRUD operations.
- */
-type CRUD = 'create' | 'read' | 'update' | 'delete';
 
-export type { UpdateInput as A, BatchResult as B, ClientConstructor as C, DateTimeFilter as D, DeleteArgs as E, FindArgs as F, DeleteManyArgs as G, CountArgs as H, CountAggregateInput as I, CountResult as J, AggregateArgs as K, AggregateResult as L, ModelResult as M, NumberFilter as N, OrderBy as O, GroupByArgs as P, GroupByResult as Q, ModelOperations as R, StringFilter as S, ToKysely as T, UpdateArgs as U, RuntimePlugin as V, WhereInput as W, OnKyselyQueryArgs as X, ClientContract as a, ClientOptions as b, CliGenerator as c, CommonPrimitiveFilter as d, BytesFilter as e, BooleanFilter as f, SortOrder as g, NullsOrder as h, WhereUniqueInput as i, SelectIncludeOmit as j, Subset as k, SelectSubset as l, MapFieldType as m, OptionalFieldsForCreate as n, OppositeRelationFields as o, OppositeRelationAndFK as p, FindUniqueArgs as q, CreateArgs as r, CreateManyArgs as s, CreateManyAndReturnArgs as t, CreateInput as u, UpdateManyArgs as v, UpdateManyAndReturnArgs as w, UpsertArgs as x, UpdateScalarInput as y, UpdateRelationInput as z };
+export { type DeleteManyArgs as A, type BatchResult as B, type ClientConstructor as C, type DateTimeFilter as D, type CountArgs as E, type FindArgs as F, type CountResult as G, type AggregateArgs as H, type IncludeInput as I, type JsonArray as J, type AggregateResult as K, type GroupByArgs as L, type ModelResult as M, type NumberFilter as N, type OrderBy as O, type GroupByResult as P, type OnKyselyQueryArgs as Q, type RuntimePlugin as R, type SimplifiedModelResult as S, type ToKysely as T, type UpdateArgs as U, type WhereInput as W, type JsonObject as a, type JsonValue as b, type ClientContract as c, type ClientOptions as d, definePlugin as e, type StringFilter as f, type BytesFilter as g, type BooleanFilter as h, type SortOrder as i, type NullsOrder as j, type WhereUniqueInput as k, type OmitInput as l, type SelectIncludeOmit as m, type SelectInput as n, type Subset as o, type SelectSubset as p, type FindManyArgs as q, type FindFirstArgs as r, type FindUniqueArgs as s, type CreateArgs as t, type CreateManyArgs as u, type CreateManyAndReturnArgs as v, type UpdateManyArgs as w, type UpdateManyAndReturnArgs as x, type UpsertArgs as y, type DeleteArgs as z };