ERC-20 token transfer example and improvements to NFT

examples/hello/package.json

@@ -59,4 +59,4 @@
     "@solana/web3.js": "^1.95.2",
     "@zetachain/protocol-contracts": "10.0.0-rc11"
   }
-}
+}

examples/hello/yarn.lock

@@ -2434,6 +2434,16 @@
     "@zetachain/networks" "^10.0.0"
     ethers "^6.13.1"
 
+"@zetachain/[email protected]":
+  version "10.0.0-rc11"
+  resolved "https://registry.yarnpkg.com/@zetachain/protocol-contracts/-/protocol-contracts-10.0.0-rc11.tgz#53f55ead492f7b5802b1feae4e51abc75730af33"
+  integrity sha512-qWazjqnIGRngf4OmyeSIv7sHICQRdMQ1CKPIQIqxA8qFR+gHhDHSfvMdRAvgWbsfkimXOIFiHVIATypyWhviJw==
+  dependencies:
+    "@openzeppelin/contracts" "^5.0.2"
+    "@openzeppelin/contracts-upgradeable" "^5.0.2"
+    "@zetachain/networks" "^10.0.0"
+    ethers "^6.13.1"
+
 "@zetachain/[email protected]":
   version "9.0.0"
   resolved "https://registry.yarnpkg.com/@zetachain/protocol-contracts/-/protocol-contracts-9.0.0.tgz#c20ad5da43f6f3676f31556b303d1cb4ea17357e"

examples/swap/yarn.lock

@@ -2172,6 +2172,16 @@
     "@zetachain/networks" "^10.0.0"
     ethers "^6.13.1"
 
+"@zetachain/[email protected]":
+  version "10.0.0-rc11"
+  resolved "https://registry.yarnpkg.com/@zetachain/protocol-contracts/-/protocol-contracts-10.0.0-rc11.tgz#53f55ead492f7b5802b1feae4e51abc75730af33"
+  integrity sha512-qWazjqnIGRngf4OmyeSIv7sHICQRdMQ1CKPIQIqxA8qFR+gHhDHSfvMdRAvgWbsfkimXOIFiHVIATypyWhviJw==
+  dependencies:
+    "@openzeppelin/contracts" "^5.0.2"
+    "@openzeppelin/contracts-upgradeable" "^5.0.2"
+    "@zetachain/networks" "^10.0.0"
+    ethers "^6.13.1"
+
 "@zetachain/[email protected]":
   version "9.0.0"
   resolved "https://registry.yarnpkg.com/@zetachain/protocol-contracts/-/protocol-contracts-9.0.0.tgz#c20ad5da43f6f3676f31556b303d1cb4ea17357e"

examples/token/.eslintignore

@@ -0,0 +1,6 @@
+.yarn
+artifacts
+cache
+coverage
+node_modules
+typechain-types

examples/token/.eslintrc.js

@@ -0,0 +1,47 @@
+const path = require("path");
+
+/**
+ * @type {import("eslint").Linter.Config}
+ */
+module.exports = {
+  env: {
+    browser: false,
+    es2021: true,
+    mocha: true,
+    node: true,
+  },
+  extends: ["plugin:prettier/recommended"],
+  parser: "@typescript-eslint/parser",
+  parserOptions: {
+    ecmaVersion: 12,
+  },
+  plugins: [
+    "@typescript-eslint",
+    "prettier",
+    "simple-import-sort",
+    "sort-keys-fix",
+    "typescript-sort-keys",
+  ],
+  rules: {
+    "@typescript-eslint/sort-type-union-intersection-members": "error",
+    camelcase: "off",
+    "simple-import-sort/exports": "error",
+    "simple-import-sort/imports": "error",
+    "sort-keys-fix/sort-keys-fix": "error",
+    "typescript-sort-keys/interface": "error",
+    "typescript-sort-keys/string-enum": "error",
+  },
+  settings: {
+    "import/parsers": {
+      "@typescript-eslint/parser": [".js", ".jsx", ".ts", ".tsx", ".d.ts"],
+    },
+    "import/resolver": {
+      node: {
+        extensions: [".js", ".jsx", ".ts", ".tsx", ".d.ts"],
+      },
+      typescript: {
+        project: path.join(__dirname, "tsconfig.json"),
+      },
+    },
+  },
+};

examples/token/.gitignore

@@ -0,0 +1,17 @@
+node_modules
+.env
+coverage
+coverage.json
+typechain
+typechain-types
+dependencies
+
+# Hardhat files
+cache
+artifacts
+
+# Foundry files
+out
+cache_forge
+
+access_token

examples/token/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 ZetaChain
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

examples/token/README.md

@@ -0,0 +1,4 @@
+# NFT Example
+
+This example currently only works with localnet `v4.0.0-rc*`, which supports
+authenticated calls and multiple EVM chains.

examples/token/contracts/Connected.sol

@@ -0,0 +1,77 @@
+// SPDX-License-Identifier: MIT
+pragma solidity 0.8.26;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
+import "@zetachain/protocol-contracts/contracts/evm/GatewayEVM.sol";
+import {RevertContext} from "@zetachain/protocol-contracts/contracts/Revert.sol";
+
+contract Connected is ERC20, Ownable {
+    GatewayEVM public immutable gateway;
+    address public counterparty;
+
+    modifier onlyGateway() {
+        require(msg.sender == address(gateway), "Caller is not the gateway");
+        _;
+    }
+
+    function setCounterparty(address contractAddress) external onlyOwner {
+        counterparty = contractAddress;
+    }
+
+    constructor(
+        address payable gatewayAddress,
+        address initialOwner
+    ) ERC20("MyToken", "MTK") Ownable(initialOwner) {
+        gateway = GatewayEVM(gatewayAddress);
+    }
+
+    function mint(address to, uint256 amount) public onlyOwner {
+        _mint(to, amount);
+    }
+
+    function transferCrossChain(
+        address receiver,
+        address destination,
+        uint256 amount
+    ) external payable {
+        _burn(msg.sender, amount);
+        bytes memory encodedData = abi.encode(receiver, amount, destination);
+
+        RevertOptions memory revertOptions = RevertOptions(
+            address(this),
+            true,
+            address(0),
+            encodedData,
+            0
+        );
+
+        if (destination == address(0)) {
+            gateway.call(counterparty, encodedData, revertOptions);
+        } else {
+            gateway.depositAndCall{value: msg.value}(
+                counterparty,
+                encodedData,
+                revertOptions
+            );
+        }
+    }
+
+    function onCall(
+        MessageContext calldata messageContext,
+        bytes calldata message
+    ) external payable onlyGateway returns (bytes4) {
+        if (messageContext.sender != counterparty) revert("Unauthorized");
+        (address receiver, uint256 amount) = abi.decode(
+            message,
+            (address, uint256)
+        );
+        _mint(receiver, amount);
+    }
+
+    function onRevert(RevertContext calldata context) external onlyGateway {}
+
+    receive() external payable {}
+
+    fallback() external payable {}
+}

examples/token/contracts/Universal.sol

@@ -0,0 +1,119 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.26;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
+import {RevertContext, RevertOptions} from "@zetachain/protocol-contracts/contracts/Revert.sol";
+import "@zetachain/protocol-contracts/contracts/zevm/interfaces/UniversalContract.sol";
+import "@zetachain/protocol-contracts/contracts/zevm/interfaces/IGatewayZEVM.sol";
+import "@zetachain/protocol-contracts/contracts/zevm/GatewayZEVM.sol";
+import {SwapHelperLib} from "@zetachain/toolkit/contracts/SwapHelperLib.sol";
+import {SystemContract} from "@zetachain/toolkit/contracts/SystemContract.sol";
+
+contract Universal is ERC20, Ownable, UniversalContract {
+    GatewayZEVM public immutable gateway;
+    SystemContract public immutable systemContract =
+        SystemContract(0xCf7Ed3AccA5a467e9e704C703E8D87F634fB0Fc9);
+    uint256 private _nextTokenId;
+    bool public isUniversal = true;
+    uint256 public gasLimit = 700000;
+
+    error TransferFailed();
+
+    mapping(address => bytes) public counterparty;
+
+    event CounterpartySet(address indexed zrc20, bytes indexed contractAddress);
+
+    modifier onlyGateway() {
+        require(msg.sender == address(gateway), "Caller is not the gateway");
+        _;
+    }
+
+    constructor(
+        address payable gatewayAddress,
+        address initialOwner
+    ) ERC20("MyToken", "MTK") Ownable(initialOwner) {
+        gateway = GatewayZEVM(gatewayAddress);
+    }
+
+    function setCounterparty(
+        address zrc20,
+        bytes memory contractAddress
+    ) external onlyOwner {
+        counterparty[zrc20] = contractAddress;
+        emit CounterpartySet(zrc20, contractAddress);
+    }
+
+    function transferCrossChain(
+        address receiver,
+        address zrc20,
+        uint256 amount
+    ) public {
+        _burn(msg.sender, amount);
+
+        (, uint256 gasFee) = IZRC20(zrc20).withdrawGasFeeWithGasLimit(gasLimit);
+        if (!IZRC20(zrc20).transferFrom(msg.sender, address(this), gasFee))
+            revert TransferFailed();
+        IZRC20(zrc20).approve(address(gateway), gasFee);
+
+        bytes memory encodedData = abi.encode(receiver, amount);
+
+        CallOptions memory callOptions = CallOptions(gasLimit, false);
+
+        RevertOptions memory revertOptions = RevertOptions(
+            address(this),
+            true,
+            address(0),
+            encodedData,
+            gasLimit
+        );
+
+        gateway.call(
+            counterparty[zrc20],
+            zrc20,
+            encodedData,
+            callOptions,
+            revertOptions
+        );
+    }
+
+    function mint(address to, uint256 amount) public onlyOwner {
+        _mint(to, amount);
+    }
+
+    function onCall(
+        MessageContext calldata messageContext,
+        address zrc20,
+        uint256 amount,
+        bytes calldata message
+    ) external override onlyGateway {
+        if (keccak256(messageContext.origin) != keccak256(counterparty[zrc20]))
+            revert("Unauthorized");
+        (address receiver, uint256 tokenAmount, address destination) = abi
+            .decode(message, (address, uint256, address));
+        if (destination == address(0)) {
+            _mint(receiver, tokenAmount);
+        } else {
+            (, uint256 gasFee) = IZRC20(destination).withdrawGasFeeWithGasLimit(
+                700000
+            );
+            SwapHelperLib.swapExactTokensForTokens(
+                systemContract,
+                zrc20,
+                amount,
+                destination,
+                0
+            );
+            IZRC20(destination).approve(address(gateway), gasFee);
+            gateway.call(
+                counterparty[destination],
+                destination,
+                abi.encode(receiver, tokenAmount),
+                CallOptions(700000, false),
+                RevertOptions(address(0), false, address(0), "", 0)
+            );
+        }
+    }
+
+    function onRevert(RevertContext calldata context) external onlyGateway {}
+}

examples/token/foundry.toml

@@ -0,0 +1,11 @@
+[profile.default]
+src = 'contracts'
+out = 'out'
+viaIR = true
+libs = ['node_modules', 'lib']
+test = 'test'
+cache_path  = 'cache_forge'
+verbosity = 3
+
+[dependencies]
+forge-std = { version = "1.9.2" }

examples/token/hardhat.config.ts

@@ -0,0 +1,21 @@
+import "./tasks/deploy";
+import "./tasks/deploy";
+import "./tasks/mint";
+import "./tasks/transfer";
+import "./tasks/universalSetCounterparty";
+import "./tasks/connectedSetCounterparty";
+import "@zetachain/localnet/tasks";
+import "@nomicfoundation/hardhat-toolbox";
+import "@zetachain/toolkit/tasks";
+
+import { getHardhatConfigNetworks } from "@zetachain/networks";
+import { HardhatUserConfig } from "hardhat/config";
+
+const config: HardhatUserConfig = {
+  networks: {
+    ...getHardhatConfigNetworks(),
+  },
+  solidity: "0.8.26",
+};
+
+export default config;