From ecc82441f04a9e16141bdd52815ea1b991463f37 Mon Sep 17 00:00:00 2001
From: eguajardo <edwin@axelar.network>
Date: Thu, 15 Aug 2024 10:51:56 -0600
Subject: [PATCH] feat(amplifier): allow custom instance labels (#335)

---
 cosmwasm/deploy-contract.js |  1 +
 cosmwasm/submit-proposal.js |  2 ++
 cosmwasm/utils.js           | 18 ++++++++++++------
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/cosmwasm/deploy-contract.js b/cosmwasm/deploy-contract.js
index 92a19e79..8e68223b 100644
--- a/cosmwasm/deploy-contract.js
+++ b/cosmwasm/deploy-contract.js
@@ -137,6 +137,7 @@ const programHandler = () => {
         ),
     );
     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 deployment prompt confirmation').env('YES'));
 
diff --git a/cosmwasm/submit-proposal.js b/cosmwasm/submit-proposal.js
index ccbd21ac..ebde299e 100644
--- a/cosmwasm/submit-proposal.js
+++ b/cosmwasm/submit-proposal.js
@@ -248,6 +248,7 @@ const programHandler = () => {
 
     program.name('submit-proposal').description('Submit governance proposals');
 
+    // TODO: combine deploy-contract and submit-proposal options to remove duplicates
     program.addOption(
         new Option('-e, --env <env>', 'environment')
             .choices(['local', 'devnet', 'devnet-amplifier', 'devnet-verifiers', 'stagenet', 'testnet', 'mainnet'])
@@ -267,6 +268,7 @@ const programHandler = () => {
         ),
     );
     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'));
 
diff --git a/cosmwasm/utils.js b/cosmwasm/utils.js
index 884e82fd..41bfe995 100644
--- a/cosmwasm/utils.js
+++ b/cosmwasm/utils.js
@@ -50,6 +50,8 @@ const calculateDomainSeparator = (chain, router, network) => keccak256(Buffer.fr
 
 const getSalt = (salt, contractName, chainNames) => fromHex(getSaltFromKey(salt || contractName.concat(chainNames)));
 
+const getLabel = ({ contractName, label }) => label || contractName;
+
 const readWasmFile = ({ artifactPath, contractName, aarch64 }) =>
     readFileSync(`${artifactPath}/${pascalToSnake(contractName)}${aarch64 ? '-aarch64' : ''}.wasm`);
 
@@ -94,7 +96,9 @@ const uploadContract = async (client, wallet, config, options) => {
         });
 };
 
-const instantiateContract = (client, wallet, initMsg, config, { contractName, salt, instantiate2, chainNames, admin }) => {
+const instantiateContract = (client, wallet, initMsg, config, options) => {
+    const { contractName, salt, instantiate2, chainNames, admin } = options;
+
     return wallet
         .getAccounts()
         .then(([account]) => {
@@ -105,17 +109,19 @@ const instantiateContract = (client, wallet, initMsg, config, { contractName, sa
             } = config;
             const initFee = gasLimit === 'auto' ? 'auto' : calculateFee(gasLimit, GasPrice.fromString(gasPrice));
 
+            const contractLabel = getLabel(options);
+
             return instantiate2
                 ? client.instantiate2(
                       account.address,
                       contractConfig.codeId,
                       getSalt(salt, contractName, chainNames),
                       initMsg,
-                      contractName,
+                      contractLabel,
                       initFee,
                       { admin },
                   )
-                : client.instantiate(account.address, contractConfig.codeId, initMsg, contractName, initFee, {
+                : client.instantiate(account.address, contractConfig.codeId, initMsg, contractLabel, initFee, {
                       admin,
                   });
         })
@@ -573,12 +579,12 @@ const getStoreCodeParams = (options) => {
 };
 
 const getStoreInstantiateParams = (config, options, msg) => {
-    const { contractName, admin } = options;
+    const { admin } = options;
 
     return {
         ...getStoreCodeParams(options),
         admin,
-        label: contractName,
+        label: getLabel(options),
         msg: Buffer.from(JSON.stringify(msg)),
     };
 };
@@ -592,7 +598,7 @@ const getInstantiateContractParams = (config, options, msg) => {
         ...getSubmitProposalParams(options),
         admin,
         codeId: contractConfig.codeId,
-        label: contractName,
+        label: getLabel(options),
         msg: Buffer.from(JSON.stringify(msg)),
     };
 };