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

Minor changes

Browse Source
This commit is contained in:
Mubaris NK 2021-04-02 10:50:25 +05:30
parent d9fdd17e3d
commit 33c5c8e0fd
No known key found for this signature in database
GPG Key ID: 9AC09AD0F8D68561
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
contracts/senders/aave-v2/main.sol
View File

@ -56,6 +56,7 @@ abstract contract AaveResolver is Helpers, Events {
contract AaveImportResolver is AaveResolver { contract AaveImportResolver is AaveResolver {
struct AaveData { struct AaveData {
bool isFinal; bool isFinal;
address targetDsa;
uint[] supplyAmts; uint[] supplyAmts;
uint[] variableBorrowAmts; uint[] variableBorrowAmts;
uint[] stableBorrowAmts; uint[] stableBorrowAmts;
@ -63,20 +64,27 @@ contract AaveImportResolver is AaveResolver {
address[] borrowTokens; address[] borrowTokens;
} }
mapping (address => AaveData) public records; mapping (address => AaveData) public positions;
function migrate( function migrate(
address targetDsa,
address[] calldata supplyTokens, address[] calldata supplyTokens,
address[] calldata borrowTokens address[] calldata borrowTokens
) external payable returns (string memory _eventName, bytes memory _eventParam) { ) external payable returns (string memory _eventName, bytes memory _eventParam) {
require(AccountInterface(address(this)).isAuth(msg.sender), "user-account-not-auth");
require(supplyTokens.length > 0, "0-length-not-allowed"); require(supplyTokens.length > 0, "0-length-not-allowed");
require(targetDsa != address(0), "invalid-address");
AaveData memory data; AaveData memory data;
AaveInterface aave = AaveInterface(aaveProvider.getLendingPool()); AaveInterface aave = AaveInterface(aaveProvider.getLendingPool());
(,,,,,uint healthFactor) = aave.getUserAccountData(address(this));
require(healthFactor > 1e18, "position-not-safe");
data.supplyAmts = new uint[](supplyTokens.length); data.supplyAmts = new uint[](supplyTokens.length);
data.supplyTokens = new address[](supplyTokens.length); data.supplyTokens = new address[](supplyTokens.length);
data.targetDsa = targetDsa;
for (uint i = 0; i < supplyTokens.length; i++) { for (uint i = 0; i < supplyTokens.length; i++) {
address _token = supplyTokens[i] == ethAddr ? wethAddr : supplyTokens[i]; address _token = supplyTokens[i] == ethAddr ? wethAddr : supplyTokens[i];
@ -113,7 +121,7 @@ contract AaveImportResolver is AaveResolver {
_Withdraw(supplyTokens.length, aave, data.supplyTokens, data.supplyAmts); _Withdraw(supplyTokens.length, aave, data.supplyTokens, data.supplyAmts);
records[msg.sender] = data; positions[msg.sender] = data;
bytes memory positionData = abi.encode(msg.sender, data); bytes memory positionData = abi.encode(msg.sender, data);
stateSender.syncState(polygonReceiver, positionData); stateSender.syncState(polygonReceiver, positionData);