Instadapp/dsa-polygon-migration
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-polygon-migration.git synced 2024-07-29 22:27:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enable collateral for multiple borrowAndTransferSpells()

Browse Source
This commit is contained in:
Thrilok Kumar 2021-04-16 04:18:05 +05:30
parent f7b351b0eb
commit f145687cc9
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
contracts/receivers/aave-v2-receiver/helpers.sol
View File

@ -106,8 +106,7 @@ abstract contract Helpers is Stores, DSMath, Variables {
uint splitAmt = data.borrowAmt/num; // TODO: Check decimal
uint finalSplit = data.borrowAmt - (splitAmt * (num - 1)); // TODO: to resolve upper decimal error
// uint spellsAmt = num <= 1 ? (2 * num) + 1 : (2 * num) + 2;
uint spellsAmt = (2 * num) + 2;
uint spellsAmt = num <= 1 ? (2 * (num + 1)) : (2 * (num + 1)) + 1;
string[] memory targets = new string[](spellsAmt);
bytes[] memory castData = new bytes[](spellsAmt);
@ -115,15 +114,19 @@ abstract contract Helpers is Stores, DSMath, Variables {
castData[0] = abi.encodeWithSignature("enableCollateral(address[])", supplyTokens);
for (uint j = 0; j < num; j++) {
uint k = j * 2 + 1;
// uint skip = 0;
uint skip = 0;
uint k = j * 2 + 1 + skip;
if (i < num - 1) {
targets[k] = "AAVE-V2-A";
castData[k] = abi.encodeWithSignature("borrow(address,uint256,uint256,uint256,uint256)", data.token, splitAmt, 2, 0, 0);
targets[k+1] = "AAVE-V2-A";
castData[k+1] = abi.encodeWithSignature("deposit(address,uint256,uint256,uint256)", data.token, splitAmt, 0, 0);
// targets[k+1] = "AAVE-V2-A"; // TODO : enableCollateral
// castData[k+1] = abi.encodeWithSignature("enableCollateral(address[])", data.token, splitAmt, 0, 0);
if (j == 0) {
skip = 1;
targets[k+1] = "AAVE-V2-A";
castData[k+1] = abi.encodeWithSignature("enableCollateral(address[])", data.token, splitAmt, 0, 0);
}
} else {
targets[k] = "AAVE-V2-A";
castData[k] = abi.encodeWithSignature("borrow(address,uint256,uint256,uint256,uint256)", data.token, finalSplit, 2, 0, 0);