unwrapWeth.ts

import {
  ethers,
  retrieveSignerFromCLIArgs,
  getProvider,
  WETH9,
  toBN,
  getNetworkName,
  TOKEN_SYMBOLS_MAP,
  assert,
} from "../src/utils";
import { askYesNoQuestion } from "./utils";
import minimist from "minimist";

const args = minimist(process.argv.slice(2), {
  string: ["amount", "chainId"],
  boolean: ["wrap"],
});

// Example run:
// ts-node ./scripts/unwrapWeth.ts
// \ --amount 3000000000000000000
// \ --chainId 1
// \ --wrap true
// \ --wallet gckms
// \ --keys bot1

export async function run(): Promise<void> {
  if (!Object.keys(args).includes("chainId")) {
    throw new Error("Define `chainId` as the chain you want to connect on");
  }
  if (!Object.keys(args).includes("amount")) {
    throw new Error("Define `amount` as how much you want to unwrap");
  }
  const baseSigner = await retrieveSignerFromCLIArgs();
  const signerAddr = await baseSigner.getAddress();
  const chainId = Number(args.chainId);
  const connectedSigner = baseSigner.connect(await getProvider(chainId));
  assert(TOKEN_SYMBOLS_MAP.WETH.addresses[chainId], "chainId does not have a defined WETH address");
  const token = TOKEN_SYMBOLS_MAP.WETH.addresses[chainId];
  const weth = new ethers.Contract(token, WETH9.abi, connectedSigner);
  const decimals = 18;
  const amountFromWei = ethers.utils.formatUnits(args.amount, decimals);

  const wrapping = args.wrap;
  if (wrapping) {
    console.log("Wrapping WETH 🎁");
    const currentBalance = ethers.utils.formatUnits(await connectedSigner.provider.getBalance(signerAddr), decimals);
    console.log(`Current ETH balance for account ${signerAddr} on ${getNetworkName(chainId)}: ${currentBalance}`);
    if ((await connectedSigner.provider.getBalance(signerAddr)).lt(toBN(args.amount))) {
      console.log(`ETH balance < ${amountFromWei}, exiting`);
      return;
    }
    console.log(`Wrap ${amountFromWei} ETH`);
    // Check the user is ok with the info provided. else abort.
    if (!(await askYesNoQuestion("\nConfirm that you want to execute this transaction?"))) {
      return;
    }
    console.log("sending...");
    const tx = await weth.deposit({ value: args.amount });
    const receipt = await tx.wait();
    console.log("Transaction hash:", receipt.transactionHash);
  } else {
    console.log("Unwrapping WETH 🎊");
    const currentBalance = ethers.utils.formatUnits(await weth.balanceOf(signerAddr), decimals);
    console.log(`Current WETH balance for account ${signerAddr} on Mainnet: ${currentBalance}`);
    if ((await weth.balanceOf(signerAddr)).lt(toBN(args.amount))) {
      console.log(`WETH balance < ${amountFromWei}, exiting`);
      return;
    }
    console.log(`Unwrap ${amountFromWei} WETH`);
    // Check the user is ok with the info provided. else abort.
    if (!(await askYesNoQuestion("\nConfirm that you want to execute this transaction?"))) {
      return;
    }
    console.log("sending...");
    const tx = await weth.withdraw(args.amount);
    const receipt = await tx.wait();
    console.log("Transaction hash:", receipt.transactionHash);
  }
}

if (require.main === module) {
  run()
    .then(async () => {
      // eslint-disable-next-line no-process-exit
      process.exit(0);
    })
    .catch(async (error) => {
      console.error("Process exited with", error);
      // eslint-disable-next-line no-process-exit
      process.exit(1);
    });
}