From e3ef78e6893e61363b63fc6bad5e283e61aa3734 Mon Sep 17 00:00:00 2001
From: Samyak Jain <34437877+KaymasJain@users.noreply.github.com>
Date: Wed, 14 Apr 2021 03:51:08 +0530
Subject: [PATCH] added isDepositEnabled variable

---
 contracts/senders/aave-v2-migrator/events.sol    | 2 ++
 contracts/senders/aave-v2-migrator/main.sol      | 9 +++++----
 contracts/senders/aave-v2-migrator/variables.sol | 1 +
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/contracts/senders/aave-v2-migrator/events.sol b/contracts/senders/aave-v2-migrator/events.sol
index 707043f..6f0b45f 100644
--- a/contracts/senders/aave-v2-migrator/events.sol
+++ b/contracts/senders/aave-v2-migrator/events.sol
@@ -26,4 +26,6 @@ contract Events {
 
     event LogAddTokensSupport(address[] _tokens);
 
+    event variablesUpdate(uint _safeRatioGap, uint _fee, bool _depositEnable);
+
 }
\ No newline at end of file
diff --git a/contracts/senders/aave-v2-migrator/main.sol b/contracts/senders/aave-v2-migrator/main.sol
index beda5ca..22315a4 100644
--- a/contracts/senders/aave-v2-migrator/main.sol
+++ b/contracts/senders/aave-v2-migrator/main.sol
@@ -11,13 +11,12 @@ import { Events } from "./events.sol";
 contract LiquidityResolver is Helpers, Events {
     using SafeERC20 for IERC20;
 
-    event variablesUpdate(uint _safeRatioGap, uint _fee);
-
-    function updateVariables(uint _safeRatioGap, uint _fee) public {
+    function updateVariables(uint _safeRatioGap, uint _fee, bool _depositEnable) public {
         require(msg.sender == instaIndex.master(), "not-master");
         safeRatioGap = _safeRatioGap;
         fee = _fee;
-        emit variablesUpdate(safeRatioGap, fee);
+        isDepositsEnabled = _depositEnable;
+        emit variablesUpdate(safeRatioGap, fee, isDepositsEnabled);
     }
 
     function addTokenSupport(address[] memory _tokens) public {
@@ -46,6 +45,7 @@ contract LiquidityResolver is Helpers, Events {
     }
 
     function deposit(address[] calldata tokens, uint[] calldata amts) external payable {
+        require(isDepositsEnabled, "deposit-not-enable");
         uint _length = tokens.length;
         require(_length == amts.length, "invalid-length");
 
@@ -80,6 +80,7 @@ contract LiquidityResolver is Helpers, Events {
     }
 
     function withdraw(address[] calldata tokens, uint[] calldata amts) external {
+        require(isDepositsEnabled, "withdraw-not-enable");
         uint _length = tokens.length;
         require(_length == amts.length, "invalid-length");
 
diff --git a/contracts/senders/aave-v2-migrator/variables.sol b/contracts/senders/aave-v2-migrator/variables.sol
index cf0a921..d4179c4 100644
--- a/contracts/senders/aave-v2-migrator/variables.sol
+++ b/contracts/senders/aave-v2-migrator/variables.sol
@@ -64,6 +64,7 @@ contract Variables {
     StateSenderInterface constant internal stateSender = StateSenderInterface(0x28e4F3a7f651294B9564800b2D01f35189A5bFbE);
 
     mapping(address => mapping(address => uint)) public deposits;
+    bool public isDepositsEnabled;
 
     // InstaIndex Address.
     IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723);