Add depositAndMint

rocket-pool · Nov 14, 2022 · 9c90a38 · 9c90a38
1 parent f50109b
commit 9c90a38
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/contracts/contract/node/RocketNodeDeposit.sol b/contracts/contract/node/RocketNodeDeposit.sol
@@ -20,6 +20,7 @@ import "../../interface/dao/node/settings/RocketDAONodeTrustedSettingsMembersInt
 import "../../types/MinipoolDeposit.sol";
 import "../../interface/node/RocketNodeManagerInterface.sol";
 import "../../interface/RocketVaultInterface.sol";
+import "../../interface/token/RocketTokenRETHInterface.sol";
 
 // Handles node deposits and minipool creation
 
@@ -63,6 +64,26 @@ contract RocketNodeDeposit is RocketBase, RocketNodeDepositInterface {
         _deposit(msg.value, _minimumNodeFee, _validatorPubkey, _validatorSignature, _depositDataRoot, _salt, _expectedMinipoolAddress);
     }
 
+    function depositAndMint(uint256 _mintEthAmount, uint256 _minimumNodeFee, bytes calldata _validatorPubkey, bytes calldata _validatorSignature, bytes32 _depositDataRoot, uint256 _salt, address _expectedMinipoolAddress) override external payable onlyLatestContract("rocketNodeDeposit", address(this)) onlyRegisteredNode(msg.sender) {
+        require(_mintEthAmount <= msg.value, "Requested to mint rETH with more ether than provided");
+        // Deposit the minipool
+        _deposit(msg.value - _mintEthAmount, _minimumNodeFee, _validatorPubkey, _validatorSignature, _depositDataRoot, _salt, _expectedMinipoolAddress);
+        // Mint rETH for the caller
+        _mint(_mintEthAmount);
+    }
+
+    function _mint(uint256 _mintEthAmount) private {
+        // Save current rETH balance (for calculating how much rETH is minted)
+        RocketTokenRETHInterface rocketTokenRETH = RocketTokenRETHInterface(getContractAddress("rocketTokenRETH"));
+        uint256 rETHBefore = rocketTokenRETH.balanceOf(address(this));
+        // Deposit ETH to mint rETH
+        RocketDepositPoolInterface rocketDepositPool = RocketDepositPoolInterface(getContractAddress("rocketDepositPool"));
+        rocketDepositPool.deposit{value: _mintEthAmount}();
+        // Transfer minted rETH back to the caller
+        rocketTokenRETH.transfer(msg.sender, rocketTokenRETH.balanceOf(address(this)) - rETHBefore);
+    }
+
+
     // Accept a node deposit and create a new minipool under the node
     // Only accepts calls from registered nodes
     function _deposit(uint256 _value, uint256 _minimumNodeFee, bytes calldata _validatorPubkey, bytes calldata _validatorSignature, bytes32 _depositDataRoot, uint256 _salt, address _expectedMinipoolAddress) private {

diff --git a/contracts/interface/node/RocketNodeDepositInterface.sol b/contracts/interface/node/RocketNodeDepositInterface.sol
@@ -8,4 +8,5 @@ interface RocketNodeDepositInterface {
     function increaseDepositCreditBalance(address _nodeOperator, uint256 _amount) external;
     function depositWithCredit(uint256 _creditUsed, uint256 _minimumNodeFee, bytes calldata _validatorPubkey, bytes calldata _validatorSignature, bytes32 _depositDataRoot, uint256 _salt, address _expectedMinipoolAddress) external payable;
     function deposit(uint256 _minimumNodeFee, bytes calldata _validatorPubkey, bytes calldata _validatorSignature, bytes32 _depositDataRoot, uint256 _salt, address _expectedMinipoolAddress) external payable;
+    function depositAndMint(uint256 _mintEthAmount, uint256 _minimumNodeFee, bytes calldata _validatorPubkey, bytes calldata _validatorSignature, bytes32 _depositDataRoot, uint256 _salt, address _expectedMinipoolAddress) external payable;
 }