@coti-io/coti-wallet-plugin 0.1.0 → 0.2.1

package/dist/index.js

@@ -596,6 +596,7 @@ var init_plugin = __esm({
       defaultNetworkId: void 0,
       debug: false,
       clearSessionKeyOnWagmiDisconnect: false,
+      waitForBalanceRefreshAfterTransfer: false,
       onboardingServices: { mode: "disabled" },
       onboardingGrantMinBalanceWei: 0,
       onboardingGrantPollIntervalMs: 2e3,
@@ -3827,6 +3828,17 @@ var usePrivateTokenBalance = () => {
       if (error instanceof CotiPluginError) {
         throw error;
       }
+      const message = error instanceof Error ? error.message : String(error?.message ?? error ?? "");
+      const code = error?.code;
+      const isDecryptPayloadFailure = message.includes("Invalid encrypted payload") || code === -32603 && /encrypt|decrypt|ciphertext/i.test(message);
+      if (isDecryptPayloadFailure || canUseSnapDecrypt) {
+        logger.error(`\u274C Failed to decrypt private balance for ${contractAddress}`, error);
+        throw new CotiPluginError(
+          "AES_KEY_MISMATCH" /* AES_KEY_MISMATCH */,
+          isDecryptPayloadFailure ? "Could not decrypt private balances. The Snap AES key may be missing or invalid \u2014 re-onboarding is required." : "Could not decrypt private balances via Snap. Re-onboarding may be required.",
+          message
+        );
+      }
       logger.error(`\u274C Failed to fetch/decrypt for ${contractAddress}`, error);
       return "0.00";
     }
@@ -4227,7 +4239,7 @@ var useBalanceUpdater = ({
                   return { symbol: token.symbol, value, isMismatch: false };
                 } catch (e) {
                   const msg = e?.message || "";
-                  const isMismatch = msg.includes("AES key mismatch") || msg.includes("onboarding") || msg.includes("ACCOUNT_NOT_ONBOARDED") || msg.includes("implausible decrypted balance");
+                  const isMismatch = e instanceof CotiPluginError && (e.code === "AES_KEY_MISMATCH" /* AES_KEY_MISMATCH */ || e.code === "ACCOUNT_NOT_ONBOARDED" /* ACCOUNT_NOT_ONBOARDED */) || msg.includes("AES key mismatch") || msg.includes("Invalid encrypted payload") || msg.includes("onboarding") || msg.includes("ACCOUNT_NOT_ONBOARDED") || msg.includes("implausible decrypted balance");
                   if (isMismatch) {
                     logger.warn(`\u26A0\uFE0F Private token decrypt mismatch for ${tokenAddress}. Falling back to 0.`);
                     return { symbol: token.symbol, value: "0", isMismatch: true };
@@ -7518,7 +7530,7 @@ var sleep2 = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
 function useAesKeyProvider(walletTypeInfo) {
   const [isOnboarding, setIsOnboarding] = (0, import_react16.useState)(false);
   const [onboardingError, setOnboardingError] = (0, import_react16.useState)(null);
-  const [onboardingWarning, setOnboardingWarning] = (0, import_react16.useState)(null);
+  const [onboardingWarnings, setOnboardingWarnings] = (0, import_react16.useState)({});
   const [wasRestoreCancelled, setWasRestoreCancelled] = (0, import_react16.useState)(false);
   const [currentStep, setCurrentStep] = (0, import_react16.useState)("idle");
   const progressCallbackRef = (0, import_react16.useRef)();
@@ -7539,7 +7551,7 @@ function useAesKeyProvider(walletTypeInfo) {
       progressCallbackRef.current = onProgress ?? options.onProgress;
       setOnboardingError(null);
       debugTraceRef.current.clear();
-      setOnboardingWarning(null);
+      setOnboardingWarnings({});
       setWasRestoreCancelled(false);
       const forceContractOnboarding = options?.forceContractOnboarding === true;
       if (!forceContractOnboarding) {
@@ -7648,7 +7660,9 @@ function useAesKeyProvider(walletTypeInfo) {
               }
             } catch (restoreError) {
               if (isUserRejection(restoreError)) {
-                setOnboardingWarning("Backup restore was cancelled. Approve the wallet signature to unlock from your encrypted backup.");
+                setOnboardingWarnings({
+                  intro: "Backup restore was cancelled. Approve the wallet signature to unlock from your encrypted backup."
+                });
                 setWasRestoreCancelled(true);
                 options.onRestoreCancelled?.();
                 emitStep("idle");
@@ -7657,7 +7671,9 @@ function useAesKeyProvider(walletTypeInfo) {
               const message = restoreError instanceof Error ? restoreError.message : "Encrypted AES backup could not be restored.";
               logger.warn("\u26A0\uFE0F AES backup restore failed, falling back to contract onboarding:", restoreError);
               restoreBackupFailed = true;
-              setOnboardingWarning(`Encrypted backup could not be restored. Continuing with onboarding. ${message}`);
+              setOnboardingWarnings({
+                intro: `Encrypted backup could not be restored. Continuing with onboarding. ${message}`
+              });
             }
           }
           if (options.restoreOnly) {
@@ -7818,9 +7834,9 @@ function useAesKeyProvider(walletTypeInfo) {
             savedToSnap = await saveAESKeyToSnap(aesKey, address);
             if (!savedToSnap) {
               logger.warn("\u26A0\uFE0F AES key retrieved but could not persist to Snap");
-              setOnboardingWarning(
-                "Onboarding succeeded, but the AES key could not be saved to MetaMask Snap. You can retry by unlocking again."
-              );
+              setOnboardingWarnings({
+                success: "Onboarding succeeded, but the AES key could not be saved to MetaMask Snap. You can retry by unlocking again."
+              });
             }
           } else if (aesKey && walletTypeInfo.walletType === "metamask" && canPersistAesKeyToSnap() && !walletTypeInfo.isMetaMaskWithSnap) {
             logger.log(
@@ -7842,14 +7858,14 @@ function useAesKeyProvider(walletTypeInfo) {
                 "\u26A0\uFE0F AES key retrieved but encrypted backup save failed:",
                 backupResult.message
               );
-              setOnboardingWarning(
-                `Onboarding succeeded, but encrypted backup was not saved. ${backupResult.message}`
-              );
+              setOnboardingWarnings({
+                success: `Onboarding succeeded, but encrypted backup was not saved. ${backupResult.message}`
+              });
             } else if (backupResult.status === "cancelled") {
               logger.warn("\u26A0\uFE0F AES key retrieved but encrypted backup save was cancelled");
-              setOnboardingWarning(
-                "Onboarding succeeded, but encrypted backup save was cancelled. You can save it later by re-entering your AES key."
-              );
+              setOnboardingWarnings({
+                success: "Onboarding succeeded, but encrypted backup save was cancelled. You can save it later by re-entering your AES key."
+              });
             }
           }
           if (aesKey && isValidAesKey(aesKey)) {
@@ -7913,7 +7929,7 @@ function useAesKeyProvider(walletTypeInfo) {
     getAesKey,
     isOnboarding,
     onboardingError,
-    onboardingWarning,
+    onboardingWarnings,
     wasRestoreCancelled,
     currentStep
   };
@@ -8021,7 +8037,7 @@ var usePrivacyBridgeSessionCore = ({
     getAesKey: getAesKeyFromProvider,
     isOnboarding,
     onboardingError,
-    onboardingWarning
+    onboardingWarnings
   } = useAesKeyProvider(walletTypeInfo);
   const { fetchPrivateBalance } = usePrivateTokenBalance();
   (0, import_react18.useEffect)(() => {
@@ -8077,7 +8093,7 @@ var usePrivacyBridgeSessionCore = ({
     fetchPrivateBalance,
     getAesKeyFromProvider,
     onboardingError,
-    onboardingWarning
+    onboardingWarnings
   };
 };
 
@@ -8711,6 +8727,7 @@ var usePrivacyBridgeWalletConnection = ({
 // src/context/privacyBridge/usePrivacyBridgeUnlockSession.ts
 var import_react23 = require("react");
 var import_wagmi10 = require("wagmi");
+init_plugin();
 init_logger();
 
 // src/hooks/privacyBridge/privateValueCrypto.ts
@@ -8980,6 +8997,13 @@ var usePrivacyBridgeUnlockSession = ({
   }, [resolveAesAccess, resolveSessionAesKey, walletAddress, aesKeyChainId, checkSnapStatus, hasAesKeyInSnap, walletTypeInfo.walletType, currentChainId]);
   const refreshPrivateBalances = (0, import_react23.useCallback)(async (aesKeyOptions) => {
     if (!walletAddress) return false;
+    const lockSessionOnAesFailure = !aesKeyOptions?.preserveSessionOnError;
+    const clearSessionAfterAesFailure = () => {
+      setSessionAesKey(null);
+      clearAesKeyValidatedForUnlock(walletAddress);
+      clearSnapCache();
+      setArePrivateBalancesHidden(true);
+    };
     logger.log("Triggering private balance fetch...");
     try {
       const chainOverride = wagmiSyncRef.current ? wagmiChainId : void 0;
@@ -9058,10 +9082,7 @@ var usePrivacyBridgeUnlockSession = ({
         );
       }
       if (errorInfo.message?.includes("ACCOUNT_NOT_ONBOARDED")) {
-        setSessionAesKey(null);
-        clearAesKeyValidatedForUnlock(walletAddress);
-        clearSnapCache();
-        setArePrivateBalancesHidden(true);
+        if (lockSessionOnAesFailure) clearSessionAfterAesFailure();
         throw new CotiPluginError(
           "ACCOUNT_NOT_ONBOARDED" /* ACCOUNT_NOT_ONBOARDED */,
           "Account has not been onboarded to the COTI network.",
@@ -9069,17 +9090,11 @@ var usePrivacyBridgeUnlockSession = ({
         );
       }
       if (err instanceof CotiPluginError && err.code === "AES_KEY_MISMATCH" /* AES_KEY_MISMATCH */) {
-        setSessionAesKey(null);
-        clearAesKeyValidatedForUnlock(walletAddress);
-        clearSnapCache();
-        setArePrivateBalancesHidden(true);
+        if (lockSessionOnAesFailure) clearSessionAfterAesFailure();
         throw err;
       }
       if (errorInfo.message?.includes("AES key") || errorInfo.message?.includes("onboarding")) {
-        setSessionAesKey(null);
-        clearAesKeyValidatedForUnlock(walletAddress);
-        clearSnapCache();
-        setArePrivateBalancesHidden(true);
+        if (lockSessionOnAesFailure) clearSessionAfterAesFailure();
         throw new CotiPluginError(
           "AES_KEY_MISMATCH" /* AES_KEY_MISMATCH */,
           "AES key mismatch or onboarding error.",
@@ -9124,7 +9139,13 @@ var usePrivacyBridgeUnlockSession = ({
       hasSnap: strategy.snapInstalled,
       getAESKeyFromSnap: strategy.snapInstalled ? getAESKeyFromSnap : void 0
     });
-    await refreshPrivateBalances();
+    if (getPluginConfig().waitForBalanceRefreshAfterTransfer) {
+      await refreshPrivateBalances();
+    } else {
+      void refreshPrivateBalances({ preserveSessionOnError: true }).catch(
+        (err) => logger.error("Background balance refresh after transfer failed", err)
+      );
+    }
     return result;
   }, [
     walletAddress,
@@ -9274,7 +9295,7 @@ var usePrivacyBridgeSession = ({ modals }) => {
     refreshPublicBalances: unlock.refreshPublicBalances,
     refreshPrivateBalances: unlock.refreshPrivateBalances,
     onboardingError: core.onboardingError,
-    onboardingWarning: core.onboardingWarning,
+    onboardingWarnings: core.onboardingWarnings,
     lockPrivateBalances: unlock.lockPrivateBalances,
     saveManualAesKey: unlock.saveManualAesKey,
     sendPrivateToken: unlock.sendPrivateToken,
@@ -9647,6 +9668,40 @@ function deriveOnboardScreen({
 
 // src/components/OnboardModal.tsx
 init_logger();
+
+// src/lib/onboardModalWarnings.ts
+function mergeOnboardModalWarnings(...sources) {
+  const merged = {};
+  for (const source of sources) {
+    if (!source) continue;
+    for (const page of Object.keys(source)) {
+      const message = source[page];
+      if (message) {
+        merged[page] = message;
+      }
+    }
+  }
+  return merged;
+}
+function getDefaultSuccessWarning(saveBackup) {
+  return saveBackup ? "Important: An encrypted backup can help restore this key later, but you should still store it safely." : "Important: This key will be lost when you refresh the page. Store it in a secure location.";
+}
+function resolveOnboardPageWarning(page, options) {
+  const runtime = options.runtimeWarnings?.[page]?.trim();
+  if (runtime) return runtime;
+  const app = options.warnings?.[page]?.trim();
+  if (app) return app;
+  if (page === "success") {
+    return getDefaultSuccessWarning(options.saveBackup ?? true);
+  }
+  return null;
+}
+function hasOnboardModalWarnings(warnings) {
+  if (!warnings) return false;
+  return Object.values(warnings).some((message) => !!message?.trim());
+}
+
+// src/components/OnboardModal.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
 var defaultStyles = {
   backdrop: {
@@ -9818,28 +9873,23 @@ var defaultStyles = {
   },
   saveOptionBody: {
     flex: 1,
-    minWidth: 0
+    minWidth: 0,
+    display: "flex",
+    alignItems: "center",
+    minHeight: "32px"
   },
   saveOptionTitleRow: {
     display: "flex",
     alignItems: "center",
-    gap: "5px",
-    marginBottom: "2px"
+    gap: "5px"
   },
   saveOptionTitle: {
     fontSize: "13px",
     fontWeight: 600,
-    lineHeight: 1.25,
-    color: "#ffffff"
-  },
-  saveOptionDescription: {
-    fontSize: "11px",
-    lineHeight: 1.3,
-    color: "rgba(255, 255, 255, 0.55)",
-    margin: 0,
-    whiteSpace: "nowrap",
-    overflow: "hidden",
-    textOverflow: "ellipsis"
+    lineHeight: "14px",
+    color: "#ffffff",
+    display: "inline-flex",
+    alignItems: "center"
   },
   saveOptionSwitchTrack: {
     position: "relative",
@@ -9881,12 +9931,13 @@ var defaultStyles = {
   tooltipButton: {
     width: "14px",
     height: "14px",
+    boxSizing: "border-box",
     borderRadius: "50%",
     border: "1px solid rgba(255, 255, 255, 0.25)",
     backgroundColor: "rgba(255, 255, 255, 0.08)",
     color: "rgba(255, 255, 255, 0.8)",
     fontSize: "9px",
-    lineHeight: 1,
+    lineHeight: "12px",
     padding: 0,
     cursor: "help",
     flexShrink: 0,
@@ -10190,8 +10241,7 @@ var MUTED_TEXT_KEYS = [
   "closeButton",
   "tooltipButton",
   "cancelButton",
-  "stepDescription",
-  "saveOptionDescription"
+  "stepDescription"
 ];
 var ACCENT_TEXT_KEYS = [
   "keyInput",
@@ -10241,7 +10291,8 @@ var LIGHT_INTERACTIVE_SURFACE_KEYS = [
   "saveOptionCard",
   "saveOptionIconWrap",
   "saveOptionSwitchTrackOff",
-  "saveOptionSwitchTrackOn"
+  "saveOptionSwitchTrackOn",
+  "tooltipButton"
 ];
 function isDefaultDarkSurfaceBackground(backgroundColor) {
   if (!backgroundColor) return true;
@@ -10325,6 +10376,11 @@ function applyLightInteractiveSurfaceGaps(merged, theme) {
     saveOptionSwitchTrackOn: {
       backgroundColor: primary,
       borderColor: colorWithAlpha(primary, 0.45)
+    },
+    tooltipButton: {
+      color: muted2,
+      backgroundColor: colorWithAlpha(foreground, 0.08),
+      border: `1px solid ${colorWithAlpha(foreground, 0.2)}`
     }
   };
   for (const key of LIGHT_INTERACTIVE_SURFACE_KEYS) {
@@ -10471,7 +10527,8 @@ var OnboardModal = ({
   showSaveBackupOption = true,
   onSaveBackupChange,
   onManualAesKeySubmit,
-  warning,
+  warnings,
+  runtimeWarnings,
   theme
 }) => {
   const [copied, setCopied] = (0, import_react27.useState)(false);
@@ -10517,6 +10574,15 @@ var OnboardModal = ({
     /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.iconContainer, children: icon }),
     /* @__PURE__ */ (0, import_jsx_runtime.jsx)("h2", { id: "onboard-modal-title", style: styles2.title, children: title })
   ] });
+  const renderPageWarning = (page) => {
+    const message = resolveOnboardPageWarning(page, {
+      warnings,
+      runtimeWarnings,
+      saveBackup
+    });
+    if (!message) return null;
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: warningStyles.box, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: warningStyles.text, children: message }) });
+  };
   const renderSaveLocallyOption = () => {
     if (!showSaveBackupOption) return null;
     return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
@@ -10528,29 +10594,26 @@ var OnboardModal = ({
         },
         children: [
           /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.saveOptionIconWrap, "aria-hidden": "true", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("svg", { width: "16", height: "16", viewBox: "0 0 24 24", fill: "none", stroke: "currentColor", strokeWidth: "2", strokeLinecap: "round", strokeLinejoin: "round", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("path", { d: "M21 2l-2 2m-7.61 7.61a5.5 5.5 0 1 1-7.778 7.778 5.5 5.5 0 0 1 7.777-7.777zm0 0L15.5 7.5m0 0 3 3L22 7l-3-3m-3.5 3.5L19 4" }) }) }),
-          /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { style: styles2.saveOptionBody, children: [
-            /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { style: styles2.saveOptionTitleRow, children: [
-              /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { style: styles2.saveOptionTitle, children: "Save Locally" }),
-              /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("span", { style: styles2.tooltipWrap, children: [
-                /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
-                  "button",
-                  {
-                    type: "button",
-                    "aria-label": "How local save works",
-                    "aria-describedby": showBackupTooltip ? "backup-details-tooltip" : void 0,
-                    onMouseEnter: () => setShowBackupTooltip(true),
-                    onMouseLeave: () => setShowBackupTooltip(false),
-                    onFocus: () => setShowBackupTooltip(true),
-                    onBlur: () => setShowBackupTooltip(false),
-                    style: styles2.tooltipButton,
-                    children: "?"
-                  }
-                ),
-                showBackupTooltip && /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { id: "backup-details-tooltip", role: "tooltip", style: styles2.tooltipBubble, children: "Only an encrypted blob is stored locally. Restoring it requires a wallet signature." })
-              ] })
-            ] }),
-            /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: styles2.saveOptionDescription, children: "Encrypted locally. Restore via wallet signature." })
-          ] }),
+          /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.saveOptionBody, children: /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { style: styles2.saveOptionTitleRow, children: [
+            /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { style: styles2.saveOptionTitle, children: "Save Locally" }),
+            /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("span", { style: styles2.tooltipWrap, children: [
+              /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+                "button",
+                {
+                  type: "button",
+                  "aria-label": "How local save works",
+                  "aria-describedby": showBackupTooltip ? "backup-details-tooltip" : void 0,
+                  onMouseEnter: () => setShowBackupTooltip(true),
+                  onMouseLeave: () => setShowBackupTooltip(false),
+                  onFocus: () => setShowBackupTooltip(true),
+                  onBlur: () => setShowBackupTooltip(false),
+                  style: styles2.tooltipButton,
+                  children: "?"
+                }
+              ),
+              showBackupTooltip && /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { id: "backup-details-tooltip", role: "tooltip", style: styles2.tooltipBubble, children: "Only an encrypted blob is stored locally. Restoring it requires a wallet signature." })
+            ] })
+          ] }) }),
           /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
             "button",
             {
@@ -10746,7 +10809,7 @@ var OnboardModal = ({
                   "Onboard User"
                 ),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { id: "onboard-modal-description", style: styles2.description, children: "This will execute a transaction on the COTI Network to retrieve your encryption key." }),
-                warning && /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: warningStyles.box, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: warningStyles.text, children: warning }) }),
+                renderPageWarning("intro"),
                 renderSaveLocallyOption(),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
                   "div",
@@ -10763,7 +10826,7 @@ var OnboardModal = ({
                             type: "text",
                             value: manualAesKey,
                             onChange: (event) => setManualAesKey(event.target.value),
-                            placeholder: "Paste AES key",
+                            placeholder: "Paste key",
                             "aria-label": "Manual AES key",
                             disabled: isSubmittingManualKey,
                             style: { ...styles2.manualKeyInput, ...styles2.actionPrimary },
@@ -10837,7 +10900,7 @@ var OnboardModal = ({
                   /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.spinner }),
                   getProgressTitle(currentStep)
                 ),
-                warning && /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: warningStyles.box, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: warningStyles.text, children: warning }) }),
+                renderPageWarning("progress"),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.stepperContainer, children: displaySteps.map((step) => {
                   const status = getOnboardingStepStatus(step.id, currentStep, !!error, includePersistStep);
                   const isActive = status === "active";
@@ -10935,12 +10998,7 @@ var OnboardModal = ({
                     )
                   ] })
                 ] }),
-                /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: warningStyles.box, children: /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("p", { style: warningStyles.text, children: [
-                  /* @__PURE__ */ (0, import_jsx_runtime.jsx)("strong", { children: "Important:" }),
-                  " ",
-                  saveBackup ? "An encrypted backup can help restore this key later, but you should still store it safely." : "This key will be lost when you refresh the page. Store it in a secure location."
-                ] }) }),
-                warning && /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: warningStyles.box, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: warningStyles.text, children: warning }) }),
+                renderPageWarning("success"),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
                   "button",
                   {
@@ -10974,6 +11032,7 @@ var OnboardModal = ({
                 ),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { id: "onboard-modal-description", style: styles2.description, children: "The onboarding process encountered an error. You can retry the signature request." }),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { style: styles2.errorBox, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("p", { style: styles2.errorText, children: error || "An unknown error occurred" }) }),
+                renderPageWarning("error"),
                 /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
                   "button",
                   {
@@ -11084,14 +11143,14 @@ function keepForwardContractProgress(previous, next) {
   return nextOrder >= previousOrder ? next : previous;
 }
 function usePrivateUnlockController(options = {}) {
-  const { theme, warning, onUnlocked, onRestoreCancelled, onOnboardingCancelled } = options;
+  const { theme, warnings, onUnlocked, onRestoreCancelled, onOnboardingCancelled } = options;
   const unlock = usePrivacyBridgeUnlock();
   const wallet = usePrivacyBridgeWallet();
   const walletTypeInfo = useWalletType();
   const [showOnboardModal, setShowOnboardModal] = (0, import_react28.useState)(false);
   const [isUnlocking, setIsUnlocking] = (0, import_react28.useState)(false);
   const [modalError, setModalError] = (0, import_react28.useState)(null);
-  const [modalWarning, setModalWarning] = (0, import_react28.useState)(null);
+  const [modalRuntimeWarnings, setModalRuntimeWarnings] = (0, import_react28.useState)({});
   const [snapConnectedInModal, setSnapConnectedInModal] = (0, import_react28.useState)(false);
   const [saveBackup, setSaveBackup] = (0, import_react28.useState)(true);
   const [currentStep, setCurrentStep] = (0, import_react28.useState)("idle");
@@ -11128,7 +11187,7 @@ function usePrivateUnlockController(options = {}) {
     pendingActionOwnerRef.current = null;
     setCurrentStep("idle");
     setModalError(null);
-    setModalWarning(null);
+    setModalRuntimeWarnings({});
     backupSaveProgressStartedAtRef.current = null;
     pendingCompleteRequestIdRef.current = null;
     contractOnboardingCancelledRef.current = false;
@@ -11157,7 +11216,7 @@ function usePrivateUnlockController(options = {}) {
     setShowOnboardModal(false);
     setCurrentStep("idle");
     setModalError(null);
-    setModalWarning(null);
+    setModalRuntimeWarnings({});
     setIsUnlockInProgress(false);
     setIsUnlocking(false);
     backupSaveProgressStartedAtRef.current = null;
@@ -11182,7 +11241,7 @@ function usePrivateUnlockController(options = {}) {
     setShowOnboardModal(false);
     setCurrentStep("idle");
     setModalError(null);
-    setModalWarning(null);
+    setModalRuntimeWarnings({});
     unlockInProgressOwnerRef.current = null;
     setIsUnlockInProgress(false);
     setIsUnlocking(false);
@@ -11218,14 +11277,18 @@ function usePrivateUnlockController(options = {}) {
     try {
       const connected = await unlock.requestSnapConnection();
       if (!connected) {
-        setModalWarning("Snap connection was skipped or rejected. Continuing without Snap storage.");
+        setModalRuntimeWarnings({
+          intro: "Snap connection was skipped or rejected. Continuing without Snap storage."
+        });
         return false;
       }
       snapConnectedInModalRef.current = true;
       setSnapConnectedInModal(true);
       return true;
     } catch {
-      setModalWarning("Snap connection failed. Continuing without Snap storage.");
+      setModalRuntimeWarnings({
+        intro: "Snap connection failed. Continuing without Snap storage."
+      });
       return false;
     }
   }, [canAttemptSnapInstall, unlock]);
@@ -11259,15 +11322,13 @@ function usePrivateUnlockController(options = {}) {
     if (failureMessage) {
       setCurrentStep("error");
       setModalError(failureMessage);
-      setModalWarning(unlock.onboardingWarning ?? null);
       releaseUnlockInProgress(requestId);
       return;
     }
     setCurrentStep("error");
     setModalError("Onboarding did not complete. Please retry.");
-    setModalWarning(unlock.onboardingWarning ?? null);
     releaseUnlockInProgress(requestId);
-  }, [dismissOnboardModal, isActiveUnlockRequest, onOnboardingCancelled, releaseUnlockInProgress, unlock.onboardingWarning]);
+  }, [dismissOnboardModal, isActiveUnlockRequest, onOnboardingCancelled, releaseUnlockInProgress]);
   const handleStaleRestoreCompletion = (0, import_react28.useCallback)((requestId, refreshSucceeded) => {
     clearPendingActionForRequest(requestId);
     const unlocked = refreshSucceeded || isPrivateUnlockedRef.current;
@@ -11283,7 +11344,7 @@ function usePrivateUnlockController(options = {}) {
       setPendingActionForRequest(requestId, pendingAction);
     }
     setModalError(null);
-    setModalWarning(null);
+    setModalRuntimeWarnings({});
     inFlightRestoreRequestIdRef.current = requestId;
     setIsUnlocking(true);
     try {
@@ -11317,7 +11378,6 @@ function usePrivateUnlockController(options = {}) {
         return false;
       }
       setCurrentStep("idle");
-      setModalWarning(unlock.onboardingWarning ?? null);
       setShowOnboardModal(true);
       return false;
     } catch (error) {
@@ -11379,7 +11439,6 @@ function usePrivateUnlockController(options = {}) {
         contractOnboardingFailureRef.current = failureMessage;
         setCurrentStep("error");
         setModalError(failureMessage);
-        setModalWarning(unlock.onboardingWarning ?? null);
         releaseUnlockInProgress(unlockRequestIdRef.current);
       }
       return;
@@ -11391,7 +11450,7 @@ function usePrivateUnlockController(options = {}) {
     if (step === "idle") {
       setModalError(null);
     }
-  }, [currentStep, showOnboardModal, releaseUnlockInProgress, unlock.onboardingWarning]);
+  }, [currentStep, showOnboardModal, releaseUnlockInProgress]);
   const showOnboardingComplete = (0, import_react28.useCallback)((requestId) => {
     if (!isActiveUnlockRequest(requestId)) return;
     pendingCompleteRequestIdRef.current = null;
@@ -11409,7 +11468,7 @@ function usePrivateUnlockController(options = {}) {
     if (!connectedAddress) return;
     const requestId = unlockRequestIdRef.current;
     setModalError(null);
-    setModalWarning(null);
+    setModalRuntimeWarnings({});
     setIsUnlocking(true);
     setShowOnboardModal(true);
     setCurrentStep("preparing-onboard");
@@ -11444,7 +11503,7 @@ function usePrivateUnlockController(options = {}) {
         currentStep,
         hasSessionAesKey: !!unlock.sessionAesKey,
         hasOnboardingError: !!unlock.onboardingError,
-        hasOnboardingWarning: !!unlock.onboardingWarning
+        hasOnboardingWarning: hasOnboardModalWarnings(unlock.onboardingWarnings)
       });
       const backupSaveProgressStartedAt = backupSaveProgressStartedAtRef.current;
       if (backupSaveProgressStartedAt !== null) {
@@ -11560,7 +11619,7 @@ function usePrivateUnlockController(options = {}) {
       onSaveBackupChange: setSaveBackup,
       onManualAesKeySubmit: async (aesKey, { saveBackup: shouldSaveBackup }) => {
         const requestId = unlockRequestIdRef.current;
-        setModalWarning(null);
+        setModalRuntimeWarnings({});
         const manualSaveResult = await unlock.saveManualAesKey(aesKey, {
           saveBackup: shouldSaveBackup,
           onProgress: (step) => {
@@ -11581,7 +11640,7 @@ function usePrivateUnlockController(options = {}) {
           return;
         }
         if (manualSaveResult.backupWarning) {
-          setModalWarning(manualSaveResult.backupWarning);
+          setModalRuntimeWarnings({ success: manualSaveResult.backupWarning });
           setShowOnboardModal(true);
           setCurrentStep("complete");
           releaseUnlockInProgress(requestId);
@@ -11590,7 +11649,8 @@ function usePrivateUnlockController(options = {}) {
         await completeUnlock(requestId);
       },
       theme,
-      warning: modalWarning ?? unlock.onboardingWarning ?? warning ?? null
+      warnings,
+      runtimeWarnings: mergeOnboardModalWarnings(unlock.onboardingWarnings, modalRuntimeWarnings)
     }
   );
   return {
@@ -11763,7 +11823,7 @@ var PrivacyBridgeProvider = ({
       decryptPrivateValue: session.decryptPrivateValue,
       refreshPrivateBalances: session.refreshPrivateBalances,
       onboardingError: session.onboardingError,
-      onboardingWarning: session.onboardingWarning,
+      onboardingWarnings: session.onboardingWarnings,
       lockPrivateBalances: session.lockPrivateBalances,
       handleOnboard: session.handleOnboard,
       saveManualAesKey: session.saveManualAesKey,