refactor scripts

Author: ali-behjati

contract_manager/scripts/common.ts

@@ -78,6 +78,7 @@ export const COMMON_DEPLOY_OPTIONS = {
   },
   chain: {
     type: "array",
+    string: true,
     demandOption: true,
     desc: "Chains to upload the contract on. Must be one of the chains available in the store",
   },
@@ -207,22 +208,6 @@ export function findEntropyContract(chain: EvmChain): EvmEntropyContract {
   throw new Error(`Entropy contract not found for chain ${chain.getId()}`);
 }
 
-/**
- * Finds an EVM chain by its name.
- * @param {string} chainName The name of the chain to find.
- * @returns The EVM chain instance.
- * @throws {Error} an error if the chain is not found or is not an EVM chain.
- */
-export function findEvmChain(chainName: string): EvmChain {
-  const chain = DefaultStore.chains[chainName];
-  if (!chain) {
-    throw new Error(`Chain ${chainName} not found`);
-  } else if (!(chain instanceof EvmChain)) {
-    throw new Error(`Chain ${chainName} is not an EVM chain`);
-  }
-  return chain;
-}
-
 /**
  * Finds the wormhole contract for a given EVM chain.
  * @param {EvmChain} chain The EVM chain to find the wormhole contract for.

contract_manager/scripts/deploy_cosmwasm.ts

@@ -34,7 +34,7 @@ async function main() {
   const argv = await parser.argv;
   const { code, wormholeContract } = argv;
   console.log(
-    await CosmWasmPriceFeedContract.deploy(
+    await CosmWasmChain.deploy(
       DefaultStore.chains[argv.chain] as CosmWasmChain,
       wormholeContract,
       argv["private-key"],

contract_manager/scripts/deploy_evm_contract.ts

@@ -33,25 +33,16 @@ const parser = yargs(hideBin(process.argv))
 async function main() {
   const argv = await parser.argv;
 
-  const chain = DefaultStore.chains[argv.chain];
-
-  if (!chain) {
-    throw new Error(`Chain ${argv.contract} not found`);
-  }
-
-  if (chain instanceof EvmChain) {
-    const artifact = JSON.parse(readFileSync(argv["std-output"], "utf8"));
-    const address = await chain.deploy(
-      toPrivateKey(argv["private-key"]),
-      artifact["abi"],
-      artifact["bytecode"],
-      argv["deploy-args"] || []
-    );
-
-    console.log(`Deployed contract at ${address}`);
-  } else {
-    throw new Error("Chain is not an EVM chain");
-  }
+  const chain = DefaultStore.getChainOrThrow(argv.chain, EvmChain);
+  const artifact = JSON.parse(readFileSync(argv["std-output"], "utf8"));
+  const address = await chain.deploy(
+    toPrivateKey(argv["private-key"]),
+    artifact["abi"],
+    artifact["bytecode"],
+    argv["deploy-args"] || []
+  );
+
+  console.log(`Deployed contract at ${address}`);
 }
 
 main();

contract_manager/scripts/deploy_evm_entropy_contracts.ts

@@ -172,13 +172,7 @@ async function topupAccountsIfNecessary(
 async function main() {
   const argv = await parser.argv;
 
-  const chainName = argv.chain;
-  const chain = DefaultStore.chains[chainName];
-  if (!chain) {
-    throw new Error(`Chain ${chainName} not found`);
-  } else if (!(chain instanceof EvmChain)) {
-    throw new Error(`Chain ${chainName} is not an EVM chain`);
-  }
+  const chain = DefaultStore.getChainOrThrow(argv.chain, EvmChain);
 
   const deploymentConfig: DeploymentConfig = {
     type: toDeploymentType(argv.deploymentType),

contract_manager/scripts/deploy_evm_pricefeed_contracts.ts

@@ -111,12 +111,7 @@ async function main() {
   const chainNames = argv.chain;
 
   for (const chainName of chainNames) {
-    const chain = DefaultStore.chains[chainName];
-    if (!chain) {
-      throw new Error(`Chain ${chainName} not found`);
-    } else if (!(chain instanceof EvmChain)) {
-      throw new Error(`Chain ${chainName} is not an EVM chain`);
-    }
+    const chain = DefaultStore.getChainOrThrow(chainName, EvmChain);
 
     console.log(`Deploying price feed contracts on ${chain.getId()}...`);
 

contract_manager/scripts/entropy_debug_reveal.ts

@@ -1,11 +1,7 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
-import { toPrivateKey } from "../src";
-import {
-  COMMON_DEPLOY_OPTIONS,
-  findEntropyContract,
-  findEvmChain,
-} from "./common";
+import { DefaultStore, EvmChain, toPrivateKey } from "../src";
+import { COMMON_DEPLOY_OPTIONS, findEntropyContract } from "./common";
 
 const parser = yargs(hideBin(process.argv))
   .usage(
@@ -29,7 +25,7 @@ const parser = yargs(hideBin(process.argv))
 
 async function main() {
   const argv = await parser.argv;
-  const chain = findEvmChain(argv.chain);
+  const chain = DefaultStore.getChainOrThrow(argv.chain, EvmChain);
   const contract = findEntropyContract(chain);
   const sequenceNumber = argv.sequenceNumber;
 

contract_manager/scripts/generate_governance_set_fee_config.sample.yaml

@@ -1,24 +1,24 @@
-- name: aurora
+- chainName: aurora
   fee: 3
   exponent: 12
-- name: avalanche
+- chainName: avalanche
   fee: 25
   exponent: 13
-- name: conflux_espace
+- chainName: conflux_espace
   fee: 1
   exponent: 17
-- name: cronos
+- chainName: cronos
   fee: 6
   exponent: 16
-- name: meter
+- chainName: meter
   fee: 2
   exponent: 16
-- name: ronin
+- chainName: ronin
   fee: 1
   exponent: 15
-- name: sei_evm_mainnet
+- chainName: sei_evm_mainnet
   fee: 1
   exponent: 16
-- name: shimmer
+- chainName: shimmer
   fee: 1
   exponent: 18

contract_manager/scripts/generate_governance_set_fee_payload.ts

@@ -7,7 +7,7 @@ import { parse } from "yaml";
 const parser = yargs(hideBin(process.argv))
   .usage("Usage: $0 --config <path/to/config.yaml>")
   .options({
-    config: {
+    "config-path": {
       type: "string",
       demandOption: true,
       desc: "Path to the config file",
@@ -26,40 +26,36 @@ const parser = yargs(hideBin(process.argv))
 
 async function main() {
   const {
-    config,
-    "ops-key-path": ops_key_path,
-    vault: vault_id,
+    "config-path": configPath,
+    "ops-key-path": opsKeyPath,
+    vault: vaultId,
   } = await parser.argv;
 
-  const config_obj = parse(readFileSync(config, "utf8"));
+  const config = parse(readFileSync(configPath, "utf8"));
 
-  let update_payloads: Buffer[] = [];
-  for (const chain of config_obj) {
-    const chain_obj = DefaultStore.chains[chain.name];
-    if (!chain_obj) {
-      throw new Error(`Chain with ID '${chain.name}' does not exist.`);
-    }
-
-    const payload = chain_obj.generateGovernanceSetFeePayload(
-      chain.fee,
-      chain.exponent
+  const updatePayloads: Buffer[] = [];
+  for (const setFeeEntry of config) {
+    const chain = DefaultStore.getChainOrThrow(setFeeEntry.chainName);
+    const payload = chain.generateGovernanceSetFeePayload(
+      setFeeEntry.fee,
+      setFeeEntry.exponent
     );
-    update_payloads.push(payload);
+    updatePayloads.push(payload);
     console.log(
-      `Generated payload for chain ${chain.name}:`,
+      `Generated payload for chain ${setFeeEntry.chainName}:`,
       payload.toString("hex")
     );
   }
 
-  const vault = DefaultStore.vaults[vault_id];
+  const vault = DefaultStore.vaults[vaultId];
 
   if (!vault) {
-    throw new Error(`Vault with ID '${vault_id}' does not exist.`);
+    throw new Error(`Vault with ID '${vaultId}' does not exist.`);
   }
 
-  const keypair = await loadHotWallet(ops_key_path);
+  const keypair = await loadHotWallet(opsKeyPath);
   vault.connect(keypair);
-  const proposal = await vault.proposeWormholeMessage(update_payloads);
+  const proposal = await vault.proposeWormholeMessage(updatePayloads);
   console.log("Proposal address:", proposal.address.toBase58());
 }
 

contract_manager/scripts/generate_upgrade_near_contract_proposal.ts

@@ -31,10 +31,7 @@ async function main() {
   const argv = await parser.argv;
 
   // Near wormhole contracts have the same id on testnet and mainnet.
-  const chain = DefaultStore.chains.near;
-  if (!(chain instanceof NearChain)) {
-    throw new Error("Near chain is missing");
-  }
+  const chain = DefaultStore.getChainOrThrow("near", NearChain);
 
   const vault =
     DefaultStore.vaults[

contract_manager/scripts/generate_upgrade_ton_contract_proposal.ts

@@ -42,10 +42,10 @@ async function main() {
   const wormholeChainName = toChainName(chainId);
 
   // Get the TON chain instance from DefaultStore based on network
-  const chain = DefaultStore.chains[isMainnet ? "ton_mainnet" : "ton_testnet"];
-  if (!chain || !(chain instanceof TonChain)) {
-    throw new Error(`Chain configuration not found for TON ${argv.network}`);
-  }
+  const chain = DefaultStore.getChainOrThrow(
+    isMainnet ? "ton_mainnet" : "ton_testnet",
+    TonChain
+  );
 
   const vault =
     DefaultStore.vaults[