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,89 @@
+// SPDX-License-Identifier: MIT
+pragma solidity 0.8.26;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+import "@openzeppelin/contracts/access/Ownable2Step.sol";
+import "@zetachain/protocol-contracts/contracts/evm/GatewayEVM.sol";
+import {RevertContext} from "@zetachain/protocol-contracts/contracts/Revert.sol";
+import "./shared/Events.sol";
+
+contract Connected is ERC20, Ownable2Step, Events {
+    GatewayEVM public immutable gateway;
+    address public counterparty;
+
+    error InvalidAddress();
+    error Unauthorized();
+
+    modifier onlyGateway() {
+        require(msg.sender == address(gateway), "Caller is not the gateway");
+        _;
+    }
+
+    function setCounterparty(address contractAddress) external onlyOwner {
+        counterparty = contractAddress;
+        emit CounterpartySet(contractAddress);
+    }
+
+    constructor(
+        address payable gatewayAddress,
+        address owner,
+        string memory name,
+        string memory symbol
+    ) ERC20(name, symbol) Ownable(owner) {
+        if (gatewayAddress == address(0) || owner == address(0))
+            revert InvalidAddress();
+        gateway = GatewayEVM(gatewayAddress);
+    }
+
+    function mint(address to, uint256 amount) public onlyOwner {
+        _mint(to, amount);
+    }
+
+    function transferCrossChain(
+        address destination,
+        address receiver,
+        uint256 amount
+    ) external payable {
+        _burn(msg.sender, amount);
+        bytes memory message = abi.encode(destination, receiver, amount);
+
+        RevertOptions memory revertOptions = RevertOptions(
+            address(this),
+            true,
+            address(0),
+            message,
+            0
+        );
+
+        if (destination == address(0)) {
+            gateway.call(counterparty, message, revertOptions);
+        } else {
+            gateway.depositAndCall{value: msg.value}(
+                counterparty,
+                message,
+                revertOptions
+            );
+        }
+        emit TokenTransfer(destination, receiver, amount);
+    }
+
+    function onCall(
+        MessageContext calldata context,
+        bytes calldata message
+    ) external payable onlyGateway returns (bytes4) {
+        if (context.sender != counterparty) revert Unauthorized();
+        (address receiver, uint256 amount) = abi.decode(
+            message,
+            (address, uint256)
+        );
+        _mint(receiver, amount);
+        emit TokenTransferReceived(receiver, amount);
+        return "";
+    }
+
+    function onRevert(RevertContext calldata context) external onlyGateway {}
+
+    receive() external payable {}
+
+    fallback() external payable {}
+}

examples/token/contracts/Universal.sol

@@ -0,0 +1,140 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.26;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+import "@openzeppelin/contracts/access/Ownable2Step.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";
+import "./shared/Events.sol";
+
+contract Universal is ERC20, Ownable2Step, UniversalContract, Events {
+    GatewayZEVM public immutable gateway;
+    SystemContract public immutable systemContract =
+        SystemContract(0xCf7Ed3AccA5a467e9e704C703E8D87F634fB0Fc9);
+    uint256 private _nextTokenId;
+    bool public isUniversal = true;
+    uint256 public gasLimit;
+
+    error TransferFailed();
+    error Unauthorized();
+    error InvalidAddress();
+    error InvalidGasLimit();
+
+    mapping(address => bytes) public counterparty;
+
+    modifier onlyGateway() {
+        require(msg.sender == address(gateway), "Caller is not the gateway");
+        _;
+    }
+
+    constructor(
+        address payable gatewayAddress,
+        address owner,
+        string memory name,
+        string memory symbol,
+        uint256 gas
+    ) ERC20(name, symbol) Ownable(owner) {
+        if (gatewayAddress == address(0) || owner == address(0))
+            revert InvalidAddress();
+        if (gas == 0) revert InvalidGasLimit();
+        gateway = GatewayZEVM(gatewayAddress);
+        gasLimit = gas;
+    }
+
+    function setCounterparty(
+        address zrc20,
+        bytes memory contractAddress
+    ) external onlyOwner {
+        counterparty[zrc20] = contractAddress;
+        emit CounterpartyMappingSet(zrc20, contractAddress);
+    }
+
+    function transferCrossChain(
+        address destination,
+        address receiver,
+        uint256 amount
+    ) public {
+        if (receiver == address(0)) revert InvalidAddress();
+        _burn(msg.sender, amount);
+
+        (, uint256 gasFee) = IZRC20(destination).withdrawGasFeeWithGasLimit(
+            gasLimit
+        );
+        if (
+            !IZRC20(destination).transferFrom(msg.sender, address(this), gasFee)
+        ) revert TransferFailed();
+        IZRC20(destination).approve(address(gateway), gasFee);
+        bytes memory message = abi.encode(receiver, amount);
+
+        CallOptions memory callOptions = CallOptions(gasLimit, false);
+
+        RevertOptions memory revertOptions = RevertOptions(
+            address(this),
+            true,
+            address(0),
+            message,
+            gasLimit
+        );
+
+        gateway.call(
+            counterparty[destination],
+            destination,
+            message,
+            callOptions,
+            revertOptions
+        );
+        emit TokenTransfer(destination, receiver, amount);
+    }
+
+    function mint(address to, uint256 amount) public onlyOwner {
+        _mint(to, amount);
+    }
+
+    function onCall(
+        MessageContext calldata context,
+        address zrc20,
+        uint256 amount,
+        bytes calldata message
+    ) external override onlyGateway {
+        if (keccak256(context.origin) != keccak256(counterparty[zrc20]))
+            revert Unauthorized();
+        (address destination, address receiver, uint256 tokenAmount) = abi
+            .decode(message, (address, address, uint256));
+        if (destination == address(0)) {
+            _mint(receiver, tokenAmount);
+        } else {
+            (, uint256 gasFee) = IZRC20(destination).withdrawGasFeeWithGasLimit(
+                gasLimit
+            );
+            SwapHelperLib.swapExactTokensForTokens(
+                systemContract,
+                zrc20,
+                amount,
+                destination,
+                0
+            );
+            IZRC20(destination).approve(address(gateway), gasFee);
+            gateway.call(
+                counterparty[destination],
+                destination,
+                abi.encode(receiver, tokenAmount),
+                CallOptions(gasLimit, false),
+                RevertOptions(address(0), false, address(0), "", 0)
+            );
+        }
+        emit TokenTransferToDestination(destination, receiver, amount);
+    }
+
+    function onRevert(RevertContext calldata context) external onlyGateway {
+        (address sender, uint256 amount) = abi.decode(
+            context.revertMessage,
+            (address, uint256)
+        );
+        _mint(sender, amount);
+        emit TokenTransferReverted(sender, amount);
+    }
+}

examples/token/contracts/shared/Events.sol

@@ -0,0 +1,24 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.26;
+
+contract Events {
+    event CounterpartyMappingSet(
+        address indexed zrc20,
+        bytes indexed contractAddress
+    );
+    event CounterpartySet(address indexed contractAddress);
+
+    event TokenMinted(address indexed to, uint256 amount);
+    event TokenTransfer(
+        address indexed destination,
+        address indexed receiver,
+        uint256 amount
+    );
+    event TokenTransferReceived(address indexed receiver, uint256 amount);
+    event TokenTransferReverted(address indexed sender, uint256 amount);
+    event TokenTransferToDestination(
+        address indexed destination,
+        address indexed sender,
+        uint256 amount
+    );
+}