Minor changes.

This commit is contained in:
Sowmayjain 2019-06-20 02:00:22 +05:30
parent 13d668e5c4
commit df0071bb08
4 changed files with 14 additions and 15 deletions

View File

@ -177,8 +177,7 @@ contract CompoundHelper is MakerHelper {
function borrowDAI(uint tokenAmt) internal { function borrowDAI(uint tokenAmt) internal {
enterMarket(getCETHAddress()); enterMarket(getCETHAddress());
enterMarket(getCDAIAddress()); enterMarket(getCDAIAddress());
CTokenInterface cDaiContract = CTokenInterface(getCDAIAddress()); require(CTokenInterface(getCDAIAddress()).borrow(tokenAmt) == 0, "got collateral?");
require(cDaiContract.borrow(tokenAmt) == 0, "got collateral?");
} }
} }

View File

@ -112,7 +112,7 @@ contract Helper is DSMath {
address public ethAddr = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE; address public ethAddr = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
address public daiAddr = 0x89d24A6b4CcB1B6fAA2625fE562bDD9a23260359; address public daiAddr = 0x89d24A6b4CcB1B6fAA2625fE562bDD9a23260359;
address public registryAddr = 0xF5DCe57282A584D2746FaF1593d3121Fcac444dC; address public registry = 0xF5DCe57282A584D2746FaF1593d3121Fcac444dC;
address public sai = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3; address public sai = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3;
address public ume = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3; // Uniswap Maker Exchange address public ume = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3; // Uniswap Maker Exchange
address public ude = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3; // Uniswap DAI Exchange address public ude = 0x448a5065aeBB8E423F0896E6c5D525C040f59af3; // Uniswap DAI Exchange
@ -139,9 +139,9 @@ contract Helper is DSMath {
modifier isUserWallet { modifier isUserWallet {
address userAdd = UserWalletInterface(msg.sender).owner(); address userAdd = UserWalletInterface(msg.sender).owner();
address walletAdd = RegistryInterface(registryAddr).proxies(userAdd); address walletAdd = RegistryInterface(registry).proxies(userAdd);
require(walletAdd != address(0), "Not-User-Wallet"); require(walletAdd != address(0), "not-user-wallet");
require(walletAdd == msg.sender, "Not-Wallet-Owner"); require(walletAdd == msg.sender, "not-wallet-owner");
_; _;
} }
@ -249,7 +249,6 @@ contract MakerResolver is Helper {
setAllowance(peth, tubAddr); setAllowance(peth, tubAddr);
tub.exit(ink); tub.exit(ink);
uint freeJam = weth.balanceOf(address(this)); // withdraw possible previous stuck WETH as well uint freeJam = weth.balanceOf(address(this)); // withdraw possible previous stuck WETH as well
weth.withdraw(freeJam); weth.withdraw(freeJam);
@ -329,15 +328,15 @@ contract CompoundResolver is MakerResolver {
/** /**
* @dev Deposit ETH/ERC20 and mint Compound Tokens * @dev Deposit ETH/ERC20 and mint Compound Tokens
*/ */
function mintCETH(uint tokenAmt) internal { function mintCETH(uint ethAmt) internal {
CETHInterface cToken = CETHInterface(cEth); CETHInterface cToken = CETHInterface(cEth);
cToken.mint.value(tokenAmt)(); cToken.mint.value(ethAmt)();
uint cEthToReturn = wdiv(tokenAmt, CTokenInterface(cEth).exchangeRateCurrent()); uint cEthToReturn = wdiv(ethAmt, CTokenInterface(cEth).exchangeRateCurrent());
cToken.transfer(msg.sender, cEthToReturn); cToken.transfer(msg.sender, cEthToReturn);
emit LogMint( emit LogMint(
ethAddr, ethAddr,
cEth, cEth,
tokenAmt, ethAmt,
msg.sender msg.sender
); );
} }
@ -373,10 +372,6 @@ contract CompoundResolver is MakerResolver {
contract Bridge is CompoundResolver { contract Bridge is CompoundResolver {
function payUsersDebt(uint daiDebt) internal {
redeemUnderlying(cDai, daiDebt);
require(CDAIInterface(cDai).repayBorrowBehalf(msg.sender, daiDebt) == 0, "Enough DAI?");
}
function takeDebtBack(uint daiDebt) external { function takeDebtBack(uint daiDebt) external {
require(ERC20Interface(daiAddr).transferFrom(msg.sender, address(this),daiDebt), "Contract Approved?"); require(ERC20Interface(daiAddr).transferFrom(msg.sender, address(this),daiDebt), "Contract Approved?");
@ -399,6 +394,11 @@ contract Bridge is CompoundResolver {
give(cdpNum, msg.sender); give(cdpNum, msg.sender);
} }
function payUsersDebt(uint daiDebt) internal {
redeemUnderlying(cDai, daiDebt);
require(CDAIInterface(cDai).repayBorrowBehalf(msg.sender, daiDebt) == 0, "Enough DAI?");
}
} }