fix chain filtering

hyperlane-xyz · Feb 12, 2024 · bd1e2df · bd1e2df
1 parent 851a035
commit bd1e2df
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/typescript/sdk/src/contracts/contracts.ts b/typescript/sdk/src/contracts/contracts.ts
@@ -189,13 +189,26 @@ export async function filterOwnableContracts(
 }
 
 export function appFromAddressesMapHelper<F extends HyperlaneFactories>(
-  addressesMap: HyperlaneAddressesMap<any>,
+  unfilteredAddressesMap: HyperlaneAddressesMap<any>,
   factories: F,
   multiProvider: MultiProvider,
+  supportedChainNames?: string[],
 ): {
   contractsMap: HyperlaneContractsMap<F>;
   multiProvider: MultiProvider;
 } {
+  let addressesMap;
+  if (supportedChainNames) {
+    addressesMap = objFilter(
+      unfilteredAddressesMap,
+      (chain, _): _ is HyperlaneAddresses<any> => {
+        return supportedChainNames.includes(chain);
+      },
+    );
+  } else {
+    addressesMap = unfilteredAddressesMap;
+  }
+
   // Hack to accommodate non-Ethereum artifacts, while still retaining their
   // presence in the addressesMap so that they are included in the list of chains
   // on the MultiProvider (needed for getting metadata). A non-Ethereum-style address

diff --git a/typescript/sdk/src/core/HyperlaneCore.ts b/typescript/sdk/src/core/HyperlaneCore.ts
@@ -55,6 +55,7 @@ export class HyperlaneCore extends HyperlaneApp<CoreFactories> {
       addressesMap,
       coreFactories,
       multiProvider,
+      supportedChainNames,
     );
     return new HyperlaneCore(
       helper.contractsMap,

diff --git a/typescript/sdk/src/gas/HyperlaneIgp.ts b/typescript/sdk/src/gas/HyperlaneIgp.ts
@@ -42,6 +42,7 @@ export class HyperlaneIgp extends HyperlaneApp<IgpFactories> {
       addressesMap,
       igpFactories,
       multiProvider,
+      supportedChainNames,
     );
     return new HyperlaneIgp(
       helper.contractsMap,

diff --git a/typescript/sdk/src/ism/HyperlaneIsmFactory.ts b/typescript/sdk/src/ism/HyperlaneIsmFactory.ts
@@ -94,6 +94,7 @@ export class HyperlaneIsmFactory extends HyperlaneApp<ProxyFactoryFactories> {
       addressesMap,
       proxyFactoryFactories,
       multiProvider,
+      supportedChainNames,
     );
     return new HyperlaneIsmFactory(
       helper.contractsMap,