add batch multichain

Uniswap · Oct 19, 2024 · 11dd7f9 · 11dd7f9
1 parent 8339afd
commit 11dd7f9
Show file tree

Hide file tree

Showing 3 changed files with 400 additions and 46 deletions.
diff --git a/src/TheCompact.sol b/src/TheCompact.sol
@@ -154,6 +154,22 @@ contract TheCompact is ITheCompact, ERC6909, Extsload {
     using HashLib for ExogenousSplitMultichainClaimWithWitness;
     using HashLib for ExogenousQualifiedSplitMultichainClaim;
     using HashLib for ExogenousQualifiedSplitMultichainClaimWithWitness;
+    using HashLib for BatchMultichainClaim;
+    using HashLib for QualifiedBatchMultichainClaim;
+    using HashLib for BatchMultichainClaimWithWitness;
+    using HashLib for QualifiedBatchMultichainClaimWithWitness;
+    using HashLib for SplitBatchMultichainClaim;
+    using HashLib for SplitBatchMultichainClaimWithWitness;
+    using HashLib for QualifiedSplitBatchMultichainClaim;
+    using HashLib for QualifiedSplitBatchMultichainClaimWithWitness;
+    using HashLib for ExogenousBatchMultichainClaim;
+    using HashLib for ExogenousQualifiedBatchMultichainClaim;
+    using HashLib for ExogenousBatchMultichainClaimWithWitness;
+    using HashLib for ExogenousQualifiedBatchMultichainClaimWithWitness;
+    using HashLib for ExogenousSplitBatchMultichainClaim;
+    using HashLib for ExogenousSplitBatchMultichainClaimWithWitness;
+    using HashLib for ExogenousQualifiedSplitBatchMultichainClaim;
+    using HashLib for ExogenousQualifiedSplitBatchMultichainClaimWithWitness;
     using IdLib for uint96;
     using IdLib for uint256;
     using IdLib for address;
@@ -590,6 +606,22 @@ contract TheCompact is ITheCompact, ERC6909, Extsload {
         return _processExogenousSplitMultichainClaim(claimPayload, _withdraw);
     }
 
+    function claim(BatchMultichainClaim calldata claimPayload) external returns (bool) {
+        return _processBatchMultichainClaim(claimPayload, _release);
+    }
+
+    function claimAndWithdraw(BatchMultichainClaim calldata claimPayload) external returns (bool) {
+        return _processBatchMultichainClaim(claimPayload, _withdraw);
+    }
+
+    function claim(ExogenousBatchMultichainClaim calldata claimPayload) external returns (bool) {
+        return _processExogenousBatchMultichainClaim(claimPayload, _release);
+    }
+
+    function claimAndWithdraw(ExogenousBatchMultichainClaim calldata claimPayload) external returns (bool) {
+        return _processExogenousBatchMultichainClaim(claimPayload, _withdraw);
+    }
+
     function enableForcedWithdrawal(uint256 id) external returns (uint256 withdrawableAt) {
         // overflow check not necessary as reset period is capped
         unchecked {
@@ -2300,6 +2332,10 @@ contract TheCompact is ITheCompact, ERC6909, Extsload {
         return usingSplitMultichainClaim(_processSimpleSplitClaim)(claimPayload.toMessageHash(), claimPayload, 0xc0, operation);
     }
 
+    function _processBatchMultichainClaim(BatchMultichainClaim calldata claimPayload, function(address, address, uint256, uint256) internal returns (bool) operation) internal returns (bool) {
+        return usingBatchMultichainClaim(_processSimpleBatchClaim)(claimPayload.toMessageHash(), claimPayload, 0xc0, operation);
+    }
+
     function _processExogenousMultichainClaim(ExogenousMultichainClaim calldata claimPayload, function(address, address, uint256, uint256) internal returns (bool) operation) internal returns (bool) {
         return usingExogenousMultichainClaim(_processClaimWithSponsorDomain)(claimPayload.toMessageHash(), claimPayload, 0x100, claimPayload.notarizedChainId.toNotarizedDomainSeparator(), operation);
     }
@@ -2342,6 +2378,15 @@ contract TheCompact is ITheCompact, ERC6909, Extsload {
         );
     }
 
+    function _processExogenousBatchMultichainClaim(ExogenousBatchMultichainClaim calldata claimPayload, function(address, address, uint256, uint256) internal returns (bool) operation)
+        internal
+        returns (bool)
+    {
+        return usingExogenousBatchMultichainClaim(_processBatchClaimWithSponsorDomain)(
+            claimPayload.toMessageHash(), claimPayload, 0x100, claimPayload.notarizedChainId.toNotarizedDomainSeparator(), operation
+        );
+    }
+
     function _processSplitClaim(SplitClaim calldata claimPayload, function(address, address, uint256, uint256) internal returns (bool) operation) internal returns (bool) {
         return usingSplitClaim(_processSimpleSplitClaim)(claimPayload.toMessageHash(), claimPayload, 0xa0, operation);
     }