fix(customAuthData): integrate CustomAuthDataSchema into utils and MintWithCustomAuthSchema for improved type consistency

Author: Ansonhkg

packages/lit-client/src/lib/LitClient/schemas/MintWithCustomAuthSchema.ts

@@ -1,4 +1,4 @@
-import { AuthData } from '@lit-protocol/schemas';
+import { AuthData, CustomAuthDataSchema } from '@lit-protocol/schemas';
 import { Optional } from '@lit-protocol/types';
 import { z } from 'zod';
 
@@ -9,7 +9,7 @@ const BaseMintWithCustomAuthSchema = z.object({
   // Account information - this will be passed from the calling context
   account: z.any(), // Account type varies by network
   // Authentication data for the user
-  authData: z.custom<Optional<AuthData, 'accessToken'>>(),
+  authData: CustomAuthDataSchema,
   scope: z.enum(['no-permissions', 'sign-anything', 'personal-sign']),
   // Optional overrides
   addPkpEthAddressAsPermittedAddress: z.boolean().default(false),

packages/lit-client/src/lib/LitClient/utils.ts

@@ -1,4 +1,4 @@
-import { CustomAuthData } from '@lit-protocol/schemas';
+import { CustomAuthData, CustomAuthDataSchema } from '@lit-protocol/schemas';
 import { hexToBigInt, keccak256, toBytes } from 'viem';
 
 export const utils = {
@@ -60,9 +60,11 @@ export const utils = {
   }): CustomAuthData => {
     const uniqueUserId = `${uniqueDappName}-${userId}`;
 
-    return {
+    const customAuthData = CustomAuthDataSchema.parse({
       authMethodType: uniqueAuthMethodType,
       authMethodId: keccak256(toBytes(uniqueUserId)),
-    };
+    });
+
+    return customAuthData;
   },
 };

packages/schemas/src/lib/auth/auth-schemas.ts

@@ -1,4 +1,4 @@
-import { AUTH_METHOD_TYPE_VALUES } from '@lit-protocol/constants';
+import { AUTH_METHOD_TYPE } from '@lit-protocol/constants';
 import { z } from 'zod';
 import {
   AuthMethodSchema,
@@ -9,6 +9,7 @@ import {
 
 export const CustomAuthDataSchema = z.object({
   authMethodId: HexPrefixedSchema,
+  // This will be a very big number, unlike our native auth
   authMethodType: z.bigint(),
 });
 
@@ -28,11 +29,7 @@ export type StrictAuthData = z.infer<typeof StrictAuthDataSchema>;
 
 export const AuthDataSchema = z.object({
   authMethodId: HexPrefixedSchema,
-  authMethodType: z.union([
-    AuthMethodSchema.shape.authMethodType,
-    z.number(),
-    z.bigint(),
-  ]),
+  authMethodType: z.coerce.number().pipe(z.nativeEnum(AUTH_METHOD_TYPE)),
   accessToken: AuthMethodSchema.shape.accessToken,
   publicKey: HexPrefixedSchema.optional(),