mirror of
https://github.com/Instadapp/smart-contract.git
synced 2024-07-29 22:08:07 +00:00
almost done with dydx import
This commit is contained in:
Thrilok Kumar
parent
d9ad9c1ce5
commit
5cb23d4ea4
429
contracts/ProxyLogics/import/InstaDydxImport.sol
Normal file
429
contracts/ProxyLogics/import/InstaDydxImport.sol
Normal file
|
|
@ -0,0 +1,429 @@
|
|||
pragma solidity ^0.5.8;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
interface ERC20Interface {
|
||||
function allowance(address, address) external view returns (uint);
|
||||
function balanceOf(address) external view returns (uint);
|
||||
function approve(address, uint) external;
|
||||
function transfer(address, uint) external returns (bool);
|
||||
function transferFrom(address, address, uint) external returns (bool);
|
||||
function deposit() external payable;
|
||||
function withdraw(uint) external;
|
||||
}
|
||||
|
||||
|
||||
contract SoloMarginContract {
|
||||
|
||||
struct Info {
|
||||
address owner; // The address that owns the account
|
||||
uint256 number; // A nonce that allows a single address to control many accounts
|
||||
}
|
||||
|
||||
enum ActionType {
|
||||
Deposit, // supply tokens
|
||||
Withdraw, // borrow tokens
|
||||
Transfer, // transfer balance between accounts
|
||||
Buy, // buy an amount of some token (externally)
|
||||
Sell, // sell an amount of some token (externally)
|
||||
Trade, // trade tokens against another account
|
||||
Liquidate, // liquidate an undercollateralized or expiring account
|
||||
Vaporize, // use excess tokens to zero-out a completely negative account
|
||||
Call // send arbitrary data to an address
|
||||
}
|
||||
|
||||
enum AssetDenomination {
|
||||
Wei, // the amount is denominated in wei
|
||||
Par // the amount is denominated in par
|
||||
}
|
||||
|
||||
enum AssetReference {
|
||||
Delta, // the amount is given as a delta from the current value
|
||||
Target // the amount is given as an exact number to end up at
|
||||
}
|
||||
|
||||
struct AssetAmount {
|
||||
bool sign; // true if positive
|
||||
AssetDenomination denomination;
|
||||
AssetReference ref;
|
||||
uint256 value;
|
||||
}
|
||||
|
||||
struct ActionArgs {
|
||||
ActionType actionType;
|
||||
uint256 accountId;
|
||||
AssetAmount amount;
|
||||
uint256 primaryMarketId;
|
||||
uint256 secondaryMarketId;
|
||||
address otherAddress;
|
||||
uint256 otherAccountId;
|
||||
bytes data;
|
||||
}
|
||||
|
||||
struct Wei {
|
||||
bool sign; // true if positive
|
||||
uint256 value;
|
||||
}
|
||||
|
||||
function operate(Info[] memory accounts, ActionArgs[] memory actions) public;
|
||||
function getAccountWei(Info memory account, uint256 marketId) public returns (Wei memory);
|
||||
function getNumMarkets() public view returns (uint256);
|
||||
function getMarketTokenAddress(uint256 marketI) public view returns (address);
|
||||
|
||||
}
|
||||
|
||||
interface PoolInterface {
|
||||
function accessToken(address[] calldata ctknAddr, uint[] calldata tknAmt, bool isCompound) external;
|
||||
function paybackToken(address[] calldata ctknAddr, bool isCompound) external payable;
|
||||
}
|
||||
|
||||
|
||||
contract DSMath {
|
||||
|
||||
function add(uint x, uint y) internal pure returns (uint z) {
|
||||
require((z = x + y) >= x, "math-not-safe");
|
||||
}
|
||||
|
||||
function sub(uint x, uint y) internal pure returns (uint z) {
|
||||
z = x - y <= x ? x - y : 0;
|
||||
}
|
||||
|
||||
function mul(uint x, uint y) internal pure returns (uint z) {
|
||||
require(y == 0 || (z = x * y) / y == x, "math-not-safe");
|
||||
}
|
||||
|
||||
uint constant WAD = 10 ** 18;
|
||||
|
||||
function wmul(uint x, uint y) internal pure returns (uint z) {
|
||||
z = add(mul(x, y), WAD / 2) / WAD;
|
||||
}
|
||||
|
||||
function wdiv(uint x, uint y) internal pure returns (uint z) {
|
||||
z = add(mul(x, WAD), y / 2) / y;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract Helpers is DSMath {
|
||||
|
||||
/**
|
||||
* @dev get ethereum address
|
||||
*/
|
||||
function getAddressETH() public pure returns (address eth) {
|
||||
eth = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev get WETH address
|
||||
*/
|
||||
function getAddressWETH() public pure returns (address weth) {
|
||||
weth = 0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev get Dydx Solo Address
|
||||
*/
|
||||
function getSoloAddress() public pure returns (address addr) {
|
||||
addr = 0x1E0447b19BB6EcFdAe1e4AE1694b0C3659614e4e;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev get InstaDApp Liquidity Address
|
||||
*/
|
||||
function getPoolAddress() public pure returns (address payable liqAddr) {
|
||||
liqAddr = 0x1564D040EC290C743F67F5cB11f3C1958B39872A;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev setting allowance to dydx for the "user proxy" if required
|
||||
*/
|
||||
function setApproval(address erc20, uint srcAmt, address to) internal {
|
||||
ERC20Interface erc20Contract = ERC20Interface(erc20);
|
||||
uint tokenAllowance = erc20Contract.allowance(address(this), to);
|
||||
if (srcAmt > tokenAllowance) {
|
||||
erc20Contract.approve(to, uint(-1));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev getting actions arg
|
||||
*/
|
||||
function getActionsArgs(address tknAccount, uint256 marketId, uint256 tokenAmt, bool sign) internal view returns (SoloMarginContract.ActionArgs[] memory) {
|
||||
SoloMarginContract.ActionArgs[] memory actions = new SoloMarginContract.ActionArgs[](1);
|
||||
SoloMarginContract.AssetAmount memory amount = SoloMarginContract.AssetAmount(
|
||||
sign,
|
||||
SoloMarginContract.AssetDenomination.Wei,
|
||||
SoloMarginContract.AssetReference.Delta,
|
||||
tokenAmt
|
||||
);
|
||||
bytes memory empty;
|
||||
// address otherAddr = (marketId == 0 && sign) ? getSoloPayableAddress() : address(this);
|
||||
SoloMarginContract.ActionType action = sign ? SoloMarginContract.ActionType.Deposit : SoloMarginContract.ActionType.Withdraw;
|
||||
actions[0] = SoloMarginContract.ActionArgs(
|
||||
action,
|
||||
0,
|
||||
amount,
|
||||
marketId,
|
||||
0,
|
||||
tknAccount,
|
||||
0,
|
||||
empty
|
||||
);
|
||||
return actions;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev getting acccount arg
|
||||
*/
|
||||
function getAccountArgs(address owner, uint accountId) internal view returns (SoloMarginContract.Info[] memory) {
|
||||
SoloMarginContract.Info[] memory accounts = new SoloMarginContract.Info[](1);
|
||||
accounts[0] = (SoloMarginContract.Info(owner, accountId));
|
||||
return accounts;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev getting dydx balance
|
||||
*/
|
||||
function getDydxBal(address owner, uint256 marketId, uint accountId) internal returns (uint tokenBal, bool tokenSign) {
|
||||
SoloMarginContract solo = SoloMarginContract(getSoloAddress());
|
||||
SoloMarginContract.Wei memory tokenWeiBal = solo.getAccountWei(getAccountArgs(owner, accountId)[0], marketId);
|
||||
tokenBal = tokenWeiBal.value;
|
||||
tokenSign = tokenWeiBal.sign;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract DydxResolver is Helpers {
|
||||
|
||||
event LogDeposit(address erc20Addr, uint tokenAmt, address owner);
|
||||
event LogWithdraw(address erc20Addr, uint tokenAmt, address owner);
|
||||
event LogBorrow(address erc20Addr, uint tokenAmt, address owner);
|
||||
event LogPayback(address erc20Addr, uint tokenAmt, address owner);
|
||||
|
||||
// /**
|
||||
// * @dev Deposit ETH/ERC20
|
||||
// */
|
||||
// function deposit(uint256 marketId, address erc20Addr, uint256 tokenAmt) public payable {
|
||||
// if (erc20Addr == getAddressETH()) {
|
||||
// ERC20Interface(getAddressWETH()).deposit.value(msg.value)();
|
||||
// setApproval(getAddressWETH(), tokenAmt, getSoloAddress());
|
||||
// } else {
|
||||
// require(ERC20Interface(erc20Addr).transferFrom(msg.sender, address(this), tokenAmt), "Allowance or not enough bal");
|
||||
// setApproval(erc20Addr, tokenAmt, getSoloAddress());
|
||||
// }
|
||||
// SoloMarginContract(getSoloAddress()).operate(getAccountArgs(), getActionsArgs(marketId, tokenAmt, true));
|
||||
// emit LogDeposit(erc20Addr, tokenAmt, address(this));
|
||||
// }
|
||||
|
||||
// /**
|
||||
// * @dev Payback ETH/ERC20
|
||||
// */
|
||||
// function payback(uint256 marketId, address erc20Addr, uint256 tokenAmt) public payable {
|
||||
// (uint toPayback, bool tokenSign) = getDydxBal(marketId);
|
||||
// require(!tokenSign, "No debt to payback");
|
||||
// toPayback = toPayback > tokenAmt ? tokenAmt : toPayback;
|
||||
// if (erc20Addr == getAddressETH()) {
|
||||
// ERC20Interface(getAddressWETH()).deposit.value(toPayback)();
|
||||
// setApproval(getAddressWETH(), toPayback, getSoloAddress());
|
||||
// msg.sender.transfer(address(this).balance);
|
||||
// } else {
|
||||
// require(ERC20Interface(erc20Addr).transferFrom(msg.sender, address(this), toPayback), "Allowance or not enough bal");
|
||||
// setApproval(erc20Addr, toPayback, getSoloAddress());
|
||||
// }
|
||||
// SoloMarginContract(getSoloAddress()).operate(getAccountArgs(), getActionsArgs(marketId, toPayback, true));
|
||||
// emit LogPayback(erc20Addr, toPayback, address(this));
|
||||
// }
|
||||
|
||||
// /**
|
||||
// * @dev Withdraw ETH/ERC20
|
||||
// */
|
||||
// function withdraw(uint256 marketId, address erc20Addr, uint256 tokenAmt) public {
|
||||
// (uint toWithdraw, bool tokenSign) = getDydxBal(marketId);
|
||||
// require(tokenSign, "token not deposited");
|
||||
// toWithdraw = toWithdraw > tokenAmt ? tokenAmt : toWithdraw;
|
||||
// SoloMarginContract solo = SoloMarginContract(getSoloAddress());
|
||||
// solo.operate(getAccountArgs(), getActionsArgs(marketId, toWithdraw, false));
|
||||
// if (erc20Addr == getAddressETH()) {
|
||||
// ERC20Interface wethContract = ERC20Interface(getAddressWETH());
|
||||
// uint wethBal = wethContract.balanceOf(address(this));
|
||||
// toWithdraw = toWithdraw < wethBal ? wethBal : toWithdraw;
|
||||
// setApproval(getAddressWETH(), toWithdraw, getAddressWETH());
|
||||
// ERC20Interface(getAddressWETH()).withdraw(toWithdraw);
|
||||
// msg.sender.transfer(toWithdraw);
|
||||
// } else {
|
||||
// require(ERC20Interface(erc20Addr).transfer(msg.sender, toWithdraw), "Allowance or not enough bal");
|
||||
// }
|
||||
// emit LogWithdraw(erc20Addr, toWithdraw, address(this));
|
||||
// }
|
||||
|
||||
// /**
|
||||
// * @dev Borrow ETH/ERC20
|
||||
// */
|
||||
// function borrow(uint256 marketId, address erc20Addr, uint256 tokenAmt) public {
|
||||
// SoloMarginContract(getSoloAddress()).operate(getAccountArgs(), getActionsArgs(marketId, tokenAmt, false));
|
||||
// if (erc20Addr == getAddressETH()) {
|
||||
// setApproval(getAddressWETH(), tokenAmt, getAddressWETH());
|
||||
// ERC20Interface(getAddressWETH()).withdraw(tokenAmt);
|
||||
// msg.sender.transfer(tokenAmt);
|
||||
// } else {
|
||||
// require(ERC20Interface(erc20Addr).transfer(msg.sender, tokenAmt), "Allowance or not enough bal");
|
||||
// }
|
||||
// emit LogBorrow(erc20Addr, tokenAmt, address(this));
|
||||
// }
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract ImportHelper is DydxResolver {
|
||||
struct BorrowData {
|
||||
uint[] borrowAmt;
|
||||
address[] borrowAddr;
|
||||
uint[] marketId;
|
||||
uint borrowCount;
|
||||
}
|
||||
|
||||
struct SupplyData {
|
||||
uint[] supplyAmt;
|
||||
address[] supplyAddr;
|
||||
uint[] marketId;
|
||||
uint supplyCount;
|
||||
}
|
||||
|
||||
function getUserData(uint accountId, uint toConvert) internal returns(SupplyData memory, BorrowData memory) {
|
||||
SoloMarginContract solo = SoloMarginContract(getSoloAddress());
|
||||
uint markets = solo.getNumMarkets();
|
||||
uint[] memory supplyArr = new uint[](markets);
|
||||
uint[] memory borrowArr = new uint[](markets);
|
||||
uint[] memory supplyId = new uint[](markets);
|
||||
uint[] memory borrowId = new uint[](markets);
|
||||
address[] memory supplyAddr = new address[](markets);
|
||||
address[] memory borrowAddr = new address[](markets);
|
||||
uint borrowCount = 0;
|
||||
uint supplyCount = 0;
|
||||
for (uint i = 0; i < markets; i++) {
|
||||
(uint tokenbal, bool tokenSign) = getDydxBal(msg.sender, i, accountId);
|
||||
if (tokenSign) {
|
||||
supplyArr[supplyCount] = wmul(tokenbal, toConvert);
|
||||
supplyAddr[supplyCount] = solo.getMarketTokenAddress(i);
|
||||
supplyId[supplyCount] = i;
|
||||
supplyCount++;
|
||||
} else {
|
||||
borrowArr[borrowCount] = wmul(tokenbal, toConvert);
|
||||
borrowAddr[borrowCount] = solo.getMarketTokenAddress(i);
|
||||
borrowId[borrowCount] = i;
|
||||
borrowCount++;
|
||||
}
|
||||
}
|
||||
return (
|
||||
SupplyData(
|
||||
supplyArr,
|
||||
supplyAddr,
|
||||
supplyId,
|
||||
supplyCount),
|
||||
BorrowData(
|
||||
borrowArr,
|
||||
borrowAddr,
|
||||
borrowId,
|
||||
borrowCount
|
||||
));
|
||||
}
|
||||
|
||||
struct ImportStruct {
|
||||
SoloMarginContract.Info[] accounts;
|
||||
SoloMarginContract.ActionArgs[] actions;
|
||||
}
|
||||
|
||||
function createArgs(
|
||||
SupplyData memory suppyArr,
|
||||
BorrowData memory borrowArr
|
||||
) internal view returns(SoloMarginContract.Info[] memory, SoloMarginContract.ActionArgs[] memory) {
|
||||
uint borrowCount = borrowArr.borrowAmt.length;
|
||||
uint supplyCount = suppyArr.supplyAmt.length;
|
||||
uint totalCount = borrowArr.borrowAmt.length + suppyArr.supplyAmt.length;
|
||||
SoloMarginContract.ActionArgs[] memory actions = new SoloMarginContract.ActionArgs[](totalCount);
|
||||
SoloMarginContract.Info[] memory accounts = new SoloMarginContract.Info[](totalCount);
|
||||
|
||||
for (uint i = 0; i < borrowCount; i++) {
|
||||
actions[i] = getActionsArgs(
|
||||
address(this),
|
||||
borrowArr.marketId[i],
|
||||
borrowArr.borrowAmt[i],
|
||||
true)[0]; // add markets , account to take token.
|
||||
accounts[i] = getAccountArgs(msg.sender, 0)[0];
|
||||
actions[i + supplyCount + supplyCount] = getActionsArgs(
|
||||
msg.sender,
|
||||
borrowArr.marketId[i],
|
||||
borrowArr.borrowAmt[i],
|
||||
false)[0];
|
||||
accounts[i + supplyCount + supplyCount] = getAccountArgs(address(this), 0)[0];
|
||||
}
|
||||
|
||||
for (uint i = 0; i < supplyCount; i++) {
|
||||
uint baseIndex = borrowCount + i;
|
||||
actions[baseIndex] = getActionsArgs(
|
||||
address(this),
|
||||
suppyArr.marketId[i],
|
||||
suppyArr.supplyAmt[i],
|
||||
false)[0];
|
||||
accounts[i] = getAccountArgs(msg.sender, 0)[0];
|
||||
actions[baseIndex + supplyCount] = getActionsArgs(
|
||||
address(this),
|
||||
suppyArr.marketId[i],
|
||||
suppyArr.supplyAmt[i],
|
||||
true)[0];
|
||||
accounts[baseIndex + supplyCount] = getAccountArgs(address(this), 0)[0];
|
||||
}
|
||||
|
||||
return (accounts, actions);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
contract ImportResolver is ImportHelper {
|
||||
event LogDydxImport(address owner, uint percentage, bool isCompound, address[] markets, address[] borrowAddr, uint[] borrowAmt);
|
||||
|
||||
function importAssets(uint toConvert, uint accountId, bool isCompound) external {
|
||||
// subtracting 0.00000001 ETH from initialPoolBal to solve Compound 8 decimal CETH error.
|
||||
uint initialPoolBal = sub(getPoolAddress().balance, 10000000000);
|
||||
(SupplyData memory supplyArr, BorrowData memory borrowArr) = getUserData(accountId, toConvert);
|
||||
|
||||
// Get liquidity assets to payback user wallet borrowed assets
|
||||
PoolInterface(getPoolAddress()).accessToken(borrowArr.borrowAddr, borrowArr.borrowAmt, isCompound);
|
||||
|
||||
SoloMarginContract solo = SoloMarginContract(getSoloAddress());
|
||||
(SoloMarginContract.Info[] memory accounts, SoloMarginContract.ActionArgs[] memory actions) = createArgs(supplyArr, borrowArr);
|
||||
solo.operate(accounts, actions);
|
||||
// // payback user wallet borrowed assets
|
||||
for (uint i = 0; i < borrowArr.borrowCount; i++) {
|
||||
address erc20 = borrowArr.borrowAddr[i];
|
||||
uint toPayback = borrowArr.borrowAmt[i];
|
||||
if (erc20 == getAddressWETH()) {
|
||||
getPoolAddress().transfer(toPayback);
|
||||
} else {
|
||||
require(ERC20Interface(erc20).transfer(getPoolAddress(), toPayback), "Not-enough-amt");
|
||||
}
|
||||
}
|
||||
|
||||
//payback InstaDApp liquidity
|
||||
PoolInterface(getPoolAddress()).paybackToken(borrowArr.borrowAddr, isCompound);
|
||||
|
||||
uint finalPoolBal = getPoolAddress().balance;
|
||||
assert(finalPoolBal >= initialPoolBal);
|
||||
|
||||
// emit LogCompoundImport(
|
||||
// msg.sender,
|
||||
// toConvert,
|
||||
// isCompound,
|
||||
// markets,
|
||||
// borrowAddr,
|
||||
// borrowAmt
|
||||
// );
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract InstaDydxImport is ImportResolver {
|
||||
function() external payable {}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user