Add callbacks to next and express auth

.changeset/curvy-bobcats-think.md

@@ -0,0 +1,5 @@
+---
+"@thirdweb-dev/auth": patch
+---
+
+Add callbacks to express and next auth

packages/auth/src/express/index.ts

@@ -34,13 +34,18 @@ 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 = await ctx.callbacks.user(address);
+        }
+
+        user = { ...user, address };
       } catch {
         // No-op
       }
     }
 
-    req.user = user;
+    req.user = user as ThirdwebAuthUser | null;
     next();
   });
 

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);
+    await 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) => (Promise<void> | void);
+    user?: (address: string) => (Promise<Omit<ThirdwebAuthUser, "address">> | Omit<ThirdwebAuthUser, "address">);
+  };
 };
 
 export type ThirdwebAuthContext = {
   sdk: ThirdwebSDK;
   domain: string;
+  callbacks?: {
+    login?: (address: string) => (Promise<void> | void);
+    user?: (address: string) => (Promise<Omit<ThirdwebAuthUser, "address">> | Omit<ThirdwebAuthUser, "address">);
+  };
 };
 
 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);
+    await ctx.callbacks.login(address);
+  }
+
   return res.status(301).redirect(req.query.redirect as string);
 }

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

@@ -1,4 +1,4 @@
-import { ThirdwebAuthContext } from "../types";
+import { ThirdwebAuthContext, ThirdwebAuthUser } from "../types";
 import { NextApiRequest, NextApiResponse } from "next";
 
 export default async function handler(
@@ -19,11 +19,16 @@ export default async function handler(
   if (token) {
     try {
       const address = await sdk.auth.authenticate(domain, token);
-      user = { address };
+
+      if (ctx.callbacks?.user) {
+        user = await ctx.callbacks.user(address);
+      }
+
+      user = { ...user, address };
     } catch {
       // No-op
     }
   }
 
-  return res.status(200).json(user);
+  return res.status(200).json(user as ThirdwebAuthUser | null);
 }

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) => (Promise<void> | void);
+    user?: (address: string) => (Promise<Omit<ThirdwebAuthUser, "address">> | Omit<ThirdwebAuthUser, "address">);
+  };
 };
 
 export type ThirdwebAuthContext = {
   sdk: ThirdwebSDK;
   domain: string;
+  callbacks?: {
+    login?: (address: string) => (Promise<void> | void);
+    user?: (address: string) => (Promise<Omit<ThirdwebAuthUser, "address">> | Omit<ThirdwebAuthUser, "address">);
+  };
 };
 
 export type ThirdwebAuthUser = {
   address: string;
+  [key: string]: any;
 };