@arcblock/did-connect-react 3.1.0

package/lib/Connect/hooks/token.js

@@ -0,0 +1,305 @@
+import { useRef as I, useEffect as S } from "react";
+import W from "lodash/get";
+import ue from "js-cookie";
+import B from "lodash/omitBy";
+import M from "lodash/isNil";
+import fe from "lodash/omit";
+import de from "lodash/cloneDeep";
+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 V from "../assets/locale.js";
+import Ie from "./security.js";
+import $e 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;
+}
+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)
+      return h.data;
+    if (h?.data?.error)
+      throw new Error(h.data.error);
+    const U = ge(a) ? a : `${i}${a}`;
+    throw new Error(`Error generating ${o} QR Code from: ${U}`);
+  }), R[p];
+}
+const Ae = () => {
+  try {
+    const a = new URL(window.location.href).searchParams.get("__connect_url__");
+    if (!a)
+      return null;
+    const d = Se(a);
+    return {
+      token: xe(d),
+      url: d
+    };
+  } 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);
+function Xe({
+  action: o,
+  checkFn: a,
+  checkInterval: d,
+  checkTimeout: w,
+  extraParams: i,
+  locale: p,
+  prefix: _,
+  tokenKey: k,
+  encKey: h,
+  onError: g,
+  onSuccess: U,
+  baseUrl: Z,
+  autoConnect: K = !0,
+  forceConnected: ee = !0,
+  saveConnect: te = !0,
+  // FIXME: @zhanghan 将来需要设置为默认 false，仅在需要的时候进行保存 login, switch/role/profile
+  useSocket: re = !0,
+  provider: ne = "wallet"
+}) {
+  const e = H({
+    checking: !1,
+    loading: !1,
+    token: "",
+    url: "",
+    store: null,
+    status: "created",
+    error: "",
+    checkCount: 0,
+    mfaCode: 0,
+    appInfo: null,
+    memberAppInfo: null,
+    connectedDid: "",
+    saveConnect: !1,
+    inExistingSession: !1,
+    nextWorkflow: "",
+    baseUrl: Z,
+    prefix: `${_}/${o}`,
+    extraParams: B(i, M),
+    checkFn: a,
+    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({
+    onSuccessCalled: !1,
+    cancelWhenScannedCounter: 0,
+    isSocketAvailable: !1
+  }), { encryptKey: se, decrypt: A } = Ie(), D = me(), O = he(), c = I(null), m = I(null), b = {
+    ...e.extraParams,
+    locale: p,
+    [k]: e.token || f?.token,
+    provider: e.provider
+  }, L = async () => {
+    if (!(f?.token && f?.token === e.token)) {
+      try {
+        b[k] && e.status === "created" && await e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
+          headers: v(e.baseUrl)
+        });
+      } catch {
+      }
+      try {
+        e.token && m.current && (await c.current.unsubscribe(P(e.token)), m.current = null);
+      } catch {
+      }
+    }
+  }, ae = E((...r) => {
+    const t = {
+      [h]: se,
+      ...e.extraParams,
+      locale: p,
+      forceConnected: ee,
+      // - autoConnect 请求参数用于控制服务端是否给钱包应用发送自动连接通知
+      // - 使用 connect 组件时明确传入了 autoConnect = false, 则 autoConnect 请求参数 为 false
+      // - 如果 cancelWhenScannedCounter > 0, 说明用户进行过 cancel 操作, 则临时禁用自动连接, autoConnect 请求参数 为 false
+      //   (避免 "无限自动连接问题")
+      // - 如果上次连接设备 (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")),
+      provider: e.provider
+    };
+    O && (t.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({
+      action: o,
+      prefix: e.prefix,
+      baseUrl: e.baseUrl,
+      checkFn: e.checkFn,
+      extraParams: t
+    })(...r);
+  }), y = E(async (r = !0) => {
+    r && 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}`;
+    }
+  }), ce = E(() => {
+    l.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(
+    `${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);
+  }), [e.baseUrl]);
+  const ie = () => !e.token || e.loading || Y(e.status) ? null : d, j = I(null);
+  S(() => {
+    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)
+    });
+  }, []);
+  const C = E(async (r = !1) => {
+    if ((e.checking || document.hidden) && !r || (e.status === "created" && e.checkCount++, l.isSocketAvailable && !r))
+      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;
+      }
+      return t.data;
+    } catch (t) {
+      const { response: n } = t;
+      if (n?.status) {
+        b[k] && e.checkFn(`${e.prefix}/timeout?${$(b)}`, {
+          headers: v(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);
+      } else
+        e.status = "error", e.checking = !1, e.error = x.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);
+    } finally {
+      e.loading = !1;
+    }
+  };
+  S(() => {
+    if (!e.token && !e.store && !e.loading && !e.error)
+      if (f)
+        le();
+      else {
+        y(!1);
+        return;
+      }
+    if (e.status !== "timeout" && e.checkCount > oe) {
+      e.status = "timeout", L();
+      return;
+    }
+    if (e.status === "succeed" && !l.onSuccessCalled) {
+      const r = be({
+        ...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)
+          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 = {
+              ...e.results,
+              [e.token]: e.store
+            }, e.checkFn = z({ baseURL: n.baseUrl, timeout: 8e3 }).get, C();
+          } catch (n) {
+            console.error(`Invalid nextWorkflow: ${t}`, n), e.status = "error", e.error = `Invalid nextWorkflow: ${t}: ${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);
+          U(s, A, {
+            sourceAppPid: i?.sourceAppPid,
+            ...r
+          });
+        }
+      }
+      return;
+    }
+    if (e.store?.nextWorkflow) {
+      const { nextWorkflow: r } = 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 = {
+          ...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}`;
+      }
+    }
+  });
+  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);
+          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);
+      }));
+    }
+  }, [e.token, l.isSocketAvailable, c.current]), ke(C, ie()), $e(() => {
+    e.token && Y(e.status) === !1 && (e.checking = !1, setTimeout(() => {
+      C(!0);
+    }, 100));
+  }), { state: e, generate: y, cancelWhenScanned: ce };
+}
+export {
+  Xe as default
+};

package/lib/Connect/hooks/use-apps.js

@@ -0,0 +1,19 @@
+import { useCreation as u } from "ahooks";
+import { getFederatedEnabled as m, getMaster as c, getApps as A } from "@arcblock/ux/lib/Util/federated";
+import p from "lodash/isUndefined";
+import { use as P } from "react";
+import { SessionContext as g } from "../../Session/context.js";
+function F({ blocklet: t, action: f, sourceAppPid: r, connectState: i, enableSwitchApp: d }) {
+  const s = P(g), n = u(() => d === !0 ? !0 : !(d === !1 || s?.session?.user && !["login", "invite", "connect-to-did-space", "connect-to-did-domain"].includes(f)), [f, s?.session?.user]), o = u(() => {
+    if (p(r)) {
+      const a = m(t), e = c(t);
+      if (a && e?.appPid)
+        return e.appPid;
+    }
+    return r;
+  }, [r, t]);
+  return { appInfoList: u(() => A(t).filter((e) => n ? !0 : p(r) ? p(i.sourceAppPid) ? s?.session?.user ? e.sourceAppPid === s.session.user.sourceAppPid : e.sourceAppPid === o : e.sourceAppPid === i.sourceAppPid : e.sourceAppPid === r), [n, r, i.sourceAppPid, o, s?.session?.user]), autoGenerateSourceAppPid: o, canSwitchApp: n };
+}
+export {
+  F as default
+};

package/lib/Connect/hooks/use-quick-connect.js

@@ -0,0 +1,97 @@
+import { useCreation as l, useMemoizedFn as E, useReactive as L, useRequest as T } from "ahooks";
+import { joinURL as u } from "ufo";
+import k from "lodash/pick";
+import A from "lodash/unionBy";
+import D from "lodash/sortBy";
+import x from "@arcblock/ux/lib/Toast";
+import { getFederatedEnabled as B, getMaster as C } from "@arcblock/ux/lib/Util/federated";
+import { BLOCKLET_SERVICE_PATH_PREFIX as P } from "../../constant.js";
+import { createAxios as _, logger as m } from "../../utils.js";
+const f = u(P, "/api/user-session");
+function O({
+  appPid: y,
+  sourceAppPid: U,
+  loginAppPid: S,
+  autoFetch: w = !0,
+  baseUrl: a = "/",
+  fetchAll: I = !1
+} = {}) {
+  const p = l(() => _(
+    {
+      baseURL: u(a, f)
+    },
+    { lazy: !0, lazyTime: 500 }
+  ), [a]), h = E(async () => {
+    if (!window.blocklet)
+      return [];
+    let r = [];
+    try {
+      ({ data: r = [] } = await p.get(""));
+    } catch (t) {
+      m.error("Failed to get user-sessions", t);
+    }
+    const s = r.map((t) => ({
+      ...t,
+      appUrl: a
+    })), e = B(), n = C();
+    if (I && e && n) {
+      let t = [];
+      try {
+        ({ data: t = [] } = await p.get(u(n.appUrl, f)));
+      } catch (c) {
+        m.error("Failed to get master user-sessions", c);
+      }
+      return s.concat(t.map((c) => ({ ...c, appUrl: n.appUrl })));
+    }
+    return !e || !n ? s.filter((t) => t?.user?.sourceAppPid === null) : s;
+  }), i = L({
+    loaded: !1,
+    loadingId: null
+  }), o = T(h, {
+    manual: !w,
+    onFinally() {
+      i.loaded = !0;
+    },
+    refreshDeps: [a]
+  }), F = async (r) => {
+    const s = k(r, ["userDid", "visitorId", "passportId", "id"]);
+    try {
+      i.loadingId = r.id;
+      const { data: e } = await p.post(
+        "login",
+        { ...s, appPid: S },
+        {
+          baseURL: u(r.appUrl || "/", f)
+        }
+      );
+      return e;
+    } catch (e) {
+      const n = e.response ? e.response?.data?.error || e.response?.data : e.message;
+      throw x.error(n), m.error("Quick login error", e), e;
+    } finally {
+      i.loadingId = null;
+    }
+  }, g = l(() => o.data ? o.data.filter((r) => r.status !== "expired") : [], [o.data]), d = l(() => {
+    const r = (/* @__PURE__ */ new Date()).getTime(), s = D(g, (e) => r - new Date(e.updatedAt).getTime());
+    return A(
+      s,
+      (e) => [
+        e.userDid
+        // 暂不以用户角色来做去重的因子
+        // x.user.role
+      ].join("_")
+    );
+  }, [g, U, y]), R = l(() => !!(d && d.length > 0), [d]);
+  return {
+    loadingId: i.loadingId,
+    userSessions: d,
+    quickLoginEnabled: R,
+    loginUserSession: F,
+    loading: o.loading,
+    loaded: o.loaded,
+    refresh: o.refreshAsync
+  };
+}
+export {
+  O as default
+};