removed "withdraw to owner" from deposit() method

This commit is contained in:
Sowmay Jain 2020-08-27 16:57:27 +10:00
parent e7e13bf44d
commit 85bd5774b8

View File

@ -47,25 +47,19 @@ contract Flusher {
address poolToken = registry.poolToken(token); address poolToken = registry.poolToken(token);
IERC20 tokenContract = IERC20(token); IERC20 tokenContract = IERC20(token);
if (poolToken != address(0)) { YieldPool poolContract = YieldPool(poolToken);
YieldPool poolContract = YieldPool(poolToken); uint amt;
uint amt; if (address(tokenContract) == address(0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)) {
if (address(tokenContract) == address(0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)) { amt = address(this).balance;
amt = address(this).balance; poolContract.deposit{value: amt}(amt);
poolContract.deposit{value: amt}(amt);
} else {
amt = tokenContract.balanceOf(address(this));
if (tokenContract.allowance(address(this), address(poolContract)) == 0)
tokenContract.approve(address(poolContract), uint(-1));
poolContract.deposit(amt);
}
emit LogDeposit(msg.sender, token, address(poolContract), amt);
} else { } else {
uint amt = tokenContract.balanceOf(address(this)); amt = tokenContract.balanceOf(address(this));
tokenContract.safeTransfer(owner, amt); if (tokenContract.allowance(address(this), address(poolContract)) == 0)
emit LogWithdrawToOwner(token, owner, amt); tokenContract.approve(address(poolContract), uint(-1));
poolContract.deposit(amt);
} }
emit LogDeposit(msg.sender, token, address(poolContract), amt);
} }
function withdraw(address token, uint amount) external isSigner returns (uint _amount) { function withdraw(address token, uint amount) external isSigner returns (uint _amount) {