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

Connect reciever and implementation

Browse Source
This commit is contained in:
Mubaris NK 2021-04-04 09:17:23 +05:30
parent dad172e4b7
commit 86e8d1d148
No known key found for this signature in database
GPG Key ID: 9AC09AD0F8D68561
1 changed files with 37 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
contracts/receivers/aave-v2-receiver/main.sol
View File

@ -4,6 +4,8 @@ pragma experimental ABIEncoderV2;
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol"; import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol"; import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import { AccountInterface } from "./interfaces.sol";
contract MigrateResolver { contract MigrateResolver {
using SafeERC20 for IERC20; using SafeERC20 for IERC20;
@ -20,15 +22,50 @@ contract MigrateResolver {
uint private lastStateId; uint private lastStateId;
mapping (address => AaveData) public positions; mapping (address => AaveData) public positions;
function _migratePosition(address owner) internal {
AaveData storage data = positions[owner];
for (uint i = 0; i < data.supplyTokens.length; i++) {
IERC20(data.supplyTokens[i]).safeTransfer(data.targetDsa, data.supplyAmts[i]);
}
AccountInterface(data.targetDsa).migrateAave(owner);
data.isFinal = true;
}
function onStateReceive(uint256 stateId, bytes calldata receivedData) external { function onStateReceive(uint256 stateId, bytes calldata receivedData) external {
// require(stateId > lastStateId, "wrong-data"); // require(stateId > lastStateId, "wrong-data");
lastStateId = stateId; lastStateId = stateId;
(address owner, AaveData memory data) = abi.decode(receivedData, (address, AaveData)); (address owner, AaveData memory data) = abi.decode(receivedData, (address, AaveData));
positions[owner] = data; positions[owner] = data;
if (canMigrate(owner)) {
_migratePosition(owner);
}
}
function migrate(address owner) external {
require(msg.sender == owner, "not-authorized");
require(canMigrate(owner), "not-enough-liquidity");
_migratePosition(owner);
} }
function getPosition(address owner) public view returns (AaveData memory data) { function getPosition(address owner) public view returns (AaveData memory data) {
data = positions[owner]; data = positions[owner];
} }
function canMigrate(address owner) public view returns (bool can) {
can = true;
AaveData memory data = getPosition(owner);
for (uint i = 0; i < data.supplyTokens.length; i++) {
IERC20 token = IERC20(data.supplyTokens[i]);
if (token.balanceOf(address(this)) < data.supplyAmts[i]) {
can = false;
}
}
}
} }