From 52f92d86509f1c80136debafedbe9eac6d31c4bb Mon Sep 17 00:00:00 2001
From: "2569718930@qq.com" <2569718930@qq.com>
Date: Sat, 30 May 2026 21:08:19 +0800
Subject: [PATCH] Speed up terminal auth snapshot loading

---
 frontend/app/api/auth/me/route.ts             | 35 +++++++++++++++++--
 .../account/__tests__/paymentShell.test.ts    |  6 ++++
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/frontend/app/api/auth/me/route.ts b/frontend/app/api/auth/me/route.ts
index 199ddf31..854637b6 100644
--- a/frontend/app/api/auth/me/route.ts
+++ b/frontend/app/api/auth/me/route.ts
@@ -21,7 +21,10 @@ import {
   buildSubscriptionRequiredAuthProfile,
   isSubscriptionRequiredBackendResponse,
 } from "@/lib/auth-profile-proxy";
-import { hasSupabaseServerEnv } from "@/lib/supabase/server";
+import {
+  hasSupabaseServerEnv,
+  hasSupabaseSessionCookieValues,
+} from "@/lib/supabase/server";
 
 const API_BASE = process.env.POLYWEATHER_API_BASE_URL;
 
@@ -171,6 +174,15 @@ function applyEntitlementSnapshotFromAuthPayload(
   return response;
 }
 
+function hasRequestSupabaseSessionCookie(req: NextRequest) {
+  return hasSupabaseSessionCookieValues(
+    req.cookies.getAll().map((item) => ({
+      name: item.name,
+      value: item.value,
+    })),
+  );
+}
+
 export async function GET(req: NextRequest) {
   const requestHost =
     req.headers.get("x-forwarded-host") || req.headers.get("host") || req.nextUrl.host;
@@ -190,9 +202,28 @@ export async function GET(req: NextRequest) {
     );
   }
 
+  const preferSnapshot = req.nextUrl.searchParams.get("prefer_snapshot") === "1";
+  if (
+    preferSnapshot &&
+    !req.headers.get("authorization") &&
+    hasRequestSupabaseSessionCookie(req)
+  ) {
+    const snapshotPayload = entitlementSnapshotToAuthPayload(
+      readEntitlementSnapshot(req),
+    );
+    if (snapshotPayload) {
+      return NextResponse.json(
+        {
+          ...snapshotPayload,
+          entitlement_snapshot_reason: "prefer_snapshot_fast_path",
+        },
+        { headers: { "Cache-Control": "no-store" } },
+      );
+    }
+  }
+
   let auth: Awaited<ReturnType<typeof buildBackendRequestHeaders>> | null = null;
   let bearerIdentity: VerifiedBearerIdentity | null | undefined;
-  const preferSnapshot = req.nextUrl.searchParams.get("prefer_snapshot") === "1";
   const getBearerIdentityOnce = async () => {
     if (bearerIdentity !== undefined) return bearerIdentity;
     bearerIdentity = await getVerifiedBearerIdentity(req);
diff --git a/frontend/components/account/__tests__/paymentShell.test.ts b/frontend/components/account/__tests__/paymentShell.test.ts
index 8ed80a93..3e5f13d1 100644
--- a/frontend/components/account/__tests__/paymentShell.test.ts
+++ b/frontend/components/account/__tests__/paymentShell.test.ts
@@ -302,6 +302,12 @@ export function runTests() {
         authMeRouteSource.indexOf("await fetch(`${API_BASE}/api/auth/me`"),
     "auth profile proxy must return unauthenticated locally for no-session Supabase requests instead of forwarding the backend entitlement token",
   );
+  assert(
+    authMeRouteSource.includes('reason: "prefer_snapshot_fast_path"') &&
+      authMeRouteSource.indexOf('reason: "prefer_snapshot_fast_path"') <
+        authMeRouteSource.indexOf("buildBackendRequestHeaders(req)"),
+    "auth profile proxy must return a valid entitlement snapshot before reading Supabase session cookies on terminal cold start",
+  );
   assert(
     backendAuthSource.includes("if (incomingAuth) {") &&
       backendAuthSource.indexOf("if (incomingAuth) {") <