h5微信授权

2026-07-14 20:09:36 +08:00
parent 9b13d02dde
commit 1ad49f1acf
14 changed files with 569 additions and 79 deletions
@@ -0,0 +1,94 @@
+import Taro, { useDidShow } from '@tarojs/taro';
+import { captureIosJssdkEntryUrl } from '@dukang/weixin-sdk';
+import { useEffect, useRef } from 'react';
+import { finishLoginNavigate, goLogin } from '../lib/auth-nav';
+import { toast } from '../lib/api';
+import { saveWechatLoginResult } from '../lib/pay-wechat';
+import { applyWechatShare } from '../lib/wechat-share';
+import { handleWechatAuthCallback } from '../lib/wechat-auth';
+import { isWechatEnv } from '../lib/weixin';
+
+function currentPagePathWithQuery(): string {
+  const pages = Taro.getCurrentPages();
+  const cur = pages[pages.length - 1] as
+    | { route?: string; options?: Record<string, string | undefined> }
+    | undefined;
+  if (!cur?.route) {
+    if (typeof window !== 'undefined') {
+      return `${window.location.pathname}${window.location.search}`;
+    }
+    return '';
+  }
+  const path = cur.route.startsWith('/') ? cur.route : `/${cur.route}`;
+  const opts = cur.options ?? {};
+  const qs = Object.entries(opts)
+    .filter(([k, v]) => v != null && v !== '' && k !== 'code' && k !== 'state')
+    .map(([k, v]) => `${encodeURIComponent(k)}=${encodeURIComponent(String(v))}`)
+    .join('&');
+  return qs ? `${path}?${qs}` : path;
+}
+
+/**
+ * H5 全局：
+ * 1. 默认分享卡片
+ * 2. 公众号 OAuth ?code= 统一回调（避免各页重复消费 code）
+ */
+export default function WechatShareBootstrap() {
+  const handlingCode = useRef(false);
+
+  useEffect(() => {
+    if (process.env.TARO_ENV === 'h5') {
+      captureIosJssdkEntryUrl();
+    }
+  }, []);
+
+  useDidShow(() => {
+    if (process.env.TARO_ENV !== 'h5') return;
+    void applyWechatShare().catch(() => {});
+
+    if (!isWechatEnv()) return;
+    if (typeof window === 'undefined') return;
+    const params = new URLSearchParams(window.location.search);
+    if (!params.get('code')) return;
+    if (handlingCode.current) return;
+    handlingCode.current = true;
+
+    const returnFromLogin = (() => {
+      const path = currentPagePathWithQuery();
+      if (path.includes('/pages/login/')) {
+        try {
+          return decodeURIComponent(params.get('return') || '') || undefined;
+        } catch {
+          return undefined;
+        }
+      }
+      return undefined;
+    })();
+
+    handleWechatAuthCallback()
+      .then((result) => {
+        if (!result) return;
+        if (result.needBindPhone && result.wxSessionKey) {
+          goLogin(returnFromLogin, {
+            bindMode: '1',
+            wxSessionKey: result.wxSessionKey,
+          });
+          return;
+        }
+        if (saveWechatLoginResult(result)) {
+          toast('微信授权成功', 'success');
+          if (returnFromLogin !== undefined || currentPagePathWithQuery().includes('/pages/login/')) {
+            finishLoginNavigate(returnFromLogin || params.get('return') || undefined);
+          }
+        }
+      })
+      .catch((e) => {
+        toast(e instanceof Error ? e.message : '微信授权失败');
+      })
+      .finally(() => {
+        handlingCode.current = false;
+      });
+  });
+
+  return null;
+}