From eabde95a32007267201ddb485a2c96a20df9767c Mon Sep 17 00:00:00 2001
From: Zai Shi <zaishi00@outlook.com>
Date: Fri, 8 Nov 2024 14:05:08 +0100
Subject: [PATCH] Added callbackUrl to server user invitation (#331)

---
 docs/fern/docs/pages/sdk/team.mdx   |  8 ++++++++
 packages/stack/src/lib/stack-app.ts | 19 +++++++++++++------
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/docs/fern/docs/pages/sdk/team.mdx b/docs/fern/docs/pages/sdk/team.mdx
index 500f04502..e56606b5f 100644
--- a/docs/fern/docs/pages/sdk/team.mdx
+++ b/docs/fern/docs/pages/sdk/team.mdx
@@ -108,6 +108,14 @@ An invitation email containing a magic link will be sent to the specified user.
       <ParamField path="email" type="string" required>
         The email of the user to invite.
       </ParamField>
+
+      <ParamField path="callbackUrl" type="string">
+        The URL where users will be redirected after accepting the team invitation.
+        
+        Required when calling `inviteUser()` in the server environment since the URL cannot be automatically determined.
+        
+        Example: `https://your-app-url.com/handler/team-invitation`
+      </ParamField>
     </div>
   </ParamField>
 </div>
diff --git a/packages/stack/src/lib/stack-app.ts b/packages/stack/src/lib/stack-app.ts
index 7fe863176..31a60bb66 100644
--- a/packages/stack/src/lib/stack-app.ts
+++ b/packages/stack/src/lib/stack-app.ts
@@ -753,12 +753,15 @@ class _StackClientAppImpl<HasTokenStore extends boolean, ProjectId extends strin
       profileImageUrl: crud.profile_image_url,
       clientMetadata: crud.client_metadata,
       clientReadOnlyMetadata: crud.client_read_only_metadata,
-      async inviteUser(options: { email: string }) {
+      async inviteUser(options: { email: string, callbackUrl?: string }) {
+        if (!options.callbackUrl && typeof window === "undefined") {
+          throw new Error("Cannot invite user without a callback URL when calling the inviteUser function on the server.");
+        }
         await app._interface.sendTeamInvitation({
           teamId: crud.id,
           email: options.email,
           session,
-          callbackUrl: constructRedirectUrl(app.urls.teamInvitation),
+          callbackUrl: options.callbackUrl ?? constructRedirectUrl(app.urls.teamInvitation),
         });
       },
       async listUsers() {
@@ -1959,12 +1962,16 @@ class _StackServerAppImpl<HasTokenStore extends boolean, ProjectId extends strin
         });
         await app._serverTeamMemberProfilesCache.refresh([crud.id]);
       },
-      async inviteUser(options: { email: string }) {
+      async inviteUser(options: { email: string, callbackUrl?: string }) {
+        if (!options.callbackUrl && typeof window === "undefined") {
+          throw new Error("Cannot invite user without a callback URL when calling the inviteUser function on the server.");
+        }
+
         await app._interface.sendTeamInvitation({
           teamId: crud.id,
           email: options.email,
           session: null,
-          callbackUrl: constructRedirectUrl(app.urls.teamInvitation),
+          callbackUrl: options.callbackUrl ?? constructRedirectUrl(app.urls.teamInvitation),
         });
       },
     };
@@ -3046,7 +3053,7 @@ export type Team = {
   profileImageUrl: string | null,
   clientMetadata: any,
   clientReadOnlyMetadata: any,
-  inviteUser(options: { email: string }): Promise<void>,
+  inviteUser(options: { email: string, callbackUrl?: string }): Promise<void>,
   listUsers(): Promise<TeamUser[]>,
   useUsers(): TeamUser[],
   update(update: TeamUpdateOptions): Promise<void>,
@@ -3092,7 +3099,7 @@ export type ServerTeam = {
   update(update: ServerTeamUpdateOptions): Promise<void>,
   delete(): Promise<void>,
   addUser(userId: string): Promise<void>,
-  inviteUser(options: { email: string }): Promise<void>,
+  inviteUser(options: { email: string, callbackUrl?: string }): Promise<void>,
   removeUser(userId: string): Promise<void>,
 } & Team;