Support success and cancel URLs

dbos-inc · Dec 5, 2024 · 9efa235 · 9efa235
1 parent dd04104
commit 9efa235
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/src/endpoints.ts b/src/endpoints.ts
@@ -45,7 +45,10 @@ export class CloudSubscription {
     const auth0UserID = ctxt.authenticatedUser;
     // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
     const userEmail = ctxt.koaContext.state.user["https://dbos.dev/email"] as string;
-    const sessionURL = await ctxt.invokeWorkflow(Utils).createSubscription(auth0UserID, userEmail);
+    const body = ctxt.request.body as {success_url: string, cancel_url: string};
+    const successUrl = body.success_url as string ?? 'https://console.dbos.dev';
+    const cancelUrl = body.cancel_url as string ?? 'https://www.dbos.dev/pricing';
+    const sessionURL = await ctxt.invokeWorkflow(Utils).createSubscription(auth0UserID, userEmail, successUrl, cancelUrl);
     if (!sessionURL) {
       throw new DBOSResponseError("Failed to create a checkout session!", 500);
     }

diff --git a/src/subscription.ts b/src/subscription.ts
@@ -44,7 +44,7 @@ export class Utils {
 
   // Workflow to create a Stripe checkout session
   @Workflow()
-  static async createSubscription(ctxt: WorkflowContext, auth0UserID: string, userEmail: string): Promise<string|null> {
+  static async createSubscription(ctxt: WorkflowContext, auth0UserID: string, userEmail: string, successUrl: string, cancelUrl: string): Promise<string|null> {
     // First, look up the customer from the accounts table
     let stripeCustomerID = await ctxt.invoke(Utils).findStripeCustomerID(auth0UserID);
 
@@ -55,7 +55,7 @@ export class Utils {
     }
 
     // Finally, create a Stripe checkout session.
-    const res = await ctxt.invoke(Utils).createStripeCheckout(stripeCustomerID);
+    const res = await ctxt.invoke(Utils).createStripeCheckout(stripeCustomerID, successUrl, cancelUrl);
     return res;
   }
 
@@ -123,7 +123,7 @@ export class Utils {
 
   // Create a Stripe checkout session for a customer
   @Communicator({intervalSeconds: 10, maxAttempts: 2})
-  static async createStripeCheckout(_ctxt: CommunicatorContext, stripeCustomerID: string): Promise<string|null> {
+  static async createStripeCheckout(_ctxt: CommunicatorContext, stripeCustomerID: string, successUrl: string, cancelUrl: string): Promise<string|null> {
     const session = await stripe.checkout.sessions.create({
       customer: stripeCustomerID,
       billing_address_collection: 'auto',
@@ -134,8 +134,8 @@ export class Utils {
         },
       ],
       mode: 'subscription',
-      success_url: `https://docs.dbos.dev`,
-      cancel_url: `https://www.dbos.dev/pricing`,
+      success_url: successUrl,
+      cancel_url: cancelUrl,
       allow_promotion_codes: true,
     });
     return session.url;