logos-co
/
optimism-bridge-snt
mirror of https://github.com/logos-co/optimism-bridge-snt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve claimTokens

This commit is contained in:
Ricardo Guilherme Schmidt 2023-10-06 12:05:20 -03:00
parent 95ebd28004
commit b0e2f1b61b
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
contracts/SNTOptimismController.sol
View File

@ -53,18 +53,20 @@ contract SNTOptimismController is TokenController, Ownable2Step {
}
/**
* @notice Extract mistakenly sent tokens to this contract.
* @notice Send tokens or ether from this contract to owner.
* @param _token Token contract to recover, 0 to extract ether.
*/
function claimTokens(MiniMeBase _token) public onlyOwner {
uint256 balance;
if (address(_token) == address(0)) {
payable(owner()).transfer(address(this).balance);
balance = address(this).balance;
payable(msg.sender).transfer(balance);
return;
} else {
balance = _token.balanceOf(address(this));
_token.transfer(msg.sender, balance);
}
uint256 balance = _token.balanceOf(address(this));
_token.transfer(owner(), balance);
emit ClaimedTokens(address(_token), owner(), balance);
emit ClaimedTokens(address(_token), msg.sender, balance);
}
event ClaimedTokens(address indexed _token, address indexed _controller, uint256 _amount);