@zenstackhq/runtime 3.0.0-alpha.4 → 3.0.0-alpha.6

package/dist/{contract-BBUC-uVq.d.ts → contract-DW8XGrtV.d.ts}

@@ -1,9 +1,8 @@
-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 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;
@@ -24,7 +23,7 @@ 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;
@@ -157,7 +156,7 @@ 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 Select<Schema extends SchemaDef, Model extends GetModels<Schema>, AllowCount extends boolean, AllowRelation extends boolean = true> = {
     [Key in NonRelationFields<Schema, Model>]?: true;
 } & (AllowRelation extends true ? Include<Schema, Model> : {}) & // relation fields
 (AllowCount extends true ? {
@@ -202,7 +201,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;
@@ -448,603 +447,82 @@ 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 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>;
     /**
-     * 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 to use a transaction for 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;
+    useTransactionForMutation?: boolean;
     /**
      * Whether entities should be loaded before the mutation.
      */
@@ -1068,9 +546,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,7 +553,6 @@ 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> = {
@@ -1109,7 +583,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 +606,695 @@ 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['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> = {
+    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>) => 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.
+     * @param tx Optional transaction client to use for the query.
+     */
+    query: (args: Parameters<ModelOperations<Schema, Model>[Operation]>[0], tx?: ClientContract<Schema>) => 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'];
+} & (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;
+
+/**
+ * ZenStack client interface.
+ */
+type ClientContract<Schema extends SchemaDef> = {
+    readonly $schema: Schema;
+    /**
+     * The client options.
+     */
+    readonly $options: ClientOptions<Schema>;
+    /**
+     * 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 a transaction.
+     */
+    $transaction<T>(callback: (tx: ClientContract<Schema>) => Promise<T>): Promise<T>;
+    /**
+     * 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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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>>): Promise<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 UpdateRelationInput as A, type BatchResult as B, type ClientConstructor as C, type DateTimeFilter as D, type UpdateInput as E, type FindArgs as F, type DeleteArgs as G, type DeleteManyArgs as H, type CountArgs as I, type CountAggregateInput as J, type CountResult as K, type AggregateArgs as L, type ModelResult as M, type NumberFilter as N, type OrderBy as O, type AggregateResult as P, type GroupByArgs as Q, type GroupByResult as R, type StringFilter as S, type ToKysely as T, type UpdateArgs as U, type RuntimePlugin as V, type WhereInput as W, type OnKyselyQueryArgs as X, type ClientContract as a, type ClientOptions as b, type CommonPrimitiveFilter as c, definePlugin as d, type BytesFilter as e, type BooleanFilter as f, type SortOrder as g, type NullsOrder as h, type WhereUniqueInput as i, type SelectIncludeOmit as j, type Subset as k, type SelectSubset as l, type MapFieldType as m, type OptionalFieldsForCreate as n, type OppositeRelationFields as o, type OppositeRelationAndFK as p, type FindUniqueArgs as q, type CreateArgs as r, type CreateManyArgs as s, type CreateManyAndReturnArgs as t, type CreateManyPayload as u, type CreateInput as v, type UpdateManyArgs as w, type UpdateManyAndReturnArgs as x, type UpsertArgs as y, type UpdateScalarInput as z };