From 54fcdc4a3c2cfed79b84da963d7bfa8514e98794 Mon Sep 17 00:00:00 2001
From: npty <78221556+npty@users.noreply.github.com>
Date: Thu, 22 Aug 2024 19:32:47 +0700
Subject: [PATCH] chore: reuse addEnvOptions (#344)

---
 common/cli-utils.js          | 19 ++++++++++++++++---
 cosmwasm/cli-utils.js        |  4 ++--
 cosmwasm/deploy-contract.js  |  5 +++++
 cosmwasm/submit-proposal.js  | 18 ++++++++++++++++++
 stellar/balances.js          |  9 ++-------
 stellar/deploy-contract.js   |  9 ++-------
 stellar/generate-bindings.js |  9 ++-------
 stellar/operators.js         | 10 ++--------
 sui/utils/cli-utils.js       |  9 ++-------
 9 files changed, 51 insertions(+), 41 deletions(-)

diff --git a/common/cli-utils.js b/common/cli-utils.js
index 1bcb7173..189aee26 100644
--- a/common/cli-utils.js
+++ b/common/cli-utils.js
@@ -2,16 +2,28 @@
 
 require('dotenv').config();
 
+const fs = require('fs');
 const { Option } = require('commander');
 
-const addBaseOptions = (program, options = {}) => {
+// A path to the chain configuration files
+const CHAIN_CONFIG_PATH = `${__dirname}/../axelar-chains-config/info`;
+
+// A list of available chain environments which are the names of the files in the CHAIN_CONFIG_PATH
+const CHAIN_ENVIRONMENTS = fs.readdirSync(CHAIN_CONFIG_PATH).map((chainName) => chainName.split('.')[0]);
+
+const addEnvOption = (program, defaultValue) => {
     program.addOption(
         new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'devnet-amplifier', 'devnet-verifiers', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
+            .choices(CHAIN_ENVIRONMENTS)
+            .default(defaultValue || 'testnet')
             .makeOptionMandatory(true)
             .env('ENV'),
     );
+};
+
+const addBaseOptions = (program, options = {}) => {
+    addEnvOption(program);
+
     program.addOption(new Option('-y, --yes', 'skip deployment prompt confirmation').env('YES'));
 
     if (!options.ignoreParallel) {
@@ -93,6 +105,7 @@ const addExtendedOptions = (program, options = {}) => {
 };
 
 module.exports = {
+    addEnvOption,
     addBaseOptions,
     addExtendedOptions,
 };
diff --git a/cosmwasm/cli-utils.js b/cosmwasm/cli-utils.js
index e1ef6a40..684f3382 100644
--- a/cosmwasm/cli-utils.js
+++ b/cosmwasm/cli-utils.js
@@ -2,7 +2,7 @@
 
 require('dotenv').config();
 
-const { addExtendedOptions } = require('../common');
+const { addExtendedOptions, addEnvOption } = require('../common');
 const { governanceAddress } = require('./utils');
 
 const { Option } = require('commander');
@@ -20,8 +20,8 @@ const addCommonAmplifierOptions = (program, options = {}) => {
         ...options,
     };
 
+    addEnvOption(program);
     addExtendedOptions(program, ops);
-
     program.addOption(new Option('-m, --mnemonic <mnemonic>', 'mnemonic').makeOptionMandatory(true).env('MNEMONIC'));
     program.addOption(new Option('-a, --artifactPath <artifactPath>', 'artifact path').makeOptionMandatory(true).env('ARTIFACT_PATH'));
     program.addOption(new Option('--aarch64', 'aarch64').env('AARCH64').default(false));
diff --git a/cosmwasm/deploy-contract.js b/cosmwasm/deploy-contract.js
index 7b722c37..da49d2f8 100644
--- a/cosmwasm/deploy-contract.js
+++ b/cosmwasm/deploy-contract.js
@@ -3,6 +3,7 @@
 require('dotenv').config();
 const { isNil } = require('lodash');
 
+const { addEnvOption } = require('../common');
 const { isNumber, printInfo, loadConfig, saveConfig, prompt } = require('../evm/utils');
 const {
     prepareWallet,
@@ -114,6 +115,10 @@ const programHandler = () => {
 
     addCommonAmplifierOptions(program);
 
+    program.addOption(new Option('-m, --mnemonic <mnemonic>', 'mnemonic').makeOptionMandatory(true).env('MNEMONIC'));
+    program.addOption(new Option('-a, --artifactPath <artifactPath>', 'artifact path').makeOptionMandatory(true).env('ARTIFACT_PATH'));
+    program.addOption(new Option('-c, --contractName <contractName>', 'contract name').makeOptionMandatory(true));
+    program.addOption(new Option('-n, --chainNames <chainNames>', 'chain names').default('none'));
     program.addOption(new Option('-r, --reuseCodeId', 'reuse code Id'));
     program.addOption(
         new Option(
diff --git a/cosmwasm/submit-proposal.js b/cosmwasm/submit-proposal.js
index c2d8059b..e56b941d 100644
--- a/cosmwasm/submit-proposal.js
+++ b/cosmwasm/submit-proposal.js
@@ -22,6 +22,7 @@ const {
     governanceAddress,
 } = require('./utils');
 const { isNumber, saveConfig, loadConfig, printInfo, prompt } = require('../evm/utils');
+const { addEnvOption } = require('../common');
 const {
     StoreCodeProposal,
     StoreAndInstantiateContractProposal,
@@ -251,6 +252,23 @@ const programHandler = () => {
 
     addCommonAmplifierOptions(program, { predictOnly: true });
 
+    // TODO: combine deploy-contract and submit-proposal options to remove duplicates
+    program.addOption(new Option('-m, --mnemonic <mnemonic>', 'mnemonic').makeOptionMandatory(true).env('MNEMONIC'));
+    program.addOption(new Option('-a, --artifactPath <artifactPath>', 'artifact path').makeOptionMandatory(true).env('ARTIFACT_PATH'));
+    program.addOption(new Option('-c, --contractName <contractName>', 'contract name').makeOptionMandatory(true));
+    program.addOption(new Option('-n, --chainNames <chainNames>', 'chain names').default('none'));
+
+    program.addOption(new Option('-s, --salt <salt>', 'salt for instantiate2. defaults to contract name').env('SALT'));
+    program.addOption(
+        new Option('--admin <address>', 'when instantiating contract, set an admin address. Defaults to governance module account').default(
+            governanceAddress,
+        ),
+    );
+    program.addOption(new Option('--instantiate2', 'use instantiate2 for constant address deployment'));
+    program.addOption(new Option('-l, --label <label>', 'contract instance label'));
+    program.addOption(new Option('--aarch64', 'aarch64').env('AARCH64').default(false));
+    program.addOption(new Option('-y, --yes', 'skip prompt confirmation').env('YES'));
+
     program.addOption(new Option('-t, --title <title>', 'title of proposal').makeOptionMandatory(true));
     program.addOption(new Option('-d, --description <description>', 'description of proposal').makeOptionMandatory(true));
     program.addOption(new Option('--deposit <deposit>', 'the proposal deposit').makeOptionMandatory(true));
diff --git a/stellar/balances.js b/stellar/balances.js
index a1a45b92..601a55b6 100644
--- a/stellar/balances.js
+++ b/stellar/balances.js
@@ -1,6 +1,7 @@
 const { Command, Option } = require('commander');
 const { getWallet } = require('./utils');
 const { loadConfig } = require('../evm/utils');
+const { addEnvOption } = require('../common');
 require('./cli-utils');
 
 async function processCommand(options, _, chain) {
@@ -12,13 +13,7 @@ if (require.main === module) {
 
     program.name('balances').description('Wallet balance');
 
-    program.addOption(
-        new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
-            .makeOptionMandatory(true)
-            .env('ENV'),
-    );
+    addEnvOption(program);
 
     program.addOption(new Option('-p, --privateKey <privateKey>', 'private key').makeOptionMandatory(true).env('PRIVATE_KEY'));
 
diff --git a/stellar/deploy-contract.js b/stellar/deploy-contract.js
index 033bd774..9a13d4f1 100644
--- a/stellar/deploy-contract.js
+++ b/stellar/deploy-contract.js
@@ -5,6 +5,7 @@ const { Command, Option } = require('commander');
 const { execSync } = require('child_process');
 const { loadConfig, printInfo, saveConfig } = require('../evm/utils');
 const { getNetworkPassphrase, getWallet, prepareTransaction, sendTransaction, buildTransaction, estimateCost } = require('./utils');
+const { addEnvOption } = require('../common');
 require('./cli-utils');
 
 function getInitializeArgs(chain, contractName, wallet, options) {
@@ -108,13 +109,7 @@ function main() {
     const program = new Command();
     program.name('deploy-contract').description('Deploy Axelar Soroban contracts on Stellar');
 
-    program.addOption(
-        new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
-            .makeOptionMandatory(true)
-            .env('ENV'),
-    );
+    addEnvOption(program);
     program.addOption(new Option('-p, --privateKey <privateKey>', 'private key').env('PRIVATE_KEY'));
     program.addOption(new Option('-v, --verbose', 'verbose output').default(false));
     program.addOption(new Option('--initialize', 'initialize the contract'));
diff --git a/stellar/generate-bindings.js b/stellar/generate-bindings.js
index 89ff4486..608fa77d 100644
--- a/stellar/generate-bindings.js
+++ b/stellar/generate-bindings.js
@@ -5,6 +5,7 @@ const { execSync } = require('child_process');
 const { loadConfig } = require('../evm/utils');
 const path = require('path');
 const { getNetworkPassphrase } = require('./utils');
+const { addEnvOption } = require('../common');
 require('./cli-utils');
 
 function processCommand(options, _, chain) {
@@ -27,13 +28,7 @@ function main() {
     const program = new Command();
     program.name('Generate TypeScript Bindings for Soroban contract').description('Generates TypeScript bindings for a Soroban contract.');
 
-    program.addOption(
-        new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
-            .makeOptionMandatory(true)
-            .env('ENV'),
-    );
+    addEnvOption(program);
     program.addOption(new Option('--wasmPath <wasmPath>', 'path to the WASM file').makeOptionMandatory(true));
     program.addOption(new Option('--contractId <contractId>', 'contract ID').makeOptionMandatory(true));
     program.addOption(
diff --git a/stellar/operators.js b/stellar/operators.js
index 1e1cb825..c99eda1c 100644
--- a/stellar/operators.js
+++ b/stellar/operators.js
@@ -2,6 +2,7 @@ const { Contract, Address, nativeToScVal } = require('@stellar/stellar-sdk');
 const { Command, Option } = require('commander');
 const { getWallet, prepareTransaction, buildTransaction, sendTransaction, estimateCost } = require('./utils');
 const { loadConfig, printInfo, parseArgs, validateParameters } = require('../evm/utils');
+const { addEnvOption } = require('../common');
 require('./cli-utils');
 
 async function processCommand(options, _, chain) {
@@ -115,14 +116,7 @@ if (require.main === module) {
 
     program.name('operators').description('Operators contract management');
 
-    program.addOption(
-        new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
-            .makeOptionMandatory(true)
-            .env('ENV'),
-    );
-
+    addEnvOption(program);
     program.addOption(new Option('-p, --privateKey <privateKey>', 'private key').makeOptionMandatory(true).env('PRIVATE_KEY'));
     program.addOption(new Option('-v, --verbose', 'verbose output').default(false));
     program.addOption(
diff --git a/sui/utils/cli-utils.js b/sui/utils/cli-utils.js
index 8cf95622..9c8aa924 100644
--- a/sui/utils/cli-utils.js
+++ b/sui/utils/cli-utils.js
@@ -4,15 +4,10 @@ require('dotenv').config();
 
 const { Option, InvalidArgumentError } = require('commander');
 const { getUnitAmount } = require('./amount-utils');
+const { addEnvOption } = require('../../common');
 
 const addBaseOptions = (program, options = {}) => {
-    program.addOption(
-        new Option('-e, --env <env>', 'environment')
-            .choices(['local', 'devnet', 'devnet-amplifier', 'devnet-verifiers', 'stagenet', 'testnet', 'mainnet'])
-            .default('testnet')
-            .makeOptionMandatory(true)
-            .env('ENV'),
-    );
+    addEnvOption(program);
     program.addOption(new Option('-y, --yes', 'skip deployment prompt confirmation').env('YES'));
     program.addOption(new Option('--gasOptions <gasOptions>', 'gas options cli override'));