modify bridge contracts

buckyos · Jan 10, 2024 · f8ea63b · f8ea63b
1 parent 5410827
commit f8ea63b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/contracts/erc721_nft_bridge.sol b/contracts/erc721_nft_bridge.sol
@@ -5,15 +5,18 @@ import "./public_data_storage.sol";
 import "@openzeppelin/contracts/token/ERC721/IERC721.sol";
 
 contract ERC721NFTBridge is IERCPublicDataContract {
+    address owner;
     IERC721 nftAddress;
 
     mapping (bytes32 => uint256) hashToTokenId;
 
     constructor(IERC721 _nftAddress) {
         nftAddress = _nftAddress;
+        owner = msg.sender;
     }
 
     function addTokenId(bytes32[] calldata dataMixedHash, uint256[] calldata tokenId) public {
+        require(msg.sender == owner);
         for (uint i = 0; i < dataMixedHash.length; i++) {
             hashToTokenId[dataMixedHash[i]] = tokenId[i];
         }

diff --git a/contracts/owned_nft_bridge.sol b/contracts/owned_nft_bridge.sol
@@ -9,8 +9,10 @@ contract OwnedNFTBridge is IERCPublicDataContract {
     constructor() {
     }
 
-    function setData(bytes32 dataMixedHash) public {
-        ownerData[dataMixedHash] = msg.sender;
+    function setData(bytes32[] calldata dataMixedHash) public {
+        for (uint i = 0; i < dataMixedHash.length; i++) {
+            ownerData[dataMixedHash[i]] = msg.sender;
+        }
     }
 
     function getDataOwner(bytes32 dataMixedHash) public view returns (address) {

diff --git a/test/test_public_data.ts b/test/test_public_data.ts
@@ -85,7 +85,7 @@ describe("PublicDataStorage", function () {
         }
 
         for (const data of TestDatas) {
-            await (await nftBridge.setData(data.hash)).wait();
+            await (await nftBridge.setData([data.hash])).wait();
         }
 
         // large balance