From 37e56f0955e0c41152b256b8df16ac3c9511aefc Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Fri, 10 Jun 2022 18:16:45 +0530
Subject: [PATCH] added deleverageAndWithdraw func

---
 .../mainnet/connectors/instaLite/events.sol   | 10 ++++
 .../connectors/instaLite/interface.sol        |  6 ++
 .../mainnet/connectors/instaLite/main.sol     | 55 ++++++++++++++++++-
 3 files changed, 69 insertions(+), 2 deletions(-)

diff --git a/contracts/mainnet/connectors/instaLite/events.sol b/contracts/mainnet/connectors/instaLite/events.sol
index 6b1e2827..4cee914b 100644
--- a/contracts/mainnet/connectors/instaLite/events.sol
+++ b/contracts/mainnet/connectors/instaLite/events.sol
@@ -24,4 +24,14 @@ contract Events {
 		uint256 getId,
 		uint256 setId
 	);
+
+	event LogDeleverageAndWithdraw(
+		address vaultAddr,
+		uint256 deleverageAmt,
+		uint256 withdrawAmount,
+		uint256 stETHAmt,
+		uint256 tokenAmt,
+		uint256[] getIds,
+		uint256[] setIds
+	);
 }
diff --git a/contracts/mainnet/connectors/instaLite/interface.sol b/contracts/mainnet/connectors/instaLite/interface.sol
index ae20fed4..49c14083 100644
--- a/contracts/mainnet/connectors/instaLite/interface.sol
+++ b/contracts/mainnet/connectors/instaLite/interface.sol
@@ -14,5 +14,11 @@ interface IInstaLite {
 	function withdraw(uint256 amount_, address to_) external returns (uint256);
 
 	function deleverage(uint amt_) external;
+	function deleverageAndWithdraw(
+        uint256 deleverageAmt_,
+        uint256 withdrawAmount_,
+        address to_
+    ) external;
+	function token() external returns(address);
 
 }
diff --git a/contracts/mainnet/connectors/instaLite/main.sol b/contracts/mainnet/connectors/instaLite/main.sol
index 89ba6582..4f780abd 100644
--- a/contracts/mainnet/connectors/instaLite/main.sol
+++ b/contracts/mainnet/connectors/instaLite/main.sol
@@ -132,8 +132,59 @@ abstract contract InstaLiteConnector is Events, Basic {
 		_eventParam = abi.encode(vaultAddr, _amt, getId, setId);
 	}
 
+	/**
+	 * @dev Deleverage and Withdraw vault. Pays back weth debt and get stETH/WETH collateral.
+	 * @notice Deleverage Instalite vault.
+	 * @param vaultAddr Address of vaultAddress Contract.
+	 * @param deleverageAmount The amount of the token to deleverage.
+     * @param withdrawAmount The amount of the token to deleverage.
+	 * @param getId ID to retrieve amt.
+	 * @param setId ID to set amt.
+	 */
+	function deleverageAndWithdraw(
+		address vaultAddr,
+		uint256 deleverageAmount,
+        uint256 withdrawAmount,
+		uint256[] memory getIds,
+		uint256[] memory setIds
+	)
+		external
+		payable
+		returns (string memory _eventName, bytes memory _eventParam)
+	{	
+		if (getIds.length > 2) {
+			uint256 _deleverageAmt = getUint(getIds[0], deleverageAmount);
+			uint256 _withdrawAmount = getUint(getIds[1], withdrawAmount);
+		}
+
+		TokenInterface tokenContract = 
+			0xc383a3833A87009fD9597F8184979AF5eDFad019 == vaultAddr ?
+				TokenInterface(wethAddr) :
+				TokenInterface(IInstaLite(vaultAddr).token());
+		
+		uint initialBalStETH = astethToken.balanceOf(address(this));
+		uint initialBalToken = tokenContract.balanceOf(address(this));
+
+		approve(TokenInterface(wethAddr), vaultAddr, _amt);
+
+		IInstaLite(vaultAddr).deleverageAndWithdraw(_deleverageAmt, _withdrawAmount, address(this));
+
+		uint _stETHAmt = astethToken.balanceOf(address(this)) - initialBalStETH;
+		uint _tokenAmt = tokenContract.balanceOf(address(this)) - initialBalToken;
+
+		// TODO: add require conditions
+
+		if (setIds.length > 2) {
+			setUint(setIds[0], _stETHAmt);
+			setUint(setIds[1], _tokenAmt);
+		}
+
+		_eventName = "LogDeleverageAndWithdraw(address,uint256,uint256,uint256,uint256,uint256[],uint256[])";
+		_eventParam = abi.encode(vaultAddr, _deleverageAmt, _withdrawAmount, _stETHAmt, _tokenAmt, getIds, setIds);
+	}
+
 }
 
 contract ConnectV2InstaLite is InstaLiteConnector {
-	string public constant name = "InstaLite-v1";
-}
+	string public constant name = "InstaLite-v1.1";
+}
\ No newline at end of file