Simplification of maxWithdraw (#33)

bgd-labs · Aug 9, 2024 · 9e3530d · 9e3530d
1 parent 1a8082c
commit 9e3530d
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/src/contracts/StakeToken.sol b/src/contracts/StakeToken.sol
@@ -163,17 +163,7 @@ contract StakeToken is
   }
 
   function maxWithdraw(address owner) public view override returns (uint256) {
-    StakeTokenStorage storage $ = _getStakeTokenStorage();
-    CooldownSnapshot memory cooldownSnapshot = $._stakersCooldowns[owner];
-
-    if (
-      block.timestamp >= cooldownSnapshot.timestamp &&
-      block.timestamp - cooldownSnapshot.timestamp <= $._smConfig.unstakeWindowSeconds
-    ) {
-      return _convertToAssets($._stakersCooldowns[owner].amount, Math.Rounding.Floor);
-    }
-
-    return 0;
+    return _convertToAssets(maxRedeem(owner), Math.Rounding.Floor);
   }
 
   function maxRedeem(address owner) public view override returns (uint256) {