Skip to content

Commit

Permalink
update script
Browse files Browse the repository at this point in the history
  • Loading branch information
@fadeev
fadeev committed Jan 10, 2024
1 parent c226039 commit 7ba85a9
Showing 1 changed file with 140 additions and 100 deletions.
240 changes: 140 additions & 100 deletions tasks/addresses.ts
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import { task } from "hardhat/config";

Check failure on line 1 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Run autofix to sort these imports!
import { HardhatRuntimeEnvironment } from "hardhat/types";
import axios, { AxiosResponse } from "axios";
import { ERC20Custody__factory } from "../typechain-types/factories/contracts/evm/ERC20Custody__factory";
import { SystemContract__factory } from "../typechain-types/factories/contracts/zevm/SystemContract.sol/SystemContract__factory";
import {
ZetaConnectorBase__factory,
Expand All @@ -9,23 +10,19 @@ import {
ZetaConnectorZEVM__factory,
} from "@typechain-types";
import { ZetaConnector__factory } from "../typechain-types";
import { zetaConnectorNonEthSol } from "typechain-types/factories/contracts/evm";
import { zetaConnectorNonEthSol } from "../typechain-types/factories/contracts/evm";
import { getEndpoints } from "@zetachain/networks";

const ATHENS_EVM_RPC = "https://zetachain-athens-evm.blockpi.network/v1/rpc/public";
const ATHENS_TENDERMINT_RPC = "http://46.4.15.110:1317";
const ATHENS_TENDERMINT_RPC = "https://zetachain-athens.blockpi.network/lcd/v1/public";

const fetchChains = async (chainObject: any) => {
const fetchChains = async () => {
const URL = `${ATHENS_TENDERMINT_RPC}/zeta-chain/observer/supportedChains`;
try {
const response: AxiosResponse<any> = await axios.get(URL);

if (response.status === 200) {
const chains: string[] = response.data.chains;
chains.forEach((chain: any) => {
chainObject[chain.chain_name.toLowerCase()] = {
chainId: chain.chain_id,
};
});
return response.data.chains;
} else {
console.error("Error fetching chains:", response.status, response.statusText);
}
Expand All @@ -34,17 +31,23 @@ const fetchChains = async (chainObject: any) => {
}
};

const fetchTssData = async (chainObject: any) => {
const fetchTssData = async (chains: any, addresses: any) => {
const URL = `${ATHENS_TENDERMINT_RPC}/zeta-chain/crosschain/get_tss_address`;
try {
const tssResponse: AxiosResponse<any> = await axios.get(URL);

if (tssResponse.status === 200) {
for (const chain in chainObject) {
chains.forEach((chain: any) => {
const { btc, eth } = tssResponse.data;
if (chain.includes("zeta")) continue;
chainObject[chain]["tss"] = chain.includes("btc") ? btc : eth;
}
if (chain.chain_name === "zeta_testnet") return;
addresses.push({
chain_id: chain.chain_id,
chain_name: chain.chain_name,
type: "tss",
category: "omnichain",

Check failure on line 47 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'category' should be before 'type'
address: chain.chain_name === "btc_testnet" ? btc : eth,

Check failure on line 48 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'address' should be before 'category'
});
});
} else {
console.error("Error fetching TSS data:", tssResponse.status, tssResponse.statusText);
}
Expand All @@ -53,17 +56,26 @@ const fetchTssData = async (chainObject: any) => {
}
};

const fetchSystemContract = async (chainObject: any) => {
const fetchSystemContract = async (addresses: any) => {
const URL = `${ATHENS_TENDERMINT_RPC}/zeta-chain/fungible/system_contract`;
try {
const systemContractResponse: AxiosResponse<any> = await axios.get(URL);

if (systemContractResponse.status === 200) {
if (!chainObject["zeta_testnet"]) {
chainObject["zeta_testnet"] = {};
}
chainObject["zeta_testnet"]["systemContract"] = systemContractResponse.data.SystemContract.system_contract;
chainObject["zeta_testnet"]["connectorZEVM"] = systemContractResponse.data.SystemContract.connector_zevm;
addresses.push({
chain_id: 7001,
chain_name: "zeta_testnet",
type: "systemContract",
category: "omnichain",

Check failure on line 69 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'category' should be before 'type'
address: systemContractResponse.data.SystemContract.system_contract,

Check failure on line 70 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'address' should be before 'category'
});
addresses.push({
chain_id: 7001,
chain_name: "zeta_testnet",
type: "connectorZEVM",
category: "messaging",

Check failure on line 76 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'category' should be before 'type'
address: systemContractResponse.data.SystemContract.connector_zevm,

Check failure on line 77 in tasks/addresses.ts

View workflow job for this annotation

GitHub Actions / build 

Expected object keys to be in ascending order. 'address' should be before 'category'
});
} else {
console.error("Error fetching system contract:", systemContractResponse.statusText);
}
Expand All @@ -72,20 +84,32 @@ const fetchSystemContract = async (chainObject: any) => {
}
};

const fetchForeignCoinsData = async (chainObject: any) => {
const fetchForeignCoinsData = async (chains: any, addresses: any) => {
const URL = `${ATHENS_TENDERMINT_RPC}/zeta-chain/fungible/foreign_coins`;
try {
const foreignCoinsResponse: AxiosResponse<any> = await axios.get(URL);
if (foreignCoinsResponse.status === 200) {
foreignCoinsResponse.data.foreignCoins.forEach((coin: any) => {
for (const key in chainObject) {
if (chainObject.hasOwnProperty(key)) {
const chain = chainObject[key];
if (chain.chainId === coin.foreign_chain_id) {
chain.zrc20ContractAddress = coin.zrc20_contract_address;
}
}
}
foreignCoinsResponse.data.foreignCoins.forEach((token: any) => {
addresses.push({
chain_id: token.foreign_chain_id,
chain_name: chains.find((c: any) => c.chain_id === token.foreign_chain_id)?.chain_name,
coin_type: token.coin_type,
address: token.asset,
symbol: token.symbol,
zrc20: token.zrc20_contract_address,
category: "omnichain",
decimals: token.decimals, // TODO: dynamically fetch from contract (to verify)
});
addresses.push({
chain_id: 7001,
chain_name: "zeta_testnet",
coin_type: "ZRC20",
address: token.zrc20_contract_address,
symbol: token.name,
asset: token.asset,
category: "omnichain",
decimals: 18, // TODO: dynamically fetch from contract
});
});
} else {
console.error("Error fetching foreign coins data:", foreignCoinsResponse.status, foreignCoinsResponse.statusText);
Expand All @@ -95,86 +119,102 @@ const fetchForeignCoinsData = async (chainObject: any) => {
}
};

const fetchAthensAddresses = async (chainObject: any, hre: HardhatRuntimeEnvironment) => {
const zeta = "zeta_testnet";
if (chainObject[zeta] && chainObject[zeta]["systemContract"]) {
const athens = chainObject[zeta];
const systemContract = athens["systemContract"] as string;
const provider = new hre.ethers.providers.JsonRpcProvider(ATHENS_EVM_RPC);
const sc = SystemContract__factory.connect(systemContract, provider);
try {
athens["uniswapv2FactoryAddress"] = await sc.uniswapv2FactoryAddress();
athens["wZetaContractAddress"] = await sc.wZetaContractAddress();
athens["uniswapv2Router02Address"] = await sc.uniswapv2Router02Address();
athens["zetaConnectorZEVMAddress"] = await sc.zetaConnectorZEVMAddress();
athens["fungibleModuleAddress"] = await sc.FUNGIBLE_MODULE_ADDRESS();
} catch (error) {
console.error("Error fetching addresses from ZetaChain:", error);
}
} else {
console.error("ZetaChain system contract not found.");
}
};

const fetchChainSpecificAddresses = async (chainObject: any) => {
const URL = `${ATHENS_TENDERMINT_RPC}/zeta-chain/observer/get_client_params_for_chain`;
const fetchAthensAddresses = async (addresses: any, hre: HardhatRuntimeEnvironment) => {
const systemContract = addresses.find((a: any) => {
return a.chain_name === "zeta_testnet" && a.type === "systemContract";
})?.address;
const provider = new hre.ethers.providers.JsonRpcProvider(ATHENS_EVM_RPC);
const sc = SystemContract__factory.connect(systemContract, provider);
const common = {
chain_id: 7001,
chain_name: "zeta_testnet",
category: "omnichain",
};
try {
for (const chain in chainObject) {
if (chain.includes("zeta") || chain.includes("btc")) continue;
const id = chainObject[chain].chainId;
const response: AxiosResponse<any> = await axios.get(`${URL}/${id}`);
if (response.status === 200) {
chainObject[chain]["zetaTokenContractAddress"] = response.data.core_params.zeta_token_contract_address;
chainObject[chain]["connectorContractAddress"] = response.data.core_params.connector_contract_address;
chainObject[chain]["erc20CustodyContractAddress"] = response.data.core_params.erc20_custody_contract_address;
} else {
console.error("Error fetching chain data:", response.status, response.statusText);
}
}
addresses.push({ ...common, type: "uniswapv2FactoryAddress", address: await sc.uniswapv2FactoryAddress() });
addresses.push({ ...common, type: "wZetaContractAddress", address: await sc.wZetaContractAddress() });
addresses.push({ ...common, type: "uniswapv2Router02Address", address: await sc.uniswapv2Router02Address() });
addresses.push({ ...common, type: "zetaConnectorZEVMAddress", address: await sc.zetaConnectorZEVMAddress() });
addresses.push({ ...common, type: "fungibleModuleAddress", address: await sc.FUNGIBLE_MODULE_ADDRESS() });
} catch (error) {
console.error("Error fetching chain data:", error);
console.error("Error fetching addresses from ZetaChain:", error);
}
};

const fetchPauserUpdater = async (chainObject: any, hre: HardhatRuntimeEnvironment) => {
try {
for (const chain in chainObject) {
const provider = new hre.ethers.providers.JsonRpcProvider(ATHENS_EVM_RPC);
}
} catch (error) {
console.error("Error fetching addresses from ZetaChain:", error);
}
const zeta = "zeta_testnet";
if (chainObject[zeta] && chainObject[zeta]["systemContract"]) {
const athens = chainObject[zeta];
const systemContract = athens["systemContract"] as string;
const provider = new hre.ethers.providers.JsonRpcProvider(ATHENS_EVM_RPC);
const sc = SystemContract__factory.connect(systemContract, provider);
try {
athens["uniswapv2FactoryAddress"] = await sc.uniswapv2FactoryAddress();
athens["wZetaContractAddress"] = await sc.wZetaContractAddress();
athens["uniswapv2Router02Address"] = await sc.uniswapv2Router02Address();
athens["zetaConnectorZEVMAddress"] = await sc.zetaConnectorZEVMAddress();
athens["fungibleModuleAddress"] = await sc.FUNGIBLE_MODULE_ADDRESS();
} catch (error) {
console.error("Error fetching addresses from ZetaChain:", error);
}
} else {
console.error("ZetaChain system contract not found.");
}
const fetchChainSpecificAddresses = async (chains: any, addresses: any) => {
await Promise.all(
chains.map(async (chain: any) => {
return axios
.get(`${ATHENS_TENDERMINT_RPC}/zeta-chain/observer/get_client_params_for_chain/${chain.chain_id}`)
.then(({ data }) => {
if (data.core_params.zeta_token_contract_address) {
addresses.push({
chain_id: chain.chain_id,
chain_name: chain.chain_name,
category: "messaging",
type: "zetaToken",
address: data.core_params.zeta_token_contract_address,
});
}
if (data.core_params.connector_contract_address) {
addresses.push({
chain_id: chain.chain_id,
chain_name: chain.chain_name,
category: "messaging",
type: "connector",
address: data.core_params.connector_contract_address,
});
}
if (data.core_params.erc20_custody_contract_address) {
addresses.push({
chain_id: chain.chain_id,
category: "omnichain",
chain_name: chain.chain_name,
type: "erc20Custody",
address: data.core_params.erc20_custody_contract_address,
});
}
});
})
);
};

const fetchTSSUpdater = async (chains: any, addresses: any) => {
await Promise.all(
chains.map(async (chain: any) => {
const erc20Custody = addresses.find((a: any) => {
return a.chain_name === chain.chain_name && a.type === "erc20Custody";
})?.address;
if (erc20Custody) {
const rpc = getEndpoints("evm", chain.chain_name)[0]?.url;
const provider = new hre.ethers.providers.JsonRpcProvider(rpc);
const custody = ERC20Custody__factory.connect(erc20Custody, provider);
return custody.TSSAddressUpdater().then((address: string) => {
addresses.push({
chain_id: chain.chain_id,
chain_name: chain.chain_name,
category: "omnichain",
type: "tssUpdater",
address,
});
});
}
})
);
};

const main = async (args: any, hre: HardhatRuntimeEnvironment) => {
const chainObject: any = {};
const addresses: any = [];

await fetchChains(chainObject);
await fetchTssData(chainObject);
await fetchSystemContract(chainObject);
await fetchForeignCoinsData(chainObject);
await fetchAthensAddresses(chainObject, hre);
await fetchChainSpecificAddresses(chainObject);
const chains = await fetchChains();
await fetchTssData(chains, addresses);
await fetchSystemContract(addresses);
await fetchForeignCoinsData(chains, addresses);
await fetchAthensAddresses(addresses, hre);
await fetchChainSpecificAddresses(chains, addresses);
await fetchTSSUpdater(chains, addresses);

console.log(JSON.stringify(chainObject, null, 2));
console.log(JSON.stringify(addresses, null, 2));
};

task("addresses", "").setAction(main);

0 comments on commit 7ba85a9

Please sign in to comment.