From 042cbdd2faa76fdc8da3ddce42adceb6368bd941 Mon Sep 17 00:00:00 2001
From: Stephen Zhou <38493346+hyoban@users.noreply.github.com>
Date: Tue, 18 Nov 2025 15:42:00 +0800
Subject: [PATCH] feat: manage subscription on stripe

---
 .../src/modules/settings/tabs/plan.tsx        | 61 +++++--------------
 1 file changed, 15 insertions(+), 46 deletions(-)

diff --git a/apps/desktop/layer/renderer/src/modules/settings/tabs/plan.tsx b/apps/desktop/layer/renderer/src/modules/settings/tabs/plan.tsx
index 85d9c7f50..a0a67e1b6 100644
--- a/apps/desktop/layer/renderer/src/modules/settings/tabs/plan.tsx
+++ b/apps/desktop/layer/renderer/src/modules/settings/tabs/plan.tsx
@@ -10,7 +10,6 @@ import { useMutation, useQuery } from "@tanstack/react-query"
 import type { TFunction } from "i18next"
 import { useState } from "react"
 import { useTranslation } from "react-i18next"
-import { toast } from "sonner"
 
 import type { PaymentFeature, PaymentPlan } from "~/atoms/server-configs"
 import { useIsPaymentEnabled, useServerConfigs } from "~/atoms/server-configs"
@@ -98,6 +97,7 @@ const useActiveSubscription = () => {
     queryKey: ["activeSubscription"],
     queryFn: async () => {
       const { data } = await subscription.list()
+      // We used to allow one time purchases, so we need to check for active subscriptions
       return data?.find(
         (sub) => (sub.status === "active" || sub.status === "trialing") && sub.stripeSubscriptionId,
       )
@@ -106,36 +106,6 @@ const useActiveSubscription = () => {
   })
 }
 
-const useCancelPlan = () => {
-  const { data: latestSubscription } = useActiveSubscription()
-  const subscriptionId = latestSubscription?.id
-  const cancelAtPeriodEnd = latestSubscription?.cancelAtPeriodEnd
-
-  const cancelMutation = useMutation({
-    mutationFn: async () => {
-      if (!subscriptionId) {
-        return
-      }
-
-      await subscription.cancel({
-        subscriptionId,
-        returnUrl: IN_ELECTRON ? `${DEEPLINK_SCHEME}refresh` : env.VITE_WEB_URL,
-        fetchOptions: {
-          onError(context) {
-            toast.error(context.error.message)
-          },
-        },
-      })
-    },
-  })
-
-  if (cancelAtPeriodEnd || !subscriptionId) {
-    return null
-  } else {
-    return cancelMutation
-  }
-}
-
 export function SettingPlan() {
   const isPaymentEnabled = useIsPaymentEnabled()
   const role = useUserRole()
@@ -250,7 +220,6 @@ const PlanCard = ({ plan, billingPeriod, isCurrentPlan, currentTier }: PlanCardP
     plan: plan.planID,
     annual: billingPeriod === "yearly",
   })
-  const cancelPlanMutation = useCancelPlan()
 
   // Calculate price and period based on billing period
   const regularPrice =
@@ -302,7 +271,7 @@ const PlanCard = ({ plan, billingPeriod, isCurrentPlan, currentTier }: PlanCardP
         <PlanAction
           actionType={actionType}
           upgradeButtonText={plan.upgradeButtonText}
-          isLoading={upgradePlanMutation.isPending || cancelPlanMutation?.isPending}
+          isLoading={upgradePlanMutation.isPending}
           onSelect={
             !plan.isComingSoon && !isCurrentPlan
               ? () => {
@@ -310,13 +279,6 @@ const PlanCard = ({ plan, billingPeriod, isCurrentPlan, currentTier }: PlanCardP
                 }
               : undefined
           }
-          onCancel={
-            isCurrentPlan && cancelPlanMutation
-              ? () => {
-                  cancelPlanMutation.mutate()
-                }
-              : undefined
-          }
         />
       </div>
 
@@ -392,15 +354,18 @@ const PlanAction = ({
   actionType,
   upgradeButtonText,
   onSelect,
-  onCancel,
   isLoading,
 }: {
   actionType: "current" | "upgrade" | "coming-soon" | "in-trial" | "switch" | "new" | null
   upgradeButtonText?: string
   onSelect?: () => void
-  onCancel?: () => void
   isLoading?: boolean
 }) => {
+  const { data: activeSubscription } = useActiveSubscription()
+  const serverConfig = useServerConfigs()
+  const stripePortalLink = serverConfig?.STRIPE_PORTAL_LINK
+  const canManageSubscription = !!activeSubscription && !!stripePortalLink
+
   const getButtonConfig = () => {
     switch (actionType) {
       case "coming-soon": {
@@ -413,11 +378,11 @@ const PlanAction = ({
       }
       case "current": {
         return {
-          text: `Current Plan${onCancel ? " | Cancel" : ""}`,
+          text: canManageSubscription ? "Manage Subscription" : "Current Plan",
           icon: undefined,
           variant: "outline" as const,
-          className: onCancel ? "" : "text-text-secondary",
-          disabled: onCancel ? false : true,
+          className: !canManageSubscription ? "text-text-secondary" : undefined,
+          disabled: !canManageSubscription,
         }
       }
       case "in-trial": {
@@ -475,7 +440,11 @@ const PlanAction = ({
         buttonConfig.className,
       )}
       disabled={buttonConfig.disabled}
-      onClick={buttonConfig.disabled ? undefined : (onCancel ?? onSelect)}
+      onClick={
+        actionType === "current" && canManageSubscription
+          ? () => window.open(stripePortalLink, "_blank")
+          : onSelect
+      }
       isLoading={isLoading}
     >
       <span className="flex items-center justify-center gap-1.5">