Adjusting gamma whitelist logic :)

usePicnic · Jul 12, 2023 · 512d24d · 512d24d
1 parent 287080c
commit 512d24d
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 13 deletions.
diff --git a/contracts/bridges/Gamma/GammaDepositBridge.sol b/contracts/bridges/Gamma/GammaDepositBridge.sol
@@ -9,26 +9,34 @@ import "./interfaces/IHypervisorRouter.sol";
 import "./interfaces/IHypervisor.sol";
 
 contract GammaDepositBridge is IGammaDeposit {
-    address constant hypervisorRouterAddress = 0xe0A61107E250f8B5B24bf272baBFCf638569830C;
-    IHypervisorRouter constant hypervisorRouter = IHypervisorRouter(hypervisorRouterAddress);
+    function _approveTokens(
+        address hypervisorAddress, 
+        address[] calldata tokens, 
+        uint256[] calldata percentages) 
+    internal returns (uint256[] memory) {
+        uint256 numTokens = uint256(tokens.length);
+        uint256[] memory amountsIn = new uint256[](numTokens);
+
+        for (uint256 i = 0; i < tokens.length; i++) { 
+            amountsIn[i] = IERC20(tokens[i]).balanceOf(address(this)) * percentages[i] / 100_000;
+            IERC20(tokens[i]).approve(hypervisorAddress, 0);
+            IERC20(tokens[i]).approve(hypervisorAddress, amountsIn[i]);
+        }   
+
+        return amountsIn;  
+    }
 
     function deposit(
         address hypervisorAddress,
         address[] calldata tokens,
         uint256[] calldata percentages,
         uint256[4] calldata minAmountsin
-    ) external override {
-        uint256 numTokens = uint256(tokens.length);
-        uint256[] memory amountsIn = new uint256[](numTokens);
+    ) external override {       
+        IHypervisorRouter hypervisorRouter = IHypervisorRouter(IHypervisor(hypervisorAddress).whitelistedAddress());
 
-        for (uint256 i = 0; i < numTokens; i++) { 
-            amountsIn[i] = IERC20(tokens[i]).balanceOf(address(this)) * percentages[i] / 100_000;
-            // Approve 0 first as a few ERC20 tokens are requiring this pattern.
-            IERC20(tokens[i]).approve(hypervisorAddress, 0);
-            IERC20(tokens[i]).approve(hypervisorAddress, amountsIn[i]);
-        }     
+        uint256[] memory amountsIn = _approveTokens(hypervisorAddress, tokens, percentages);
 
-        (uint256 depositA, uint256 depositB) = capRatios(tokens, amountsIn, hypervisorAddress);
+        (uint256 depositA, uint256 depositB) = capRatios(tokens, amountsIn, hypervisorAddress, hypervisorRouter);
 
         uint256 amountOut = hypervisorRouter.deposit(
             depositA,
@@ -64,7 +72,8 @@ contract GammaDepositBridge is IGammaDeposit {
     function capRatios( 
         address[] calldata tokens, 
         uint256[] memory amountsIn, 
-        address hypervisorAddress
+        address hypervisorAddress,
+        IHypervisorRouter hypervisorRouter
     ) internal view returns (uint256, uint256) {    
         (uint256 startA, uint256 endA) = hypervisorRouter.getDepositAmount(
             hypervisorAddress,

diff --git a/contracts/bridges/Gamma/interfaces/IHypervisor.sol b/contracts/bridges/Gamma/interfaces/IHypervisor.sol
@@ -13,4 +13,7 @@ interface IHypervisor {
   function balanceOf(
     address account
   ) external view returns (uint256);  
+
+  function whitelistedAddress(
+  ) external view returns (address);  
 }