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

package/dist/index.mjs

@@ -3673,6 +3673,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 +4084,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 };