read gateway address from networks file

package.json

@@ -100,7 +100,7 @@
     "@solana/web3.js": "^1.95.3",
     "@uniswap/v2-periphery": "^1.1.0-beta.0",
     "@zetachain/faucet-cli": "^4.1.1",
-    "@zetachain/networks": "v10.0.0",
+    "@zetachain/networks": "v10.0.0-rc1",
     "@zetachain/protocol-contracts": "11.0.0-rc3",
     "axios": "^1.4.0",
     "bech32": "^2.0.0",

packages/client/src/client.ts

@@ -2,6 +2,8 @@ import type { Wallet as SolanaWallet } from "@coral-xyz/anchor";
 import type { WalletContextState } from "@solana/wallet-adapter-react";
 import { PublicKey } from "@solana/web3.js";
 import { networks } from "@zetachain/networks";
+import mainnetAddresses from "@zetachain/protocol-contracts/dist/data/addresses.mainnet.json";
+import testnetAddresses from "@zetachain/protocol-contracts/dist/data/addresses.testnet.json";
 import type { Signer, Wallet } from "ethers";
 import merge from "lodash/merge";
 
@@ -31,6 +33,7 @@ import {
 
 export interface ZetaChainClientParamsBase {
   chains?: { [key: string]: any };
+  contracts?: LocalnetAddress[] | MainnetTestnetAddress[];
   network?: string;
 }
 
@@ -68,13 +71,28 @@ export type ZetaChainClientParams = ZetaChainClientParamsBase &
       }
   );
 
+interface MainnetTestnetAddress {
+  address: string;
+  category: string;
+  chain_id: number;
+  chain_name: string;
+  type: string;
+}
+
+interface LocalnetAddress {
+  address: string;
+  chain: string;
+  type: string;
+}
+
 export class ZetaChainClient {
   public chains: { [key: string]: any };
   public network: string;
   public wallet: Wallet | undefined;
   public signer: any | undefined;
   public solanaWallet: SolanaWallet | undefined;
   public solanaAdapter: WalletContextState | undefined;
+  private contracts: LocalnetAddress[] | MainnetTestnetAddress[];
 
   /**
    * Initializes ZetaChainClient instance.
@@ -136,6 +154,16 @@ export class ZetaChainClient {
     this.chains = { ...networks };
     this.network = params.network || "";
 
+    if (params.contracts) {
+      this.contracts = params.contracts;
+    } else if (this.network === "localnet") {
+      throw new Error("Localnet contracts are required");
+    } else {
+      this.contracts = this.network.includes("test")
+        ? testnetAddresses
+        : mainnetAddresses;
+    }
+
     this.mergeChains(params.chains);
   }
 
@@ -147,6 +175,49 @@ export class ZetaChainClient {
     });
   }
 
+  public getGatewayAddress(): string {
+    if (this.network === "localnet") {
+      const gateway = (this.contracts as LocalnetAddress[]).find(
+        (item) => item.type === "gatewayZEVM"
+      );
+
+      if (!gateway) {
+        throw new Error("Gateway address not found in localnet configuration");
+      }
+
+      return gateway.address;
+    } else {
+      let gateway;
+      if (this.wallet) {
+        try {
+          gateway = (this.contracts as MainnetTestnetAddress[]).find(
+            async (item) =>
+              (await this.wallet!.getChainId()) === item.chain_id &&
+              item.type === "gateway"
+          );
+        } catch (error) {
+          throw new Error("Failed to get gateway address: " + error);
+        }
+      } else {
+        try {
+          gateway = (this.contracts as MainnetTestnetAddress[]).find(
+            async (item) =>
+              (await this.signer!.getChainId()) === item.chain_id &&
+              item.type === "gateway"
+          );
+        } catch (error) {
+          throw new Error("Failed to get gateway address: " + error);
+        }
+      }
+
+      if (!gateway) {
+        throw new Error(`Gateway address not found in signer or wallet`);
+      }
+
+      return gateway.address;
+    }
+  }
+
   public getChains(): { [key: string]: any } {
     return this.chains;
   }

packages/client/src/evmCall.ts

@@ -24,7 +24,7 @@ import type { revertOptions, txOptions } from "./types";
 export const evmCall = async function (
   this: ZetaChainClient,
   args: {
-    gatewayEvm: string;
+    gatewayEvm?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
@@ -34,7 +34,12 @@ export const evmCall = async function (
 ) {
   const signer = this.signer;
   const { utils } = ethers;
-  const gateway = new ethers.Contract(args.gatewayEvm, GatewayABI.abi, signer);
+  const gatewayEvmAddress = args.gatewayEvm || this.getGatewayAddress();
+  const gateway = new ethers.Contract(
+    gatewayEvmAddress,
+    GatewayABI.abi,
+    signer
+  );
 
   const valuesArray = args.values.map((value, index) => {
     const type = args.types[index];

packages/client/src/evmDeposit.ts

@@ -27,15 +27,20 @@ export const evmDeposit = async function (
   args: {
     amount: string;
     erc20: string;
-    gatewayEvm: string;
+    gatewayEvm?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
   }
 ) {
   const signer = this.signer;
   const { utils } = ethers;
-  const gateway = new ethers.Contract(args.gatewayEvm, GatewayABI.abi, signer);
+  const gatewayEvmAddress = args.gatewayEvm || this.getGatewayAddress();
+  const gateway = new ethers.Contract(
+    gatewayEvmAddress,
+    GatewayABI.abi,
+    signer
+  );
 
   const revertOptions = {
     abortAddress: "0x0000000000000000000000000000000000000000", // not used

packages/client/src/evmDepositAndCall.ts

@@ -29,7 +29,7 @@ export const evmDepositAndCall = async function (
   args: {
     amount: string;
     erc20: string;
-    gatewayEvm: string;
+    gatewayEvm?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
@@ -39,7 +39,12 @@ export const evmDepositAndCall = async function (
 ) {
   const signer = this.signer;
   const { utils } = ethers;
-  const gateway = new ethers.Contract(args.gatewayEvm, GatewayABI.abi, signer);
+  const gatewayEvmAddress = args.gatewayEvm || this.getGatewayAddress();
+  const gateway = new ethers.Contract(
+    gatewayEvmAddress,
+    GatewayABI.abi,
+    signer
+  );
 
   const revertOptions = {
     abortAddress: "0x0000000000000000000000000000000000000000", // not used

packages/client/src/zetachainCall.ts

@@ -32,7 +32,7 @@ export const zetachainCall = async function (
   args: {
     callOptions: any;
     function: string;
-    gatewayZetaChain: string;
+    gatewayZetaChain?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
@@ -43,9 +43,10 @@ export const zetachainCall = async function (
 ) {
   const signer = this.signer;
   const { utils } = ethers;
-
+  const gatewayZetaChainAddress =
+    args.gatewayZetaChain || this.getGatewayAddress();
   const gateway = new ethers.Contract(
-    args.gatewayZetaChain,
+    gatewayZetaChainAddress,
     GatewayABI.abi,
     signer
   );

packages/client/src/zetachainWithdraw.ts

@@ -28,7 +28,7 @@ export const zetachainWithdraw = async function (
   this: ZetaChainClient,
   args: {
     amount: string;
-    gatewayZetaChain: string;
+    gatewayZetaChain?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
@@ -38,8 +38,10 @@ export const zetachainWithdraw = async function (
   const signer = this.signer;
   const { utils } = ethers;
 
+  const gatewayZetaChainAddress =
+    args.gatewayZetaChain || this.getGatewayAddress();
   const gateway = new ethers.Contract(
-    args.gatewayZetaChain,
+    gatewayZetaChainAddress,
     GatewayABI.abi,
     signer
   );

packages/client/src/zetachainWithdrawAndCall.ts

@@ -34,7 +34,7 @@ export const zetachainWithdrawAndCall = async function (
     amount: string;
     callOptions: any;
     function?: string;
-    gatewayZetaChain: string;
+    gatewayZetaChain?: string;
     receiver: string;
     revertOptions: revertOptions;
     txOptions: txOptions;
@@ -46,8 +46,10 @@ export const zetachainWithdrawAndCall = async function (
   const signer = this.signer;
   const { utils } = ethers;
 
+  const gatewayZetaChainAddress =
+    args.gatewayZetaChain || this.getGatewayAddress();
   const gateway = new ethers.Contract(
-    args.gatewayZetaChain,
+    gatewayZetaChainAddress,
     GatewayABI.abi,
     signer
   );

packages/tasks/src/evmCall.ts

@@ -6,7 +6,8 @@ import { ZetaChainClient } from "../../client/src/";
 export const evmCall = async (args: any, hre: HardhatRuntimeEnvironment) => {
   try {
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const tx = await client.evmCall({
       gatewayEvm: args.gatewayEvm,
       receiver: args.receiver,
@@ -32,11 +33,7 @@ export const evmCall = async (args: any, hre: HardhatRuntimeEnvironment) => {
 
 task("evm-call", "Call a universal app", evmCall)
   .addParam("receiver", "Receiver address on ZetaChain")
-  .addOptionalParam(
-    "gatewayEvm",
-    "contract address of gateway on EVM",
-    "0x9fE46736679d2D9a65F0992F2272dE9f3c7fa6e0"
-  )
+  .addOptionalParam("gatewayEvm", "contract address of gateway on EVM")
   .addFlag("callOnRevert", "Whether to call on revert")
   .addOptionalParam(
     "revertAddress",

packages/tasks/src/evmDeposit.ts

@@ -6,7 +6,8 @@ import { ZetaChainClient } from "../../client/src/";
 export const evmDeposit = async (args: any, hre: HardhatRuntimeEnvironment) => {
   try {
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const tx = await client.evmDeposit({
       amount: args.amount,
       erc20: args.erc20,
@@ -34,11 +35,7 @@ export const evmDeposit = async (args: any, hre: HardhatRuntimeEnvironment) => {
 
 task("evm-deposit", "Deposit tokens", evmDeposit)
   .addParam("receiver", "Receiver address on ZetaChain")
-  .addOptionalParam(
-    "gatewayEvm",
-    "contract address of gateway on EVM",
-    "0x9fE46736679d2D9a65F0992F2272dE9f3c7fa6e0"
-  )
+  .addOptionalParam("gatewayEvm", "contract address of gateway on EVM")
   .addFlag("callOnRevert", "Whether to call on revert")
   .addOptionalParam(
     "revertAddress",

packages/tasks/src/evmDepositAndCall.ts

@@ -9,7 +9,8 @@ export const evmDepositAndCall = async (
 ) => {
   try {
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const tx = await client.evmDepositAndCall({
       amount: args.amount,
       erc20: args.erc20,
@@ -39,11 +40,7 @@ export const evmDepositAndCall = async (
 
 task("evm-deposit-and-call", "Deposit tokens", evmDepositAndCall)
   .addParam("receiver", "Receiver address on ZetaChain")
-  .addOptionalParam(
-    "gatewayEvm",
-    "contract address of gateway on EVM",
-    "0x9fE46736679d2D9a65F0992F2272dE9f3c7fa6e0"
-  )
+  .addOptionalParam("gatewayEvm", "contract address of gateway on EVM")
   .addFlag("callOnRevert", "Whether to call on revert")
   .addOptionalParam(
     "revertAddress",

packages/tasks/src/zetachainCall.ts

@@ -14,7 +14,8 @@ export const zetachainCall = async (
 
   try {
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const response = await client.zetachainCall({
       callOptions,
       function: args.function,
@@ -44,8 +45,7 @@ export const zetachainCall = async (
 task("zetachain-call", "Call a contract on a connected chain", zetachainCall)
   .addOptionalParam(
     "gatewayZetaChain",
-    "contract address of gateway on ZetaChain",
-    "0xA51c1fc2f0D1a1b8494Ed1FE312d7C3a78Ed91C0"
+    "contract address of gateway on ZetaChain"
   )
   .addParam("zrc20", "The address of ZRC-20 to pay fees")
   .addFlag("callOnRevert", "Whether to call on revert")

packages/tasks/src/zetachainWithdraw.ts

@@ -9,7 +9,8 @@ export const zetachainWithdraw = async (
 ) => {
   try {
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const response = await client.zetachainWithdraw({
       amount: args.amount,
       gatewayZetaChain: args.gatewayZetaChain,
@@ -37,8 +38,7 @@ export const zetachainWithdraw = async (
 task("zetachain-withdraw", "Withdraw tokens from ZetaChain", zetachainWithdraw)
   .addOptionalParam(
     "gatewayZetaChain",
-    "contract address of gateway on ZetaChain",
-    "0xA51c1fc2f0D1a1b8494Ed1FE312d7C3a78Ed91C0"
+    "contract address of gateway on ZetaChain"
   )
   .addOptionalParam("zrc20", "The address of the ZRC20 token")
   .addFlag("callOnRevert", "Whether to call on revert")

packages/tasks/src/zetachainWithdrawAndCall.ts

@@ -14,7 +14,8 @@ export const zetachainWithdrawAndCall = async (
     };
 
     const [signer] = await hre.ethers.getSigners();
-    const client = new ZetaChainClient({ network: "testnet", signer });
+    const network = hre.network.name;
+    const client = new ZetaChainClient({ network, signer });
     const response = await client.zetachainWithdrawAndCall({
       amount: args.amount,
       callOptions,
@@ -50,8 +51,7 @@ task(
 )
   .addOptionalParam(
     "gatewayZetaChain",
-    "contract address of gateway on ZetaChain",
-    "0xA51c1fc2f0D1a1b8494Ed1FE312d7C3a78Ed91C0"
+    "contract address of gateway on ZetaChain"
   )
   .addOptionalParam("zrc20", "The address of the ZRC20 token")
   .addFlag("callOnRevert", "Whether to call on revert")