From 8d521bb7035c6b704bb46a27bd59538320b34cbd Mon Sep 17 00:00:00 2001
From: mantikoros <sgrugett@gmail.com>
Date: Sun, 19 Dec 2021 21:17:32 -0600
Subject: [PATCH] add funds referer url

---
 functions/src/stripe.ts             | 7 ++++---
 web/components/add-funds-button.tsx | 6 +++++-
 web/lib/service/stripe.ts           | 9 +++++++--
 3 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/functions/src/stripe.ts b/functions/src/stripe.ts
index d274d4ae..c67da656 100644
--- a/functions/src/stripe.ts
+++ b/functions/src/stripe.ts
@@ -37,7 +37,8 @@ export const createCheckoutSession = functions
       return
     }
 
-    const referrer = req.headers.referer || 'https://mantic.markets'
+    const referrer =
+      req.query.referer || req.headers.referer || 'https://mantic.markets'
 
     const session = await stripe.checkout.sessions.create({
       metadata: {
@@ -54,8 +55,8 @@ export const createCheckoutSession = functions
         },
       ],
       mode: 'payment',
-      success_url: `${referrer}/?funding-success`,
-      cancel_url: `${referrer}/?funding-failiure`,
+      success_url: `${referrer}?funding-success`,
+      cancel_url: `${referrer}?funding-failiure`,
     })
 
     res.redirect(303, session.url || '')
diff --git a/web/components/add-funds-button.tsx b/web/components/add-funds-button.tsx
index 2acc1ad7..eb603593 100644
--- a/web/components/add-funds-button.tsx
+++ b/web/components/add-funds-button.tsx
@@ -52,7 +52,11 @@ export function AddFundsButton() {
             </label>
 
             <form
-              action={checkoutURL(user?.id || '', amountSelected)}
+              action={checkoutURL(
+                user?.id || '',
+                amountSelected,
+                window.location.href
+              )}
               method="POST"
             >
               <button
diff --git a/web/lib/service/stripe.ts b/web/lib/service/stripe.ts
index dac138f3..129492b2 100644
--- a/web/lib/service/stripe.ts
+++ b/web/lib/service/stripe.ts
@@ -1,5 +1,10 @@
-export const checkoutURL = (userId: string, manticDollarQuantity: number) => {
+export const checkoutURL = (
+  userId: string,
+  manticDollarQuantity: number,
+  referer = ''
+) => {
   const endpoint =
     'https://us-central1-mantic-markets.cloudfunctions.net/createCheckoutSession'
-  return `${endpoint}?userId=${userId}&manticDollarQuantity=${manticDollarQuantity}`
+
+  return `${endpoint}?userId=${userId}&manticDollarQuantity=${manticDollarQuantity}&referer=${referer}`
 }