fix: enhance auth type

packages/orm/src/client/contract.ts

@@ -1,5 +1,14 @@
 import type Decimal from 'decimal.js';
-import { type GetModels, type IsDelegateModel, type ProcedureDef, type SchemaDef } from '../schema';
+import {
+    type FieldIsArray,
+    type GetModels,
+    type IsDelegateModel,
+    type NonRelationFields,
+    type ProcedureDef,
+    type RelationFields,
+    type RelationFieldType,
+    type SchemaDef,
+} from '../schema';
 import type { AnyKysely } from '../utils/kysely-utils';
 import type { OrUndefinedIf, Simplify, UnwrapTuplePromises } from '../utils/type-utils';
 import type { TRANSACTION_UNSUPPORTED_METHODS } from './constants';
@@ -19,6 +28,7 @@ import type {
     FindUniqueArgs,
     GroupByArgs,
     GroupByResult,
+    MapModelFieldType,
     ModelResult,
     SelectSubset,
     SimplifiedModelResult,
@@ -810,11 +820,23 @@ export type ModelOperations<Schema extends SchemaDef, Model extends GetModels<Sc
 
 //#region Supporting types
 
+/**
+ * Type for auth context that includes both scalar and relation fields.
+ * Relations are recursively included to allow nested auth data like { user: { profile: { ... } } }
+ */
+type AuthModelType<Schema extends SchemaDef, Model extends GetModels<Schema>> = {
+    [Key in NonRelationFields<Schema, Model>]?: MapModelFieldType<Schema, Model, Key>;
+} & {
+    [Key in RelationFields<Schema, Model>]?: FieldIsArray<Schema, Model, Key> extends true
+        ? AuthModelType<Schema, RelationFieldType<Schema, Model, Key>>[]
+        : AuthModelType<Schema, RelationFieldType<Schema, Model, Key>>;
+};
+
 export type AuthType<Schema extends SchemaDef> =
     string extends GetModels<Schema>
         ? Record<string, unknown>
         : Schema['authType'] extends GetModels<Schema>
-          ? Partial<ModelResult<Schema, Schema['authType']>>
+          ? AuthModelType<Schema, Schema['authType']>
           : never;
 
 //#endregion

packages/orm/src/client/crud-types.ts

@@ -551,7 +551,7 @@ type RelationFilter<
 
 //#region Field utils
 
-type MapModelFieldType<
+export type MapModelFieldType<
     Schema extends SchemaDef,
     Model extends GetModels<Schema>,
     Field extends GetModelFields<Schema, Model>,