Instadapp/dsa-connectors-old
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors-old.git synced 2024-07-29 22:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated COMP connector

Browse Source
This commit is contained in:
Thrilok Kumar 2020-07-12 17:39:54 +05:30
parent 220e394caf
commit 83482abf97
1 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
contracts/connectors/COMP.sol
View File

@ -99,7 +99,7 @@ contract Helpers is DSMath {
* @dev Connector Details
*/
function connectorID() public pure returns(uint _type, uint _id) {
(_type, _id) = (1, 24);
(_type, _id) = (1, 31);
}
}
@ -125,6 +125,32 @@ contract COMPHelpers is Helpers {
function getMappingAddr() internal pure returns (address) {
return 0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88; // InstaMapping Address
}
function mergeTokenArr(address[] memory supplyTokens, address[] memory borrowTokens)
internal
view
returns (address[] memory ctokens, bool isBorrow, bool isSupply)
{
uint _supplyLen = supplyTokens.length;
uint _borrowLen = borrowTokens.length;
uint _totalLen = add(_supplyLen, _borrowLen);
ctokens = new address[](_totalLen);
isBorrow;
isSupply;
if(_supplyLen > 0) {
for (uint i = 0; i < _supplyLen; i++) {
ctokens[i] = InstaMapping(getMappingAddr()).cTokenMapping(supplyTokens[i]);
}
isSupply = true;
}
if(_borrowLen > 0) {
for (uint i = 0; i < _borrowLen; i++) {
ctokens[_supplyLen + i] = InstaMapping(getMappingAddr()).cTokenMapping(borrowTokens[i]);
}
isBorrow = true;
}
}
}
@ -154,6 +180,7 @@ contract BasicResolver is COMPHelpers {
/**
* @dev Claim Accrued COMP Token.
* @param tokens Array of tokens supplied and borrowed.
* @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
*/
function ClaimCompTwo(address[] calldata tokens, uint setId) external payable {
@ -180,28 +207,12 @@ contract BasicResolver is COMPHelpers {
/**
* @dev Claim Accrued COMP Token.
* @param supplyTokens Array of tokens supplied.
* @param borrowTokens Array of tokens borrowed.
* @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
*/
function ClaimCompThree(address[] calldata supplyTokens, address[] calldata borrowTokens, uint setId) external payable {
uint _supplyLen = supplyTokens.length;
uint _borrowLen = borrowTokens.length;
uint _totalLen = add(_supplyLen, _borrowLen);
address[] memory ctokens = new address[](_totalLen);
bool isBorrow;
bool isSupply;
if(_supplyLen > 0) {
for (uint i = 0; i < _supplyLen; i++) {
ctokens[i] = InstaMapping(getMappingAddr()).cTokenMapping(supplyTokens[i]);
}
isSupply = true;
}
if(_borrowLen > 0) {
for (uint i = 0; i < _borrowLen; i++) {
ctokens[_supplyLen + i] = InstaMapping(getMappingAddr()).cTokenMapping(borrowTokens[i]);
}
isBorrow = true;
}
(address[] memory ctokens, bool isBorrow, bool isSupply) = mergeTokenArr(supplyTokens, borrowTokens);
address[] memory holders = new address[](1);
holders[0] = address(this);
@ -239,7 +250,6 @@ contract BasicResolver is COMPHelpers {
}
}
contract ConnectCOMP is BasicResolver {
string public name = "COMP-v1";
}