feat: restore backward compatibility with single bulk functions to allow an staged upgrade

Author: abarmat

contracts/staking/IStaking.sol

@@ -94,6 +94,14 @@ interface IStaking is IStakingData {
 
     // -- Channel management and allocations --
 
+    function allocate(
+        bytes32 _subgraphDeploymentID,
+        uint256 _tokens,
+        address _allocationID,
+        bytes32 _metadata,
+        bytes calldata _proof
+    ) external;
+
     function allocateFrom(
         address _indexer,
         bytes32 _subgraphDeploymentID,
@@ -105,10 +113,25 @@ interface IStaking is IStakingData {
 
     function closeAllocation(address _allocationID, bytes32 _poi) external;
 
+    function closeAllocationMany(CloseAllocationRequest[] calldata _requests) external;
+
+    function closeAndAllocate(
+        address _oldAllocationID,
+        bytes32 _poi,
+        address _indexer,
+        bytes32 _subgraphDeploymentID,
+        uint256 _tokens,
+        address _allocationID,
+        bytes32 _metadata,
+        bytes calldata _proof
+    ) external;
+
     function collect(uint256 _tokens, address _allocationID) external;
 
     function claim(address _allocationID, bool _restake) external;
 
+    function claimMany(address[] calldata _allocationID, bool _restake) external;
+
     // -- Getters and calculations --
 
     function hasStake(address _indexer) external view returns (bool);

contracts/staking/Staking.sol

@@ -859,6 +859,24 @@ contract Staking is StakingV2Storage, GraphUpgradeable, IStaking, Multicall {
         return _withdrawDelegated(msg.sender, _indexer, _delegateToIndexer);
     }
 
+    /**
+     * @dev Allocate available tokens to a subgraph deployment.
+     * @param _subgraphDeploymentID ID of the SubgraphDeployment where tokens will be allocated
+     * @param _tokens Amount of tokens to allocate
+     * @param _allocationID The allocation identifier
+     * @param _metadata IPFS hash for additional information about the allocation
+     * @param _proof A 65-bytes Ethereum signed message of `keccak256(indexerAddress,allocationID)`
+     */
+    function allocate(
+        bytes32 _subgraphDeploymentID,
+        uint256 _tokens,
+        address _allocationID,
+        bytes32 _metadata,
+        bytes calldata _proof
+    ) external override notPaused {
+        _allocate(msg.sender, _subgraphDeploymentID, _tokens, _allocationID, _metadata, _proof);
+    }
+
     /**
      * @dev Allocate available tokens to a subgraph deployment.
      * @param _indexer Indexer address to allocate funds from.
@@ -891,6 +909,49 @@ contract Staking is StakingV2Storage, GraphUpgradeable, IStaking, Multicall {
         _closeAllocation(_allocationID, _poi);
     }
 
+    /**
+     * @dev Close multiple allocations and free the staked tokens.
+     * To be eligible for rewards a proof of indexing must be presented.
+     * Presenting a bad proof is subject to slashable condition.
+     * To opt out for rewards set _poi to 0x0
+     * @param _requests An array of CloseAllocationRequest
+     */
+    function closeAllocationMany(CloseAllocationRequest[] calldata _requests)
+        external
+        override
+        notPaused
+    {
+        for (uint256 i = 0; i < _requests.length; i++) {
+            _closeAllocation(_requests[i].allocationID, _requests[i].poi);
+        }
+    }
+
+    /**
+     * @dev Close and allocate. This will perform a close and then create a new Allocation
+     * atomically on the same transaction.
+     * @param _closingAllocationID The identifier of the allocation to be closed
+     * @param _poi Proof of indexing submitted for the allocated period
+     * @param _indexer Indexer address to allocate funds from.
+     * @param _subgraphDeploymentID ID of the SubgraphDeployment where tokens will be allocated
+     * @param _tokens Amount of tokens to allocate
+     * @param _allocationID The allocation identifier
+     * @param _metadata IPFS hash for additional information about the allocation
+     * @param _proof A 65-bytes Ethereum signed message of `keccak256(indexerAddress,allocationID)`
+     */
+    function closeAndAllocate(
+        address _closingAllocationID,
+        bytes32 _poi,
+        address _indexer,
+        bytes32 _subgraphDeploymentID,
+        uint256 _tokens,
+        address _allocationID,
+        bytes32 _metadata,
+        bytes calldata _proof
+    ) external override notPaused {
+        _closeAllocation(_closingAllocationID, _poi);
+        _allocate(_indexer, _subgraphDeploymentID, _tokens, _allocationID, _metadata, _proof);
+    }
+
     /**
      * @dev Collect query fees from state channels and assign them to an allocation.
      * Funds received are only accepted from a valid sender.
@@ -976,6 +1037,21 @@ contract Staking is StakingV2Storage, GraphUpgradeable, IStaking, Multicall {
         _claim(_allocationID, _restake);
     }
 
+    /**
+     * @dev Claim tokens from the rebate pool for many allocations.
+     * @param _allocationID Array of allocations from where we are claiming tokens
+     * @param _restake True if restake fees instead of transfer to indexer
+     */
+    function claimMany(address[] calldata _allocationID, bool _restake)
+        external
+        override
+        notPaused
+    {
+        for (uint256 i = 0; i < _allocationID.length; i++) {
+            _claim(_allocationID[i], _restake);
+        }
+    }
+
     /**
      * @dev Stake tokens on the indexer.
      * This function does not check minimum indexer stake requirement to allow