fix(schemas): Custom Auth Data types

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,3 +1,4 @@
+import { CustomAuthData, CustomAuthDataSchema } from '@lit-protocol/schemas';
 import { hexToBigInt, keccak256, toBytes } from 'viem';
 
 export const utils = {
@@ -56,12 +57,14 @@ export const utils = {
     uniqueDappName: string;
     uniqueAuthMethodType: bigint;
     userId: string;
-  }) => {
+  }): 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

@@ -11,6 +11,14 @@ import {
 } from '../schemas';
 import { ScopeSchemaRaw } from './ScopeSchema';
 
+export const CustomAuthDataSchema = z.object({
+  authMethodId: HexPrefixedSchema,
+  // This will be a very big number, unlike our native auth
+  authMethodType: z.bigint(),
+});
+
+export type CustomAuthData = z.infer<typeof CustomAuthDataSchema>;
+
 export const AuthDataSchema = z.object({
   authMethodId: HexPrefixedSchema,
   authMethodType: z.coerce.number().pipe(z.nativeEnum(AUTH_METHOD_TYPE)),