@oydual31/more-vaults-sdk 0.2.1 → 0.2.3

package/dist/react/index.cjs

@@ -1031,84 +1031,6 @@ async function isAsyncMode(publicClient, vault) {
   });
   return !oraclesEnabled;
 }
-var OMNI_FACTORY_ADDRESS = "0x7bDB8B17604b03125eFAED33cA0c55FBf856BB0C";
-var FACTORY_ABI = [
-  {
-    name: "localEid",
-    type: "function",
-    inputs: [],
-    outputs: [{ type: "uint32" }],
-    stateMutability: "view"
-  },
-  {
-    name: "isCrossChainVault",
-    type: "function",
-    inputs: [{ name: "__eid", type: "uint32" }, { name: "_vault", type: "address" }],
-    outputs: [{ type: "bool" }],
-    stateMutability: "view"
-  },
-  {
-    name: "hubToSpokes",
-    type: "function",
-    inputs: [{ name: "__eid", type: "uint32" }, { name: "_hubVault", type: "address" }],
-    outputs: [{ name: "eids", type: "uint32[]" }, { name: "vaults", type: "address[]" }],
-    stateMutability: "view"
-  },
-  {
-    name: "spokeToHub",
-    type: "function",
-    inputs: [{ name: "__eid", type: "uint32" }, { name: "_spokeVault", type: "address" }],
-    outputs: [{ name: "eid", type: "uint32" }, { name: "vault", type: "address" }],
-    stateMutability: "view"
-  }
-];
-async function getVaultTopology(publicClient, vault, factoryAddress = OMNI_FACTORY_ADDRESS) {
-  const v = viem.getAddress(vault);
-  const f = viem.getAddress(factoryAddress);
-  const localEid = await publicClient.readContract({
-    address: f,
-    abi: FACTORY_ABI,
-    functionName: "localEid"
-  });
-  const isHub = await publicClient.readContract({
-    address: f,
-    abi: FACTORY_ABI,
-    functionName: "isCrossChainVault",
-    args: [localEid, v]
-  });
-  if (isHub) {
-    const [spokeEids] = await publicClient.readContract({
-      address: f,
-      abi: FACTORY_ABI,
-      functionName: "hubToSpokes",
-      args: [localEid, v]
-    });
-    const localChainId2 = EID_TO_CHAIN_ID[localEid] ?? Number(publicClient.chain?.id ?? 0);
-    const spokeChainIds = spokeEids.map((eid) => EID_TO_CHAIN_ID[eid]).filter((id) => id !== void 0);
-    return { role: "hub", hubChainId: localChainId2, spokeChainIds };
-  }
-  const [hubEid, hubVault] = await publicClient.readContract({
-    address: f,
-    abi: FACTORY_ABI,
-    functionName: "spokeToHub",
-    args: [localEid, v]
-  });
-  if (hubEid !== 0 && hubVault !== "0x0000000000000000000000000000000000000000") {
-    const hubChainId = EID_TO_CHAIN_ID[hubEid] ?? 0;
-    const spokeChainIds = [];
-    const localChainId2 = EID_TO_CHAIN_ID[localEid];
-    if (localChainId2 !== void 0) spokeChainIds.push(localChainId2);
-    return { role: "spoke", hubChainId, spokeChainIds };
-  }
-  const localChainId = EID_TO_CHAIN_ID[localEid] ?? Number(publicClient.chain?.id ?? 0);
-  return { role: "local", hubChainId: localChainId, spokeChainIds: [] };
-}
-function isOnHubChain(currentChainId, topology) {
-  return currentChainId === topology.hubChainId;
-}
-function getAllVaultChainIds(topology) {
-  return [topology.hubChainId, ...topology.spokeChainIds];
-}
 var PUBLIC_RPCS = {
   1: [
     "https://ethereum-rpc.publicnode.com",
@@ -1299,6 +1221,144 @@ async function getUserBalancesForRoutes(routes, userAddress) {
   );
 }
 
+// src/viem/topology.ts
+var OMNI_FACTORY_ADDRESS = "0x7bDB8B17604b03125eFAED33cA0c55FBf856BB0C";
+var FACTORY_ABI = [
+  {
+    name: "localEid",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "uint32" }],
+    stateMutability: "view"
+  },
+  {
+    name: "isCrossChainVault",
+    type: "function",
+    inputs: [{ name: "__eid", type: "uint32" }, { name: "_vault", type: "address" }],
+    outputs: [{ type: "bool" }],
+    stateMutability: "view"
+  },
+  {
+    name: "hubToSpokes",
+    type: "function",
+    inputs: [{ name: "__eid", type: "uint32" }, { name: "_hubVault", type: "address" }],
+    outputs: [{ name: "eids", type: "uint32[]" }, { name: "vaults", type: "address[]" }],
+    stateMutability: "view"
+  },
+  {
+    name: "spokeToHub",
+    type: "function",
+    inputs: [{ name: "__eid", type: "uint32" }, { name: "_spokeVault", type: "address" }],
+    outputs: [{ name: "eid", type: "uint32" }, { name: "vault", type: "address" }],
+    stateMutability: "view"
+  }
+];
+async function getVaultTopology(publicClient, vault, factoryAddress = OMNI_FACTORY_ADDRESS) {
+  const v = viem.getAddress(vault);
+  const f = viem.getAddress(factoryAddress);
+  const localEid = await publicClient.readContract({
+    address: f,
+    abi: FACTORY_ABI,
+    functionName: "localEid"
+  });
+  const isHub = await publicClient.readContract({
+    address: f,
+    abi: FACTORY_ABI,
+    functionName: "isCrossChainVault",
+    args: [localEid, v]
+  });
+  if (isHub) {
+    const [spokeEids] = await publicClient.readContract({
+      address: f,
+      abi: FACTORY_ABI,
+      functionName: "hubToSpokes",
+      args: [localEid, v]
+    });
+    const localChainId2 = EID_TO_CHAIN_ID[localEid] ?? Number(publicClient.chain?.id ?? 0);
+    const spokeChainIds = spokeEids.map((eid) => EID_TO_CHAIN_ID[eid]).filter((id) => id !== void 0);
+    return { role: "hub", hubChainId: localChainId2, spokeChainIds };
+  }
+  const [hubEid, hubVault] = await publicClient.readContract({
+    address: f,
+    abi: FACTORY_ABI,
+    functionName: "spokeToHub",
+    args: [localEid, v]
+  });
+  if (hubEid !== 0 && hubVault !== "0x0000000000000000000000000000000000000000") {
+    const hubChainId = EID_TO_CHAIN_ID[hubEid] ?? 0;
+    const spokeChainIds = [];
+    const localChainId2 = EID_TO_CHAIN_ID[localEid];
+    if (localChainId2 !== void 0) spokeChainIds.push(localChainId2);
+    return { role: "spoke", hubChainId, spokeChainIds };
+  }
+  const localChainId = EID_TO_CHAIN_ID[localEid] ?? Number(publicClient.chain?.id ?? 0);
+  return { role: "local", hubChainId: localChainId, spokeChainIds: [] };
+}
+async function getFullVaultTopology(hubChainClient, vault, factoryAddress = OMNI_FACTORY_ADDRESS) {
+  const topo = await getVaultTopology(hubChainClient, vault, factoryAddress);
+  if (topo.role !== "hub") {
+    throw new Error(
+      `getFullVaultTopology: client must be connected to the hub chain (${topo.hubChainId}), but got role="${topo.role}". Connect to chainId ${topo.hubChainId} instead.`
+    );
+  }
+  return topo;
+}
+var DISCOVERY_CHAIN_IDS = Object.values(CHAIN_IDS).filter(
+  (id) => id !== 545
+  // exclude testnet
+);
+async function discoverVaultTopology(vault, publicClient, factoryAddress = OMNI_FACTORY_ADDRESS) {
+  const v = viem.getAddress(vault);
+  let triedChainId;
+  if (publicClient) {
+    try {
+      const topo = await getVaultTopology(publicClient, v, factoryAddress);
+      if (topo.role !== "local") {
+        if (topo.role === "spoke") {
+          const hubClient = createChainClient(topo.hubChainId);
+          if (hubClient) {
+            try {
+              return await getFullVaultTopology(hubClient, v, factoryAddress);
+            } catch {
+            }
+          }
+        }
+        return topo;
+      }
+      triedChainId = publicClient.chain?.id;
+    } catch {
+    }
+  }
+  for (const chainId of DISCOVERY_CHAIN_IDS) {
+    if (chainId === triedChainId) continue;
+    const client = createChainClient(chainId);
+    if (!client) continue;
+    try {
+      const topo = await getVaultTopology(client, v, factoryAddress);
+      if (topo.role === "hub") return topo;
+      if (topo.role === "spoke") {
+        const hubClient = createChainClient(topo.hubChainId);
+        if (hubClient) {
+          try {
+            return await getFullVaultTopology(hubClient, v, factoryAddress);
+          } catch {
+            return topo;
+          }
+        }
+        return topo;
+      }
+    } catch {
+    }
+  }
+  return { role: "local", hubChainId: 0, spokeChainIds: [] };
+}
+function isOnHubChain(currentChainId, topology) {
+  return currentChainId === topology.hubChainId;
+}
+function getAllVaultChainIds(topology) {
+  return [topology.hubChainId, ...topology.spokeChainIds];
+}
+
 // src/viem/preflight.ts
 async function preflightAsync(publicClient, vault, escrow) {
   const v = viem.getAddress(vault);
@@ -1787,9 +1847,14 @@ function useVaultTopology(vault, factoryAddress = OMNI_FACTORY_ADDRESS) {
   const currentChainId = wagmi.useChainId();
   const publicClient = wagmi.usePublicClient();
   const { data: topology, isLoading } = reactQuery.useQuery({
-    queryKey: ["vaultTopology", vault, currentChainId, factoryAddress],
-    queryFn: () => getVaultTopology(asSdkClient(publicClient), vault, factoryAddress),
-    enabled: !!vault && !!publicClient,
+    // Key does NOT include currentChainId — topology is chain-independent
+    queryKey: ["vaultTopology", vault, factoryAddress],
+    queryFn: () => discoverVaultTopology(
+      vault,
+      publicClient ? asSdkClient(publicClient) : null,
+      factoryAddress
+    ),
+    enabled: !!vault,
     staleTime: 5 * 60 * 1e3
     // topology rarely changes — 5 min cache
   });
@@ -1974,9 +2039,12 @@ function useRedeemShares(vault, chainId) {
   };
 }
 function useVaultDistribution(vault) {
-  const chainId = wagmi.useChainId();
-  const publicClient = wagmi.usePublicClient();
-  const { topology } = useVaultTopology(vault);
+  const { data: topology } = reactQuery.useQuery({
+    queryKey: ["vaultTopology", vault],
+    queryFn: () => discoverVaultTopology(vault),
+    enabled: !!vault,
+    staleTime: 5 * 60 * 1e3
+  });
   const spokeClients = react.useMemo(() => {
     if (!topology) return {};
     const clients = {};
@@ -1987,13 +2055,13 @@ function useVaultDistribution(vault) {
     return clients;
   }, [topology]);
   const { data: distribution, isLoading } = reactQuery.useQuery({
-    queryKey: ["vaultDistribution", vault, chainId],
-    queryFn: () => getVaultDistribution(
-      asSdkClient(publicClient),
-      vault,
-      spokeClients
-    ),
-    enabled: !!vault && !!publicClient,
+    queryKey: ["vaultDistribution", vault, topology?.hubChainId],
+    queryFn: () => {
+      const hubClient = createChainClient(topology.hubChainId);
+      if (!hubClient) throw new Error(`No public RPC for hub chainId ${topology.hubChainId}`);
+      return getVaultDistribution(hubClient, vault, spokeClients);
+    },
+    enabled: !!vault && !!topology && topology.role !== "local",
     staleTime: 3e4
   });
   return { distribution, isLoading };