add events

zeta-chain · Nov 7, 2024 · f191e37 · f191e37
1 parent bddf609
commit f191e37
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/examples/nft/contracts/Connected.sol b/examples/nft/contracts/Connected.sol
@@ -17,9 +17,29 @@ contract Connected is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
     error InvalidAddress();
     error Unauthorized();
 
+    event SetCounterparty(address indexed newCounterparty);
+    event TokenMinted(address indexed to, uint256 indexed tokenId, string uri);
+    event TokenTransfer(
+        uint256 indexed tokenId,
+        address indexed receiver,
+        address indexed destination,
+        string uri
+    );
+    event TokenTransferReceived(
+        uint256 indexed tokenId,
+        address indexed receiver,
+        string uri
+    );
+    event TokenTransferReverted(
+        uint256 indexed tokenId,
+        address indexed sender,
+        string uri
+    );
+
     function setCounterparty(address contractAddress) external onlyOwner {
         if (contractAddress == address(0)) revert InvalidAddress();
         counterparty = contractAddress;
+        emit SetCounterparty(contractAddress);
     }
 
     modifier onlyGateway() {
@@ -45,6 +65,7 @@ contract Connected is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
 
         _safeMint(to, tokenId);
         _setTokenURI(tokenId, uri);
+        emit TokenMinted(to, tokenId, uri);
     }
 
     function transferCrossChain(
@@ -75,6 +96,8 @@ contract Connected is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
                 revertOptions
             );
         }
+
+        emit TokenTransfer(tokenId, receiver, destination, uri);
     }
 
     function onCall(
@@ -90,6 +113,7 @@ contract Connected is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
 
         _safeMint(receiver, tokenId);
         _setTokenURI(tokenId, uri);
+        emit TokenTransferReceived(tokenId, receiver, uri);
         return "";
     }
 
@@ -101,6 +125,7 @@ contract Connected is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
 
         _safeMint(sender, tokenId);
         _setTokenURI(tokenId, uri);
+        emit TokenTransferReverted(tokenId, sender, uri);
     }
 
     receive() external payable {}

diff --git a/examples/nft/contracts/Universal.sol b/examples/nft/contracts/Universal.sol
@@ -34,6 +34,29 @@ contract Universal is
     mapping(address => bytes) public counterparty;
 
     event CounterpartySet(address indexed zrc20, bytes indexed contractAddress);
+    event TokenTransfer(
+        uint256 indexed tokenId,
+        address indexed receiver,
+        address indexed destination,
+        string uri
+    );
+    event TokenTransferReceived(
+        uint256 indexed tokenId,
+        address indexed receiver,
+        string uri
+    );
+    event TokenTransferReverted(
+        uint256 indexed tokenId,
+        address indexed sender,
+        string uri
+    );
+
+    event TokenTransferToDestination(
+        uint256 indexed tokenId,
+        address indexed sender,
+        address indexed destination,
+        string uri
+    );
 
     modifier onlyGateway() {
         if (msg.sender != address(gateway)) revert Unauthorized();
@@ -96,6 +119,8 @@ contract Universal is
             callOptions,
             revertOptions
         );
+
+        emit TokenTransfer(tokenId, receiver, destination, uri);
     }
 
     function safeMint(address to, string memory uri) public onlyOwner {
@@ -130,6 +155,7 @@ contract Universal is
         if (destination == address(0)) {
             _safeMint(sender, tokenId);
             _setTokenURI(tokenId, uri);
+            emit TokenTransferReceived(tokenId, sender, uri);
         } else {
             (, uint256 gasFee) = IZRC20(destination).withdrawGasFeeWithGasLimit(
                 gasLimit
@@ -151,6 +177,7 @@ contract Universal is
                 CallOptions(gasLimit, false),
                 RevertOptions(address(0), false, address(0), "", 0)
             );
+            emit TokenTransferToDestination(tokenId, sender, destination, uri);
         }
     }
 
@@ -162,6 +189,7 @@ contract Universal is
 
         _safeMint(sender, tokenId);
         _setTokenURI(tokenId, uri);
+        emit TokenTransferReverted(tokenId, sender, uri);
     }
 
     // The following functions are overrides required by Solidity.