@arcblock/did-connect-react 3.1.34 → 3.1.36

package/lib/Session/hooks/use-session-token.js

@@ -1,173 +1,181 @@
-import { useState as K } from "react";
-import { useCreation as M, useMemoizedFn as c, useInterval as Z } from "ahooks";
-import { joinURL as U, withQuery as H } from "ufo";
-import k from "js-cookie";
-import { setVisitorId as ee, getCookieOptions as re } from "@arcblock/ux/lib/Util";
-import oe from "jwt-decode";
-import ne from "@arcblock/ux/lib/Toast";
-import se from "p-retry";
-import te from "lodash/noop";
-import { ReactGA as ie } from "@arcblock/ux/lib/withTracker";
-import { GA_LAST_SOURCE_PROVIDER as ce } from "@arcblock/ux/lib/withTracker/constant";
-import fe from "../../Service/index.js";
-import { createAxios as le, getBrowserLang as ue, logger as A, debug as u, sleep as ae, decrypt as de } from "../../utils.js";
-import { CHECK_INTERVAL_TIME as m, LANG_COOKIE_NAME as he } from "../../constant.js";
+import { useState as M } from "react";
+import { useCreation as N, useMemoizedFn as c, useInterval as oe } from "ahooks";
+import { joinURL as A, withQuery as ne } from "ufo";
+import h from "js-cookie";
+import { setVisitorId as se, getCookieOptions as te } from "@arcblock/ux/lib/Util";
+import ie from "jwt-decode";
+import ce from "@arcblock/ux/lib/Toast";
+import fe from "p-retry";
+import le from "lodash/noop";
+import { ReactGA as ue } from "@arcblock/ux/lib/withTracker";
+import { GA_LAST_SOURCE_PROVIDER as ae } from "@arcblock/ux/lib/withTracker/constant";
+import de from "../../Service/index.js";
+import { createAxios as ke, getBrowserLang as he, logger as O, debug as u, sleep as me, decrypt as Te } from "../../utils.js";
+import { CHECK_INTERVAL_TIME as m, LANG_COOKIE_NAME as pe } from "../../constant.js";
 class T extends Error {
 }
-const ke = 10, v = (e) => de(e, localStorage.getItem("__encKey"), localStorage.getItem("__decKey"));
-function Ee({
+const Re = 10, p = (e) => Te(e, localStorage.getItem("__encKey"), localStorage.getItem("__decKey"));
+function Ve({
   state: e,
   pageState: i,
-  serviceHost: p,
-  sessionTokenStorage: O,
-  refreshTokenStorage: V,
-  lazyRefreshToken: N,
-  apiOptions: b = {},
-  onRefresh: z = te
+  serviceHost: R,
+  sessionTokenStorage: V,
+  csrfTokenStorage: b,
+  refreshTokenStorage: q,
+  lazyRefreshToken: z,
+  apiOptions: P = {},
+  onRefresh: j = le
 }) {
-  const [q, R] = K(0), [P, d] = K(m), { getToken: y, setToken: C, removeToken: j } = O, { getToken: w, setToken: g, removeToken: B } = V, $ = M(() => le({
-    baseURL: p,
+  const [D, y] = M(0), [B, d] = M(m), { getToken: w, setToken: v, removeToken: $ } = V, { setToken: Q } = b, { getToken: C, setToken: g, removeToken: X } = q, Y = N(() => ke({
+    baseURL: R,
     timeout: 10 * 1e3,
     secure: !0
-  }), [p]), h = () => {
-    j(), B();
-  }, _ = M(() => fe(
+  }), [R]), k = () => {
+    $(), X();
+  }, _ = N(() => de(
     {
-      sessionTokenStorage: O,
-      refreshTokenStorage: V,
-      serviceHost: p,
+      sessionTokenStorage: V,
+      refreshTokenStorage: q,
+      serviceHost: R,
       authServicePrefix: i.prefix,
       onRefreshTokenError() {
-        h(), e.user && (e.user = null);
+        k(), e.user && (e.user = null);
       }
     },
-    b
-  ), [p, i.prefix]), x = () => {
-    const o = re({ returnDomain: !1 });
-    k.remove("connected_did", o), k.remove("connected_pk", o), k.remove("connected_app", o), k.remove("connected_wallet_os", o), h();
-  }, Q = c(() => {
-    const o = k.get(he) || ue();
-    o !== i.currentLocale && (i.currentLocale = o);
-  }), X = c(async () => {
-    const o = () => _.get(U(i.notificationPrefix, "/unread-count"));
+    P
+  ), [R, i.prefix]), x = () => {
+    const r = te({ returnDomain: !1 });
+    h.remove("connected_did", r), h.remove("connected_pk", r), h.remove("connected_app", r), h.remove("connected_wallet_os", r), k();
+  }, J = c(() => {
+    const r = h.get(pe) || he();
+    r !== i.currentLocale && (i.currentLocale = r);
+  }), W = c(async () => {
+    const r = () => _.get(A(i.notificationPrefix, "/unread-count"));
     try {
-      const { data: n } = await o();
+      const { data: n } = await r();
       return n.unReadCount;
     } catch (n) {
       return console.error("getUnreadCount error", n), 0;
     }
-  }), D = c(
-    async ({ showProgress: o = !1, requestFn: n = null, onlyRefreshToken: s = !1, type: t = "refreshToken" }) => {
+  }), F = c(
+    async ({ showProgress: r = !1, requestFn: n = null, onlyRefreshToken: t = !1, type: s = "refreshToken" }) => {
       try {
         if (e.loading)
           return console.warn("SessionProvider.refresh is currently in progress, call it will be noop"), { loading: !0 };
-        o && (e.loading = !0);
-        const { data: r, status: f } = await n(), a = r?.user?.did;
-        if (a && ie.set({
+        r && (e.loading = !0);
+        const { data: o, status: f } = await n(), a = o?.user?.did;
+        if (a && ue.set({
           user_id: a
-        }), r?.user?.sourceProvider && localStorage.setItem(ce, r.user.sourceProvider), f === 400 && (h(), e.user = null, e.error = ""), r.error)
-          throw e.error = r.error, e.open = !1, e.unReadCount = 0, new Error(r.error);
-        if (r.user) {
+        }), o?.user?.sourceProvider && localStorage.setItem(ae, o.user.sourceProvider), f === 400 && (k(), e.user = null, e.error = ""), o.error)
+          throw e.error = o.error, e.open = !1, e.unReadCount = 0, new Error(o.error);
+        if (o.user) {
           try {
-            const l = await X();
+            const l = await W();
             e.unReadCount = l || 0;
           } catch (l) {
-            A.error("getUnreadCount error", l), e.unReadCount = 0;
+            O.error("getUnreadCount error", l), e.unReadCount = 0;
           }
-          r.nextToken && (C(r.nextToken), z({
-            type: t,
-            sessionToken: r.nextToken,
-            refreshToken: r.nextRefreshToken,
-            user: r.user
-          }), r.nextRefreshToken && g(r.nextRefreshToken)), s || (e.open = !1, e.user = r.user, e.provider = r.provider, e.walletOS = r.walletOS), y() || ne.error("Can't write session token, please use https url to access this page");
+          o.nextToken && (v(o.nextToken), j({
+            type: s,
+            sessionToken: o.nextToken,
+            refreshToken: o.nextRefreshToken,
+            user: o.user
+          }), o.nextRefreshToken && g(o.nextRefreshToken)), t || (e.open = !1, e.user = o.user, e.provider = o.provider, e.walletOS = o.walletOS), w() || ce.error("Can't write session token, please use https url to access this page");
         } else
-          throw o && (e.loading = i.autoConnect), e.open = i.autoConnect, e.user = null, e.provider = "", e.walletOS = "", e.unReadCount = 0, new T("Invalid token");
-        return { data: r, status: f };
-      } catch (r) {
-        throw A.error("SessionProvider.refresh error", r), e.open = !1, e.error = r.message, r?.response?.status === 400 ? new T("Invalid token") : r;
+          throw r && (e.loading = i.autoConnect), e.open = i.autoConnect, e.user = null, e.provider = "", e.walletOS = "", e.unReadCount = 0, new T("Invalid token");
+        return { data: o, status: f };
+      } catch (o) {
+        throw O.error("SessionProvider.refresh error", o), e.open = !1, e.error = o.message, o?.response?.status === 400 ? new T("Invalid token") : o;
       } finally {
-        o && (e.loading = !1);
+        r && (e.loading = !1);
       }
     }
-  ), I = c(async (o = !1, n = !1) => {
-    u("handleRefreshToken", { showProgress: o, onlyRefreshToken: n });
-    const s = () => $.post(H(U(i.prefix, "/refreshSession")), null, {
+  ), I = c(async (r = !1, n = !1) => {
+    u("handleRefreshToken", { showProgress: r, onlyRefreshToken: n });
+    const t = () => Y.post(ne(A(i.prefix, "/refreshSession")), null, {
       headers: {
-        authorization: `Bearer ${encodeURIComponent(w())}`
+        authorization: `Bearer ${encodeURIComponent(C())}`
       }
     });
     d(void 0);
     try {
-      await D({ showProgress: o, requestFn: s, onlyRefreshToken: n, type: "refreshToken" }), R(0), d(m);
-    } catch (t) {
-      t instanceof T ? (console.warn("refresh token failed, remove all tokens"), R(0), h(), d(m)) : (A.error("refresh token failed, unexpected error:", t), R(q + 1), await ae(m), q >= ke ? (R(0), d(void 0)) : d(m));
+      await F({ showProgress: r, requestFn: t, onlyRefreshToken: n, type: "refreshToken" }), y(0), d(m);
+    } catch (s) {
+      s instanceof T ? (console.warn("refresh token failed, remove all tokens"), y(0), k(), d(m)) : (O.error("refresh token failed, unexpected error:", s), y(D + 1), await me(m), D >= Re ? (y(0), d(void 0)) : d(m));
     }
-  }), F = c((o = !1) => {
-    if (N)
+  }), G = c((r = !1) => {
+    if (z)
       return;
-    const n = y();
-    let s = o;
-    const t = (/* @__PURE__ */ new Date()).getTime();
+    const n = w();
+    let t = r;
+    const s = (/* @__PURE__ */ new Date()).getTime();
     if (n)
       try {
-        const { exp: r, iat: f } = oe(n);
-        r * 1e3 - t < t - f * 1e3 && (s = !0);
+        const { exp: o, iat: f } = ie(n);
+        o * 1e3 - s < s - f * 1e3 && (t = !0);
       } catch {
-        s = !0;
+        t = !0;
       }
     else
-      s = !0;
-    s && (w() ? I(!1, !0) : (x(), e.user && (e.user = null)));
-  }), G = c(async ({ showProgress: o = !1, forceRefreshToken: n = !1 } = {}) => {
-    const s = () => se(() => _.get(U(i.prefix, "/session"), { secure: !0 }), {
+      t = !0;
+    t && (C() ? I(!1, !0) : (x(), e.user && (e.user = null)));
+  }), K = c(async ({ showProgress: r = !1, forceRefreshToken: n = !1 } = {}) => {
+    const t = () => fe(() => _.get(A(i.prefix, "/session"), { secure: !0 }), {
       retries: 2
     });
-    u("handleRefreshUser", { showProgress: o, forceRefreshToken: n });
+    u("handleRefreshUser", { showProgress: r, forceRefreshToken: n });
     try {
       if (n)
         throw new T("need force refresh token");
-      await D({ showProgress: o, requestFn: s, type: "refreshUser" });
-    } catch (t) {
-      t instanceof T && (u("handleRefreshUser failed, try to refresh token", { err: t }), await I());
+      await F({ showProgress: r, requestFn: t, type: "refreshUser" });
+    } catch (s) {
+      s instanceof T && (u("handleRefreshUser failed, try to refresh token", { err: s }), await I());
     }
-  }), Y = c(async () => {
-    const o = y(), n = w();
-    e.user || (o || n ? await G() : (x(), e.user && (e.user = null)));
-  }), J = c(() => {
+  }), Z = c(async () => {
+    const r = w(), n = C();
+    e.user || (r || n ? await K() : (x(), e.user && (e.user = null)));
+  }), H = c(() => {
     if (e.initialized && e.user)
       try {
-        F();
+        G();
       } catch {
       }
-  }), W = c((o) => {
-    const n = Array.isArray(o) ? o[0] : o;
-    u("handleLoginResult", { loginResult: n, result: o });
-    const { loginToken: s, sessionToken: t, refreshToken: r, visitorId: f, encrypted: a = !0 } = n, l = s || t;
-    let L, S, E;
-    l && (L = a ? v(l) : l, u("handleLoginResult: setSessionToken", { decryptSessionToken: L }), C(L), r && (S = a ? v(r) : r, u("handleLoginResult: setRefreshToken", { decryptRefreshToken: S }), g(S))), f && (E = a ? v(f) : f, u("handleLoginResult: setVisitorId", { decryptVisitorId: E }), ee(E));
+  }), ee = c((r) => {
+    const n = Array.isArray(r) ? r[0] : r;
+    u("handleLoginResult", { loginResult: n, result: r });
+    const { loginToken: t, csrfToken: s, sessionToken: o, refreshToken: f, visitorId: a, encrypted: l = !0 } = n, L = t || o;
+    let S, E, U;
+    if (L) {
+      if (S = l ? p(L) : L, u("handleLoginResult: setSessionToken", { decryptSessionToken: S }), v(S), s) {
+        const re = l ? p(s) : s;
+        Q(re);
+      }
+      f && (E = l ? p(f) : f, u("handleLoginResult: setRefreshToken", { decryptRefreshToken: E }), g(E));
+    }
+    a && (U = l ? p(a) : a, u("handleLoginResult: setVisitorId", { decryptVisitorId: U }), se(U));
   });
-  return Z(
+  return oe(
     () => {
-      Q(), F();
+      J(), G();
     },
-    P,
+    B,
     { immediate: !0 }
   ), {
-    renewToken: J,
-    handleRefreshUser: G,
+    renewToken: H,
+    handleRefreshUser: K,
     handleRefreshToken: I,
-    syncSessionSate: Y,
-    handleLoginResult: W,
-    decrypt: v,
-    removeToken: h,
+    syncSessionSate: Z,
+    handleLoginResult: ee,
+    decrypt: p,
+    removeToken: k,
     clearSession: x,
-    getSessionToken: y,
-    getRefreshToken: w,
+    getSessionToken: w,
+    getRefreshToken: C,
     setRefreshToken: g,
-    setSessionToken: C,
+    setSessionToken: v,
     service: _
   };
 }
 export {
-  Ee as default
+  Ve as default
 };