@coti-io/coti-wallet-plugin 0.2.0 → 0.2.2

package/dist/index.mjs

@@ -546,10 +546,10 @@ function configureCotiPlugin(config) {
       ...config.onboardingServices
     }
   };
-  if ("snapInstallEnabled" in config && _config.debug) {
-    console.log("[CotiPlugin] snapInstallEnabled:", {
-      snapInstallEnabled: config.snapInstallEnabled,
-      effective: isSnapInstallEnabled()
+  if ("snapEnabled" in config && _config.debug) {
+    console.log("[CotiPlugin] snapEnabled:", {
+      snapEnabled: config.snapEnabled,
+      effective: isSnapEnabled()
     });
   }
 }
@@ -560,8 +560,8 @@ function getSnapRequestParams(snapId = _config.snapId, snapVersion) {
 function getPluginConfig() {
   return _config;
 }
-function isSnapInstallEnabled() {
-  return getPluginConfig().snapInstallEnabled !== false;
+function isSnapEnabled() {
+  return getPluginConfig().snapEnabled !== false;
 }
 var _config;
 var init_plugin = __esm({
@@ -570,7 +570,7 @@ var init_plugin = __esm({
     init_chains();
     _config = {
       snapId: "npm:@coti-io/coti-snap",
-      snapInstallEnabled: true,
+      snapEnabled: true,
       defaultNetworkId: void 0,
       debug: false,
       clearSessionKeyOnWagmiDisconnect: false,
@@ -1400,6 +1400,7 @@ var resolveBridgeAddress = (chainId, symbol, pubTok) => {
   if (symbol === "USDC.e") return addresses.PrivacyBridgeUSDCe;
   if (symbol === "WADA") return addresses.PrivacyBridgeWADA;
   if (symbol === "gCOTI") return addresses.PrivacyBridgegCOTI;
+  if (symbol === "NIGHT") return addresses.PrivacyBridgeNight;
   return addresses.PrivacyBridgeCotiNative;
 };
 var quoteCotiBridgeFees = async (params) => {
@@ -1431,7 +1432,7 @@ var quoteCotiBridgeFees = async (params) => {
     if (isWbtc) {
       publicDecimals = 8;
       privateDecimals = 8;
-    } else if (isUsdt || isUsdcE || isWada) {
+    } else if (isUsdt || isUsdcE || isWada || isNight) {
       publicDecimals = 6;
       privateDecimals = 6;
     }
@@ -3673,6 +3674,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";
     }
@@ -4073,7 +4085,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 };
@@ -4157,8 +4169,10 @@ var SYMBOL_TO_ORACLE_BASE = {
   USDT: "USDT",
   USDC: "USDC",
   "USDC.e": "USDC",
-  WADA: "ADA"
-  // gCOTI is not in the on-chain price consumer
+  WADA: "ADA",
+  // Consumer base is uppercase GCOTI (lowercase gCOTI reverts).
+  gCOTI: "GCOTI",
+  NIGHT: "NIGHT"
 };
 function getPriceConsumerAddress(chainId = 7082400) {
   const addresses = CONTRACT_ADDRESSES[chainId];
@@ -4266,7 +4280,8 @@ var TOKEN_TO_ORACLE_SYMBOL = {
   USDT: "USDT",
   "USDC.e": "USDC",
   WADA: "ADA",
-  gCOTI: "GCOTI"
+  gCOTI: "GCOTI",
+  NIGHT: "NIGHT"
 };
 var TOKEN_DECIMALS = {
   COTI: 18,
@@ -4275,7 +4290,8 @@ var TOKEN_DECIMALS = {
   USDT: 6,
   "USDC.e": 6,
   WADA: 6,
-  gCOTI: 18
+  gCOTI: 18,
+  NIGHT: 6
 };
 async function simulateFeeOnChain(bridgeAddress, amount, fixedFee, percentageBps, maxFee, oracleSymbol, tokenDecimals, provider) {
   try {
@@ -4788,6 +4804,10 @@ var useSnap = (setSnapError) => {
     }
   }, [resolveProvider]);
   const isSnapInstalled = useCallback5(async () => {
+    if (!isSnapEnabled()) {
+      logger.log("\u2139\uFE0F Snap disabled via plugin config \u2014 treating as not installed");
+      return false;
+    }
     const provider = await resolveProvider();
     if (!provider) {
       logger.log("\u274C isSnapInstalled: No MetaMask provider");
@@ -4817,8 +4837,8 @@ var useSnap = (setSnapError) => {
     }
   }, [detectFlask, resolveProvider]);
   const connectToSnap = useCallback5(async () => {
-    if (!isSnapInstallEnabled()) {
-      logger.log("\u2139\uFE0F Snap install disabled via plugin config");
+    if (!isSnapEnabled()) {
+      logger.log("\u2139\uFE0F Snap disabled via plugin config");
       return false;
     }
     const provider = await resolveProvider();
@@ -4965,6 +4985,13 @@ var useSnap = (setSnapError) => {
     return parseSnapItUint256(result?.value);
   }, [connectedAddress, invokeSnapOperation]);
   const getAESKeyFromSnap = useCallback5(async (accountAddress, options) => {
+    if (!isSnapEnabled()) {
+      logger.log("\u2139\uFE0F Snap disabled via plugin config \u2014 treating as unavailable");
+      throw new CotiPluginError(
+        "SNAP_CONNECT_FAILED" /* SNAP_CONNECT_FAILED */,
+        "COTI Snap is disabled for this app"
+      );
+    }
     if (setSnapError) setSnapError(null);
     const skipCache = options?.skipCache === true;
     if (isSnapRequestPending.current) {
@@ -5101,6 +5128,7 @@ var useSnap = (setSnapError) => {
     }
   }, [isSnapInstalled, setSnapError, resolveProvider, syncEnvironment]);
   const hasAesKeyInSnap = useCallback5(async (accountAddress) => {
+    if (!isSnapEnabled()) return null;
     const provider = await resolveProvider();
     if (!provider) return null;
     const installed = await isSnapInstalled();
@@ -5133,6 +5161,10 @@ var useSnap = (setSnapError) => {
     }
   }, [isSnapInstalled, resolveProvider, syncEnvironment]);
   const saveAESKeyToSnap = useCallback5(async (key, accountAddress = "") => {
+    if (!isSnapEnabled()) {
+      logger.log("\u2139\uFE0F Snap disabled via plugin config \u2014 skipping AES key save");
+      return false;
+    }
     const provider = await resolveProvider();
     if (!provider) return false;
     try {
@@ -6571,7 +6603,7 @@ function useWalletType() {
   const walletType = mapConnectorIdToWalletType(connectorId);
   useEffect6(() => {
     let cancelled = false;
-    if (walletType !== "metamask" || !connector) {
+    if (walletType !== "metamask" || !connector || !isSnapEnabled()) {
       setIsMetaMaskWithSnap(false);
       return;
     }
@@ -7660,7 +7692,8 @@ function useAesKeyProvider(walletTypeInfo) {
             }
           }
           let savedToSnap = false;
-          const canSaveToConnectedSnap = aesKey && walletTypeInfo.walletType === "metamask" && walletTypeInfo.isMetaMaskWithSnap && canPersistAesKeyToSnap();
+          const shouldPersistToSnap = walletTypeInfo.isMetaMaskWithSnap || options.persistToSnap === true;
+          const canSaveToConnectedSnap = !!aesKey && walletTypeInfo.walletType === "metamask" && shouldPersistToSnap && canPersistAesKeyToSnap();
           const canSaveEncryptedBackup = aesKey && isValidAesKey(aesKey) && options.saveBackup && servicesEnabled && (services?.saveEncryptedAesBackup || services?.replaceEncryptedAesBackup);
           if (canSaveToConnectedSnap) {
             emitStep("persisting-key");
@@ -7672,7 +7705,7 @@ function useAesKeyProvider(walletTypeInfo) {
                 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) {
+          } else if (aesKey && walletTypeInfo.walletType === "metamask" && canPersistAesKeyToSnap() && !shouldPersistToSnap) {
             logger.log(
               "\u2139\uFE0F Skipping Snap AES persist \u2014 Snap is not connected to this origin"
             );
@@ -7757,7 +7790,17 @@ function useAesKeyProvider(walletTypeInfo) {
         }
       }
     },
-    [walletTypeInfo.walletType, getAESKeyFromSnap, saveAESKeyToSnap, clearSnapCache, connector, connectedChainId, emitStep, reportOnboardingFailure]
+    [
+      walletTypeInfo.walletType,
+      walletTypeInfo.isMetaMaskWithSnap,
+      getAESKeyFromSnap,
+      saveAESKeyToSnap,
+      clearSnapCache,
+      connector,
+      connectedChainId,
+      emitStep,
+      reportOnboardingFailure
+    ]
   );
   return {
     getAesKey,
@@ -8140,6 +8183,7 @@ var usePrivacyBridgeNetworkSession = ({
 };
 
 // src/context/privacyBridge/usePrivacyBridgeAccountSync.ts
+init_plugin();
 import { useCallback as useCallback16, useEffect as useEffect10, useRef as useRef9 } from "react";
 import { useAccount as useAccount8 } from "wagmi";
 init_logger();
@@ -8188,13 +8232,13 @@ var usePrivacyBridgeAccountSync = ({
   );
   const getAESKeyForCurrentNetwork = useCallback16(
     async (accountAddress, options) => {
-      if (options?.skipCache && walletTypeInfo.walletType === "metamask" && !options.forceContractOnboarding && !options.restoreOnly) {
+      if (options?.skipCache && walletTypeInfo.walletType === "metamask" && isSnapEnabled() && !options.forceContractOnboarding && !options.restoreOnly) {
         return getAESKeyFromSnap(accountAddress, { skipCache: true });
       }
       if (sessionAesKey && !options?.forceContractOnboarding) return sessionAesKey;
       return options === void 0 ? getAesKeyFromProvider(accountAddress) : getAesKeyFromProvider(accountAddress, options.onProgress, options);
     },
-    [sessionAesKey, getAesKeyFromProvider, getAESKeyFromSnap]
+    [sessionAesKey, getAesKeyFromProvider, getAESKeyFromSnap, walletTypeInfo.walletType]
   );
   const { updateAccountState } = useBalanceUpdater({
     setWalletAddress,
@@ -11007,8 +11051,8 @@ function usePrivateUnlockController(options = {}) {
   isPrivateUnlockedRef.current = unlock.isPrivateUnlocked;
   const connectedAddress = wallet.walletAddress || "";
   const isMetaMaskWallet = walletTypeInfo.walletType === "metamask";
-  const canAttemptSnapInstall = isMetaMaskWallet && !isMetaMaskMobileBrowser() && isSnapInstallEnabled();
-  const usesSnapStorage = canAttemptSnapInstall && (walletTypeInfo.isMetaMaskWithSnap || snapConnectedInModal);
+  const canUseSnap = isMetaMaskWallet && !isMetaMaskMobileBrowser() && isSnapEnabled();
+  const usesSnapStorage = canUseSnap && (walletTypeInfo.isMetaMaskWithSnap || snapConnectedInModal);
   const isActiveUnlockRequest = useCallback19(
     (requestId) => requestId === unlockRequestIdRef.current,
     []
@@ -11106,7 +11150,7 @@ function usePrivateUnlockController(options = {}) {
     dismissOnboardModal();
   }, [dismissOnboardModal, wallet.walletAddress]);
   const connectSnap = useCallback19(async () => {
-    if (!canAttemptSnapInstall) return false;
+    if (!canUseSnap) return false;
     setModalError(null);
     try {
       const connected = await unlock.requestSnapConnection();
@@ -11125,7 +11169,7 @@ function usePrivateUnlockController(options = {}) {
       });
       return false;
     }
-  }, [canAttemptSnapInstall, unlock]);
+  }, [canUseSnap, unlock]);
   const handleRestoreUnlockProgress = useCallback19((step) => {
     if (step === "signing-backup") {
       setCurrentStep(step);
@@ -11310,7 +11354,7 @@ function usePrivateUnlockController(options = {}) {
     contractOnboardingFailureRef.current = null;
     try {
       let useSnapStorageForOnboarding = usesSnapStorage || snapConnectedInModalRef.current;
-      if (!useSnapStorageForOnboarding && canAttemptSnapInstall) {
+      if (!useSnapStorageForOnboarding && canUseSnap) {
         useSnapStorageForOnboarding = await connectSnap();
       }
       if (!isActiveUnlockRequest(requestId)) {
@@ -11319,6 +11363,8 @@ function usePrivateUnlockController(options = {}) {
       const ok = await unlock.refreshPrivateBalances({
         forceContractOnboarding: true,
         saveBackup: useSnapStorageForOnboarding ? false : saveBackup,
+        // Explicit: don't rely on isMetaMaskWithSnap which can lag after connectSnap.
+        persistToSnap: useSnapStorageForOnboarding,
         onProgress: (step, details) => {
           if (isActiveUnlockRequest(requestId)) {
             handleContractOnboardingProgress(step, details);
@@ -11374,7 +11420,7 @@ function usePrivateUnlockController(options = {}) {
         setIsUnlocking(false);
       }
     }
-  }, [canAttemptSnapInstall, connectSnap, connectedAddress, currentStep, dismissOnboardModal, handleContractOnboardingProgress, handleOnboardingIncomplete, isActiveUnlockRequest, onOnboardingCancelled, saveBackup, showOnboardingComplete, unlock, usesSnapStorage]);
+  }, [canUseSnap, connectSnap, connectedAddress, currentStep, dismissOnboardModal, handleContractOnboardingProgress, handleOnboardingIncomplete, isActiveUnlockRequest, onOnboardingCancelled, saveBackup, showOnboardingComplete, unlock, usesSnapStorage]);
   const handleOnboardModalClose = useCallback19(() => {
     if (currentStep === "complete" || currentStep === "validating-key" && pendingCompleteRequestIdRef.current !== null) {
       void finishSuccessfulOnboarding();
@@ -12564,7 +12610,7 @@ export {
   isChainUpdatesMuted,
   isCotiPluginError,
   isMultipleWalletsError,
-  isSnapInstallEnabled,
+  isSnapEnabled,
   loadPodRequests,
   logger,
   mobileZerionWallet,