Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: check refund fee in omnichain interact task #150

Merged
merged 1 commit into from
Jun 14, 2024
Merged

feat: check refund fee in omnichain interact task #150

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 32 additions & 6 deletions packages/tasks/templates/omnichain/tasks/interact.ts.hbs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,22 +3,48 @@ 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,
[{{#each arguments.types}}"{{this}}", {{/each}}],
[{{#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(
Expand All @@ -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);
Expand All @@ -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}}")
Expand Down
Loading