s'c1'

app/lib/payment/usePayStatusPoll.ts

@@ -1,13 +1,10 @@
-/**
- * 微信 H5 支付完成后可能不跳转，通过 cookie 中的订单号轮询检测支付状态
- * cookie 格式：order_id|timestamp，仅对 15 分钟内设置的 cookie 轮询，避免陈旧 cookie 误触发
- */
 import { useEffect, useRef, useState } from "react";
 
 const COOKIE_NAME = "join_pay_order";
-const POLL_INTERVAL = 2000;
-const MAX_POLLS = 150; // 约 5 分钟
-const MAX_AGE_MS = 15 * 60 * 1000; // 15 分钟
+const STORAGE_NAME = "join_pay_order";
+const POLL_INTERVAL_MS = 1200;
+const MAX_POLLS = 60;
+const MAX_AGE_MS = 15 * 60 * 1000;
 
 function getCookie(name: string): string | null {
   if (typeof document === "undefined") return null;
@@ -20,15 +17,40 @@ function clearCookie(name: string): void {
   document.cookie = `${name}=; path=/; max-age=0`;
 }
 
-/** 解析 cookie：order_id|timestamp，返回 [orderId, isValid]。无时间戳视为过期并清除 */
-function parsePayOrderCookie(raw: string | null): [string | null, boolean] {
+function getStorage(name: string): string | null {
+  if (typeof window === "undefined") return null;
+  return window.localStorage.getItem(name);
+}
+
+function removeStorage(name: string): void {
+  if (typeof window === "undefined") return;
+  window.localStorage.removeItem(name);
+}
+
+function clearPendingOrder(): void {
+  clearCookie(COOKIE_NAME);
+  removeStorage(STORAGE_NAME);
+}
+
+function persistPendingOrder(raw: string): void {
+  if (typeof document !== "undefined") {
+    document.cookie = `${COOKIE_NAME}=${encodeURIComponent(raw)}; path=/; max-age=900`;
+  }
+  if (typeof window !== "undefined") {
+    try {
+      window.localStorage.setItem(STORAGE_NAME, raw);
+    } catch {}
+  }
+}
+
+function parsePendingOrder(raw: string | null): [string | null, boolean] {
   if (!raw?.trim()) return [null, false];
   const parts = raw.split("|");
   const orderId = parts[0]?.trim() || raw.trim();
   const ts = parts[1] ? parseInt(parts[1], 10) : NaN;
   if (!orderId) return [null, false];
-  if (Number.isNaN(ts)) return [orderId, false]; // 旧格式无时间戳，视为过期
-  if (Date.now() - ts > MAX_AGE_MS) return [orderId, false]; // 超过 15 分钟
+  if (Number.isNaN(ts)) return [orderId, false];
+  if (Date.now() - ts > MAX_AGE_MS) return [orderId, false];
   return [orderId, true];
 }
 
@@ -38,38 +60,110 @@ export function usePayStatusPoll(onPaid: () => void): boolean {
   onPaidRef.current = onPaid;
 
   useEffect(() => {
-    const raw = getCookie(COOKIE_NAME);
-    const [orderId, isValid] = parsePayOrderCookie(raw);
-    if (!orderId || !isValid) {
-      if (raw) clearCookie(COOKIE_NAME);
+    const cookieRaw = getCookie(COOKIE_NAME);
+    const [cookieOrderId, cookieValid] = parsePendingOrder(cookieRaw);
+    const storageRaw = getStorage(STORAGE_NAME);
+    const [storageOrderId, storageValid] = parsePendingOrder(storageRaw);
+    const orderId = cookieValid ? cookieOrderId : storageValid ? storageOrderId : null;
+    const activeRaw = cookieValid ? cookieRaw : storageValid ? storageRaw : null;
+
+    if (!orderId) {
+      if (cookieRaw || storageRaw) {
+        clearPendingOrder();
+      }
       return;
     }
 
+    if (activeRaw) {
+      persistPendingOrder(activeRaw);
+    }
+
     setPolling(true);
-    let count = 0;
-    const timer = setInterval(async () => {
-      count++;
-      if (count > MAX_POLLS) {
-        clearInterval(timer);
-        clearCookie(COOKIE_NAME);
+    let attempts = 0;
+    let stopped = false;
+    let timer: number | undefined;
+    let inFlight = false;
+
+    const stop = () => {
+      stopped = true;
+      if (timer) window.clearTimeout(timer);
+    };
+
+    const scheduleNext = () => {
+      if (stopped) return;
+      timer = window.setTimeout(tick, POLL_INTERVAL_MS);
+    };
+
+    const tick = async () => {
+      if (stopped) return;
+      if (inFlight) {
+        scheduleNext();
+        return;
+      }
+
+      attempts += 1;
+      if (attempts > MAX_POLLS) {
+        stop();
+        clearPendingOrder();
         setPolling(false);
         return;
       }
+
+      inFlight = true;
       try {
-        const res = await fetch(`/api/pay/status?order_id=${encodeURIComponent(orderId)}`);
-        const data = await res.json();
+        const res = await fetch(
+          `/api/pay/status?order_id=${encodeURIComponent(orderId)}`,
+          { cache: "no-store" }
+        );
+        const data = await res.json().catch(() => ({}));
         if (data?.ok && data?.paid) {
-          clearInterval(timer);
-          clearCookie(COOKIE_NAME);
+          stop();
+          clearPendingOrder();
           setPolling(false);
+          // 对齐 nomadvip：调用 confirm 写入 PocketBase site_vip（不依赖 payjsapi 异步 notify）
+          try {
+            const confirmRes = await fetch("/api/pay/confirm", {
+              method: "POST",
+              headers: { "Content-Type": "application/json" },
+              body: JSON.stringify({ order_id: orderId }),
+              cache: "no-store",
+            });
+            if (confirmRes.ok && typeof window !== "undefined") {
+              window.dispatchEvent(new CustomEvent("vip:updated"));
+            }
+          } catch {
+            /* 忽略，不影响跳转 */
+          }
           onPaidRef.current();
+          return;
         }
       } catch {
-        // 继续轮询
+        // keep polling
+      } finally {
+        inFlight = false;
       }
-    }, POLL_INTERVAL);
 
-    return () => clearInterval(timer);
+      scheduleNext();
+    };
+
+    const handleResume = () => {
+      if (typeof document !== "undefined" && document.visibilityState === "hidden") {
+        return;
+      }
+      void tick();
+    };
+
+    document.addEventListener("visibilitychange", handleResume);
+    window.addEventListener("pageshow", handleResume);
+    window.addEventListener("focus", handleResume);
+    void tick();
+
+    return () => {
+      document.removeEventListener("visibilitychange", handleResume);
+      window.removeEventListener("pageshow", handleResume);
+      window.removeEventListener("focus", handleResume);
+      stop();
+    };
   }, []);
 
   return polling;