feat: add option to iterate over chain's type

axelarnetwork · Sep 24, 2024 · d3ff44d · d3ff44d
1 parent b34985a
commit d3ff44d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/common/cli-utils.js b/common/cli-utils.js
@@ -30,16 +30,15 @@ const addBaseOptions = (program, options = {}) => {
     program.addOption(new Option('--gasOptions <gasOptions>', 'gas options cli override'));
 
     if (!options.ignoreChainNames) {
-        program.addOption(
-            new Option('-n, --chainNames <chainNames>', 'chains to run the script over').makeOptionMandatory(true).env('CHAINS'),
-        );
+        program.addOption(new Option('-n, --chainNames <chainNames>', 'chains to run the script over').env('CHAINS'));
         program.addOption(new Option('--skipChains <skipChains>', 'chains to skip over'));
         program.addOption(
             new Option(
                 '--startFromChain <startFromChain>',
                 'start from a specific chain onwards in the config, useful when a cmd fails for an intermediate chain',
             ),
         );
+        program.addOption(new Option('--chainType <chainType>', 'chain type to run script over'));
     }
 
     if (!options.ignorePrivateKey) {

diff --git a/evm/utils.js b/evm/utils.js
@@ -636,18 +636,27 @@ const mainProcessor = async (options, processCommand, save = true, catchErr = fa
         throw new Error('Environment was not provided');
     }
 
-    if (!options.chainName && !options.chainNames) {
-        throw new Error('Chain names were not provided');
-    }
-
     printInfo('Environment', options.env);
 
     const config = loadConfig(options.env);
-    let chains = options.chainName ? [options.chainName] : options.chainNames.split(',');
     const chainsToSkip = (options.skipChains || '').split(',').map((str) => str.trim().toLowerCase());
 
+    let chains = [];
+
     if (options.chainNames === 'all') {
         chains = Object.keys(config.chains);
+        chains = chains.map((chain) => chain.trim().toLowerCase());
+    } else if (options.chainType) {
+        chains = Object.keys(config.chains);
+        chains.filter((chain) => config.chains[chain].chainType === options.chainType);
+    } else if (options.chainName) {
+        chains = [options.chainName];
+    } else if (options.chainNames) {
+        chains = options.chainNames.split(',');
+    }
+
+    if (chains.length === 0) {
+        throw new Error('Chain names were not provided');
     }
 
     chains = chains.map((chain) => chain.trim().toLowerCase());