feat: add first implem for whitelisted withdraw.

OriginProtocol · Jul 11, 2024 · 1e3d01f · 1e3d01f
1 parent 144819b
commit 1e3d01f
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 2 deletions.
diff --git a/contracts/contracts/vault/OETHVaultCore.sol b/contracts/contracts/vault/OETHVaultCore.sol
@@ -171,6 +171,7 @@ contract OETHVaultCore is VaultCore {
      */
     function requestWithdrawal(uint256 _amount)
         external
+        onlyWhitelistedWithdrawers
         whenNotCapitalPaused
         nonReentrant
         returns (uint256 requestId, uint256 queued)

diff --git a/contracts/contracts/vault/VaultCore.sol b/contracts/contracts/vault/VaultCore.sol
@@ -53,6 +53,27 @@ contract VaultCore is VaultInitializer {
         _;
     }
 
+    modifier onlyWhitelistedWithdrawers() {
+        if (whitelistEnabled) {
+            require(
+                whitelistedWithdrawers[msg.sender],
+                "Only whitelisted withdrawers"
+            );
+        }
+        _;
+    }
+
+    function setEnableWhitelist(bool _enabled) external onlyGovernor {
+        whitelistEnabled = _enabled;
+    }
+
+    function setWhitelistedWithdrawer(address _user, bool _enabled)
+        external
+        onlyGovernor
+    {
+        whitelistedWithdrawers[_user] = _enabled;
+    }
+
     /**
      * @notice Deposit a supported asset and mint OTokens.
      * @param _asset Address of the asset being deposited

diff --git a/contracts/contracts/vault/VaultInitializer.sol b/contracts/contracts/vault/VaultInitializer.sol
@@ -35,5 +35,7 @@ contract VaultInitializer is VaultStorage {
         rebaseThreshold = 1000e18;
         // Initialize all strategies
         allStrategies = new address[](0);
+        // Enable whitelist by default
+        whitelistEnabled = true;
     }
 }
diff --git a/contracts/contracts/vault/VaultStorage.sol b/contracts/contracts/vault/VaultStorage.sol
@@ -221,10 +221,17 @@ contract VaultStorage is Initializable, Governable {
     /// @notice Mapping of withdrawal request indices to the user withdrawal request data
     mapping(uint256 => WithdrawalRequest) public withdrawalRequests;
 
-    uint256 constant CLAIM_DELAY = 30 minutes;
+    /// @notice Minimum time between withdrawal requests and claims
+    uint256 public constant CLAIM_DELAY = 30 minutes;
+
+    /// @notice Mapping of addresses that are allowed to withdraw, when the whitelist is enabled
+    mapping(address => bool) public whitelistedWithdrawers;
+
+    /// @notice Status of the whitelist
+    bool public whitelistEnabled;
 
     // For future use
-    uint256[46] private __gap;
+    uint256[44] private __gap;
 
     /**
      * @notice set the implementation for the admin, this needs to be in a base class else we cannot set it