Feat/leveraged steth

src/vault/adapter/lido/LeveragedWstETHAdapter.sol

@@ -8,7 +8,7 @@ import {IwstETH} from "./IwstETH.sol";
 import {Math} from "openzeppelin-contracts/utils/math/Math.sol";
 import {IWETH} from "../../../interfaces/external/IWETH.sol";
 import {ICurveMetapool} from "../../../interfaces/external/curve/ICurveMetapool.sol";
-import {ILendingPool, IAaveIncentives, IAToken, IFlashLoanReceiver, IProtocolDataProvider, DataTypes, IPoolAddressesProvider} from "../aave/aaveV3/IAaveV3.sol";
+import {ILendingPool, IAToken, IFlashLoanReceiver, IProtocolDataProvider, IPoolAddressesProvider} from "../aave/aaveV3/IAaveV3.sol";
 
 /// @title Leveraged wstETH yield adapter
 /// @author Andrea Di Nenno
@@ -27,7 +27,9 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
     ILendingPool public lendingPool;
     IPoolAddressesProvider public poolAddressesProvider;
 
-    IWETH public weth;
+    IWETH public constant weth = IWETH(0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2);
+    address public constant stETH = address(0xae7ab96520DE3A18E5e111B5EaAb095312D7fE84);
+
     IERC20 public debtToken; // aave eth debt token
     IERC20 public interestToken; // aave awstETH
 
@@ -52,8 +54,6 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
      * @param adapterInitData Encoded data for the base adapter initialization.
      * @param _initData Encoded data for the adapter initialization.
      * @dev `_slippage` - allowed slippage in 1e18
-     * @dev `_weth` - Weth address.
-     * @dev `_stEth` - stETH address.
      * @dev `_poolAddressesProvider` - aave Pool Addresses Provider contract address.
      * @dev `_targetLTV` - The desired loan to value of the vault CDP.
      * @dev `_maxLTV` - The max loan to value allowed before a automatic de-leverage
@@ -67,15 +67,13 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         __AdapterBase_init(adapterInitData);
 
         (
-            address _wETH,
-            address _stETH,
             address _poolAddressesProvider,
             uint256 _slippage,
             uint256 _targetLTV,
             uint256 _maxLTV
         ) = abi.decode(
                 _initData,
-                (address, address, address, uint256, uint256, uint256)
+                (address, uint256, uint256, uint256)
             );
 
         address baseAsset = asset();
@@ -84,7 +82,6 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         maxLTV = _maxLTV;
 
         slippage = _slippage;
-        weth = IWETH(_wETH);
 
         // retrieve and set wstETH aToken, lending pool
         (address _aToken, , ) = IProtocolDataProvider(aaveDataProvider)
@@ -97,7 +94,8 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         // retrieve and set WETH variable debt token
         (, , address _variableDebtToken) = IProtocolDataProvider(
             aaveDataProvider
-        ).getReserveTokensAddresses(_wETH);
+        ).getReserveTokensAddresses(address(weth));
+
         debtToken = IERC20(_variableDebtToken); // variable debt WETH token
 
         _name = string.concat(
@@ -114,7 +112,7 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         IERC20(address(weth)).approve(address(lendingPool), type(uint256).max);
 
         // approve curve router to pull stETH for swapping
-        IERC20(address(_stETH)).approve(
+        IERC20(stETH).approve(
             address(StableSwapSTETH),
             type(uint256).max
         );
@@ -172,6 +170,8 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
     /*//////////////////////////////////////////////////////////////
                           FLASH LOAN LOGIC
     //////////////////////////////////////////////////////////////*/
+
+    error NotFlashLoan();
 
     function ADDRESSES_PROVIDER()
         external
@@ -193,6 +193,9 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         address initiator,
         bytes calldata params
     ) external override returns (bool) {
+        if(initiator != address(this) && msg.sender != address(lendingPool))
+            revert NotFlashLoan();
+
         (bool isWithdraw, bool isFullWithdraw, uint256 assetsToWithdraw) = abi
             .decode(params, (bool, bool, uint256));
 
@@ -269,16 +272,16 @@ contract LeveragedWstETHAdapter is AdapterBase, IFlashLoanReceiver {
         address wstETH = asset();
 
         // account for eventual eth dust
-        borrowAmount += address(this).balance;
+        uint256 ethDust = address(this).balance;
 
         // unwrap into ETH
         weth.withdraw(borrowAmount);
-
+       
         // get amount of wstETH the vault receives
-        uint256 wstETHAmount = IwstETH(wstETH).getWstETHByStETH(borrowAmount);
+        uint256 wstETHAmount = IwstETH(wstETH).getWstETHByStETH(borrowAmount + ethDust);
 
         // stake borrowed eth and receive wstETH
-        (bool sent, ) = wstETH.call{value: borrowAmount}("");
+        (bool sent, ) = wstETH.call{value: borrowAmount + ethDust}("");
         require(sent, "Fail to send eth to wstETH");
 
         // deposit wstETH into lending protocol

src/vault/adapter/morpho/IMorpho.sol

@@ -0,0 +1,68 @@
+// SPDX-License-Identifier: GPL-3.0
+
+pragma solidity ^0.8.15;
+
+struct MarketParams {
+    address loanToken;
+    address collateralToken;
+    address oracle;
+    address irm;
+    uint256 lltv;
+}
+
+interface IMorpho {
+    function supply(
+        MarketParams memory marketParams,
+        uint256 assets,
+        uint256 shares,
+        address onBehalf,
+        bytes memory data
+    ) external returns (uint256 assetsSupplied, uint256 sharesSupplied);
+
+    function withdraw(
+        MarketParams memory marketParams,
+        uint256 assets,
+        uint256 shares,
+        address onBehalf,
+        address receiver
+    ) external returns (uint256 assetsWithdrawn, uint256 sharesWithdrawn);
+
+    function borrow(
+        MarketParams memory marketParams,
+        uint256 assets,
+        uint256 shares,
+        address onBehalf,
+        address receiver
+    ) external returns (uint256 assetsBorrowed, uint256 sharesBorrowed);
+
+    function repay(
+        MarketParams memory marketParams,
+        uint256 assets,
+        uint256 shares,
+        address onBehalf,
+        bytes memory data
+    ) external returns (uint256 assetsRepaid, uint256 sharesRepaid);
+
+    function flashLoan(address token, uint256 assets, bytes calldata data) external;
+
+    function idToMarketParams(bytes32 id)
+        external
+        view
+        returns (
+            address loanToken,
+            address collateralToken, 
+            address oracle, 
+            address irm, 
+            uint256 lltv
+        );
+}
+
+/// @title IMorphoFlashLoanCallback
+/// @notice Interface that users willing to use `flashLoan`'s callback must implement.
+interface IMorphoFlashLoanCallback {
+    /// @notice Callback called when a flash loan occurs.
+    /// @dev The callback is called only if data is not empty.
+    /// @param assets The amount of assets that was flash loaned.
+    /// @param data Arbitrary data passed to the `flashLoan` function.
+    function onMorphoFlashLoan(uint256 assets, bytes calldata data) external;
+}