Liquidity.Party/tycho-execution
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add withdraw methods

Browse Source
This commit is contained in:
royvardhan
2025-01-22 20:33:24 +05:30
parent 113a996c16
commit 78fa890cd3
60 changed files with 50 additions and 27724 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
foundry/src/TychoRouter.sol
View File

@@ -3,9 +3,16 @@ pragma solidity ^0.8.28;
import "@openzeppelin/contracts/access/AccessControl.sol";
import "@permit2/src/interfaces/IAllowanceTransfer.sol";
import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
error TychoRouter__WithdrawalFailed();
error TychoRouter__InvalidReceiver();
contract TychoRouter is AccessControl {
IAllowanceTransfer public immutable permit2;
using SafeERC20 for IERC20;
//keccak256("NAME_OF_ROLE") : save gas on deployment
bytes32 public constant EXECUTOR_SETTER_ROLE =
@@ -17,6 +24,8 @@ contract TychoRouter is AccessControl {
bytes32 public constant FUND_RESCUER_ROLE =
0x912e45d663a6f4cc1d0491d8f046e06c616f40352565ea1cdb86a0e1aaefa41b;
event Withdrawal(address indexed token, uint256 amount);
constructor(address _permit2) {
permit2 = IAllowanceTransfer(_permit2);
_grantRole(DEFAULT_ADMIN_ROLE, msg.sender);
@@ -59,10 +68,49 @@ contract TychoRouter is AccessControl {
{
// TODO
}
/**
* @dev Allows withdrawing any ERC20 funds if funds get stuck in case of a bug.
*/
function withdraw(IERC20[] memory tokens, address receiver)
external
onlyRole(FUND_RESCUER_ROLE)
{
if (receiver == address(0)) revert TychoRouter__InvalidReceiver();
for (uint256 i = 0; i < tokens.length; i++) {
uint256 tokenBalance = tokens[i].balanceOf(address(this));
if (tokenBalance > 0) {
tokens[i].safeTransfer(receiver, tokenBalance);
emit Withdrawal(address(tokens[i]), tokenBalance);
}
}
}
/**
* @dev Allows withdrawing any NATIVE funds if funds get stuck in case of a bug.
* The contract should never hold any NATIVE tokens for security reasons.
*/
function withdrawNative(address receiver)
external
onlyRole(FUND_RESCUER_ROLE)
{
if (receiver == address(0)) revert TychoRouter__InvalidReceiver();
uint256 amount = address(this).balance;
if (amount > 0) {
// slither-disable-next-line arbitrary-send-eth
(bool success,) = receiver.call{value: amount}("");
if (!success) revert TychoRouter__WithdrawalFailed();
emit Withdrawal(address(0), amount);
}
}
/**
* @dev Allows this contract to receive native token
*/
// TODO Uncomment once withdraw method is implemented - or else Slither fails
// receive() external payable {}
receive() external payable {}
}