Add callbacks to next and express auth

packages/auth/src/express/index.ts

@@ -34,7 +34,12 @@ export function ThirdwebAuth(app: Express, cfg: ThirdwebAuthConfig) {
     if (token) {
       try {
         const address = await sdk.auth.authenticate(domain, token);
-        user = { address };
+
+        if (ctx.callbacks?.user) {
+          user = ctx.callbacks.user(address);
+        }
+
+        user = { ...user, address };
       } catch {
         // No-op
       }

packages/auth/src/express/routes/login.ts

@@ -47,5 +47,10 @@ export default async function handler(
     })
   );
 
+  if (ctx.callbacks?.login) {
+    const address = sdk.auth.verify(domain, payload);
+    ctx.callbacks.login(address);
+  }
+
   return res.status(301).redirect(req.query.redirect as string);
 }

packages/auth/src/express/types/index.ts

@@ -7,15 +7,24 @@ export type ThirdwebAuthConfig = {
   privateKey: string;
   domain: string;
   authUrl?: string;
+  callbacks?: {
+    login?: (address: string) => void;
+    user?: (address: string) => ThirdwebAuthUser;
+  };
 };
 
 export type ThirdwebAuthContext = {
   sdk: ThirdwebSDK;
   domain: string;
+  callbacks?: {
+    login?: (address: string) => void;
+    user?: (address: string) => ThirdwebAuthUser;
+  };
 };
 
 export type ThirdwebAuthUser = {
   address: string;
+  [key: string]: any;
 };
 
 export type RequestWithUser = Request & {

packages/auth/src/next/routes/login.ts

@@ -51,5 +51,10 @@ export default async function handler(
     })
   );
 
+  if (ctx.callbacks?.login) {
+    const address = sdk.auth.verify(domain, payload);
+    ctx.callbacks.login(address);
+  }
+
   return res.status(301).redirect(req.query.redirect as string);
 }

packages/auth/src/next/routes/user.ts

@@ -19,7 +19,12 @@ export default async function handler(
   if (token) {
     try {
       const address = await sdk.auth.authenticate(domain, token);
-      user = { address };
+
+      if (ctx.callbacks?.user) {
+        user = ctx.callbacks.user(address);
+      }
+
+      user = { ...user, address };
     } catch {
       // No-op
     }

packages/auth/src/next/types/index.ts

@@ -5,13 +5,22 @@ export type ThirdwebAuthRoute = "login" | "logout" | "user";
 export type ThirdwebAuthConfig = {
   privateKey: string;
   domain: string;
+  callbacks?: {
+    login?: (address: string) => void;
+    user?: (address: string) => ThirdwebAuthUser;
+  };
 };
 
 export type ThirdwebAuthContext = {
   sdk: ThirdwebSDK;
   domain: string;
+  callbacks?: {
+    login?: (address: string) => void;
+    user?: (address: string) => ThirdwebAuthUser;
+  };
 };
 
 export type ThirdwebAuthUser = {
   address: string;
+  [key: string]: any;
 };