Merge pull request #2 from distributed-lab/feature/optimizations

Feature/optimizations

Author: Hrom131

contracts/SPVContract.sol

@@ -0,0 +1,60 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.28;
+
+import {BlockHeaderData} from "../libs/BlockHeader.sol";
+
+interface ISPVContract {
+    error InvalidInitialBlockHeight(uint256 blockHeight);
+    error PrevBlockDoesNotExist(bytes32 prevBlockHash);
+    error BlockAlreadyExists(bytes32 blockHash);
+
+    error EmptyBlockHeaderArray();
+    error InvalidBlockHeadersOrder();
+
+    error InvalidTarget(bytes32 blockTarget, bytes32 networkTarget);
+    error InvalidBlockHash(bytes32 actualBlockHash, bytes32 blockTarget);
+    error InvalidBlockTime(uint32 blockTime, uint32 medianTime);
+
+    event MainchainHeadUpdated(
+        uint256 indexed newMainchainHeight,
+        bytes32 indexed newMainchainHead
+    );
+    event BlockHeaderAdded(uint256 indexed blockHeight, bytes32 indexed blockHash);
+
+    struct BlockData {
+        BlockHeaderData header;
+        uint256 blockHeight;
+    }
+
+    struct BlockInfo {
+        BlockData mainBlockData;
+        bool isInMainchain;
+        uint256 cumulativeWork;
+    }
+
+    function addBlockHeaderBatch(bytes[] calldata blockHeaderRawArray_) external;
+
+    function addBlockHeader(bytes calldata blockHeaderRaw_) external;
+
+    function getLastEpochCumulativeWork() external view returns (uint256);
+
+    function getBlockMerkleRoot(bytes32 blockHash_) external view returns (bytes32);
+
+    function getBlockInfo(bytes32 blockHash_) external view returns (BlockInfo memory blockInfo_);
+
+    function getBlockData(bytes32 blockHash_) external view returns (BlockData memory);
+
+    function getMainchainHead() external view returns (bytes32);
+
+    function getBlockHeight(bytes32 blockHash_) external view returns (uint256);
+
+    function getBlockHash(uint256 blockHeight_) external view returns (bytes32);
+
+    function getBlockTarget(bytes32 blockHash_) external view returns (bytes32);
+
+    function blockExists(bytes32 blockHash_) external view returns (bool);
+
+    function getMainchainBlockHeight() external view returns (uint256);
+
+    function isInMainchain(bytes32 blockHash_) external view returns (bool);
+}

contracts/interfaces/ISPVContract.sol

@@ -22,7 +22,7 @@ library BlockHeader {
     function getBlockHeaderHash(bytes calldata blockHeaderRaw_) internal pure returns (bytes32) {
         bytes32 rawBlockHash_ = sha256(abi.encode(sha256(blockHeaderRaw_)));
 
-        return reverseHash(rawBlockHash_);
+        return _reverseHash(rawBlockHash_);
     }
 
     function parseBlockHeaderData(
@@ -35,8 +35,8 @@ library BlockHeader {
 
         headerData_ = BlockHeaderData({
             version: uint32(_reverseBytes(blockHeaderRaw_[0:4])),
-            prevBlockHash: reverseHash(bytes32(blockHeaderRaw_[4:36])),
-            merkleRoot: reverseHash(bytes32(blockHeaderRaw_[36:68])),
+            prevBlockHash: _reverseHash(bytes32(blockHeaderRaw_[4:36])),
+            merkleRoot: _reverseHash(bytes32(blockHeaderRaw_[36:68])),
             time: uint32(_reverseBytes(blockHeaderRaw_[68:72])),
             bits: bytes4(uint32(_reverseBytes(blockHeaderRaw_[72:76]))),
             nonce: uint32(_reverseBytes(blockHeaderRaw_[76:80]))
@@ -48,15 +48,15 @@ library BlockHeader {
         return
             abi.encodePacked(
                 _reverseUint32(headerData_.version),
-                reverseHash(headerData_.prevBlockHash),
-                reverseHash(headerData_.merkleRoot),
+                _reverseHash(headerData_.prevBlockHash),
+                _reverseHash(headerData_.merkleRoot),
                 _reverseUint32(headerData_.time),
                 _reverseUint32(uint32(headerData_.bits)),
                 _reverseUint32(headerData_.nonce)
             );
     }
 
-    function reverseHash(bytes32 blockHash_) internal pure returns (bytes32) {
+    function _reverseHash(bytes32 blockHash_) private pure returns (bytes32) {
         return bytes32(uint256(blockHash_).reverseBytes());
     }