@danielhritcu/zenstack-orm 3.5.16 → 3.5.18

package/dist/index.d.cts

@@ -120,6 +120,10 @@ type JsonObject = {
 };
 type JsonArray = ReadonlyArray<JsonValue | null>;
 type JsonNullValues = DbNull | JsonNull | AnyNull;
+declare const __zenstackBrand: unique symbol;
+type Brand<T extends string> = string & {
+    [__zenstackBrand]: T;
+};
 declare class DbNullClass {
     private __brand;
 }
@@ -1201,7 +1205,7 @@ type WhereInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Optio
     OR?: WhereInput<Schema, Model, Options, ScalarOnly>[];
     NOT?: OrArray<WhereInput<Schema, Model, Options, ScalarOnly>>;
 };
-type FieldFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>, Options extends QueryOptions<Schema>, WithAggregations extends boolean, AllowedKinds extends FilterKind = GetSlicedFilterKindsForField<Schema, Model, Field, Options>> = Field extends RelationFields<Schema, Model> ? RelationFilter<Schema, Model, Field, Options, AllowedKinds> : FieldIsArray<Schema, Model, Field> extends true ? ArrayFilter<Schema, GetModelFieldType<Schema, Model, Field>, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetEnums<Schema> ? EnumFilter<Schema, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetModelFieldType<Schema, Model, Field>, FieldIsArray<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, AllowedKinds> : PrimitiveFilter<GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds>;
+type FieldFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>, Options extends QueryOptions<Schema>, WithAggregations extends boolean, AllowedKinds extends FilterKind = GetSlicedFilterKindsForField<Schema, Model, Field, Options>> = Field extends RelationFields<Schema, Model> ? RelationFilter<Schema, Model, Field, Options, AllowedKinds> : FieldIsArray<Schema, Model, Field> extends true ? ArrayFilter<Schema, GetModelFieldType<Schema, Model, Field>, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetEnums<Schema> ? EnumFilter<Schema, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetModelFieldType<Schema, Model, Field>, FieldIsArray<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, AllowedKinds> : PrimitiveFilter<MapModelFieldType<Schema, Model, Field>, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds>;
 type EnumFilter<Schema extends SchemaDef, T extends GetEnums<Schema>, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<keyof GetEnum<Schema, T>, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified enum value.
@@ -1258,7 +1262,7 @@ type ArrayFilter<Schema extends SchemaDef, Type extends string, AllowedKinds ext
     isEmpty?: boolean;
 } : {});
 type MapScalarType<Schema extends SchemaDef, Type extends string> = Type extends GetEnums<Schema> ? keyof GetEnum<Schema, Type> : MapBaseType$1<Type>;
-type PrimitiveFilter<T extends string, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = T extends 'String' ? StringFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable, WithAggregations, AllowedKinds> : T extends 'Boolean' ? BooleanFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'DateTime' ? DateTimeFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Bytes' ? BytesFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Json' ? JsonFilter<AllowedKinds> : never;
+type PrimitiveFilter<DataType, T extends string, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = T extends 'String' ? StringFilter<Nullable, WithAggregations, AllowedKinds, DataType & string> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable, WithAggregations, AllowedKinds, DataType & (number | bigint)> : T extends 'Boolean' ? BooleanFilter<Nullable, WithAggregations, AllowedKinds, DataType & boolean> : T extends 'DateTime' ? DateTimeFilter<Nullable, WithAggregations, AllowedKinds, DataType & (Date | string)> : T extends 'Bytes' ? BytesFilter<Nullable, WithAggregations, AllowedKinds, DataType & (Uint8Array | Buffer)> : T extends 'Json' ? JsonFilter<AllowedKinds> : never;
 type CommonPrimitiveFilter<DataType, T extends BuiltinType, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = ('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
@@ -1297,9 +1301,9 @@ type CommonPrimitiveFilter<DataType, T extends BuiltinType, Nullable extends boo
     /**
      * Builds a negated filter.
      */
-    not?: PrimitiveFilter<T, Nullable, WithAggregations, AllowedKinds>;
+    not?: PrimitiveFilter<DataType, T, Nullable, WithAggregations, AllowedKinds>;
 };
-type StringFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<string, Nullable> : never) | (CommonPrimitiveFilter<string, 'String', Nullable, WithAggregations, AllowedKinds> & ('Like' extends AllowedKinds ? {
+type StringFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends string = string> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, 'String', Nullable, WithAggregations, AllowedKinds> & ('Like' extends AllowedKinds ? {
     /**
      * Checks if the string contains the specified substring.
      */
@@ -1330,7 +1334,7 @@ type StringFilter<Nullable extends boolean, WithAggregations extends boolean, Al
      */
     _max?: StringFilter<false, false, AllowedKinds>;
 } : {}));
-type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<number | bigint, Nullable> : never) | (CommonPrimitiveFilter<number, T, Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
+type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends number | bigint = number | bigint> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, T, Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
      */
@@ -1338,21 +1342,21 @@ type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable ext
     /**
      * Filters against the average value.
      */
-    _avg?: NumberFilter<T, false, false, AllowedKinds>;
+    _avg?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the sum value.
      */
-    _sum?: NumberFilter<T, false, false, AllowedKinds>;
+    _sum?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the minimum value.
      */
-    _min?: NumberFilter<T, false, false, AllowedKinds>;
+    _min?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the maximum value.
      */
-    _max?: NumberFilter<T, false, false, AllowedKinds>;
+    _max?: NumberFilter<T, false, false, AllowedKinds, DataType>;
 } : {}));
-type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<Date | string, Nullable> : never) | (CommonPrimitiveFilter<Date | string, 'DateTime', Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
+type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends Date | string = Date | string> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, 'DateTime', Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
      */
@@ -1366,24 +1370,24 @@ type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean,
      */
     _max?: DateTimeFilter<false, false, AllowedKinds>;
 } : {}));
-type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<Uint8Array | Buffer, Nullable> : never) | (('Equality' extends AllowedKinds ? {
+type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends Uint8Array | Buffer = Uint8Array | Buffer> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
      */
-    equals?: NullableIf<Uint8Array, Nullable>;
+    equals?: NullableIf<DataType, Nullable>;
     /**
      * Checks if the value is in the specified list of values.
      */
-    in?: Uint8Array[];
+    in?: DataType[];
     /**
      * Checks if the value is not in the specified list of values.
      */
-    notIn?: Uint8Array[];
+    notIn?: DataType[];
 } : {}) & {
     /**
      * Builds a negated filter.
      */
-    not?: BytesFilter<Nullable, WithAggregations, AllowedKinds>;
+    not?: BytesFilter<Nullable, WithAggregations, AllowedKinds, DataType>;
 } & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
@@ -1398,16 +1402,16 @@ type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, All
      */
     _max?: BytesFilter<false, false, AllowedKinds>;
 } : {}));
-type BooleanFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<boolean, Nullable> : never) | (('Equality' extends AllowedKinds ? {
+type BooleanFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends boolean = boolean> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
      */
-    equals?: NullableIf<boolean, Nullable>;
+    equals?: NullableIf<DataType, Nullable>;
 } : {}) & {
     /**
      * Builds a negated filter.
      */
-    not?: BooleanFilter<Nullable, WithAggregations, AllowedKinds>;
+    not?: BooleanFilter<Nullable, WithAggregations, AllowedKinds, DataType>;
 } & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
@@ -1477,7 +1481,7 @@ type NonArrayTypedJsonFilter<Schema extends SchemaDef, TypeDefName extends GetTy
     isNot?: TypedJsonFieldsFilter<Schema, TypeDefName, AllowedKinds>;
 } | TypedJsonFieldsFilter<Schema, TypeDefName, AllowedKinds>;
 type TypedJsonFieldsFilter<Schema extends SchemaDef, TypeDefName extends GetTypeDefs<Schema>, AllowedKinds extends FilterKind> = {
-    [Key in GetTypeDefFields<Schema, TypeDefName>]?: GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsArray<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, AllowedKinds> : TypeDefFieldIsArray<Schema, TypeDefName, Key> extends true ? ArrayFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, AllowedKinds> : GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetEnums<Schema> ? EnumFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds> : PrimitiveFilter<GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds>;
+    [Key in GetTypeDefFields<Schema, TypeDefName>]?: GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsArray<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, AllowedKinds> : TypeDefFieldIsArray<Schema, TypeDefName, Key> extends true ? ArrayFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, AllowedKinds> : GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetEnums<Schema> ? EnumFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds> : PrimitiveFilter<MapFieldDefType<Schema, GetTypeDefField<Schema, TypeDefName, Key>>, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds>;
 };
 type SortOrder = 'asc' | 'desc';
 type NullsOrder = 'first' | 'last';
@@ -1523,8 +1527,8 @@ type OrderBy<Schema extends SchemaDef, Model extends GetModels<Schema>, WithRela
     _sum?: SumAvgInput<Schema, Model, SortOrder>;
 }) : {});
 type WhereUniqueInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Options extends QueryOptions<Schema>> = AtLeast<{
-    [Key in keyof GetModel<Schema, Model>['uniqueFields']]?: GetModel<Schema, Model>['uniqueFields'][Key] extends Pick<FieldDef, 'type'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key]> : {
-        [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;
+    [Key in keyof GetModel<Schema, Model>['uniqueFields']]?: GetModel<Schema, Model>['uniqueFields'][Key] extends Pick<FieldDef, 'type' | 'brand'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key]> : {
+        [Key1 in keyof GetModel<Schema, Model>['uniqueFields'][Key]]: GetModel<Schema, Model>['uniqueFields'][Key][Key1] extends Pick<FieldDef, 'type' | 'brand'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key][Key1]> : never;
     };
 } & WhereInput<Schema, Model, Options>, Extract<keyof GetModel<Schema, Model>['uniqueFields'], string>>;
 type OmitInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
@@ -1589,7 +1593,7 @@ type ToOneRelationFilter<Schema extends SchemaDef, Model extends GetModels<Schem
 }, ModelFieldIsOptional<Schema, Model, Field>>;
 type RelationFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>, Options extends QueryOptions<Schema>, AllowedKinds extends FilterKind> = 'Relation' extends AllowedKinds ? FieldIsArray<Schema, Model, Field> extends true ? ToManyRelationFilter<Schema, Model, Field, Options> : ToOneRelationFilter<Schema, Model, Field, Options> : never;
 type MapModelFieldType<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>> = MapFieldDefType<Schema, GetModelField<Schema, Model, Field>>;
-type MapFieldDefType<Schema extends SchemaDef, T extends Pick<FieldDef, 'type' | 'optional' | 'array'>, Partial extends boolean = false> = WrapType<T['type'] extends GetEnums<Schema> ? keyof GetEnum<Schema, T['type']> : T['type'] extends GetTypeDefs<Schema> ? TypeDefResult<Schema, T['type'], Partial> & Record<string, unknown> : MapBaseType$1<T['type']>, T['optional'], T['array']>;
+type MapFieldDefType<Schema extends SchemaDef, T extends Pick<FieldDef, 'type' | 'optional' | 'array'> & globalThis.Partial<Pick<FieldDef, 'brand'>>, IsPartial extends boolean = false> = WrapType<T['brand'] extends string ? Brand<T['brand']> : T['type'] extends GetEnums<Schema> ? keyof GetEnum<Schema, T['type']> : T['type'] extends GetTypeDefs<Schema> ? TypeDefResult<Schema, T['type'], IsPartial> & Record<string, unknown> : MapBaseType$1<T['type']>, T['optional'], T['array']>;
 type OptionalFieldsForCreate<Schema extends SchemaDef, Model extends GetModels<Schema>> = keyof {
     [Key in GetModelFields<Schema, Model> as ModelFieldIsOptional<Schema, Model, Key> extends true ? Key : FieldHasDefault<Schema, Model, Key> extends true ? Key : FieldIsArray<Schema, Model, Key> extends true ? Key : GetModelField<Schema, Model, Key>['updatedAt'] extends true | UpdatedAtInfo ? Key : never]: GetModelField<Schema, Model, Key>;
 };
@@ -3482,4 +3486,4 @@ declare namespace schemaUtils {
   export { schemaUtils_ExpressionVisitor as ExpressionVisitor, schemaUtils_MatchingExpressionVisitor as MatchingExpressionVisitor, type schemaUtils_VisitResult as VisitResult };
 }
 
-export { type AfterEntityMutationCallback, type AggregateArgs, type AggregateResult, AllCrudOperations, type AllModelOperations, AllReadOperations, AnyNull, AnyNullClass, type AnyPlugin, type AuthType, BaseCrudDialect, type BatchResult, type BeforeEntityMutationCallback, type BooleanFilter, type BytesFilter, CRUD, CRUD_EXT, type ClientConstructor, type ClientContract, type ClientOptions, type ComputedFieldsOptions, CoreCreateOperations, CoreCrudOperations, CoreDeleteOperations, CoreReadOperations, CoreUpdateOperations, CoreWriteOperations, type CountArgs, type CountResult, type CreateArgs, type CreateInput, type CreateManyAndReturnArgs, type CreateManyArgs, type CreateManyInput, type DateTimeFilter, DbNull, DbNullClass, type DefaultModelResult, type DefaultWhereConfig, type DeleteArgs, type DeleteManyArgs, type EntityMutationHooksDef, type ExistsArgs, type ExtClientMembersBase, type ExtQueryArgsBase, type ExtResultBase, type ExtResultInferenceArgs, type ExtResultSelectOmitFields, type ExtractExtResult, type FilterKind, type FindArgs, type FindFirstArgs, type FindManyArgs, type FindUniqueArgs, type GetAllIncludedOperations, type GetIncludedOperations, type GetProcedure, type GetProcedureNames, type GetProcedureParams, type GetQueryOptions, type GetSlicedFilterKindsForField, type GetSlicedModels, type GetSlicedOperations, type GetSlicedProcedures, type GroupByArgs, type GroupByResult, type HasComputedFields, type HasProcedures, type IncludeInput, InputValidator, type JsonArray, type JsonFilter, JsonNull, JsonNullClass, type JsonNullValues, type JsonObject, type JsonValue, kyselyUtils as KyselyUtils, LOGICAL_COMBINATORS, type MapModelFieldType, type ModelAllowsCreate, type ModelHasRequiredUnsupportedField, type ModelOperations, type ModelResult, type ModelSlicingOptions, type NullsOrder, type NumberFilter, ORMError, ORMErrorReason, type OmitConfig, type OmitInput, type OnKyselyQueryArgs, type OnKyselyQueryCallback, type OnProcedureHookContext, type OperationsRequiringCreate, type OrderBy, type PaginatedCursor, type PaginatedOffset, type PluginAfterEntityMutationArgs, type PluginBeforeEntityMutationArgs, type ProcedureEnvelope, type ProcedureFunc, type ProcedureHandlerFunc, type ProcedureOperations, type ProceduresOptions, type ProceedKyselyQueryFunction, type QueryOptions, queryUtils as QueryUtils, RejectedByPolicyReason, type RuntimePlugin, type SchemaExtResult, schemaUtils as SchemaUtils, type SearchFieldMap, type SearchMode, type SearchPayload, type SearchStrategy, type SelectAwareExtResult, type SelectIncludeOmit, type SelectInput, type SelectSubset, type SimplifiedPlainResult, type SimplifiedResult, type SlicingOptions, type SortOrder, type StringFilter, type Subset, type ToKysely, type TransactionClientContract, TransactionIsolationLevel, type TypeDefResult, type TypedJsonFilter, type UpdateArgs, type UpdateInput, type UpdateManyAndReturnArgs, type UpdateManyArgs, type UpsertArgs, type WhereInput, type WhereUniqueInput, ZENSTACK_QUERY_KIND_SYMBOL, ZENSTACK_QUERY_SYMBOL, type ZModelFunction, type ZModelFunctionContext, ZenStackClient, type ZenStackPromise, type ZenStackQueryKind, createQuerySchemaFactory, definePlugin, getCrudDialect };
+export { type AfterEntityMutationCallback, type AggregateArgs, type AggregateResult, AllCrudOperations, type AllModelOperations, AllReadOperations, AnyNull, AnyNullClass, type AnyPlugin, type AuthType, BaseCrudDialect, type BatchResult, type BeforeEntityMutationCallback, type BooleanFilter, type Brand, type BytesFilter, CRUD, CRUD_EXT, type ClientConstructor, type ClientContract, type ClientOptions, type ComputedFieldsOptions, CoreCreateOperations, CoreCrudOperations, CoreDeleteOperations, CoreReadOperations, CoreUpdateOperations, CoreWriteOperations, type CountArgs, type CountResult, type CreateArgs, type CreateInput, type CreateManyAndReturnArgs, type CreateManyArgs, type CreateManyInput, type DateTimeFilter, DbNull, DbNullClass, type DefaultModelResult, type DefaultWhereConfig, type DeleteArgs, type DeleteManyArgs, type EntityMutationHooksDef, type ExistsArgs, type ExtClientMembersBase, type ExtQueryArgsBase, type ExtResultBase, type ExtResultInferenceArgs, type ExtResultSelectOmitFields, type ExtractExtResult, type FilterKind, type FindArgs, type FindFirstArgs, type FindManyArgs, type FindUniqueArgs, type GetAllIncludedOperations, type GetIncludedOperations, type GetProcedure, type GetProcedureNames, type GetProcedureParams, type GetQueryOptions, type GetSlicedFilterKindsForField, type GetSlicedModels, type GetSlicedOperations, type GetSlicedProcedures, type GroupByArgs, type GroupByResult, type HasComputedFields, type HasProcedures, type IncludeInput, InputValidator, type JsonArray, type JsonFilter, JsonNull, JsonNullClass, type JsonNullValues, type JsonObject, type JsonValue, kyselyUtils as KyselyUtils, LOGICAL_COMBINATORS, type MapModelFieldType, type ModelAllowsCreate, type ModelHasRequiredUnsupportedField, type ModelOperations, type ModelResult, type ModelSlicingOptions, type NullsOrder, type NumberFilter, ORMError, ORMErrorReason, type OmitConfig, type OmitInput, type OnKyselyQueryArgs, type OnKyselyQueryCallback, type OnProcedureHookContext, type OperationsRequiringCreate, type OrderBy, type PaginatedCursor, type PaginatedOffset, type PluginAfterEntityMutationArgs, type PluginBeforeEntityMutationArgs, type ProcedureEnvelope, type ProcedureFunc, type ProcedureHandlerFunc, type ProcedureOperations, type ProceduresOptions, type ProceedKyselyQueryFunction, type QueryOptions, queryUtils as QueryUtils, RejectedByPolicyReason, type RuntimePlugin, type SchemaExtResult, schemaUtils as SchemaUtils, type SearchFieldMap, type SearchMode, type SearchPayload, type SearchStrategy, type SelectAwareExtResult, type SelectIncludeOmit, type SelectInput, type SelectSubset, type SimplifiedPlainResult, type SimplifiedResult, type SlicingOptions, type SortOrder, type StringFilter, type Subset, type ToKysely, type TransactionClientContract, TransactionIsolationLevel, type TypeDefResult, type TypedJsonFilter, type UpdateArgs, type UpdateInput, type UpdateManyAndReturnArgs, type UpdateManyArgs, type UpsertArgs, type WhereInput, type WhereUniqueInput, ZENSTACK_QUERY_KIND_SYMBOL, ZENSTACK_QUERY_SYMBOL, type ZModelFunction, type ZModelFunctionContext, ZenStackClient, type ZenStackPromise, type ZenStackQueryKind, createQuerySchemaFactory, definePlugin, getCrudDialect };

package/dist/index.d.ts

@@ -120,6 +120,10 @@ type JsonObject = {
 };
 type JsonArray = ReadonlyArray<JsonValue | null>;
 type JsonNullValues = DbNull | JsonNull | AnyNull;
+declare const __zenstackBrand: unique symbol;
+type Brand<T extends string> = string & {
+    [__zenstackBrand]: T;
+};
 declare class DbNullClass {
     private __brand;
 }
@@ -1201,7 +1205,7 @@ type WhereInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Optio
     OR?: WhereInput<Schema, Model, Options, ScalarOnly>[];
     NOT?: OrArray<WhereInput<Schema, Model, Options, ScalarOnly>>;
 };
-type FieldFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>, Options extends QueryOptions<Schema>, WithAggregations extends boolean, AllowedKinds extends FilterKind = GetSlicedFilterKindsForField<Schema, Model, Field, Options>> = Field extends RelationFields<Schema, Model> ? RelationFilter<Schema, Model, Field, Options, AllowedKinds> : FieldIsArray<Schema, Model, Field> extends true ? ArrayFilter<Schema, GetModelFieldType<Schema, Model, Field>, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetEnums<Schema> ? EnumFilter<Schema, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetModelFieldType<Schema, Model, Field>, FieldIsArray<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, AllowedKinds> : PrimitiveFilter<GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds>;
+type FieldFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>, Options extends QueryOptions<Schema>, WithAggregations extends boolean, AllowedKinds extends FilterKind = GetSlicedFilterKindsForField<Schema, Model, Field, Options>> = Field extends RelationFields<Schema, Model> ? RelationFilter<Schema, Model, Field, Options, AllowedKinds> : FieldIsArray<Schema, Model, Field> extends true ? ArrayFilter<Schema, GetModelFieldType<Schema, Model, Field>, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetEnums<Schema> ? EnumFilter<Schema, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds> : GetModelFieldType<Schema, Model, Field> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetModelFieldType<Schema, Model, Field>, FieldIsArray<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, AllowedKinds> : PrimitiveFilter<MapModelFieldType<Schema, Model, Field>, GetModelFieldType<Schema, Model, Field>, ModelFieldIsOptional<Schema, Model, Field>, WithAggregations, AllowedKinds>;
 type EnumFilter<Schema extends SchemaDef, T extends GetEnums<Schema>, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<keyof GetEnum<Schema, T>, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified enum value.
@@ -1258,7 +1262,7 @@ type ArrayFilter<Schema extends SchemaDef, Type extends string, AllowedKinds ext
     isEmpty?: boolean;
 } : {});
 type MapScalarType<Schema extends SchemaDef, Type extends string> = Type extends GetEnums<Schema> ? keyof GetEnum<Schema, Type> : MapBaseType$1<Type>;
-type PrimitiveFilter<T extends string, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = T extends 'String' ? StringFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable, WithAggregations, AllowedKinds> : T extends 'Boolean' ? BooleanFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'DateTime' ? DateTimeFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Bytes' ? BytesFilter<Nullable, WithAggregations, AllowedKinds> : T extends 'Json' ? JsonFilter<AllowedKinds> : never;
+type PrimitiveFilter<DataType, T extends string, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = T extends 'String' ? StringFilter<Nullable, WithAggregations, AllowedKinds, DataType & string> : T extends 'Int' | 'Float' | 'Decimal' | 'BigInt' ? NumberFilter<T, Nullable, WithAggregations, AllowedKinds, DataType & (number | bigint)> : T extends 'Boolean' ? BooleanFilter<Nullable, WithAggregations, AllowedKinds, DataType & boolean> : T extends 'DateTime' ? DateTimeFilter<Nullable, WithAggregations, AllowedKinds, DataType & (Date | string)> : T extends 'Bytes' ? BytesFilter<Nullable, WithAggregations, AllowedKinds, DataType & (Uint8Array | Buffer)> : T extends 'Json' ? JsonFilter<AllowedKinds> : never;
 type CommonPrimitiveFilter<DataType, T extends BuiltinType, Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind> = ('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
@@ -1297,9 +1301,9 @@ type CommonPrimitiveFilter<DataType, T extends BuiltinType, Nullable extends boo
     /**
      * Builds a negated filter.
      */
-    not?: PrimitiveFilter<T, Nullable, WithAggregations, AllowedKinds>;
+    not?: PrimitiveFilter<DataType, T, Nullable, WithAggregations, AllowedKinds>;
 };
-type StringFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<string, Nullable> : never) | (CommonPrimitiveFilter<string, 'String', Nullable, WithAggregations, AllowedKinds> & ('Like' extends AllowedKinds ? {
+type StringFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends string = string> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, 'String', Nullable, WithAggregations, AllowedKinds> & ('Like' extends AllowedKinds ? {
     /**
      * Checks if the string contains the specified substring.
      */
@@ -1330,7 +1334,7 @@ type StringFilter<Nullable extends boolean, WithAggregations extends boolean, Al
      */
     _max?: StringFilter<false, false, AllowedKinds>;
 } : {}));
-type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<number | bigint, Nullable> : never) | (CommonPrimitiveFilter<number, T, Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
+type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends number | bigint = number | bigint> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, T, Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
      */
@@ -1338,21 +1342,21 @@ type NumberFilter<T extends 'Int' | 'Float' | 'Decimal' | 'BigInt', Nullable ext
     /**
      * Filters against the average value.
      */
-    _avg?: NumberFilter<T, false, false, AllowedKinds>;
+    _avg?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the sum value.
      */
-    _sum?: NumberFilter<T, false, false, AllowedKinds>;
+    _sum?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the minimum value.
      */
-    _min?: NumberFilter<T, false, false, AllowedKinds>;
+    _min?: NumberFilter<T, false, false, AllowedKinds, DataType>;
     /**
      * Filters against the maximum value.
      */
-    _max?: NumberFilter<T, false, false, AllowedKinds>;
+    _max?: NumberFilter<T, false, false, AllowedKinds, DataType>;
 } : {}));
-type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<Date | string, Nullable> : never) | (CommonPrimitiveFilter<Date | string, 'DateTime', Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
+type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends Date | string = Date | string> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (CommonPrimitiveFilter<DataType, 'DateTime', Nullable, WithAggregations, AllowedKinds> & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
      */
@@ -1366,24 +1370,24 @@ type DateTimeFilter<Nullable extends boolean, WithAggregations extends boolean,
      */
     _max?: DateTimeFilter<false, false, AllowedKinds>;
 } : {}));
-type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<Uint8Array | Buffer, Nullable> : never) | (('Equality' extends AllowedKinds ? {
+type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends Uint8Array | Buffer = Uint8Array | Buffer> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
      */
-    equals?: NullableIf<Uint8Array, Nullable>;
+    equals?: NullableIf<DataType, Nullable>;
     /**
      * Checks if the value is in the specified list of values.
      */
-    in?: Uint8Array[];
+    in?: DataType[];
     /**
      * Checks if the value is not in the specified list of values.
      */
-    notIn?: Uint8Array[];
+    notIn?: DataType[];
 } : {}) & {
     /**
      * Builds a negated filter.
      */
-    not?: BytesFilter<Nullable, WithAggregations, AllowedKinds>;
+    not?: BytesFilter<Nullable, WithAggregations, AllowedKinds, DataType>;
 } & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
@@ -1398,16 +1402,16 @@ type BytesFilter<Nullable extends boolean, WithAggregations extends boolean, All
      */
     _max?: BytesFilter<false, false, AllowedKinds>;
 } : {}));
-type BooleanFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind> = ('Equality' extends AllowedKinds ? NullableIf<boolean, Nullable> : never) | (('Equality' extends AllowedKinds ? {
+type BooleanFilter<Nullable extends boolean, WithAggregations extends boolean, AllowedKinds extends FilterKind = FilterKind, DataType extends boolean = boolean> = ('Equality' extends AllowedKinds ? NullableIf<DataType, Nullable> : never) | (('Equality' extends AllowedKinds ? {
     /**
      * Checks for equality with the specified value.
      */
-    equals?: NullableIf<boolean, Nullable>;
+    equals?: NullableIf<DataType, Nullable>;
 } : {}) & {
     /**
      * Builds a negated filter.
      */
-    not?: BooleanFilter<Nullable, WithAggregations, AllowedKinds>;
+    not?: BooleanFilter<Nullable, WithAggregations, AllowedKinds, DataType>;
 } & (WithAggregations extends true ? {
     /**
      * Filters against the count of records.
@@ -1477,7 +1481,7 @@ type NonArrayTypedJsonFilter<Schema extends SchemaDef, TypeDefName extends GetTy
     isNot?: TypedJsonFieldsFilter<Schema, TypeDefName, AllowedKinds>;
 } | TypedJsonFieldsFilter<Schema, TypeDefName, AllowedKinds>;
 type TypedJsonFieldsFilter<Schema extends SchemaDef, TypeDefName extends GetTypeDefs<Schema>, AllowedKinds extends FilterKind> = {
-    [Key in GetTypeDefFields<Schema, TypeDefName>]?: GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsArray<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, AllowedKinds> : TypeDefFieldIsArray<Schema, TypeDefName, Key> extends true ? ArrayFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, AllowedKinds> : GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetEnums<Schema> ? EnumFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds> : PrimitiveFilter<GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds>;
+    [Key in GetTypeDefFields<Schema, TypeDefName>]?: GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetTypeDefs<Schema> ? TypedJsonFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsArray<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, AllowedKinds> : TypeDefFieldIsArray<Schema, TypeDefName, Key> extends true ? ArrayFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, AllowedKinds> : GetTypeDefFieldType<Schema, TypeDefName, Key> extends GetEnums<Schema> ? EnumFilter<Schema, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds> : PrimitiveFilter<MapFieldDefType<Schema, GetTypeDefField<Schema, TypeDefName, Key>>, GetTypeDefFieldType<Schema, TypeDefName, Key>, TypeDefFieldIsOptional<Schema, TypeDefName, Key>, false, AllowedKinds>;
 };
 type SortOrder = 'asc' | 'desc';
 type NullsOrder = 'first' | 'last';
@@ -1523,8 +1527,8 @@ type OrderBy<Schema extends SchemaDef, Model extends GetModels<Schema>, WithRela
     _sum?: SumAvgInput<Schema, Model, SortOrder>;
 }) : {});
 type WhereUniqueInput<Schema extends SchemaDef, Model extends GetModels<Schema>, Options extends QueryOptions<Schema>> = AtLeast<{
-    [Key in keyof GetModel<Schema, Model>['uniqueFields']]?: GetModel<Schema, Model>['uniqueFields'][Key] extends Pick<FieldDef, 'type'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key]> : {
-        [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;
+    [Key in keyof GetModel<Schema, Model>['uniqueFields']]?: GetModel<Schema, Model>['uniqueFields'][Key] extends Pick<FieldDef, 'type' | 'brand'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key]> : {
+        [Key1 in keyof GetModel<Schema, Model>['uniqueFields'][Key]]: GetModel<Schema, Model>['uniqueFields'][Key][Key1] extends Pick<FieldDef, 'type' | 'brand'> ? MapFieldDefType<Schema, GetModel<Schema, Model>['uniqueFields'][Key][Key1]> : never;
     };
 } & WhereInput<Schema, Model, Options>, Extract<keyof GetModel<Schema, Model>['uniqueFields'], string>>;
 type OmitInput<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
@@ -1589,7 +1593,7 @@ type ToOneRelationFilter<Schema extends SchemaDef, Model extends GetModels<Schem
 }, ModelFieldIsOptional<Schema, Model, Field>>;
 type RelationFilter<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends RelationFields<Schema, Model>, Options extends QueryOptions<Schema>, AllowedKinds extends FilterKind> = 'Relation' extends AllowedKinds ? FieldIsArray<Schema, Model, Field> extends true ? ToManyRelationFilter<Schema, Model, Field, Options> : ToOneRelationFilter<Schema, Model, Field, Options> : never;
 type MapModelFieldType<Schema extends SchemaDef, Model extends GetModels<Schema>, Field extends GetModelFields<Schema, Model>> = MapFieldDefType<Schema, GetModelField<Schema, Model, Field>>;
-type MapFieldDefType<Schema extends SchemaDef, T extends Pick<FieldDef, 'type' | 'optional' | 'array'>, Partial extends boolean = false> = WrapType<T['type'] extends GetEnums<Schema> ? keyof GetEnum<Schema, T['type']> : T['type'] extends GetTypeDefs<Schema> ? TypeDefResult<Schema, T['type'], Partial> & Record<string, unknown> : MapBaseType$1<T['type']>, T['optional'], T['array']>;
+type MapFieldDefType<Schema extends SchemaDef, T extends Pick<FieldDef, 'type' | 'optional' | 'array'> & globalThis.Partial<Pick<FieldDef, 'brand'>>, IsPartial extends boolean = false> = WrapType<T['brand'] extends string ? Brand<T['brand']> : T['type'] extends GetEnums<Schema> ? keyof GetEnum<Schema, T['type']> : T['type'] extends GetTypeDefs<Schema> ? TypeDefResult<Schema, T['type'], IsPartial> & Record<string, unknown> : MapBaseType$1<T['type']>, T['optional'], T['array']>;
 type OptionalFieldsForCreate<Schema extends SchemaDef, Model extends GetModels<Schema>> = keyof {
     [Key in GetModelFields<Schema, Model> as ModelFieldIsOptional<Schema, Model, Key> extends true ? Key : FieldHasDefault<Schema, Model, Key> extends true ? Key : FieldIsArray<Schema, Model, Key> extends true ? Key : GetModelField<Schema, Model, Key>['updatedAt'] extends true | UpdatedAtInfo ? Key : never]: GetModelField<Schema, Model, Key>;
 };
@@ -3482,4 +3486,4 @@ declare namespace schemaUtils {
   export { schemaUtils_ExpressionVisitor as ExpressionVisitor, schemaUtils_MatchingExpressionVisitor as MatchingExpressionVisitor, type schemaUtils_VisitResult as VisitResult };
 }
 
-export { type AfterEntityMutationCallback, type AggregateArgs, type AggregateResult, AllCrudOperations, type AllModelOperations, AllReadOperations, AnyNull, AnyNullClass, type AnyPlugin, type AuthType, BaseCrudDialect, type BatchResult, type BeforeEntityMutationCallback, type BooleanFilter, type BytesFilter, CRUD, CRUD_EXT, type ClientConstructor, type ClientContract, type ClientOptions, type ComputedFieldsOptions, CoreCreateOperations, CoreCrudOperations, CoreDeleteOperations, CoreReadOperations, CoreUpdateOperations, CoreWriteOperations, type CountArgs, type CountResult, type CreateArgs, type CreateInput, type CreateManyAndReturnArgs, type CreateManyArgs, type CreateManyInput, type DateTimeFilter, DbNull, DbNullClass, type DefaultModelResult, type DefaultWhereConfig, type DeleteArgs, type DeleteManyArgs, type EntityMutationHooksDef, type ExistsArgs, type ExtClientMembersBase, type ExtQueryArgsBase, type ExtResultBase, type ExtResultInferenceArgs, type ExtResultSelectOmitFields, type ExtractExtResult, type FilterKind, type FindArgs, type FindFirstArgs, type FindManyArgs, type FindUniqueArgs, type GetAllIncludedOperations, type GetIncludedOperations, type GetProcedure, type GetProcedureNames, type GetProcedureParams, type GetQueryOptions, type GetSlicedFilterKindsForField, type GetSlicedModels, type GetSlicedOperations, type GetSlicedProcedures, type GroupByArgs, type GroupByResult, type HasComputedFields, type HasProcedures, type IncludeInput, InputValidator, type JsonArray, type JsonFilter, JsonNull, JsonNullClass, type JsonNullValues, type JsonObject, type JsonValue, kyselyUtils as KyselyUtils, LOGICAL_COMBINATORS, type MapModelFieldType, type ModelAllowsCreate, type ModelHasRequiredUnsupportedField, type ModelOperations, type ModelResult, type ModelSlicingOptions, type NullsOrder, type NumberFilter, ORMError, ORMErrorReason, type OmitConfig, type OmitInput, type OnKyselyQueryArgs, type OnKyselyQueryCallback, type OnProcedureHookContext, type OperationsRequiringCreate, type OrderBy, type PaginatedCursor, type PaginatedOffset, type PluginAfterEntityMutationArgs, type PluginBeforeEntityMutationArgs, type ProcedureEnvelope, type ProcedureFunc, type ProcedureHandlerFunc, type ProcedureOperations, type ProceduresOptions, type ProceedKyselyQueryFunction, type QueryOptions, queryUtils as QueryUtils, RejectedByPolicyReason, type RuntimePlugin, type SchemaExtResult, schemaUtils as SchemaUtils, type SearchFieldMap, type SearchMode, type SearchPayload, type SearchStrategy, type SelectAwareExtResult, type SelectIncludeOmit, type SelectInput, type SelectSubset, type SimplifiedPlainResult, type SimplifiedResult, type SlicingOptions, type SortOrder, type StringFilter, type Subset, type ToKysely, type TransactionClientContract, TransactionIsolationLevel, type TypeDefResult, type TypedJsonFilter, type UpdateArgs, type UpdateInput, type UpdateManyAndReturnArgs, type UpdateManyArgs, type UpsertArgs, type WhereInput, type WhereUniqueInput, ZENSTACK_QUERY_KIND_SYMBOL, ZENSTACK_QUERY_SYMBOL, type ZModelFunction, type ZModelFunctionContext, ZenStackClient, type ZenStackPromise, type ZenStackQueryKind, createQuerySchemaFactory, definePlugin, getCrudDialect };
+export { type AfterEntityMutationCallback, type AggregateArgs, type AggregateResult, AllCrudOperations, type AllModelOperations, AllReadOperations, AnyNull, AnyNullClass, type AnyPlugin, type AuthType, BaseCrudDialect, type BatchResult, type BeforeEntityMutationCallback, type BooleanFilter, type Brand, type BytesFilter, CRUD, CRUD_EXT, type ClientConstructor, type ClientContract, type ClientOptions, type ComputedFieldsOptions, CoreCreateOperations, CoreCrudOperations, CoreDeleteOperations, CoreReadOperations, CoreUpdateOperations, CoreWriteOperations, type CountArgs, type CountResult, type CreateArgs, type CreateInput, type CreateManyAndReturnArgs, type CreateManyArgs, type CreateManyInput, type DateTimeFilter, DbNull, DbNullClass, type DefaultModelResult, type DefaultWhereConfig, type DeleteArgs, type DeleteManyArgs, type EntityMutationHooksDef, type ExistsArgs, type ExtClientMembersBase, type ExtQueryArgsBase, type ExtResultBase, type ExtResultInferenceArgs, type ExtResultSelectOmitFields, type ExtractExtResult, type FilterKind, type FindArgs, type FindFirstArgs, type FindManyArgs, type FindUniqueArgs, type GetAllIncludedOperations, type GetIncludedOperations, type GetProcedure, type GetProcedureNames, type GetProcedureParams, type GetQueryOptions, type GetSlicedFilterKindsForField, type GetSlicedModels, type GetSlicedOperations, type GetSlicedProcedures, type GroupByArgs, type GroupByResult, type HasComputedFields, type HasProcedures, type IncludeInput, InputValidator, type JsonArray, type JsonFilter, JsonNull, JsonNullClass, type JsonNullValues, type JsonObject, type JsonValue, kyselyUtils as KyselyUtils, LOGICAL_COMBINATORS, type MapModelFieldType, type ModelAllowsCreate, type ModelHasRequiredUnsupportedField, type ModelOperations, type ModelResult, type ModelSlicingOptions, type NullsOrder, type NumberFilter, ORMError, ORMErrorReason, type OmitConfig, type OmitInput, type OnKyselyQueryArgs, type OnKyselyQueryCallback, type OnProcedureHookContext, type OperationsRequiringCreate, type OrderBy, type PaginatedCursor, type PaginatedOffset, type PluginAfterEntityMutationArgs, type PluginBeforeEntityMutationArgs, type ProcedureEnvelope, type ProcedureFunc, type ProcedureHandlerFunc, type ProcedureOperations, type ProceduresOptions, type ProceedKyselyQueryFunction, type QueryOptions, queryUtils as QueryUtils, RejectedByPolicyReason, type RuntimePlugin, type SchemaExtResult, schemaUtils as SchemaUtils, type SearchFieldMap, type SearchMode, type SearchPayload, type SearchStrategy, type SelectAwareExtResult, type SelectIncludeOmit, type SelectInput, type SelectSubset, type SimplifiedPlainResult, type SimplifiedResult, type SlicingOptions, type SortOrder, type StringFilter, type Subset, type ToKysely, type TransactionClientContract, TransactionIsolationLevel, type TypeDefResult, type TypedJsonFilter, type UpdateArgs, type UpdateInput, type UpdateManyAndReturnArgs, type UpdateManyArgs, type UpsertArgs, type WhereInput, type WhereUniqueInput, ZENSTACK_QUERY_KIND_SYMBOL, ZENSTACK_QUERY_SYMBOL, type ZModelFunction, type ZModelFunctionContext, ZenStackClient, type ZenStackPromise, type ZenStackQueryKind, createQuerySchemaFactory, definePlugin, getCrudDialect };

package/dist/index.js

@@ -5327,6 +5327,27 @@ var FindOperationHandler = class extends BaseOperationHandler {
   static {
     __name(this, "FindOperationHandler");
   }
+  buildBatchQuery(operation, args, validateArgs = true) {
+    const normalizedArgs = this.normalizeArgs(args);
+    const findOne = operation === "findFirst" || operation === "findUnique";
+    let parsedArgs = validateArgs ? this.inputValidator.validateFindArgs(this.model, normalizedArgs, operation) : normalizedArgs;
+    if (findOne) {
+      parsedArgs = parsedArgs ?? {};
+      parsedArgs.take = 1;
+    }
+    const defaultWhereArgs = this.applyDefaultWhere(this.model, parsedArgs);
+    const defaultedArgs = this.applySchemaPluginDefaults(this.model, defaultWhereArgs);
+    const searchExpandedArgs = this.applySearchExpansion(this.model, defaultedArgs);
+    const { args: rewrittenArgs, plan } = this.rewriteVirtualRelations(this.model, searchExpandedArgs);
+    return {
+      query: this.buildReadQuery(this.model, rewrittenArgs),
+      args: parsedArgs,
+      plan
+    };
+  }
+  applyBatchVirtualRelationPlan(data, plan) {
+    return plan ? this.applyVirtualRelationPlan(data, plan) : data;
+  }
   buildQuery(operation, args, validateArgs = true) {
     const normalizedArgs = this.normalizeArgs(args);
     const findOne = operation === "findFirst" || operation === "findUnique";
@@ -10066,20 +10087,26 @@ var ClientImpl = class _ClientImpl {
         return {};
       }
       const { jsonArrayFrom, jsonObjectFrom } = await import("kysely/helpers/postgres");
+      const postProcessors = /* @__PURE__ */ new Map();
       const selections = await Promise.all(flat.map(async ([name, value]) => {
-        const query = await value[ZENSTACK_QUERY_SYMBOL](this);
+        const built = await value[ZENSTACK_QUERY_SYMBOL](this);
+        const query = built && typeof built === "object" && "query" in built ? built.query : built;
+        if (built && typeof built === "object" && "postProcess" in built && typeof built.postProcess === "function") {
+          postProcessors.set(name, built.postProcess);
+        }
         const kind = value[ZENSTACK_QUERY_KIND_SYMBOL];
         return kind === "many" ? jsonArrayFrom(query).as(name) : jsonObjectFrom(query).as(name);
       }));
       const row = await this.$qb.selectNoFrom(selections).executeTakeFirstOrThrow();
       const result = {};
       for (const [name, value] of Object.entries(row)) {
+        const processedValue = postProcessors.get(name)?.(value) ?? value;
         const path = name.split("__");
         let current = result;
         for (const key of path.slice(0, -1)) {
           current = current[key] ??= {};
         }
-        current[path[path.length - 1]] = value;
+        current[path[path.length - 1]] = processedValue;
       }
       return result;
     });
@@ -10205,35 +10232,127 @@ function createModelCrudHandler(client, model, inputValidator, resultProcessor)
     findUnique: /* @__PURE__ */ __name((args) => {
       const handler = new FindOperationHandler(client, model, inputValidator);
       return createPromise("findUnique", "findUnique", args, handler, true, false, {
-        [ZENSTACK_QUERY_SYMBOL]: () => handler.buildQuery("findUnique", args),
+        [ZENSTACK_QUERY_SYMBOL]: () => {
+          const shouldApplyExtResult = hasAnyExtResult && EXT_RESULT_OPERATIONS.has("findUnique");
+          const processedArgs = shouldApplyExtResult ? prepareArgsForExtResult(args, model, schema, plugins) : args;
+          const batchHandler = handler;
+          const { query, args: batchArgs, plan } = batchHandler.buildBatchQuery("findUnique", processedArgs);
+          return {
+            query,
+            postProcess: /* @__PURE__ */ __name((value) => {
+              let result = value;
+              if (result) {
+                result = batchHandler.applyBatchVirtualRelationPlan(result, plan);
+                result = resultProcessor.processResult(result, model, batchArgs);
+              }
+              if (result && shouldApplyExtResult) {
+                result = applyExtResult(result, model, args, schema, plugins);
+              }
+              return result ?? null;
+            }, "postProcess")
+          };
+        },
         [ZENSTACK_QUERY_KIND_SYMBOL]: "maybeOne"
       });
     }, "findUnique"),
     findUniqueOrThrow: /* @__PURE__ */ __name((args) => {
       const handler = new FindOperationHandler(client, model, inputValidator);
       return createPromise("findUnique", "findUniqueOrThrow", args, handler, true, true, {
-        [ZENSTACK_QUERY_SYMBOL]: () => handler.buildQuery("findUnique", args),
+        [ZENSTACK_QUERY_SYMBOL]: () => {
+          const shouldApplyExtResult = hasAnyExtResult && EXT_RESULT_OPERATIONS.has("findUnique");
+          const processedArgs = shouldApplyExtResult ? prepareArgsForExtResult(args, model, schema, plugins) : args;
+          const batchHandler = handler;
+          const { query, args: batchArgs, plan } = batchHandler.buildBatchQuery("findUnique", processedArgs);
+          return {
+            query,
+            postProcess: /* @__PURE__ */ __name((value) => {
+              if (!value) {
+                throw createNotFoundError(model);
+              }
+              let result = batchHandler.applyBatchVirtualRelationPlan(value, plan);
+              result = resultProcessor.processResult(result, model, batchArgs);
+              if (shouldApplyExtResult) {
+                result = applyExtResult(result, model, args, schema, plugins);
+              }
+              return result;
+            }, "postProcess")
+          };
+        },
         [ZENSTACK_QUERY_KIND_SYMBOL]: "one"
       });
     }, "findUniqueOrThrow"),
     findFirst: /* @__PURE__ */ __name((args) => {
       const handler = new FindOperationHandler(client, model, inputValidator);
       return createPromise("findFirst", "findFirst", args, handler, true, false, {
-        [ZENSTACK_QUERY_SYMBOL]: () => handler.buildQuery("findFirst", args),
+        [ZENSTACK_QUERY_SYMBOL]: () => {
+          const shouldApplyExtResult = hasAnyExtResult && EXT_RESULT_OPERATIONS.has("findFirst");
+          const processedArgs = shouldApplyExtResult ? prepareArgsForExtResult(args, model, schema, plugins) : args;
+          const batchHandler = handler;
+          const { query, args: batchArgs, plan } = batchHandler.buildBatchQuery("findFirst", processedArgs);
+          return {
+            query,
+            postProcess: /* @__PURE__ */ __name((value) => {
+              let result = value;
+              if (result) {
+                result = batchHandler.applyBatchVirtualRelationPlan(result, plan);
+                result = resultProcessor.processResult(result, model, batchArgs);
+              }
+              if (result && shouldApplyExtResult) {
+                result = applyExtResult(result, model, args, schema, plugins);
+              }
+              return result ?? null;
+            }, "postProcess")
+          };
+        },
         [ZENSTACK_QUERY_KIND_SYMBOL]: "maybeOne"
       });
     }, "findFirst"),
     findFirstOrThrow: /* @__PURE__ */ __name((args) => {
       const handler = new FindOperationHandler(client, model, inputValidator);
       return createPromise("findFirst", "findFirstOrThrow", args, handler, true, true, {
-        [ZENSTACK_QUERY_SYMBOL]: () => handler.buildQuery("findFirst", args),
+        [ZENSTACK_QUERY_SYMBOL]: () => {
+          const shouldApplyExtResult = hasAnyExtResult && EXT_RESULT_OPERATIONS.has("findFirst");
+          const processedArgs = shouldApplyExtResult ? prepareArgsForExtResult(args, model, schema, plugins) : args;
+          const batchHandler = handler;
+          const { query, args: batchArgs, plan } = batchHandler.buildBatchQuery("findFirst", processedArgs);
+          return {
+            query,
+            postProcess: /* @__PURE__ */ __name((value) => {
+              if (!value) {
+                throw createNotFoundError(model);
+              }
+              let result = batchHandler.applyBatchVirtualRelationPlan(value, plan);
+              result = resultProcessor.processResult(result, model, batchArgs);
+              if (shouldApplyExtResult) {
+                result = applyExtResult(result, model, args, schema, plugins);
+              }
+              return result;
+            }, "postProcess")
+          };
+        },
         [ZENSTACK_QUERY_KIND_SYMBOL]: "one"
       });
     }, "findFirstOrThrow"),
     findMany: /* @__PURE__ */ __name((args) => {
       const handler = new FindOperationHandler(client, model, inputValidator);
       return createPromise("findMany", "findMany", args, handler, true, false, {
-        [ZENSTACK_QUERY_SYMBOL]: () => handler.buildQuery("findMany", args),
+        [ZENSTACK_QUERY_SYMBOL]: () => {
+          const shouldApplyExtResult = hasAnyExtResult && EXT_RESULT_OPERATIONS.has("findMany");
+          const processedArgs = shouldApplyExtResult ? prepareArgsForExtResult(args, model, schema, plugins) : args;
+          const batchHandler = handler;
+          const { query, args: batchArgs, plan } = batchHandler.buildBatchQuery("findMany", processedArgs);
+          return {
+            query,
+            postProcess: /* @__PURE__ */ __name((value) => {
+              let result = batchHandler.applyBatchVirtualRelationPlan(value, plan);
+              result = resultProcessor.processResult(result, model, batchArgs);
+              if (shouldApplyExtResult) {
+                result = applyExtResult(result, model, args, schema, plugins);
+              }
+              return result;
+            }, "postProcess")
+          };
+        },
         [ZENSTACK_QUERY_KIND_SYMBOL]: "many"
       });
     }, "findMany"),