restore script

packages/zevm-app-contracts/scripts/zeta-swap/add-liquidity-zeta-uniswap.ts

@@ -0,0 +1,179 @@
+import { MaxUint256 } from "@ethersproject/constants";
+import { formatUnits, parseUnits } from "@ethersproject/units";
+import { SignerWithAddress } from "@nomiclabs/hardhat-ethers/signers";
+import { getChainId } from "@zetachain/addresses";
+import { NetworkName } from "@zetachain/addresses";
+import { getAddress } from "@zetachain/addresses";
+import { getGasSymbolByNetwork, getSystemContractAddress } from "@zetachain/addresses-tools";
+import { BigNumber } from "ethers";
+import { ethers } from "hardhat";
+
+import {
+  ERC20,
+  ERC20__factory,
+  IUniswapV2Factory__factory,
+  IUniswapV2Pair__factory,
+  IUniswapV2Router02,
+  SystemContract__factory,
+  UniswapV2Router02__factory
+} from "../../typechain-types";
+
+const SYSTEM_CONTRACT = getSystemContractAddress();
+
+const BTC_TO_ADD = parseUnits("0", 8);
+const ETH_TO_ADD = parseUnits("1500");
+const MATIC_TO_ADD = parseUnits("1500");
+const BNB_TO_ADD = parseUnits("100");
+
+const ZETA_TO_ADD = parseUnits("0");
+
+interface Pair {
+  TokenA: string;
+  TokenB: string;
+}
+
+export const getNow = async () => {
+  const block = await ethers.provider.getBlock("latest");
+  return block.timestamp;
+};
+
+export const sortPair = (token1: string, token2: string): Pair => {
+  if (token1 < token2) {
+    return { TokenA: token1, TokenB: token2 };
+  }
+  return { TokenA: token2, TokenB: token1 };
+};
+
+const addTokenEthLiquidity = async (
+  tokenContract: ERC20,
+  tokenAmountToAdd: BigNumber,
+  ETHToAdd: BigNumber,
+  uniswapRouter: IUniswapV2Router02,
+  deployer: SignerWithAddress
+) => {
+  const tx1 = await tokenContract.approve(uniswapRouter.address, MaxUint256);
+  await tx1.wait();
+
+  const tx2 = await uniswapRouter.addLiquidityETH(
+    tokenContract.address,
+    tokenAmountToAdd,
+    0,
+    0,
+    deployer.address,
+    (await getNow()) + 360,
+    { gasLimit: 10_000_000, value: ETHToAdd }
+  );
+  await tx2.wait();
+};
+
+const estimateZetaForToken = async (
+  network: NetworkName,
+  WZETAAddress: string,
+  uniswapFactoryAddress: string,
+  tokenContract: ERC20,
+  tokenAmountToAdd: BigNumber,
+  deployer: SignerWithAddress
+) => {
+  const uniswapV2Factory = IUniswapV2Factory__factory.connect(uniswapFactoryAddress, deployer);
+
+  const pair = sortPair(tokenContract.address, WZETAAddress);
+
+  const poolAddress = await uniswapV2Factory.getPair(pair.TokenA, pair.TokenB);
+
+  const pool = IUniswapV2Pair__factory.connect(poolAddress, deployer);
+
+  const reserves = await pool.getReserves();
+
+  const reservesZETA = WZETAAddress < tokenContract.address ? reserves.reserve0 : reserves.reserve1;
+  const reservesToken = WZETAAddress > tokenContract.address ? reserves.reserve0 : reserves.reserve1;
+
+  const ZETAValue = reservesZETA.mul(tokenAmountToAdd).div(reservesToken);
+
+  const tokenDecimals = await tokenContract.decimals();
+  console.log(
+    `Zeta/${getGasSymbolByNetwork(network)} reserves ${formatUnits(reservesZETA)}/${formatUnits(
+      reservesToken,
+      tokenDecimals
+    )}`
+  );
+  return ZETAValue;
+};
+
+async function addLiquidity(
+  network: NetworkName,
+  tokenAmountToAdd: BigNumber,
+  WZETAAddress: string,
+  uniswapFactoryAddress: string,
+  uniswapRouterAddress: string
+) {
+  console.log(`Adding liquidity for: ${network}`);
+  const initLiquidityPool = !ZETA_TO_ADD.isZero();
+
+  const [deployer] = await ethers.getSigners();
+
+  const systemContract = await SystemContract__factory.connect(SYSTEM_CONTRACT, deployer);
+  const uniswapRouter = await UniswapV2Router02__factory.connect(uniswapRouterAddress, deployer);
+
+  const tokenAddress = await systemContract.gasCoinZRC20ByChainId(getChainId(network));
+  const tokenContract = ERC20__factory.connect(tokenAddress, deployer);
+  const tokenDecimals = await tokenContract.decimals();
+
+  const zetaToAdd = initLiquidityPool
+    ? ZETA_TO_ADD
+    : await estimateZetaForToken(
+        network,
+        WZETAAddress,
+        uniswapFactoryAddress,
+        tokenContract,
+        tokenAmountToAdd,
+        deployer
+      );
+
+  console.log(
+    `Zeta/${getGasSymbolByNetwork(network)} to add ${formatUnits(zetaToAdd)}/${formatUnits(
+      tokenAmountToAdd,
+      tokenDecimals
+    )}`
+  );
+
+  // await addTokenEthLiquidity(tokenContract, tokenAmountToAdd, zetaToAdd, uniswapRouter, deployer);
+}
+async function main() {
+  const WZETA_ADDRESS = getAddress({
+    address: "weth9",
+    networkName: "athens",
+    zetaNetwork: "athens"
+  });
+
+  const UNISWAP_FACTORY_ADDRESS = getAddress({
+    address: "uniswapV2Factory",
+    networkName: "athens",
+    zetaNetwork: "athens"
+  });
+
+  const UNISWAP_ROUTER_ADDRESS = getAddress({
+    address: "uniswapV2Router02",
+    networkName: "athens",
+    zetaNetwork: "athens"
+  });
+
+  if (!ETH_TO_ADD.isZero()) {
+    await addLiquidity("goerli", ETH_TO_ADD, WZETA_ADDRESS, UNISWAP_FACTORY_ADDRESS, UNISWAP_ROUTER_ADDRESS);
+  }
+  if (!MATIC_TO_ADD.isZero()) {
+    await addLiquidity("polygon-mumbai", MATIC_TO_ADD, WZETA_ADDRESS, UNISWAP_FACTORY_ADDRESS, UNISWAP_ROUTER_ADDRESS);
+  }
+  if (!BNB_TO_ADD.isZero()) {
+    await addLiquidity("bsc-testnet", BNB_TO_ADD, WZETA_ADDRESS, UNISWAP_FACTORY_ADDRESS, UNISWAP_ROUTER_ADDRESS);
+  }
+  if (!BTC_TO_ADD.isZero()) {
+    await addLiquidity("bitcoin-test", BTC_TO_ADD, WZETA_ADDRESS, UNISWAP_FACTORY_ADDRESS, UNISWAP_ROUTER_ADDRESS);
+  }
+}
+
+main()
+  .then(() => process.exit(0))
+  .catch(error => {
+    console.error(error);
+    process.exit(1);
+  });