@embarkai/ui-kit 0.4.7 → 0.4.9

package/dist/iframe/index.html

@@ -15,7 +15,7 @@
     <meta http-equiv="X-Content-Type-Options" content="nosniff" />
     <meta http-equiv="Referrer-Policy" content="strict-origin-when-cross-origin" />
 
-    <title>EmbarkAI Secure Wallet - iframe version 0.4.7</title>
+    <title>EmbarkAI Secure Wallet - iframe version 0.4.9</title>
 
     <!-- Styles will be injected by build process -->
     <style>

package/dist/iframe/main.js

@@ -6262,7 +6262,7 @@ var SigningManager = class extends TokenRefreshApiClient {
 };
 
 // src/iframe/main.ts
-var IFRAME_VERSION = "0.4.7";
+var IFRAME_VERSION = "0.4.9";
 var IframeWallet = class {
   constructor() {
     console.log("=".repeat(60));

package/dist/index.cjs

@@ -15180,7 +15180,7 @@ function mapBlockscoutToAsset(balance) {
   }
   const asset = {
     type,
-    address: token.address,
+    address: token.address_hash,
     name: token.name || "Unknown Token",
     symbol: token.symbol || "???",
     balance: value,
@@ -15211,7 +15211,7 @@ function mapBlockscoutNftToAsset(nft) {
   const imageUrl = image_url || (metadata?.image ? resolveIpfsUri(metadata.image) : void 0);
   const asset = {
     type,
-    address: token.address,
+    address: token.address_hash,
     name: token.name || "Unknown NFT",
     symbol: token.symbol || "???",
     balance: value || "1",
@@ -15334,7 +15334,7 @@ function useBlockscoutAssets(options) {
   });
   const erc20TokenAddresses = (0, import_react57.useMemo)(() => {
     if (!tokenBalances) return [];
-    return tokenBalances.filter((tb) => tb.token.type === "ERC-20").map((tb) => tb.token.address);
+    return tokenBalances.filter((tb) => tb.token.type === "ERC-20").map((tb) => tb.token.address_hash);
   }, [tokenBalances]);
   const { data: erc3643Results } = (0, import_react_query39.useQuery)({
     queryKey: [ERC3643_DETECTION_QUERY_KEY, address, erc20TokenAddresses],
@@ -19167,7 +19167,7 @@ var init_package = __esm({
   "package.json"() {
     package_default = {
       name: "@embarkai/ui-kit",
-      version: "0.4.7",
+      version: "0.4.9",
       description: "React UI components and hooks for EmbarkAI authentication and Account Abstraction",
       type: "module",
       main: "./dist/index.cjs",
@@ -20255,10 +20255,10 @@ function DirectWalletHandler() {
     }
   }, [walletMode, setConfigMode, setActiveMode]);
   (0, import_react76.useEffect)(() => {
-    if (isConnected && address && !isExternalConnected && connector) {
+    if (isConnected && address && !isExternalConnected && connector && chain?.id) {
       setExternalWallet({
         address,
-        chainId: chain?.id || 1,
+        chainId: chain.id,
         connectorName: connector.name || "Unknown Wallet"
       });
     }
@@ -20274,10 +20274,10 @@ function DirectWalletHandler() {
     }
   }, [chain?.id, isConnected, isExternalConnected, setExternalChainId]);
   (0, import_react76.useEffect)(() => {
-    if (isExternalConnected && isConnected && address && address !== externalAddress && connector) {
+    if (isExternalConnected && isConnected && address && address !== externalAddress && connector && chain?.id) {
       setExternalWallet({
         address,
-        chainId: chain?.id || 1,
+        chainId: chain.id,
         connectorName: connector.name || "Unknown Wallet"
       });
     }
@@ -22742,11 +22742,13 @@ var init_iframe_manager = __esm({
 });
 
 // src/config/rainbowkit.ts
-var import_rainbowkit2, import_wallets, import_chains, getProjectId2, WALLET_MAP, createRainbowConfig, rainbowTheme;
+var import_read17, import_rainbowkit2, import_wallets, import_wagmi4, import_chains, getProjectId2, WALLET_MAP, getWagmiStorageKey, buildRainbowChains, createRainbowConfig, rainbowTheme;
 var init_rainbowkit = __esm({
   "src/config/rainbowkit.ts"() {
+    import_read17 = require("@embarkai/core/read");
     import_rainbowkit2 = require("@rainbow-me/rainbowkit");
     import_wallets = require("@rainbow-me/rainbowkit/wallets");
+    import_wagmi4 = require("wagmi");
     import_chains = require("wagmi/chains");
     getProjectId2 = (configProjectId) => {
       if (configProjectId) return configProjectId;
@@ -22761,12 +22763,41 @@ var init_rainbowkit = __esm({
       coinbase: import_wallets.coinbaseWallet,
       rainbow: import_wallets.rainbowWallet
     };
+    getWagmiStorageKey = (options) => {
+      const projectId = options?.projectId || "default";
+      const preferredChainId = options?.preferredChainId || "default";
+      return `embark-wagmi-v2-${projectId}-${preferredChainId}`;
+    };
+    buildRainbowChains = (supportedChains, preferredChainId) => {
+      const availableChains = [
+        import_read17.lumiaMainnetChain,
+        import_read17.lumiaTestnetChain,
+        import_chains.mainnet,
+        import_chains.polygon,
+        import_chains.bsc,
+        import_chains.arbitrum,
+        import_chains.optimism,
+        import_chains.avalanche,
+        import_chains.base,
+        import_chains.zora
+      ];
+      const chains = (supportedChains?.length ? availableChains.filter((chain) => supportedChains.includes(chain.id)) : availableChains).filter((chain, index, collection) => collection.findIndex((candidate) => candidate.id === chain.id) === index);
+      if (preferredChainId == null) {
+        return chains;
+      }
+      const preferredChain = chains.find((chain) => chain.id === preferredChainId);
+      if (!preferredChain) {
+        return chains;
+      }
+      return [preferredChain, ...chains.filter((chain) => chain.id !== preferredChainId)];
+    };
     createRainbowConfig = (options) => {
       const opts = typeof options === "string" ? { projectId: options } : options || {};
       const config = {
         appName: "EmbarkAI Wallet",
         projectId: getProjectId2(opts.projectId),
-        chains: [import_chains.mainnet, import_chains.polygon, import_chains.bsc, import_chains.arbitrum, import_chains.optimism, import_chains.avalanche, import_chains.base, import_chains.zora],
+        chains: buildRainbowChains(opts.supportedChains, opts.preferredChainId),
+        storage: (0, import_wagmi4.createStorage)({ key: getWagmiStorageKey(opts) }),
         ssr: false
       };
       if (opts.allowedWallets && opts.allowedWallets.length > 0) {
@@ -22976,9 +23007,16 @@ function RainbowKitProvider({ children }) {
   const rainbowConfig = (0, import_react78.useMemo)(
     () => createRainbowConfig({
       projectId: config.current?.wallet?.walletConnectProjectId,
-      allowedWallets: config.current?.wallet?.allowedWallets
+      allowedWallets: config.current?.wallet?.allowedWallets,
+      supportedChains: config.current?.wallet?.supportedChains,
+      preferredChainId: config.current?.network?.chainId
     }),
-    []
+    [
+      config.current?.network?.chainId,
+      config.current?.wallet?.allowedWallets,
+      config.current?.wallet?.supportedChains,
+      config.current?.wallet?.walletConnectProjectId
+    ]
   );
   const customTheme = (0, import_react78.useMemo)(() => {
     if (!config.current?.wallet?.enabled) return {};
@@ -22995,14 +23033,14 @@ function RainbowKitProvider({ children }) {
     };
   }, [colorMode]);
   if (!config.current?.wallet?.enabled) return /* @__PURE__ */ (0, import_jsx_runtime110.jsx)(import_jsx_runtime110.Fragment, { children });
-  return /* @__PURE__ */ (0, import_jsx_runtime110.jsx)(import_wagmi4.WagmiProvider, { config: rainbowConfig, children: /* @__PURE__ */ (0, import_jsx_runtime110.jsx)(import_rainbowkit3.RainbowKitProvider, { theme: customTheme, modalSize: "compact", showRecentTransactions: true, children }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime110.jsx)(import_wagmi5.WagmiProvider, { config: rainbowConfig, children: /* @__PURE__ */ (0, import_jsx_runtime110.jsx)(import_rainbowkit3.RainbowKitProvider, { theme: customTheme, modalSize: "compact", showRecentTransactions: true, children }) });
 }
-var import_rainbowkit3, import_react78, import_wagmi4, import_styles, import_jsx_runtime110;
+var import_rainbowkit3, import_react78, import_wagmi5, import_styles, import_jsx_runtime110;
 var init_RainbowKitContext = __esm({
   "src/context/RainbowKitContext.tsx"() {
     import_rainbowkit3 = require("@rainbow-me/rainbowkit");
     import_react78 = require("react");
-    import_wagmi4 = require("wagmi");
+    import_wagmi5 = require("wagmi");
     init_rainbowkit();
     init_useLayoutStore();
     init_ProviderContext();
@@ -23029,7 +23067,7 @@ var init_utils10 = __esm({
 // src/config/wagmi.ts
 function buildTransport(urls) {
   const transports = urls.map(
-    (url) => (0, import_wagmi5.http)(url, {
+    (url) => (0, import_wagmi6.http)(url, {
       timeout: 1e4,
       retryCount: 0,
       batch: false,
@@ -23045,36 +23083,52 @@ function buildTransport(urls) {
     retryCount: 0
   });
 }
-var import_read17, import_viem17, import_wagmi5, wagmiConfig;
+var import_read18, import_viem17, import_wagmi6, wagmiConfig;
 var init_wagmi = __esm({
   "src/config/wagmi.ts"() {
-    import_read17 = require("@embarkai/core/read");
+    import_read18 = require("@embarkai/core/read");
     import_viem17 = require("viem");
-    import_wagmi5 = require("wagmi");
-    wagmiConfig = (0, import_wagmi5.createConfig)({
-      chains: [import_read17.lumiaMainnetChain, import_read17.lumiaTestnetChain],
+    import_wagmi6 = require("wagmi");
+    wagmiConfig = (0, import_wagmi6.createConfig)({
+      chains: [import_read18.lumiaMainnetChain, import_read18.lumiaTestnetChain],
+      storage: (0, import_wagmi6.createStorage)({ key: "embark-wagmi-static-v2" }),
       transports: {
-        [import_read17.lumiaTestnetChain.id]: buildTransport(import_read17.lumiaTestnetChain.rpcUrls.default.http),
-        [import_read17.lumiaMainnetChain.id]: buildTransport(import_read17.lumiaMainnetChain.rpcUrls.default.http)
+        [import_read18.lumiaTestnetChain.id]: buildTransport(import_read18.lumiaTestnetChain.rpcUrls.default.http),
+        [import_read18.lumiaMainnetChain.id]: buildTransport(import_read18.lumiaMainnetChain.rpcUrls.default.http)
       }
     });
   }
 });
 
 // src/context/WagmiContext.tsx
-var import_wagmi6, import_jsx_runtime111, WagmiProvider2;
+var import_wagmi7, import_jsx_runtime111, WagmiProvider2;
 var init_WagmiContext = __esm({
   "src/context/WagmiContext.tsx"() {
-    import_wagmi6 = require("wagmi");
+    import_wagmi7 = require("wagmi");
     init_wagmi();
     import_jsx_runtime111 = require("react/jsx-runtime");
     WagmiProvider2 = ({ children }) => {
-      return /* @__PURE__ */ (0, import_jsx_runtime111.jsx)(import_wagmi6.WagmiProvider, { config: wagmiConfig, children });
+      return /* @__PURE__ */ (0, import_jsx_runtime111.jsx)(import_wagmi7.WagmiProvider, { config: wagmiConfig, children });
     };
   }
 });
 
 // src/context/ProviderContext.tsx
+function clearLegacyWagmiStorage() {
+  if (typeof window === "undefined" || !window.localStorage) return;
+  const legacyKeys = [
+    "wagmi.state",
+    "wagmi.recentConnectorId",
+    "embark-wagmi-static-v1.state",
+    "embark-wagmi-static-v1.recentConnectorId"
+  ];
+  for (const key of legacyKeys) {
+    try {
+      window.localStorage.removeItem(key);
+    } catch {
+    }
+  }
+}
 function Provider(props) {
   const { children, projectId, initialConfig = {}, callbacks } = props;
   const callbacksRef = (0, import_react79.useRef)(callbacks);
@@ -23122,6 +23176,9 @@ function Provider(props) {
     if (typeof window === "undefined" || !projectId) return;
     const mergedConfig = (0, import_lodash_es8.merge)(DEFAULT_PROVIDER_CONFIG, initialConfig);
     updateConfig(mergedConfig);
+    if (mergedConfig.wallet?.enabled) {
+      clearLegacyWagmiStorage();
+    }
     try {
       if (typeof window !== "undefined") {
         window.__EMBARK_SERVICES__ = mergedConfig.services;
@@ -23534,7 +23591,7 @@ var init_LangToggle = __esm({
 // src/clients/bundler/getUserOperationByHash.ts
 async function getUserOperationByHash(userOpHash, chainId) {
   const targetChainId = chainId ?? requireActiveChainId();
-  const chainConfig = (0, import_read18.getChainConfig)(targetChainId);
+  const chainConfig = (0, import_read19.getChainConfig)(targetChainId);
   if (!chainConfig) {
     throw new Error(`Chain ${targetChainId} is not supported`);
   }
@@ -23562,19 +23619,19 @@ async function getUserOperationByHash(userOpHash, chainId) {
     return null;
   }
 }
-var import_core3, import_read18;
+var import_core3, import_read19;
 var init_getUserOperationByHash = __esm({
   "src/clients/bundler/getUserOperationByHash.ts"() {
     init_SessionContext();
     import_core3 = require("@embarkai/core");
-    import_read18 = require("@embarkai/core/read");
+    import_read19 = require("@embarkai/core/read");
   }
 });
 
 // src/clients/bundler/getUserOperationReceipt.ts
 async function getUserOperationReceipt(userOpHash, chainId) {
   const targetChainId = chainId ?? requireActiveChainId();
-  const chainConfig = (0, import_read19.getChainConfig)(targetChainId);
+  const chainConfig = (0, import_read20.getChainConfig)(targetChainId);
   if (!chainConfig) {
     throw new Error(`Chain ${targetChainId} is not supported`);
   }
@@ -23608,12 +23665,12 @@ async function getUserOperationReceipt(userOpHash, chainId) {
     return null;
   }
 }
-var import_core4, import_read19;
+var import_core4, import_read20;
 var init_getUserOperationReceipt = __esm({
   "src/clients/bundler/getUserOperationReceipt.ts"() {
     init_SessionContext();
     import_core4 = require("@embarkai/core");
-    import_read19 = require("@embarkai/core/read");
+    import_read20 = require("@embarkai/core/read");
   }
 });
 
@@ -24362,7 +24419,7 @@ function useDirectWallet() {
   const isConnected = useExternalWalletStore((st) => st.isExternalConnected);
   const connectorName = useExternalWalletStore((st) => st.connectorName);
   const clearExternalWallet = useExternalWalletStore((st) => st.clearExternalWallet);
-  const { disconnect: wagmiDisconnect } = (0, import_wagmi8.useDisconnect)();
+  const { disconnect: wagmiDisconnect } = (0, import_wagmi9.useDisconnect)();
   const { openConnectModal } = (0, import_rainbowkit5.useConnectModal)();
   const setIsWalletLinking = useLayoutDataStore((st) => st.setIsWalletLinking);
   const connect = (0, import_react84.useCallback)(() => {
@@ -24385,12 +24442,12 @@ function useDirectWallet() {
     disconnect
   };
 }
-var import_react84, import_rainbowkit5, import_wagmi8;
+var import_react84, import_rainbowkit5, import_wagmi9;
 var init_useDirectWallet = __esm({
   "src/hooks/useDirectWallet.ts"() {
     import_react84 = require("react");
     import_rainbowkit5 = require("@rainbow-me/rainbowkit");
-    import_wagmi8 = require("wagmi");
+    import_wagmi9 = require("wagmi");
     init_useExternalWalletStore();
     init_useLayoutDataStore();
   }
@@ -24400,8 +24457,8 @@ var init_useDirectWallet = __esm({
 function useSendDirectTransaction() {
   const isConnected = useExternalWalletStore((st) => st.isExternalConnected);
   const externalChainId = useExternalWalletStore((st) => st.externalChainId);
-  const { sendTransactionAsync } = (0, import_wagmi9.useSendTransaction)();
-  const { switchChainAsync } = (0, import_wagmi9.useSwitchChain)();
+  const { sendTransactionAsync } = (0, import_wagmi10.useSendTransaction)();
+  const { switchChainAsync } = (0, import_wagmi10.useSwitchChain)();
   const [isPending, setIsPending] = (0, import_react85.useState)(false);
   const [error, setError] = (0, import_react85.useState)(null);
   const [txHash, setTxHash] = (0, import_react85.useState)(null);
@@ -24446,12 +24503,12 @@ function useSendDirectTransaction() {
   }, []);
   return { sendTransaction, isPending, error, txHash, reset };
 }
-var import_react85, import_viem18, import_wagmi9;
+var import_react85, import_viem18, import_wagmi10;
 var init_useSendDirectTransaction = __esm({
   "src/hooks/useSendDirectTransaction.ts"() {
     import_react85 = require("react");
     import_viem18 = require("viem");
-    import_wagmi9 = require("wagmi");
+    import_wagmi10 = require("wagmi");
     init_useExternalWalletStore();
   }
 });
@@ -24549,8 +24606,8 @@ var init_useTransferToLinkedWallet = __esm({
 
 // src/hooks/useNativeWalletSignature.ts
 function useNativeWalletSignature() {
-  const signMessage = (0, import_wagmi10.useSignMessage)();
-  const signTypedMessage = (0, import_wagmi10.useSignTypedData)();
+  const signMessage = (0, import_wagmi11.useSignMessage)();
+  const signTypedMessage = (0, import_wagmi11.useSignTypedData)();
   return {
     signMessage,
     /** EIP-712 signature */
@@ -24558,10 +24615,10 @@ function useNativeWalletSignature() {
     fetchProjectNameById: fetchProjectName
   };
 }
-var import_wagmi10;
+var import_wagmi11;
 var init_useNativeWalletSignature = __esm({
   "src/hooks/useNativeWalletSignature.ts"() {
-    import_wagmi10 = require("wagmi");
+    import_wagmi11 = require("wagmi");
     init_providers();
   }
 });
@@ -24664,10 +24721,10 @@ var init_clients2 = __esm({
 // src/modules/transactions.ts
 function useTransactions() {
   const chainId = requireActiveChainId();
-  const viemChain = (0, import_read20.getViemChain)(chainId);
-  const publicClient = (0, import_wagmi11.usePublicClient)({ chainId });
-  const { data: walletClient } = (0, import_wagmi11.useWalletClient)();
-  const { address } = (0, import_wagmi11.useAccount)();
+  const viemChain = (0, import_read21.getViemChain)(chainId);
+  const publicClient = (0, import_wagmi12.usePublicClient)({ chainId });
+  const { data: walletClient } = (0, import_wagmi12.useWalletClient)();
+  const { address } = (0, import_wagmi12.useAccount)();
   const sendTransaction = async (params) => {
     if (!walletClient) {
       throw new Error("Wallet not connected");
@@ -24722,13 +24779,13 @@ function useTransactions() {
     isConnected: !!address && !!walletClient
   };
 }
-var import_read20, import_viem20, import_wagmi11;
+var import_read21, import_viem20, import_wagmi12;
 var init_transactions = __esm({
   "src/modules/transactions.ts"() {
     init_SessionContext();
-    import_read20 = require("@embarkai/core/read");
+    import_read21 = require("@embarkai/core/read");
     import_viem20 = require("viem");
-    import_wagmi11 = require("wagmi");
+    import_wagmi12 = require("wagmi");
   }
 });
 
@@ -24739,7 +24796,7 @@ function useAssets(address) {
     data: nativeBalance,
     isLoading: nativeBalanceLoading,
     refetch: refetchNativeBalance
-  } = (0, import_wagmi12.useBalance)({
+  } = (0, import_wagmi13.useBalance)({
     address,
     chainId,
     query: {
@@ -24753,7 +24810,7 @@ function useAssets(address) {
     args: [address],
     chainId
   }));
-  const readContractsResult = (0, import_wagmi12.useReadContracts)({
+  const readContractsResult = (0, import_wagmi13.useReadContracts)({
     contracts: tokenContracts,
     query: {
       enabled: !!address && COMMON_TOKENS.length > 0
@@ -24834,7 +24891,7 @@ function useAssets(address) {
 }
 function useTokenInfo(tokenAddress) {
   const chainId = requireActiveChainId();
-  const readContractsResult = (0, import_wagmi12.useReadContracts)({
+  const readContractsResult = (0, import_wagmi13.useReadContracts)({
     contracts: [
       { address: tokenAddress, abi: import_viem21.erc20Abi, functionName: "name", chainId },
       { address: tokenAddress, abi: import_viem21.erc20Abi, functionName: "symbol", chainId },
@@ -24865,7 +24922,7 @@ function useTokenBalance(tokenAddress, userAddress) {
     data: balance,
     isLoading,
     refetch
-  } = (0, import_wagmi12.useReadContract)({
+  } = (0, import_wagmi13.useReadContract)({
     address: tokenAddress,
     abi: import_viem21.erc20Abi,
     functionName: "balanceOf",
@@ -24888,13 +24945,13 @@ function useTokenBalance(tokenAddress, userAddress) {
     refetch
   };
 }
-var import_react88, import_viem21, import_wagmi12, COMMON_TOKENS;
+var import_react88, import_viem21, import_wagmi13, COMMON_TOKENS;
 var init_assets2 = __esm({
   "src/modules/assets.ts"() {
     init_SessionContext();
     import_react88 = __toESM(require("react"), 1);
     import_viem21 = require("viem");
-    import_wagmi12 = require("wagmi");
+    import_wagmi13 = require("wagmi");
     COMMON_TOKENS = [
       // Add real token addresses here when available
       // {

package/dist/index.js

@@ -15117,7 +15117,7 @@ function mapBlockscoutToAsset(balance) {
   }
   const asset = {
     type,
-    address: token.address,
+    address: token.address_hash,
     name: token.name || "Unknown Token",
     symbol: token.symbol || "???",
     balance: value,
@@ -15148,7 +15148,7 @@ function mapBlockscoutNftToAsset(nft) {
   const imageUrl = image_url || (metadata?.image ? resolveIpfsUri(metadata.image) : void 0);
   const asset = {
     type,
-    address: token.address,
+    address: token.address_hash,
     name: token.name || "Unknown NFT",
     symbol: token.symbol || "???",
     balance: value || "1",
@@ -15271,7 +15271,7 @@ function useBlockscoutAssets(options) {
   });
   const erc20TokenAddresses = useMemo5(() => {
     if (!tokenBalances) return [];
-    return tokenBalances.filter((tb) => tb.token.type === "ERC-20").map((tb) => tb.token.address);
+    return tokenBalances.filter((tb) => tb.token.type === "ERC-20").map((tb) => tb.token.address_hash);
   }, [tokenBalances]);
   const { data: erc3643Results } = useQuery16({
     queryKey: [ERC3643_DETECTION_QUERY_KEY, address, erc20TokenAddresses],
@@ -19086,7 +19086,7 @@ var init_package = __esm({
   "package.json"() {
     package_default = {
       name: "@embarkai/ui-kit",
-      version: "0.4.7",
+      version: "0.4.9",
       description: "React UI components and hooks for EmbarkAI authentication and Account Abstraction",
       type: "module",
       main: "./dist/index.cjs",
@@ -20172,10 +20172,10 @@ function DirectWalletHandler() {
     }
   }, [walletMode, setConfigMode, setActiveMode]);
   useEffect42(() => {
-    if (isConnected && address && !isExternalConnected && connector) {
+    if (isConnected && address && !isExternalConnected && connector && chain?.id) {
       setExternalWallet({
         address,
-        chainId: chain?.id || 1,
+        chainId: chain.id,
         connectorName: connector.name || "Unknown Wallet"
       });
     }
@@ -20191,10 +20191,10 @@ function DirectWalletHandler() {
     }
   }, [chain?.id, isConnected, isExternalConnected, setExternalChainId]);
   useEffect42(() => {
-    if (isExternalConnected && isConnected && address && address !== externalAddress && connector) {
+    if (isExternalConnected && isConnected && address && address !== externalAddress && connector && chain?.id) {
       setExternalWallet({
         address,
-        chainId: chain?.id || 1,
+        chainId: chain.id,
         connectorName: connector.name || "Unknown Wallet"
       });
     }
@@ -22676,15 +22676,12 @@ var init_iframe_manager = __esm({
 });
 
 // src/config/rainbowkit.ts
+import { lumiaMainnetChain as lumiaMainnetChain2, lumiaTestnetChain as lumiaTestnetChain2 } from "@embarkai/core/read";
 import { getDefaultConfig } from "@rainbow-me/rainbowkit";
-import {
-  coinbaseWallet,
-  metaMaskWallet,
-  rainbowWallet,
-  walletConnectWallet
-} from "@rainbow-me/rainbowkit/wallets";
+import { coinbaseWallet, metaMaskWallet, rainbowWallet, walletConnectWallet } from "@rainbow-me/rainbowkit/wallets";
+import { createStorage } from "wagmi";
 import { arbitrum, avalanche, base, bsc, mainnet, optimism, polygon, zora } from "wagmi/chains";
-var getProjectId2, WALLET_MAP, createRainbowConfig, rainbowTheme;
+var getProjectId2, WALLET_MAP, getWagmiStorageKey, buildRainbowChains, createRainbowConfig, rainbowTheme;
 var init_rainbowkit = __esm({
   "src/config/rainbowkit.ts"() {
     getProjectId2 = (configProjectId) => {
@@ -22700,12 +22697,41 @@ var init_rainbowkit = __esm({
       coinbase: coinbaseWallet,
       rainbow: rainbowWallet
     };
+    getWagmiStorageKey = (options) => {
+      const projectId = options?.projectId || "default";
+      const preferredChainId = options?.preferredChainId || "default";
+      return `embark-wagmi-v2-${projectId}-${preferredChainId}`;
+    };
+    buildRainbowChains = (supportedChains, preferredChainId) => {
+      const availableChains = [
+        lumiaMainnetChain2,
+        lumiaTestnetChain2,
+        mainnet,
+        polygon,
+        bsc,
+        arbitrum,
+        optimism,
+        avalanche,
+        base,
+        zora
+      ];
+      const chains = (supportedChains?.length ? availableChains.filter((chain) => supportedChains.includes(chain.id)) : availableChains).filter((chain, index, collection) => collection.findIndex((candidate) => candidate.id === chain.id) === index);
+      if (preferredChainId == null) {
+        return chains;
+      }
+      const preferredChain = chains.find((chain) => chain.id === preferredChainId);
+      if (!preferredChain) {
+        return chains;
+      }
+      return [preferredChain, ...chains.filter((chain) => chain.id !== preferredChainId)];
+    };
     createRainbowConfig = (options) => {
       const opts = typeof options === "string" ? { projectId: options } : options || {};
       const config = {
         appName: "EmbarkAI Wallet",
         projectId: getProjectId2(opts.projectId),
-        chains: [mainnet, polygon, bsc, arbitrum, optimism, avalanche, base, zora],
+        chains: buildRainbowChains(opts.supportedChains, opts.preferredChainId),
+        storage: createStorage({ key: getWagmiStorageKey(opts) }),
         ssr: false
       };
       if (opts.allowedWallets && opts.allowedWallets.length > 0) {
@@ -22920,9 +22946,16 @@ function RainbowKitProvider({ children }) {
   const rainbowConfig = useMemo9(
     () => createRainbowConfig({
       projectId: config.current?.wallet?.walletConnectProjectId,
-      allowedWallets: config.current?.wallet?.allowedWallets
+      allowedWallets: config.current?.wallet?.allowedWallets,
+      supportedChains: config.current?.wallet?.supportedChains,
+      preferredChainId: config.current?.network?.chainId
     }),
-    []
+    [
+      config.current?.network?.chainId,
+      config.current?.wallet?.allowedWallets,
+      config.current?.wallet?.supportedChains,
+      config.current?.wallet?.walletConnectProjectId
+    ]
   );
   const customTheme = useMemo9(() => {
     if (!config.current?.wallet?.enabled) return {};
@@ -22965,9 +22998,9 @@ var init_utils10 = __esm({
 });
 
 // src/config/wagmi.ts
-import { lumiaMainnetChain as lumiaMainnetChain2, lumiaTestnetChain as lumiaTestnetChain2 } from "@embarkai/core/read";
+import { lumiaMainnetChain as lumiaMainnetChain3, lumiaTestnetChain as lumiaTestnetChain3 } from "@embarkai/core/read";
 import { fallback } from "viem";
-import { createConfig, http as http4 } from "wagmi";
+import { createConfig, createStorage as createStorage2, http as http4 } from "wagmi";
 function buildTransport(urls) {
   const transports = urls.map(
     (url) => http4(url, {
@@ -22990,10 +23023,11 @@ var wagmiConfig;
 var init_wagmi = __esm({
   "src/config/wagmi.ts"() {
     wagmiConfig = createConfig({
-      chains: [lumiaMainnetChain2, lumiaTestnetChain2],
+      chains: [lumiaMainnetChain3, lumiaTestnetChain3],
+      storage: createStorage2({ key: "embark-wagmi-static-v2" }),
       transports: {
-        [lumiaTestnetChain2.id]: buildTransport(lumiaTestnetChain2.rpcUrls.default.http),
-        [lumiaMainnetChain2.id]: buildTransport(lumiaMainnetChain2.rpcUrls.default.http)
+        [lumiaTestnetChain3.id]: buildTransport(lumiaTestnetChain3.rpcUrls.default.http),
+        [lumiaMainnetChain3.id]: buildTransport(lumiaMainnetChain3.rpcUrls.default.http)
       }
     });
   }
@@ -23024,6 +23058,21 @@ import {
   useRef as useRef18
 } from "react";
 import { jsx as jsx112 } from "react/jsx-runtime";
+function clearLegacyWagmiStorage() {
+  if (typeof window === "undefined" || !window.localStorage) return;
+  const legacyKeys = [
+    "wagmi.state",
+    "wagmi.recentConnectorId",
+    "embark-wagmi-static-v1.state",
+    "embark-wagmi-static-v1.recentConnectorId"
+  ];
+  for (const key of legacyKeys) {
+    try {
+      window.localStorage.removeItem(key);
+    } catch {
+    }
+  }
+}
 function Provider(props) {
   const { children, projectId, initialConfig = {}, callbacks } = props;
   const callbacksRef = useRef18(callbacks);
@@ -23071,6 +23120,9 @@ function Provider(props) {
     if (typeof window === "undefined" || !projectId) return;
     const mergedConfig = merge3(DEFAULT_PROVIDER_CONFIG, initialConfig);
     updateConfig(mergedConfig);
+    if (mergedConfig.wallet?.enabled) {
+      clearLegacyWagmiStorage();
+    }
     try {
       if (typeof window !== "undefined") {
         window.__EMBARK_SERVICES__ = mergedConfig.services;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@embarkai/ui-kit",
-  "version": "0.4.7",
+  "version": "0.4.9",
   "description": "React UI components and hooks for EmbarkAI authentication and Account Abstraction",
   "type": "module",
   "main": "./dist/index.cjs",