fix: add setGasLimit

zeta-chain · Dec 18, 2024 · 4952436 · 4952436
1 parent fd3d640
commit 4952436
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/contracts/nft/contracts/evm/UniversalNFT.sol b/contracts/nft/contracts/evm/UniversalNFT.sol
@@ -62,6 +62,11 @@ contract UniversalNFT is
         gateway = GatewayEVM(gatewayAddress);
     }
 
+    function setGasLimit(uint256 gas) external onlyOwner {
+        if (gas <= 0) revert InvalidGasLimit();
+        gasLimitAmount = gas;
+    }
+
     function setUniversal(address contractAddress) external onlyOwner {
         if (contractAddress == address(0)) revert InvalidAddress();
         universal = contractAddress;

diff --git a/contracts/nft/contracts/zetachain/UniversalNFT.sol b/contracts/nft/contracts/zetachain/UniversalNFT.sol
@@ -75,6 +75,11 @@ contract UniversalNFT is
         gasLimitAmount = gas;
     }
 
+    function setGasLimit(uint256 gas) external onlyOwner {
+        if (gas <= 0) revert InvalidGasLimit();
+        gasLimitAmount = gas;
+    }
+
     function setConnected(
         address zrc20,
         address contractAddress

diff --git a/contracts/token/contracts/evm/UniversalToken.sol b/contracts/token/contracts/evm/UniversalToken.sol
@@ -51,6 +51,11 @@ contract UniversalToken is
         gateway = GatewayEVM(gatewayAddress);
     }
 
+    function setGasLimit(uint256 gas) external onlyOwner {
+        if (gas <= 0) revert InvalidGasLimit();
+        gasLimitAmount = gas;
+    }
+
     function setUniversal(address contractAddress) external onlyOwner {
         if (contractAddress == address(0)) revert InvalidAddress();
         universal = contractAddress;

diff --git a/contracts/token/contracts/zetachain/UniversalToken.sol b/contracts/token/contracts/zetachain/UniversalToken.sol
@@ -66,6 +66,11 @@ contract UniversalToken is
         gasLimitAmount = gas;
     }
 
+    function setGasLimit(uint256 gas) external onlyOwner {
+        if (gas <= 0) revert InvalidGasLimit();
+        gasLimitAmount = gas;
+    }
+
     function setConnected(
         address zrc20,
         address contractAddress