Merge pull request DMDcoin#244 from MSalman6/dev

Staking: added method to check if pool is valid (is active or inactive)
SurfingNerd · Sep 6, 2024 · 5c03fd1 · 5c03fd1
2 parents 8320ba1 + fe551a8
commit 5c03fd1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/contracts/StakingHbbft.sol b/contracts/StakingHbbft.sol
@@ -976,6 +976,12 @@ contract StakingHbbft is Initializable, OwnableUpgradeable, ReentrancyGuardUpgra
         return _pools.contains(_stakingAddress);
     }
 
+    /// @dev Returns a flag indicating whether a specified address is in the `_pools` or `poolsInactive` array.
+    /// @param _stakingAddress The staking address of the pool.
+    function isPoolValid(address _stakingAddress) public view returns (bool) {
+        return _pools.contains(_stakingAddress) || _poolsInactive.contains(_stakingAddress);
+    }
+
     /// @dev Returns the maximum amount which can be withdrawn from the specified pool by the specified staker
     /// at the moment. Used by the `withdraw` and `moveStake` functions.
     /// @param _poolStakingAddress The pool staking address from which the withdrawal will be made.

diff --git a/contracts/interfaces/IStakingHbbft.sol b/contracts/interfaces/IStakingHbbft.sol
@@ -48,6 +48,8 @@ interface IStakingHbbft {
 
     function isPoolActive(address) external view returns (bool);
 
+    function isPoolValid(address) external view returns (bool);
+
     function MAX_CANDIDATES() external pure returns (uint256); // solhint-disable-line func-name-mixedcase
 
     function orderedWithdrawAmount(address, address)