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

Update COMP

Browse Source
This commit is contained in:
Mubaris NK 2021-02-08 20:57:45 +05:30
parent 7e69631620
commit fb0c7ff62f
No known key found for this signature in database
GPG Key ID: 9AC09AD0F8D68561
2 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
contracts/connectors/COMP/helpers.sol
View File

@ -2,21 +2,18 @@ pragma solidity ^0.7.0;
import { DSMath } from "../../common/math.sol";
import { Basic } from "../../common/basic.sol";
import { ComptrollerInterface, COMPInterface } from "./interface.sol";
abstract contract Helpers is DSMath, Basic {
/**
* @dev Return Compound Comptroller Address
* @dev Compound Comptroller
*/
function getComptrollerAddress() internal pure returns (address) {
return 0x3d9819210A31b4961b30EF54bE2aeD79B9c9Cd3B;
}
ComptrollerInterface internal constant troller = ComptrollerInterface(0x3d9819210A31b4961b30EF54bE2aeD79B9c9Cd3B);
/**
* @dev Return COMP Token Address.
* @dev COMP Token
*/
function getCompTokenAddress() internal pure returns (address) {
return 0xc00e94Cb662C3520282E6f5717214004A7f26888;
}
COMPInterface internal constant compToken = COMPInterface(0xc00e94Cb662C3520282E6f5717214004A7f26888);
function mergeTokenArr(address[] memory supplyTokens, address[] memory borrowTokens)
internal

26
contracts/connectors/COMP/main.sol
View File

@ -2,7 +2,6 @@ pragma solidity ^0.7.0;
import { TokenInterface } from "../../common/interfaces.sol";
import { Stores } from "../../common/stores.sol";
import { ComptrollerInterface, COMPInterface } from "./interface.sol";
import { Helpers } from "./helpers.sol";
import { Events } from "./events.sol";
@ -13,10 +12,10 @@ abstract contract CompResolver is Events, Helpers {
* @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
*/
function ClaimComp(uint setId) external payable returns (string memory _eventName, bytes memory _eventParam) {
TokenInterface compToken = TokenInterface(getCompTokenAddress());
uint intialBal = compToken.balanceOf(address(this));
ComptrollerInterface(getComptrollerAddress()).claimComp(address(this));
uint finalBal = compToken.balanceOf(address(this));
TokenInterface _compToken = TokenInterface(address(compToken));
uint intialBal = _compToken.balanceOf(address(this));
troller.claimComp(address(this));
uint finalBal = _compToken.balanceOf(address(this));
uint amt = sub(finalBal, intialBal);
setUint(setId, amt);
@ -37,10 +36,10 @@ abstract contract CompResolver is Events, Helpers {
ctokens[i] = instaMapping.cTokenMapping(tokens[i]);
}
TokenInterface compToken = TokenInterface(getCompTokenAddress());
uint intialBal = compToken.balanceOf(address(this));
ComptrollerInterface(getComptrollerAddress()).claimComp(address(this), ctokens);
uint finalBal = compToken.balanceOf(address(this));
TokenInterface _compToken = TokenInterface(address(compToken));
uint intialBal = _compToken.balanceOf(address(this));
troller.claimComp(address(this), ctokens);
uint finalBal = _compToken.balanceOf(address(this));
uint amt = sub(finalBal, intialBal);
setUint(setId, amt);
@ -61,10 +60,10 @@ abstract contract CompResolver is Events, Helpers {
address[] memory holders = new address[](1);
holders[0] = address(this);
TokenInterface compToken = TokenInterface(getCompTokenAddress());
uint intialBal = compToken.balanceOf(address(this));
ComptrollerInterface(getComptrollerAddress()).claimComp(holders, ctokens, isBorrow, isSupply);
uint finalBal = compToken.balanceOf(address(this));
TokenInterface _compToken = TokenInterface(address(compToken));
uint intialBal = _compToken.balanceOf(address(this));
troller.claimComp(holders, ctokens, isBorrow, isSupply);
uint finalBal = _compToken.balanceOf(address(this));
uint amt = sub(finalBal, intialBal);
setUint(setId, amt);
@ -78,7 +77,6 @@ abstract contract CompResolver is Events, Helpers {
* @param delegatee The address to delegate votes to.
*/
function delegate(address delegatee) external payable returns (string memory _eventName, bytes memory _eventParam) {
COMPInterface compToken = COMPInterface(getCompTokenAddress());
require(compToken.delegates(address(this)) != delegatee, "Already delegated to same delegatee.");
compToken.delegate(delegatee);