mirror of
https://github.com/Instadapp/dsa-connectors.git
synced 2024-07-29 22:37:00 +00:00
added connector on polygon
This commit is contained in:
bhavik-m
parent
018a1d28cc
commit
c793ce0238
14
contracts/polygon/connectors/aave/v2-to-v3-import/events.sol
Normal file
14
contracts/polygon/connectors/aave/v2-to-v3-import/events.sol
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
pragma solidity ^0.7.0;
|
||||
|
||||
contract Events {
|
||||
event LogAaveV2ImportToV3(
|
||||
address indexed user,
|
||||
bool convertStable,
|
||||
address[] supplyTokens,
|
||||
address[] borrowTokens,
|
||||
uint256[] flashLoanFees,
|
||||
uint256[] supplyAmts,
|
||||
uint256[] stableBorrowAmts,
|
||||
uint256[] variableBorrowAmts
|
||||
);
|
||||
}
|
||||
326
contracts/polygon/connectors/aave/v2-to-v3-import/helpers.sol
Normal file
326
contracts/polygon/connectors/aave/v2-to-v3-import/helpers.sol
Normal file
|
|
@ -0,0 +1,326 @@
|
|||
pragma solidity ^0.7.0;
|
||||
|
||||
import { DSMath } from "../../../common/math.sol";
|
||||
import { Basic } from "../../../common/basic.sol";
|
||||
import { TokenInterface, AccountInterface } from "../../../common/interfaces.sol";
|
||||
import "./events.sol";
|
||||
import "./interfaces.sol";
|
||||
|
||||
abstract contract Helper is DSMath, Basic {
|
||||
/**
|
||||
* @dev Aave referal code
|
||||
*/
|
||||
uint16 internal constant referalCode = 3228;
|
||||
|
||||
/**
|
||||
* @dev AaveV2 Lending Pool Provider
|
||||
*/
|
||||
AaveV2LendingPoolProviderInterface internal constant aaveV2Provider =
|
||||
AaveV2LendingPoolProviderInterface(
|
||||
0xB53C1a33016B2DC2fF3653530bfF1848a515c8c5 // v2 address TODO: need to update this
|
||||
);
|
||||
|
||||
/**
|
||||
* @dev AaveV3 Lending Pool Provider
|
||||
*/
|
||||
AaveV3PoolProviderInterface internal constant aaveV3Provider =
|
||||
AaveV3PoolProviderInterface(
|
||||
0xB53C1a33016B2DC2fF3653530bfF1848a515c8c5 // v2 address TODO: need to update this
|
||||
);
|
||||
|
||||
/**
|
||||
* @dev Aave Protocol Data Provider
|
||||
*/
|
||||
AaveV2DataProviderInterface internal constant aaveData =
|
||||
AaveV2DataProviderInterface(0x057835Ad21a177dbdd3090bB1CAE03EaCF78Fc6d); // TODO: need to update this
|
||||
|
||||
function getIsColl(address token, address user)
|
||||
internal
|
||||
view
|
||||
returns (bool isCol)
|
||||
{
|
||||
(, , , , , , , , isCol) = aaveData.getUserReserveData(token, user);
|
||||
}
|
||||
|
||||
struct ImportData {
|
||||
address[] _supplyTokens;
|
||||
address[] _borrowTokens;
|
||||
ATokenV2Interface[] aTokens;
|
||||
uint256[] supplyAmts;
|
||||
uint256[] variableBorrowAmts;
|
||||
uint256[] variableBorrowAmtsWithFee;
|
||||
uint256[] stableBorrowAmts;
|
||||
uint256[] stableBorrowAmtsWithFee;
|
||||
uint256[] totalBorrowAmts;
|
||||
uint256[] totalBorrowAmtsWithFee;
|
||||
bool convertStable;
|
||||
}
|
||||
|
||||
struct ImportInputData {
|
||||
address[] supplyTokens;
|
||||
address[] borrowTokens;
|
||||
bool convertStable;
|
||||
uint256[] flashLoanFees;
|
||||
}
|
||||
}
|
||||
|
||||
contract _AaveHelper is Helper {
|
||||
function getBorrowAmount(address _token, address userAccount)
|
||||
internal
|
||||
view
|
||||
returns (uint256 stableBorrow, uint256 variableBorrow)
|
||||
{
|
||||
(
|
||||
,
|
||||
address stableDebtTokenAddress,
|
||||
address variableDebtTokenAddress
|
||||
) = aaveData.getReserveTokensAddresses(_token);
|
||||
|
||||
stableBorrow = ATokenV2Interface(stableDebtTokenAddress).balanceOf(
|
||||
userAccount
|
||||
);
|
||||
variableBorrow = ATokenV2Interface(variableDebtTokenAddress).balanceOf(
|
||||
userAccount
|
||||
);
|
||||
}
|
||||
|
||||
function getBorrowAmounts(
|
||||
address userAccount,
|
||||
AaveV2Interface aaveV2,
|
||||
ImportInputData memory inputData,
|
||||
ImportData memory data
|
||||
) internal returns (ImportData memory) {
|
||||
if (inputData.borrowTokens.length > 0) {
|
||||
data._borrowTokens = new address[](inputData.borrowTokens.length);
|
||||
data.variableBorrowAmts = new uint256[](
|
||||
inputData.borrowTokens.length
|
||||
);
|
||||
data.stableBorrowAmts = new uint256[](
|
||||
inputData.borrowTokens.length
|
||||
);
|
||||
data.totalBorrowAmts = new uint256[](inputData.borrowTokens.length);
|
||||
for (uint256 i = 0; i < inputData.borrowTokens.length; i++) {
|
||||
for (uint256 j = i; j < inputData.borrowTokens.length; j++) {
|
||||
if (j != i) {
|
||||
require(
|
||||
inputData.borrowTokens[i] !=
|
||||
inputData.borrowTokens[j],
|
||||
"token-repeated"
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
for (uint256 i = 0; i < inputData.borrowTokens.length; i++) {
|
||||
address _token = inputData.borrowTokens[i] == maticAddr
|
||||
? wmaticAddr
|
||||
: inputData.borrowTokens[i];
|
||||
data._borrowTokens[i] = _token;
|
||||
|
||||
(
|
||||
data.stableBorrowAmts[i],
|
||||
data.variableBorrowAmts[i]
|
||||
) = getBorrowAmount(_token, userAccount);
|
||||
|
||||
if (data.variableBorrowAmts[i] != 0) {
|
||||
data.variableBorrowAmtsWithFee[i] = add(
|
||||
data.variableBorrowAmts[i],
|
||||
inputData.flashLoanFees[i]
|
||||
);
|
||||
data.stableBorrowAmtsWithFee[i] = data.stableBorrowAmts[i];
|
||||
} else {
|
||||
data.stableBorrowAmtsWithFee[i] = add(
|
||||
data.stableBorrowAmts[i],
|
||||
inputData.flashLoanFees[i]
|
||||
);
|
||||
}
|
||||
|
||||
data.totalBorrowAmts[i] = add(
|
||||
data.stableBorrowAmts[i],
|
||||
data.variableBorrowAmts[i]
|
||||
);
|
||||
data.totalBorrowAmtsWithFee[i] = add(
|
||||
data.stableBorrowAmtsWithFee[i],
|
||||
data.variableBorrowAmtsWithFee[i]
|
||||
);
|
||||
|
||||
if (data.totalBorrowAmts[i] > 0) {
|
||||
uint256 _amt = data.totalBorrowAmts[i];
|
||||
TokenInterface(_token).approve(address(aaveV2), _amt);
|
||||
}
|
||||
}
|
||||
}
|
||||
return data;
|
||||
}
|
||||
|
||||
function getSupplyAmounts(
|
||||
address userAccount,
|
||||
ImportInputData memory inputData,
|
||||
ImportData memory data
|
||||
) internal view returns (ImportData memory) {
|
||||
data.supplyAmts = new uint256[](inputData.supplyTokens.length);
|
||||
data._supplyTokens = new address[](inputData.supplyTokens.length);
|
||||
data.aTokens = new ATokenV2Interface[](inputData.supplyTokens.length);
|
||||
|
||||
for (uint256 i = 0; i < inputData.supplyTokens.length; i++) {
|
||||
for (uint256 j = i; j < inputData.supplyTokens.length; j++) {
|
||||
if (j != i) {
|
||||
require(
|
||||
inputData.supplyTokens[i] != inputData.supplyTokens[j],
|
||||
"token-repeated"
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
for (uint256 i = 0; i < inputData.supplyTokens.length; i++) {
|
||||
address _token = inputData.supplyTokens[i] == maticAddr
|
||||
? wmaticAddr
|
||||
: inputData.supplyTokens[i];
|
||||
(address _aToken, , ) = aaveData.getReserveTokensAddresses(_token);
|
||||
data._supplyTokens[i] = _token;
|
||||
data.aTokens[i] = ATokenV2Interface(_aToken);
|
||||
data.supplyAmts[i] = data.aTokens[i].balanceOf(userAccount);
|
||||
}
|
||||
|
||||
return data;
|
||||
}
|
||||
|
||||
function _paybackBehalfOne(
|
||||
AaveV2Interface aaveV2,
|
||||
address token,
|
||||
uint256 amt,
|
||||
uint256 rateMode,
|
||||
address user
|
||||
) private {
|
||||
aaveV2.repay(token, amt, rateMode, user);
|
||||
}
|
||||
|
||||
function _PaybackStable(
|
||||
uint256 _length,
|
||||
AaveV2Interface aaveV2,
|
||||
address[] memory tokens,
|
||||
uint256[] memory amts,
|
||||
address user
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
_paybackBehalfOne(aaveV2, tokens[i], amts[i], 1, user);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _PaybackVariable(
|
||||
uint256 _length,
|
||||
AaveV2Interface aaveV2,
|
||||
address[] memory tokens,
|
||||
uint256[] memory amts,
|
||||
address user
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
_paybackBehalfOne(aaveV2, tokens[i], amts[i], 2, user);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _TransferAtokens(
|
||||
uint256 _length,
|
||||
AaveV2Interface aaveV2,
|
||||
ATokenV2Interface[] memory atokenContracts,
|
||||
uint256[] memory amts,
|
||||
address[] memory tokens,
|
||||
address userAccount
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
uint256 _amt = amts[i];
|
||||
require(
|
||||
atokenContracts[i].transferFrom(
|
||||
userAccount,
|
||||
address(this),
|
||||
_amt
|
||||
),
|
||||
"allowance?"
|
||||
);
|
||||
|
||||
if (!getIsColl(tokens[i], address(this))) {
|
||||
aaveV2.setUserUseReserveAsCollateral(tokens[i], true);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _WithdrawTokensFromV2(
|
||||
uint256 _length,
|
||||
AaveV2Interface aaveV2,
|
||||
uint256[] memory amts,
|
||||
address[] memory tokens
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
uint256 _amt = amts[i];
|
||||
address _token = tokens[i];
|
||||
aaveV2.withdraw(_token, _amt, address(this));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _depositTokensInV3(
|
||||
uint256 _length,
|
||||
AaveV3Interface aaveV3,
|
||||
uint256[] memory amts,
|
||||
address[] memory tokens
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
uint256 _amt = amts[i];
|
||||
address _token = tokens[i];
|
||||
TokenInterface tokenContract = TokenInterface(_token);
|
||||
require(
|
||||
tokenContract.balanceOf(address(this)) >= _amt,
|
||||
"Insufficient funds to deposit in v3"
|
||||
);
|
||||
approve(tokenContract, address(aaveV3), _amt);
|
||||
aaveV3.supply(_token, _amt, address(this), referalCode);
|
||||
|
||||
if (!getIsColl(_token, address(this))) {
|
||||
aaveV3.setUserUseReserveAsCollateral(_token, true);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _BorrowVariable(
|
||||
uint256 _length,
|
||||
AaveV3Interface aaveV3,
|
||||
address[] memory tokens,
|
||||
uint256[] memory amts
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
_borrowOne(aaveV3, tokens[i], amts[i], 2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _BorrowStable(
|
||||
uint256 _length,
|
||||
AaveV3Interface aaveV3,
|
||||
address[] memory tokens,
|
||||
uint256[] memory amts
|
||||
) internal {
|
||||
for (uint256 i = 0; i < _length; i++) {
|
||||
if (amts[i] > 0) {
|
||||
_borrowOne(aaveV3, tokens[i], amts[i], 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function _borrowOne(
|
||||
AaveV3Interface aaveV3,
|
||||
address token,
|
||||
uint256 amt,
|
||||
uint256 rateMode
|
||||
) private {
|
||||
aaveV3.borrow(token, amt, rateMode, referalCode, address(this));
|
||||
}
|
||||
}
|
||||
150
contracts/polygon/connectors/aave/v2-to-v3-import/interfaces.sol
Normal file
150
contracts/polygon/connectors/aave/v2-to-v3-import/interfaces.sol
Normal file
|
|
@ -0,0 +1,150 @@
|
|||
pragma solidity ^0.7.0;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
// aave v2
|
||||
interface AaveV2Interface {
|
||||
function withdraw(
|
||||
address _asset,
|
||||
uint256 _amount,
|
||||
address _to
|
||||
) external;
|
||||
|
||||
function borrow(
|
||||
address _asset,
|
||||
uint256 _amount,
|
||||
uint256 _interestRateMode,
|
||||
uint16 _referralCode,
|
||||
address _onBehalfOf
|
||||
) external;
|
||||
|
||||
function repay(
|
||||
address _asset,
|
||||
uint256 _amount,
|
||||
uint256 _rateMode,
|
||||
address _onBehalfOf
|
||||
) external;
|
||||
|
||||
function setUserUseReserveAsCollateral(
|
||||
address _asset,
|
||||
bool _useAsCollateral
|
||||
) external;
|
||||
|
||||
function getUserAccountData(address user)
|
||||
external
|
||||
view
|
||||
returns (
|
||||
uint256 totalCollateralETH,
|
||||
uint256 totalDebtETH,
|
||||
uint256 availableBorrowsETH,
|
||||
uint256 currentLiquidationThreshold,
|
||||
uint256 ltv,
|
||||
uint256 healthFactor
|
||||
);
|
||||
}
|
||||
|
||||
interface AaveV2LendingPoolProviderInterface {
|
||||
function getLendingPool() external view returns (address);
|
||||
}
|
||||
|
||||
// Aave Protocol Data Provider
|
||||
interface AaveV2DataProviderInterface {
|
||||
function getUserReserveData(address _asset, address _user)
|
||||
external
|
||||
view
|
||||
returns (
|
||||
uint256 currentATokenBalance,
|
||||
uint256 currentStableDebt,
|
||||
uint256 currentVariableDebt,
|
||||
uint256 principalStableDebt,
|
||||
uint256 scaledVariableDebt,
|
||||
uint256 stableBorrowRate,
|
||||
uint256 liquidityRate,
|
||||
uint40 stableRateLastUpdated,
|
||||
bool usageAsCollateralEnabled
|
||||
);
|
||||
|
||||
// function getReserveConfigurationData(address asset)
|
||||
// external
|
||||
// view
|
||||
// returns (
|
||||
// uint256 decimals,
|
||||
// uint256 ltv,
|
||||
// uint256 liquidationThreshold,
|
||||
// uint256 liquidationBonus,
|
||||
// uint256 reserveFactor,
|
||||
// bool usageAsCollateralEnabled,
|
||||
// bool borrowingEnabled,
|
||||
// bool stableBorrowRateEnabled,
|
||||
// bool isActive,
|
||||
// bool isFrozen
|
||||
// );
|
||||
|
||||
function getReserveTokensAddresses(address asset)
|
||||
external
|
||||
view
|
||||
returns (
|
||||
address aTokenAddress,
|
||||
address stableDebtTokenAddress,
|
||||
address variableDebtTokenAddress
|
||||
);
|
||||
}
|
||||
|
||||
interface ATokenV2Interface {
|
||||
function scaledBalanceOf(address _user) external view returns (uint256);
|
||||
|
||||
function isTransferAllowed(address _user, uint256 _amount)
|
||||
external
|
||||
view
|
||||
returns (bool);
|
||||
|
||||
function balanceOf(address _user) external view returns (uint256);
|
||||
|
||||
function transferFrom(
|
||||
address,
|
||||
address,
|
||||
uint256
|
||||
) external returns (bool);
|
||||
|
||||
function allowance(address, address) external returns (uint256);
|
||||
}
|
||||
|
||||
// aave v3
|
||||
interface AaveV3Interface {
|
||||
function supply(
|
||||
address asset,
|
||||
uint256 amount,
|
||||
address onBehalfOf,
|
||||
uint16 referralCode
|
||||
) external;
|
||||
|
||||
function withdraw(
|
||||
address asset,
|
||||
uint256 amount,
|
||||
address to
|
||||
) external returns (uint256);
|
||||
|
||||
function borrow(
|
||||
address asset,
|
||||
uint256 amount,
|
||||
uint256 interestRateMode,
|
||||
uint16 referralCode,
|
||||
address onBehalfOf
|
||||
) external;
|
||||
|
||||
function repay(
|
||||
address asset,
|
||||
uint256 amount,
|
||||
uint256 interestRateMode,
|
||||
address onBehalfOf
|
||||
) external returns (uint256);
|
||||
|
||||
function setUserUseReserveAsCollateral(address asset, bool useAsCollateral)
|
||||
external;
|
||||
|
||||
function swapBorrowRateMode(address asset, uint256 interestRateMode)
|
||||
external;
|
||||
}
|
||||
|
||||
interface AaveV3PoolProviderInterface {
|
||||
function getPool() external view returns (address);
|
||||
}
|
||||
130
contracts/polygon/connectors/aave/v2-to-v3-import/main.sol
Normal file
130
contracts/polygon/connectors/aave/v2-to-v3-import/main.sol
Normal file
|
|
@ -0,0 +1,130 @@
|
|||
pragma solidity ^0.7.0;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
import { TokenInterface, AccountInterface } from "../../../common/interfaces.sol";
|
||||
import "./interfaces.sol";
|
||||
import "./helpers.sol";
|
||||
import "./events.sol";
|
||||
|
||||
contract _AaveV2ToV3MigrationResolver is _AaveHelper {
|
||||
function _importAave(address userAccount, ImportInputData memory inputData)
|
||||
internal
|
||||
returns (string memory _eventName, bytes memory _eventParam)
|
||||
{
|
||||
require(
|
||||
AccountInterface(address(this)).isAuth(userAccount),
|
||||
"user-account-not-auth"
|
||||
);
|
||||
|
||||
require(inputData.supplyTokens.length > 0, "0-length-not-allowed");
|
||||
|
||||
ImportData memory data;
|
||||
|
||||
AaveV2Interface aaveV2 = AaveV2Interface(
|
||||
aaveV2Provider.getLendingPool()
|
||||
);
|
||||
AaveV3Interface aaveV3 = AaveV3Interface(aaveV3Provider.getPool());
|
||||
|
||||
data = getBorrowAmounts(userAccount, aaveV2, inputData, data);
|
||||
data = getSupplyAmounts(userAccount, inputData, data);
|
||||
|
||||
// payback borrowed amount;
|
||||
_PaybackStable(
|
||||
data._borrowTokens.length,
|
||||
aaveV2,
|
||||
data._borrowTokens,
|
||||
data.stableBorrowAmts,
|
||||
userAccount
|
||||
);
|
||||
_PaybackVariable(
|
||||
data._borrowTokens.length,
|
||||
aaveV2,
|
||||
data._borrowTokens,
|
||||
data.variableBorrowAmts,
|
||||
userAccount
|
||||
);
|
||||
|
||||
// transfer atokens to this address;
|
||||
_TransferAtokens(
|
||||
data._supplyTokens.length,
|
||||
aaveV2,
|
||||
data.aTokens,
|
||||
data.supplyAmts,
|
||||
data._supplyTokens,
|
||||
userAccount
|
||||
);
|
||||
|
||||
// withdraw v2 supplied tokens
|
||||
_WithdrawTokensFromV2(
|
||||
data._supplyTokens.length,
|
||||
aaveV2,
|
||||
data.supplyAmts,
|
||||
data._supplyTokens
|
||||
);
|
||||
// deposit tokens in v3
|
||||
_depositTokensInV3(
|
||||
data._supplyTokens.length,
|
||||
aaveV3,
|
||||
data.supplyAmts,
|
||||
data._supplyTokens
|
||||
);
|
||||
|
||||
// borrow assets in aave v3 after migrating position
|
||||
if (data.convertStable) {
|
||||
_BorrowVariable(
|
||||
data._borrowTokens.length,
|
||||
aaveV3,
|
||||
data._borrowTokens,
|
||||
data.totalBorrowAmtsWithFee
|
||||
);
|
||||
} else {
|
||||
_BorrowStable(
|
||||
data._borrowTokens.length,
|
||||
aaveV3,
|
||||
data._borrowTokens,
|
||||
data.stableBorrowAmtsWithFee
|
||||
);
|
||||
_BorrowVariable(
|
||||
data._borrowTokens.length,
|
||||
aaveV3,
|
||||
data._borrowTokens,
|
||||
data.variableBorrowAmtsWithFee
|
||||
);
|
||||
}
|
||||
|
||||
_eventName = "LogAaveV2ImportToV3(address,bool,address[],address[],uint256,uint256[],uint256[],uint256[])";
|
||||
_eventParam = abi.encode(
|
||||
userAccount,
|
||||
inputData.convertStable,
|
||||
inputData.supplyTokens,
|
||||
inputData.borrowTokens,
|
||||
inputData.flashLoanFees,
|
||||
data.supplyAmts,
|
||||
data.stableBorrowAmts,
|
||||
data.variableBorrowAmts
|
||||
);
|
||||
}
|
||||
|
||||
function importAaveV2ToV3(
|
||||
address userAccount,
|
||||
ImportInputData memory inputData
|
||||
)
|
||||
external
|
||||
payable
|
||||
returns (string memory _eventName, bytes memory _eventParam)
|
||||
{
|
||||
(_eventName, _eventParam) = _importAave(userAccount, inputData);
|
||||
}
|
||||
|
||||
function migrateAaveV2ToV3(ImportInputData memory inputData)
|
||||
external
|
||||
payable
|
||||
returns (string memory _eventName, bytes memory _eventParam)
|
||||
{
|
||||
(_eventName, _eventParam) = _importAave(msg.sender, inputData);
|
||||
}
|
||||
}
|
||||
|
||||
contract AaveV2ToV3MigrationResolverPolygon is _AaveV2ToV3MigrationResolver {
|
||||
string public constant name = "Aave-v2-to-v3-import";
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user