@arcblock/did-connect-react 3.3.9 → 3.4.0

package/lib/Connect/hooks/token.js

@@ -1,5 +1,5 @@
-import { useRef as I, useEffect as S } from "react";
-import W from "lodash/get";
+import { useRef as I, useEffect as C } from "react";
+import T from "lodash/get";
 import ue from "js-cookie";
 import B from "lodash/omitBy";
 import M from "lodash/isNil";
@@ -9,71 +9,63 @@ import { useReactive as H, useCreation as pe, useMemoizedFn as E } from "ahooks"
 import ke from "@arcblock/react-hooks/lib/useInterval";
 import me from "@arcblock/react-hooks/lib/useBrowser";
 import { getVisitorId as he, stringifyQuery as $, isUrl as ge } from "@arcblock/ux/lib/Util";
-import { WsClient as we } from "@arcblock/ws";
-import { sleep as q, getConnectedInfo as be, updateConnectedInfo as Ce, parseNextWorkflow as Q, createAxios as z, decodeConnectUrl as Se, parseTokenFromConnectUrl as xe } from "../../utils.js";
+import { WsClient as be } from "@arcblock/ws";
+import { sleep as q, getExtraHeaders as x, getConnectedInfo as we, updateConnectedInfo as Ce, parseNextWorkflow as Q, createAxios as z, decodeConnectUrl as Se, parseTokenFromConnectUrl as Ie } from "../../utils.js";
 import V from "../assets/locale.js";
-import Ie from "./security.js";
-import $e from "./page-show.js";
+import $e from "./security.js";
+import xe from "./page-show.js";
 import { RELAY_SOCKET_PREFIX as G } from "../../constant.js";
 function ve(o) {
-  const a = o || window.location.href;
-  return new URL(a).host;
+  const l = o || window.location.href;
+  return new URL(l).host;
 }
-function v(o) {
-  const a = {}, d = o || window.location.href;
-  if (d) {
-    const { hostname: w, protocol: i, port: p } = new URL(d);
-    a["x-real-hostname"] = w, a["x-real-port"] = p, a["x-real-protocol"] = i.endsWith(":") ? i.substring(0, i.length - 1) : i;
-  }
-  return a;
-}
-const R = {};
-function Ue({ action: o, prefix: a, checkFn: d, extraParams: w, baseUrl: i }) {
-  const p = `${a}/token?${$(w)}`;
-  return R[p] || (R[p] = async function() {
-    const k = +/* @__PURE__ */ new Date(), h = await d(p, { headers: v(i) }), g = +/* @__PURE__ */ new Date();
-    if (g - k < 500 && await q(500 - (g - k)), h?.data?.token)
+const P = {};
+function Ue({ action: o, prefix: l, checkFn: k, extraParams: v, baseUrl: m }) {
+  const d = `${l}/token?${$(v)}`;
+  return P[d] || (P[d] = async function() {
+    const f = +/* @__PURE__ */ new Date(), h = await k(d, { headers: x(m) }), g = +/* @__PURE__ */ new Date();
+    if (g - f < 500 && await q(500 - (g - f)), h?.data?.token)
       return h.data;
     if (h?.data?.error)
       throw new Error(h.data.error);
-    const U = ge(a) ? a : `${i}${a}`;
+    const U = ge(l) ? l : `${m}${l}`;
     throw new Error(`Error generating ${o} QR Code from: ${U}`);
-  }), R[p];
+  }), P[d];
 }
 const Ae = () => {
   try {
-    const a = new URL(window.location.href).searchParams.get("__connect_url__");
-    if (!a)
+    const l = new URL(window.location.href).searchParams.get("__connect_url__");
+    if (!l)
       return null;
-    const d = Se(a);
+    const k = Se(l);
     return {
-      token: xe(d),
-      url: d
+      token: Ie(k),
+      url: k
     };
   } catch (o) {
     return {
       error: o
     };
   }
-}, J = () => W(globalThis, "blocklet.appPid") || W(globalThis, "blocklet.appId") || W(globalThis, "env.appId") || "", ye = () => (W(globalThis, "env.apiPrefix") || "/").replace(/\/$/, "").replace(G, ""), P = (o) => `relay:${J()}:${o}`, Ee = () => window.location.protocol === "https:" ? "wss:" : "ws:", X = (o) => !o || new URL(o).origin === window.location.origin, Y = (o) => ["succeed", "error", "timeout", "busy"].includes(o);
+}, J = () => T(globalThis, "blocklet.appPid") || T(globalThis, "blocklet.appId") || T(globalThis, "env.appId") || "", ye = () => (T(globalThis, "env.apiPrefix") || "/").replace(/\/$/, "").replace(G, ""), R = (o) => `relay:${J()}:${o}`, Ee = () => window.location.protocol === "https:" ? "wss:" : "ws:", X = (o) => !o || new URL(o).origin === window.location.origin, Y = (o) => ["succeed", "error", "timeout", "busy"].includes(o);
 function Xe({
   action: o,
-  checkFn: a,
-  checkInterval: d,
-  checkTimeout: w,
-  extraParams: i,
-  locale: p,
+  checkFn: l,
+  checkInterval: k,
+  checkTimeout: v,
+  extraParams: m,
+  locale: d,
   prefix: _,
-  tokenKey: k,
+  tokenKey: f,
   encKey: h,
   onError: g,
   onSuccess: U,
   baseUrl: Z,
   autoConnect: K = !0,
   forceConnected: ee = !0,
-  saveConnect: te = !0,
+  saveConnect: re = !0,
   // FIXME: @zhanghan 将来需要设置为默认 false，仅在需要的时候进行保存 login, switch/role/profile
-  useSocket: re = !0,
+  useSocket: te = !0,
   provider: ne = "wallet"
 }) {
   const e = H({
@@ -94,41 +86,41 @@ function Xe({
     nextWorkflow: "",
     baseUrl: Z,
     prefix: `${_}/${o}`,
-    extraParams: B(i, M),
-    checkFn: a,
+    extraParams: B(m, M),
+    checkFn: l,
     results: {},
     action: o,
     provider: ne,
     reset() {
       this.error = "", this.status = "created";
     }
-  }), F = I(null), x = V[p] || V.en, f = pe(() => Ae(), []), oe = Math.ceil(w / d), l = H({
+  }), F = I(null), S = V[d] || V.en, u = pe(() => Ae(), []), oe = Math.ceil(v / k), c = H({
     onSuccessCalled: !1,
     cancelWhenScannedCounter: 0,
     isSocketAvailable: !1
-  }), { encryptKey: se, decrypt: A } = Ie(), D = me(), O = he(), c = I(null), m = I(null), b = {
+  }), { encryptKey: se, decrypt: A } = $e(), D = me(), O = he(), a = I(null), p = I(null), b = {
     ...e.extraParams,
-    locale: p,
-    [k]: e.token || f?.token,
+    locale: d,
+    [f]: e.token || u?.token,
     provider: e.provider
   }, L = async () => {
-    if (!(f?.token && f?.token === e.token)) {
+    if (!(u?.token && u?.token === e.token)) {
       try {
-        b[k] && e.status === "created" && await e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
-          headers: v(e.baseUrl)
+        b[f] && e.status === "created" && await e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
+          headers: x(e.baseUrl)
         });
       } catch {
       }
       try {
-        e.token && m.current && (await c.current.unsubscribe(P(e.token)), m.current = null);
+        e.token && p.current && (await a.current.unsubscribe(R(e.token)), p.current = null);
       } catch {
       }
     }
-  }, ae = E((...r) => {
-    const t = {
+  }, ae = E((...t) => {
+    const r = {
       [h]: se,
       ...e.extraParams,
-      locale: p,
+      locale: d,
       forceConnected: ee,
       // - autoConnect 请求参数用于控制服务端是否给钱包应用发送自动连接通知
       // - 使用 connect 组件时明确传入了 autoConnect = false, 则 autoConnect 请求参数 为 false
@@ -136,97 +128,97 @@ function Xe({
       //   (避免 "无限自动连接问题")
       // - 如果上次连接设备 (connected_wallet_os) 不是 ios/android, 则禁用自动连接
       autoConnect: K && // 如果是 wallet webview 环境, 不发送通知, 避免 wallet 连续弹出 auth 窗口 2 次 (#341)
-      !(D.wallet || D.arcSphere) && !l.cancelWhenScannedCounter && ["ios", "android"].includes(ue.get("connected_wallet_os")),
+      !(D.wallet || D.arcSphere) && !c.cancelWhenScannedCounter && ["ios", "android"].includes(ue.get("connected_wallet_os")),
       provider: e.provider
     };
-    O && (t.visitorId = O);
+    O && (r.visitorId = O);
     const n = fe(F.current || {}, ["sourceAppPid", "verbose"]);
     return Object.keys(n).forEach((s) => {
-      (!t[s] || s === h && n[s]) && (t[s] = n[s]);
-    }), f?.token && !t.sourceToken && (t.sourceToken = f?.token), Ue({
+      (!r[s] || s === h && n[s]) && (r[s] = n[s]);
+    }), u?.token && !r.sourceToken && (r.sourceToken = u?.token), Ue({
       action: o,
       prefix: e.prefix,
       baseUrl: e.baseUrl,
       checkFn: e.checkFn,
-      extraParams: t
-    })(...r);
-  }), y = E(async (r = !0) => {
-    r && L();
+      extraParams: r
+    })(...t);
+  }), y = E(async (t = !0) => {
+    t && L();
     try {
-      e.loading = !0, e.token = "", e.url = "", e.store = null, e.extraParams = B(i, M), await q();
-      const t = await ae(), n = t.extra || {};
-      e.loading = !1, e.token = t.token, e.url = t.url, e.status = "created", e.error = "", e.checkCount = 0, e.appInfo = t.appInfo, e.memberAppInfo = t.memberAppInfo, e.connectedDid = n.connectedDid, e.saveConnect = n.saveConnect;
-    } catch (t) {
-      e.loading = !1, e.status = "error", e.error = `${x.generateError}: ${t.message}`;
+      e.loading = !0, e.token = "", e.url = "", e.store = null, e.extraParams = B(m, M), await q();
+      const r = await ae(), n = r.extra || {};
+      e.loading = !1, e.token = r.token, e.url = r.url, e.status = "created", e.error = "", e.checkCount = 0, e.appInfo = r.appInfo, e.memberAppInfo = r.memberAppInfo, e.connectedDid = n.connectedDid, e.saveConnect = n.saveConnect;
+    } catch (r) {
+      e.loading = !1, e.status = "error", e.error = `${S.generateError}: ${r.message}`;
     }
   }), ce = E(() => {
-    l.cancelWhenScannedCounter++;
+    c.cancelWhenScannedCounter++;
   });
-  S(() => {
-    l.cancelWhenScannedCounter > 0 && (i.nw ? g(new Error(x.retryForbidden)) : y());
-  }, [l.cancelWhenScannedCounter]), S(() => (!(/^((?!chrome|android).)*safari/i.test(globalThis.navigator.userAgent) || Object.hasOwn(globalThis?.blocklet || {}, "DID_CONNECT_DISABLE_SOCKET")) && re && X(e.baseUrl) && J() ? (!c.current || c.current.isConnected() === !1 || c.current.baseUrl !== e.baseUrl) && (c.current = new we(
+  C(() => {
+    c.cancelWhenScannedCounter > 0 && (m.nw ? g(new Error(S.retryForbidden)) : y());
+  }, [c.cancelWhenScannedCounter]), C(() => (!(/^((?!chrome|android).)*safari/i.test(globalThis.navigator.userAgent) || Object.hasOwn(globalThis?.blocklet || {}, "DID_CONNECT_DISABLE_SOCKET")) && te && X(e.baseUrl) && J() ? (!a.current || a.current.isConnected() === !1 || a.current.baseUrl !== e.baseUrl) && (a.current = new be(
     `${Ee()}//${ve(e.baseUrl)}${ye()}${G}/relay`,
     {
       longpollerTimeout: 5e3,
       // connection timeout
       heartbeatIntervalMs: 30 * 1e3
     }
-  ), c.current.baseUrl = e.baseUrl, c.current.onOpen(() => {
-    l.isSocketAvailable = !0;
-  }), c.current.connect()) : l.isSocketAvailable && (l.isSocketAvailable = !1), () => {
-    c.current && (c.current.disconnect(), c.current = null);
+  ), a.current.baseUrl = e.baseUrl, a.current.onOpen(() => {
+    c.isSocketAvailable = !0;
+  }), a.current.connect()) : c.isSocketAvailable && (c.isSocketAvailable = !1), () => {
+    a.current && (a.current.disconnect(), a.current = null);
   }), [e.baseUrl]);
-  const ie = () => !e.token || e.loading || Y(e.status) ? null : d, j = I(null);
-  S(() => {
+  const ie = () => !e.token || e.loading || Y(e.status) ? null : k, j = I(null);
+  C(() => {
     j.current = { state: e, params: b };
-  }), S(() => () => {
-    const { state: r, params: t } = j.current;
-    r.status === "created" && t[k] && r.checkFn(`${r.prefix}/timeout?${$(t)}`, {
-      headers: v(r.baseUrl)
+  }), C(() => () => {
+    const { state: t, params: r } = j.current;
+    t.status === "created" && r[f] && t.checkFn(`${t.prefix}/timeout?${$(r)}`, {
+      headers: x(t.baseUrl)
     });
   }, []);
-  const C = E(async (r = !1) => {
-    if ((e.checking || document.hidden) && !r || (e.status === "created" && e.checkCount++, l.isSocketAvailable && !r))
+  const w = E(async (t = !1) => {
+    if ((e.checking || document.hidden) && !t || (e.status === "created" && e.checkCount++, c.isSocketAvailable && !t))
       return null;
     try {
       e.checking = !0;
-      const t = await e.checkFn(`${e.prefix}/status?${$(b)}`, {
-        headers: v(e.baseUrl)
-      }), { status: n, error: u, mfaCode: s = 0 } = t.data;
-      if (e.store = t.data, e.mfaCode = s, e.checking = !1, e.status === "scanned" && n === "created" || (e.status = n), n === "error" && u) {
-        const N = new Error(u);
-        throw N.response = t, N;
+      const r = await e.checkFn(`${e.prefix}/status?${$(b)}`, {
+        headers: x(e.baseUrl)
+      }), { status: n, error: i, mfaCode: s = 0 } = r.data;
+      if (e.store = r.data, e.mfaCode = s, e.checking = !1, e.status === "scanned" && n === "created" || (e.status = n), n === "error" && i) {
+        const N = new Error(i);
+        throw N.response = r, N;
       }
-      return t.data;
-    } catch (t) {
-      const { response: n } = t;
+      return r.data;
+    } catch (r) {
+      const { response: n } = r;
       if (n?.status) {
-        b[k] && e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
-          headers: v(e.baseUrl)
+        b[f] && e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
+          headers: x(e.baseUrl)
         });
-        const u = n?.data?.error ? n.data.error : t.message, s = new Error(u);
-        s.code = n.status, e.status = "error", e.checking = !1, e.error = u, g(s, e?.store, A);
+        const i = n?.data?.error ? n.data.error : r.message, s = new Error(i);
+        s.code = n.status, e.status = "error", e.checking = !1, e.error = i, g(s, e?.store, A);
       } else
-        e.status = "error", e.checking = !1, e.error = x.generateError;
+        e.status = "error", e.checking = !1, e.error = S.generateError;
     }
     return null;
   }), le = async () => {
     try {
-      if (f.error)
-        throw f.error;
-      e.loading = !0, e.token = f.token, e.url = f.url, e.inExistingSession = !0;
-      let r = await C();
-      if ((["scanned"].includes(r?.status) || r?.sourceToken === f?.token) && (await y(!1), r = await C()), F.current = r?.extraParams || null, r?.status && !["created"].includes(r?.status))
-        throw new Error(`${x.invalidSessionStatus} [${r.status}]`);
-    } catch (r) {
-      e.status = "error", e.error = r.message, g(r, e?.store, A);
+      if (u.error)
+        throw u.error;
+      e.loading = !0, e.token = u.token, e.url = u.url, e.inExistingSession = !0;
+      let t = await w();
+      if ((["scanned"].includes(t?.status) || t?.sourceToken === u?.token) && (await y(!1), t = await w()), F.current = t?.extraParams || null, t?.status && !["created"].includes(t?.status))
+        throw new Error(`${S.invalidSessionStatus} [${t.status}]`);
+    } catch (t) {
+      e.status = "error", e.error = t.message, g(t, e?.store, A);
     } finally {
       e.loading = !1;
     }
   };
-  S(() => {
+  C(() => {
     if (!e.token && !e.store && !e.loading && !e.error)
-      if (f)
+      if (u)
         le();
       else {
         y(!1);
@@ -236,67 +228,67 @@ function Xe({
       e.status = "timeout", L();
       return;
     }
-    if (e.status === "succeed" && !l.onSuccessCalled) {
-      const r = be({
+    if (e.status === "succeed" && !c.onSuccessCalled) {
+      const t = we({
         ...e.store,
         appInfo: e.appInfo,
         memberAppInfo: e.memberAppInfo
       });
-      if (X(e.baseUrl) && te && e.saveConnect && e.store.did && Ce(r, !0), e.store) {
-        const { nextWorkflow: t } = e.store;
-        if (t)
+      if (X(e.baseUrl) && re && e.saveConnect && e.store.did && Ce(t, !0), e.store) {
+        const { nextWorkflow: r } = e.store;
+        if (r)
           try {
-            const n = Q(t, k);
-            e.nextWorkflow = n?.nextWorkflow || t, Object.assign(e, n), e.store = null, e.error = "", e.url = "", e.appInfo = null, e.memberAppInfo = null, e.checkCount = 0, e.status = "scanned", e.results = {
+            const n = Q(r, f);
+            e.nextWorkflow = n?.nextWorkflow || r, Object.assign(e, n), e.store = null, e.error = "", e.url = "", e.appInfo = null, e.memberAppInfo = null, e.checkCount = 0, e.status = "scanned", e.results = {
               ...e.results,
               [e.token]: e.store
-            }, e.checkFn = z({ baseURL: n.baseUrl, timeout: 8e3 }).get, C();
+            }, e.checkFn = z({ baseURL: n.baseUrl, timeout: 8e3 }).get, w();
           } catch (n) {
-            console.error(`Invalid nextWorkflow: ${t}`, n), e.status = "error", e.error = `Invalid nextWorkflow: ${t}: ${n.message}`;
+            console.error(`Invalid nextWorkflow: ${r}`, n), e.status = "error", e.error = `Invalid nextWorkflow: ${r}: ${n.message}`;
           }
         else if (e.nextWorkflow && (e.nextWorkflow = "", e.url = ""), typeof U == "function") {
           const n = Object.values({ ...e.results, [e.token]: e.store });
-          l.onSuccessCalled = !0;
-          const u = n.length > 1 ? n : n[0], s = de(u);
+          c.onSuccessCalled = !0;
+          const i = n.length > 1 ? n : n[0], s = de(i);
           U(s, A, {
-            sourceAppPid: i?.sourceAppPid,
-            ...r
+            sourceAppPid: m?.sourceAppPid,
+            ...t
           });
         }
       }
       return;
     }
     if (e.store?.nextWorkflow) {
-      const { nextWorkflow: r } = e.store;
+      const { nextWorkflow: t } = e.store;
       try {
-        const t = Q(r, k);
-        e.nextWorkflow = t?.nextWorkflow || r, Object.assign(e, t), e.store = null, e.error = "", e.url = "", e.appInfo = null, e.memberAppInfo = null, e.checkCount = 0, e.status = "scanned", e.results = {
+        const r = Q(t, f);
+        e.nextWorkflow = r?.nextWorkflow || t, Object.assign(e, r), e.store = null, e.error = "", e.url = "", e.appInfo = null, e.memberAppInfo = null, e.checkCount = 0, e.status = "scanned", e.results = {
           ...e.results,
           [e.token]: e.store
-        }, e.checkFn = z({ baseURL: t.baseUrl, timeout: 8e3 }).get, C();
-      } catch (t) {
-        console.error(`Invalid nextWorkflow: ${r}`, t), e.status = "error", e.error = `Invalid nextWorkflow: ${r}: ${t.message}`;
+        }, e.checkFn = z({ baseURL: r.baseUrl, timeout: 8e3 }).get, w();
+      } catch (r) {
+        console.error(`Invalid nextWorkflow: ${t}`, r), e.status = "error", e.error = `Invalid nextWorkflow: ${t}: ${r.message}`;
       }
     }
   });
-  const T = I(0);
-  return S(() => {
-    if (e.token && l.isSocketAvailable && c.current) {
-      let r = !1;
-      m.current ? m.current.token !== e.token && (c.current.unsubscribe(P(m.current.token)), r = !0) : r = !0, r && (m.current = c.current.subscribe(P(e.token)), T.current = 0, m.current.token = e.token, m.current.on("updated", ({ response: t }) => {
-        const n = +new Date(t.updatedAt);
-        if (n <= T.current) {
-          console.warn("Ignore outdated message", t);
+  const W = I(0);
+  return C(() => {
+    if (e.token && c.isSocketAvailable && a.current) {
+      let t = !1;
+      p.current ? p.current.token !== e.token && (a.current.unsubscribe(R(p.current.token)), t = !0) : t = !0, t && (p.current = a.current.subscribe(R(e.token)), W.current = 0, p.current.token = e.token, p.current.on("updated", ({ response: r }) => {
+        const n = +new Date(r.updatedAt);
+        if (n <= W.current) {
+          console.warn("Ignore outdated message", r);
           return;
         }
-        T.current = n;
-        let { status: u, error: s } = t;
-        u === "forbidden" && (s = x.forbidden, u = "error"), e.status = u, e.mfaCode = t.mfaCode || 0, e.store = t, e.error = s, s && g(new Error(s), e?.store, A);
+        W.current = n;
+        let { status: i, error: s } = r;
+        i === "forbidden" && (s = S.forbidden, i = "error"), e.status = i, e.mfaCode = r.mfaCode || 0, e.store = r, e.error = s, s && g(new Error(s), e?.store, A);
       }));
     }
-  }, [e.token, l.isSocketAvailable, c.current]), ke(C, ie()), $e(() => {
+  }, [e.token, c.isSocketAvailable, a.current]), ke(w, ie()), xe(() => {
     e.token && Y(e.status) === !1 && (e.checking = !1, setTimeout(() => {
-      C(!0);
+      w(!0);
     }, 100));
   }), { state: e, generate: y, cancelWhenScanned: ce };
 }