diff --git a/packages/tasks/templates/omnichain/tasks/interact.ts.hbs b/packages/tasks/templates/omnichain/tasks/interact.ts.hbs index 9c9a9b9..82aa0ce 100644 --- a/packages/tasks/templates/omnichain/tasks/interact.ts.hbs +++ b/packages/tasks/templates/omnichain/tasks/interact.ts.hbs @@ -3,12 +3,13 @@ import { HardhatRuntimeEnvironment } from "hardhat/types"; import { parseUnits } from "@ethersproject/units"; import { getAddress } from "@zetachain/protocol-contracts"; import ERC20Custody from "@zetachain/protocol-contracts/abi/evm/ERC20Custody.sol/ERC20Custody.json"; -import { prepareData } from "@zetachain/toolkit/client"; +import { prepareData, ZetaChainClient } from "@zetachain/toolkit/client"; import { utils, ethers } from "ethers"; import ERC20 from "@openzeppelin/contracts/build/contracts/ERC20.json"; const main = async (args: any, hre: HardhatRuntimeEnvironment) => { const [signer] = await hre.ethers.getSigners(); + const client = new ZetaChainClient({ network: "testnet", signer }); const data = prepareData( args.contract, @@ -16,9 +17,34 @@ const main = async (args: any, hre: HardhatRuntimeEnvironment) => { [{{#each arguments.names}}args.{{this}}, {{/each}}] ); + let decimals = 18; + + if (args.erc20) { + const contract = new ethers.Contract(args.erc20, ERC20.abi, signer); + decimals = await contract.decimals(); + } + + const value = ethers.utils.parseUnits(args.amount, decimals); + + let inputToken = args.erc20 + ? await client.getZRC20FromERC20(args.erc20) + : await client.getZRC20GasToken(hre.network.name); + + const refundFee = await client.getRefundFee(inputToken); + const refundFeeAmount = ethers.utils.formatUnits( + refundFee.amount, + refundFee.decimals + ); + + if (value.lt(refundFee.amount)) { + throw new Error( + `Amount ${args.amount} is less than refund fee ${refundFeeAmount}. This means if this transaction fails, you will not be able to get the refund of deposited tokens. Consider increasing the amount.` + ); + } + let tx; - if (args.token) { + if (args.erc20) { const custodyAddress = getAddress("erc20Custody", hre.network.name as any); if (!custodyAddress) { throw new Error( @@ -31,13 +57,13 @@ const main = async (args: any, hre: HardhatRuntimeEnvironment) => { ERC20Custody.abi, signer ); - const tokenContract = new ethers.Contract(args.token, ERC20.abi, signer); + const tokenContract = new ethers.Contract(args.erc20, ERC20.abi, signer); const decimals = await tokenContract.decimals(); const value = parseUnits(args.amount, decimals); const approve = await tokenContract.approve(custodyAddress, value); await approve.wait(); - tx = await custodyContract.deposit(signer.address, args.token, value, data); + tx = await custodyContract.deposit(signer.address, args.erc20, value, data); tx.wait(); } else { const value = parseUnits(args.amount, 18); @@ -57,9 +83,9 @@ const main = async (args: any, hre: HardhatRuntimeEnvironment) => { }; task("interact", "Interact with the contract", main) - .addParam("contract", "The address of the withdraw contract on ZetaChain") + .addParam("contract", "The address of a universal app contract on ZetaChain") .addParam("amount", "Amount of tokens to send") - .addOptionalParam("token", "The address of the token to send") + .addOptionalParam("erc20", "Send an ERC-20 token") .addFlag("json", "Output in JSON") {{#each arguments.names}} .addParam("{{this}}")