From 374aaa8c64785363eb70465c50fcf9ae2e298e00 Mon Sep 17 00:00:00 2001
From: chenhang <averych@126.com>
Date: Tue, 30 Jun 2026 10:17:29 +0800
Subject: [PATCH] fix(subscription): return to chat from paywall

---
 .../subscription/components/subscription-back-link.tsx    | 8 ++++++--
 src/app/subscription/subscription-screen.tsx              | 3 ++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/app/subscription/components/subscription-back-link.tsx b/src/app/subscription/components/subscription-back-link.tsx
index ba2186ce..7411d657 100644
--- a/src/app/subscription/components/subscription-back-link.tsx
+++ b/src/app/subscription/components/subscription-back-link.tsx
@@ -16,12 +16,16 @@ import styles from "./subscription-back-link.module.css";
 
 export interface SubscriptionBackLinkProps {
   className?: string;
+  href?: string;
 }
 
-export function SubscriptionBackLink({ className }: SubscriptionBackLinkProps) {
+export function SubscriptionBackLink({
+  className,
+  href = ROUTES.sidebar,
+}: SubscriptionBackLinkProps) {
   return (
     <Link
-      href={ROUTES.sidebar}
+      href={href}
       aria-label="Back"
       className={[styles.backLink, className].filter(Boolean).join(" ")}
     >
diff --git a/src/app/subscription/subscription-screen.tsx b/src/app/subscription/subscription-screen.tsx
index ca9ef45f..b35462ad 100644
--- a/src/app/subscription/subscription-screen.tsx
+++ b/src/app/subscription/subscription-screen.tsx
@@ -156,6 +156,7 @@ export function SubscriptionScreen({
     payment.isPollingOrder;
   const canActivate =
     selectedPlan !== null && payment.agreed && !isPaymentBusy;
+  const backHref = returnTo === "chat" ? ROUTES.chat : ROUTES.sidebar;
 
   const finishPaymentSuccessClose = () => {
     setShowPaymentSuccessDialog(false);
@@ -215,7 +216,7 @@ export function SubscriptionScreen({
     <MobileShell>
       <div className={styles.shell}>
         <header className={styles.header}>
-          <SubscriptionBackLink className={styles.backSlot} />
+          <SubscriptionBackLink className={styles.backSlot} href={backHref} />
         </header>
 
         <div className={styles.offerStack}>