dsa-polygon-migration/contracts/receivers/aave-v2-receiver/helpers.sol

pragma solidity >=0.7.0;

import { DSMath } from "../../common/math.sol";
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import { Stores } from "../../common/stores-polygon.sol";
import { Variables } from "./variables.sol";

import {
    TokenMappingInterface,
    AaveData,
    AaveDataProviderInterface,
    AaveInterface,
    AccountInterface
} from "./interfaces.sol";

abstract contract Helpers is Stores, DSMath, Variables {
    using SafeERC20 for IERC20;

    function remapTokens(AaveData memory data) internal view returns (AaveData memory) {
        for (uint i = 0; i < data.supplyTokens.length; i++) {
            data.supplyTokens[i] = tokenMapping.getMapping(data.supplyTokens[i]);
        }

        for (uint i = 0; i < data.borrowTokens.length; i++) {
            data.borrowTokens[i] = tokenMapping.getMapping(data.borrowTokens[i]);
        }

        return data;
    }

    function isPositionSafe() internal returns (bool isOk) {
        AaveInterface aave = AaveInterface(aaveProvider.getLendingPool());
        (,,,,,uint healthFactor) = aave.getUserAccountData(address(this));
        uint minLimit = wdiv(1e18, safeRatioGap);
        isOk = healthFactor > minLimit;
        require(isOk, "position-at-risk");
    }

    function transferAtokens(AaveInterface aave, address dsa, address[] memory supplyTokens, uint[] memory supplyAmts) internal {
        for (uint i = 0; i < supplyTokens.length; i++) {
            address _token = supplyTokens[i] == maticAddr ? wmaticAddr : supplyTokens[i];
            (address _aToken, ,) = aaveData.getReserveTokensAddresses(_token);
            IERC20 _atokenContract = IERC20(_aToken);
            uint _atokenBal = _atokenContract.balanceOf(address(this));
            uint _supplyAmt = supplyAmts[i];
            bool isFlash;
            uint _flashAmt;

            (uint tokenLiq,,,,,,,,,) = aaveData.getReserveData(_token);

            if (_atokenBal < _supplyAmt) {
                uint _reqAmt = _supplyAmt - _atokenBal;
                if (tokenLiq < _reqAmt) {
                    _flashAmt = flashAmts[_token];
                    if (_flashAmt > 0) {
                        aave.deposit(_token, _flashAmt, address(this), 3288); // TODO: what is our ID on Polygon?
                        tokenLiq += _flashAmt;
                        isFlash = true;
                    }
                }

                uint num = _reqAmt/tokenLiq + 1; // TODO: Is this right
                uint splitAmt = _reqAmt/num; // TODO: Check decimal
                uint finalSplit = _reqAmt - (splitAmt * (num - 1)); // TODO: to resolve upper decimal error

                for (uint j = 0; j < num; j++) {
                    if (j < num - 1) {
                        aave.borrow(_token, splitAmt, 2, 3288, address(this));
                        aave.deposit(_token, splitAmt, address(this), 3288);
                    } else {
                        aave.borrow(_token, finalSplit, 2, 3288, address(this));
                        aave.deposit(_token, finalSplit, address(this), 3288);
                    }
                }
            }

            if (isFlash) {
                aave.withdraw(_token, _flashAmt, address(this));
            }

            _atokenContract.safeTransfer(dsa, _supplyAmt);
        }
    }

    function borrowAndTransferSpells(AaveInterface aave, address dsa, address[] memory borrowTokens, uint[] memory borrowAmts) internal {
        for (uint i = 0; i < borrowTokens.length; i++) {
            address _token = borrowTokens[i] == maticAddr ? wmaticAddr : borrowTokens[i];
            (address _atoken, ,) = aaveData.getReserveTokensAddresses(_token);
            (uint tokenLiq,,,,,,,,,) = aaveData.getReserveData(_token);
            uint _borrowAmt = borrowAmts[i];

            uint _flashAmt;
            bool isFlash;
            if (tokenLiq < _borrowAmt) {
                _flashAmt = flashAmts[_token];
                aave.deposit(_token, _flashAmt, address(this), 3288); // TODO: what is our ID on Polygon?
                isFlash = true;
                tokenLiq += _flashAmt;
            }
            // TODO: Check number of loops needed. Borrow and supply on user's account.
            uint num = _borrowAmt/tokenLiq + 1; // TODO: Is this right
            uint splitAmt = _borrowAmt/num; // TODO: Check decimal
            uint finalSplit = _borrowAmt - (splitAmt * (num - 1)); // TODO: to resolve upper decimal error

            uint spellsAmt = (2 * num) + 1;
            string[] memory targets = new string[](spellsAmt);
            bytes[] memory castData = new bytes[](spellsAmt);
            for (uint j = 0; j < num; j++) {
                uint k = j * 2;
                if (i < num - 1) {
                    targets[k] = "AAVE-V2-A";
                    castData[k] = abi.encodeWithSignature("borrow(address,uint256,uint256,uint256,uint256)", _token, splitAmt, 2, 0, 0);
                    targets[k+1] = "AAVE-V2-A";
                    castData[k+1] = abi.encodeWithSignature("deposit(address,uint256,uint256,uint256)", _token, splitAmt, 0, 0);
                } else {
                    targets[k] = "AAVE-V2-A";
                    castData[k] = abi.encodeWithSignature("borrow(address,uint256,uint256,uint256,uint256)", _token, finalSplit, 2, 0, 0);
                    targets[k+1] = "AAVE-V2-A";
                    castData[k+1] = abi.encodeWithSignature("deposit(address,uint256,uint256,uint256)", _token, finalSplit, 0, 0);
                }
            }

            if (isFlash) {
                aave.withdraw(_token, _flashAmt, address(this));
            }

            targets[spellsAmt] = "BASIC-A"; // TODO: right spell?
            castData[spellsAmt] = abi.encode("withdraw(address,uint256,address,uint256,uint256)", _atoken, _borrowAmt, address(this), 0, 0); // encode the data of atoken withdrawal
            AccountInterface(dsa).castMigrate(targets, castData, address(this));
        }

    }

}