feat: add a portal redirect endpoint (#23375)

Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Eric Duong <[email protected]>
Co-authored-by: Dylan Martin <[email protected]>
Co-authored-by: Sandy Spicer <[email protected]>
Co-authored-by: Robbie <[email protected]>
Co-authored-by: Xavier Vello <[email protected]>
Co-authored-by: Paul D'Ambra <[email protected]>
Co-authored-by: Marcus Hof <[email protected]>
Co-authored-by: Juraj Majerik <[email protected]>
Co-authored-by: Tom Owers <[email protected]>
Co-authored-by: Phani Raj <[email protected]>

cypress/fixtures/api/billing/billing-unsubscribed-product-analytics.json

@@ -1955,5 +1955,5 @@
     "discount_amount_usd": null,
     "amount_off_expires_at": null,
     "never_drop_data": null,
-    "stripe_portal_url": "https://billing.stripe.com/p/session/test_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9PdXhKRW1kYVNRb3BOcFpVVHl6emVCUjdSUDJrQTJs0100akrliUs0"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

cypress/fixtures/api/billing/billing-unsubscribed.json

@@ -3657,5 +3657,5 @@
         "session_replay": 3,
         "product_analytics": 3
     },
-    "stripe_portal_url": "https://billing.stripe.com/p/session/test_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9QaEVaQ0hCTUE0aE8wUFhlVWVqd29MaElGd3lwRjFa010044U4IxJp"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

cypress/fixtures/api/billing/billing.json

@@ -3777,5 +3777,5 @@
         "session_replay": 3,
         "product_analytics": 3
     },
-    "stripe_portal_url": "https://billing.stripe.com/p/session/test_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9QaEVpVHFJNXdKYk9DaG04SVhMaUV4TDlxOTR1WEZi0100SMJCDr2e"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

ee/api/billing.py

@@ -180,6 +180,20 @@ def deactivate(self, request: Request, *args: Any, **kwargs: Any) -> HttpRespons
 
         return self.list(request, *args, **kwargs)
 
+    @action(methods=["GET"], detail=False)
+    def portal(self, request: Request, *args: Any, **kwargs: Any) -> HttpResponse:
+        license = get_cached_instance_license()
+        if not license:
+            return Response(
+                {"sucess": True},
+                status=status.HTTP_200_OK,
+            )
+
+        organization = self._get_org_required()
+
+        res = BillingManager(license)._get_stripe_portal_url(organization)
+        return redirect(res)
+
     @action(methods=["GET"], detail=False)
     def get_invoices(self, request: Request, *args: Any, **kwargs: Any) -> HttpResponse:
         license = get_cached_instance_license()

ee/api/test/test_billing.py

@@ -359,7 +359,7 @@ def mock_implementation(url: str, headers: Any = None, params: Any = None) -> Ma
             "available_product_features": [],
             "custom_limits_usd": {},
             "has_active_subscription": True,
-            "stripe_portal_url": "https://billing.stripe.com/p/session/test_1234",
+            "stripe_portal_url": "http://localhost:8000/api/billing/portal",
             "current_total_amount_usd": "100.00",
             "deactivated": False,
             "products": [
@@ -562,7 +562,7 @@ def mock_implementation(url: str, headers: Any = None, params: Any = None) -> Ma
             "free_trial_until": None,
             "current_total_amount_usd": "0.00",
             "deactivated": False,
-            "stripe_portal_url": "https://billing.stripe.com/p/session/test_1234",
+            "stripe_portal_url": "http://localhost:8000/api/billing/portal",
         }
 
     @patch("ee.api.billing.requests.get")
@@ -839,6 +839,18 @@ def mock_implementation(url: str, headers: Any = None, params: Any = None) -> Ma
         assert self.organization.customer_trust_scores == {"recordings": 0, "events": 15, "rows_synced": 0}
 
 
+class TestPortalBillingAPI(APILicensedTest):
+    @patch("ee.api.billing.requests.get")
+    def test_portal_success(self, mock_request):
+        mock_request.return_value.status_code = 200
+        mock_request.return_value.json.return_value = {"url": "https://billing.stripe.com/p/session/test_1234"}
+
+        response = self.client.get("/api/billing/portal")
+
+        self.assertEqual(response.status_code, status.HTTP_302_FOUND)
+        self.assertIn("https://billing.stripe.com/p/session/test_1234", response.url)
+
+
 class TestActivateBillingAPI(APILicensedTest):
     def test_activate_success(self):
         url = "/api/billing/activate"

ee/billing/billing_manager.py

@@ -1,3 +1,4 @@
+from django.conf import settings
 from datetime import datetime, timedelta
 from enum import Enum
 from typing import Any, Optional, cast
@@ -85,8 +86,7 @@ def get_billing(self, organization: Optional[Organization], plan_keys: Optional[
                 products = self.get_default_products(organization)
                 response["products"] = products["products"]
 
-            stripe_portal_url = self._get_stripe_portal_url(organization)
-            response["stripe_portal_url"] = stripe_portal_url
+            response["stripe_portal_url"] = f"{settings.SITE_URL}/api/billing/portal"
 
             # Extend the products with accurate usage_limit info
             for product in response["products"]:
@@ -207,7 +207,7 @@ def _get_billing(self, organization: Organization) -> BillingStatus:
 
         return data
 
-    def _get_stripe_portal_url(self, organization: Organization) -> BillingStatus:
+    def _get_stripe_portal_url(self, organization: Organization) -> str:
         """
         Retrieves stripe protal url
         """

frontend/src/mocks/fixtures/_billing_unsubscribed.json

@@ -1872,5 +1872,5 @@
     "discount_amount_usd": null,
     "amount_off_expires_at": null,
     "never_drop_data": null,
-    "stripe_portal_url": "https://billing.stripe.com/p/session/test_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9QN0ltZVQ3RmpLbTZacXgzYWo3Q0FFbFpITHZydlpK0100iKmkfAZi"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

frontend/src/mocks/fixtures/_billing_with_100_percent_discount.json

@@ -2769,5 +2769,5 @@
             ]
         }
     ],
-    "stripe_portal_url": "https://billing.stripe.com/p/session/XYZ"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

frontend/src/mocks/fixtures/_billing_with_billing_limits.json

@@ -4683,5 +4683,5 @@
         "session_replay": 3,
         "product_analytics": 3
     },
-    "stripe_portal_url": "https://billing.stripe.com/p/session/test_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9RNE9kdEg5elRmWE1wZkY1SGlsTFFFQXRKRmVkdnpJ0100M2MwqzHs"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

frontend/src/mocks/fixtures/_billing_with_discount.json

@@ -2767,5 +2767,5 @@
             ]
         }
     ],
-    "stripe_portal_url": "https://billing.stripe.com/p/session/XYZ"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

frontend/src/mocks/fixtures/_billing_with_flat_fee.json

@@ -1447,5 +1447,5 @@
     "discount_amount_usd": null,
     "amount_off_expires_at": null,
     "never_drop_data": false,
-    "stripe_portal_url": "https://billing.stripe.com/p/session/live_YWNjdF8xSElNRERFdUlhdFJYU2R6LF9PeG9wUmZuUFpmcTlvaWVjeUd0ZlhFN2hHbjJUWDR30100sm6GDMjx"
+    "stripe_portal_url": "http://localhost:8000/api/billing/portal"
 }

frontend/src/scenes/billing/Billing.tsx

@@ -220,6 +220,7 @@ export function Billing(): JSX.Element {
                                 htmlType="submit"
                                 to={billing.stripe_portal_url}
                                 disableClientSideRouting
+                                targetBlank
                                 center
                             >
                                 Manage card details and view past invoices