Tokemak

1contract Pool is ILiquidityPool, Initializable, ERC20, Ownable, Pausable {
2    using SafeMath for uint256;
3    using SafeERC20 for ERC20;
4​
5    ERC20 public override underlyer; // Underlying ERC20 token
6    IManager public manager;
7​
8    // implied: deployableLiquidity = underlyer.balanceOf(this) - withheldLiquidity
9    uint256 public override withheldLiquidity;
10​
11    / fAsset holder -> WithdrawalInfo
12    mapping(address => WithdrawalInfo) public override requestedWithdrawals;
13​
14    / NonReentrant
15    bool private _entered;
16    bool public _eventSend;
17    Destinations public destinations;
18​
19    modifier nonReentrant() {
20        require(!_entered, "ReentrancyGuard: reentrant call");
21        _entered = true;
22        _;
23        _entered = false;
24    }
25​
26    modifier onEventSend() {
27        if(_eventSend) {
28            _;
29        }
30    }
31​
32    function initialize(
33        ERC20 _underlyer,
34        IManager _manager,
35        string memory _name,
36        string memory _symbol
37    ) public initializer {
38        require(address(_underlyer) != address(0), "ZERO_ADDRESS");
39        require(address(_manager) != address(0), "ZERO_ADDRESS");
40​
41        __Context_init_unchained();
42        __Ownable_init_unchained();
43        __Pausable_init_unchained();
44        __ERC20_init_unchained(_name, _symbol);
45​
46        underlyer = _underlyer;
47        manager = _manager;
48    }
49​
50    ///@notice Gets decimals of underlyer so that tAsset decimals will match
51    function decimals() public view override returns (uint8) {
52        return underlyer.decimals();
53    }
54​
55    function deposit(uint256 amount) external override whenNotPaused {
56        _deposit(msg.sender, msg.sender, amount);
57    }
58​
59    function depositFor(address account, uint256 amount) external override whenNotPaused {
60        _deposit(msg.sender, account, amount);
61    }
62​
63    /// @dev References the WithdrawalInfo for how much the user is permitted to withdraw
64    /// @dev No withdrawal permitted unless currentCycle >= minCycle
65    /// @dev Decrements withheldLiquidity by the withdrawn amount
66    /// @dev TODO Update rewardsContract with proper accounting
67    function withdraw(uint256 requestedAmount) external override whenNotPaused nonReentrant {
68        require(
69            requestedAmount <= requestedWithdrawals[msg.sender].amount,
70            "WITHDRAW_INSUFFICIENT_BALANCE"
71        );
72        require(requestedAmount > 0, "NO_WITHDRAWAL");
73        require(underlyer.balanceOf(address(this)) >= requestedAmount, "INSUFFICIENT_POOL_BALANCE");
74​
75        // Checks for manager cycle and if user is allowed to withdraw based on their minimum withdrawal cycle
76        require(
77            requestedWithdrawals[msg.sender].minCycle <= manager.getCurrentCycleIndex(),
78            "INVALID_CYCLE"
79        );
80​
81        requestedWithdrawals[msg.sender].amount = requestedWithdrawals[msg.sender].amount.sub(
82            requestedAmount
83        );
84​
85        // If full amount withdrawn delete from mapping
86        if (requestedWithdrawals[msg.sender].amount == 0) {
87            delete requestedWithdrawals[msg.sender];
88        }
89​
90        withheldLiquidity = withheldLiquidity.sub(requestedAmount);
91​
92        _burn(msg.sender, requestedAmount);
93        underlyer.safeTransfer(msg.sender, requestedAmount);
94​
95        bytes32 eventSig = "Withdraw";
96        encodeAndSendData(eventSig, msg.sender);
97    }
98​
99    /// @dev Adjusts the withheldLiquidity as necessary
100    /// @dev Updates the WithdrawalInfo for when a user can withdraw and for what requested amount
101    function requestWithdrawal(uint256 amount) external override {
102        require(amount > 0, "INVALID_AMOUNT");
103        require(amount <= balanceOf(msg.sender), "INSUFFICIENT_BALANCE");
104​
105        //adjust withheld liquidity by removing the original withheld amount and adding the new amount
106        withheldLiquidity = withheldLiquidity.sub(requestedWithdrawals[msg.sender].amount).add(
107            amount
108        );
109        requestedWithdrawals[msg.sender].amount = amount;
110        if (manager.getRolloverStatus()) { // If manger is currently rolling over add two to min withdrawal cycle
111            requestedWithdrawals[msg.sender].minCycle = manager.getCurrentCycleIndex().add(2);
112        } else { // If manager is not rolling over add one to minimum withdrawal cycle
113            requestedWithdrawals[msg.sender].minCycle = manager.getCurrentCycleIndex().add(1);
114        }
115​
116        emit WithdrawalRequested(msg.sender, amount);
117    }
118​
119    function preTransferAdjustWithheldLiquidity(address sender, uint256 amount) internal {
120        if (requestedWithdrawals[sender].amount > 0) {
121            //reduce requested withdraw amount by transferred amount;
122            uint256 newRequestedWithdrawl = requestedWithdrawals[sender].amount.sub(
123                Math.min(amount, requestedWithdrawals[sender].amount)
124            );
125​
126            //subtract from global withheld liquidity (reduce) by removing the delta of (requestedAmount - newRequestedAmount)
127            withheldLiquidity = withheldLiquidity.sub(
128                requestedWithdrawals[sender].amount.sub(newRequestedWithdrawl)
129            );
130​
131            //update the requested withdraw for user
132            requestedWithdrawals[sender].amount = newRequestedWithdrawl;
133​
134            //if the withdraw request is 0, empty it out
135            if (requestedWithdrawals[sender].amount == 0) {
136                delete requestedWithdrawals[sender];
137            }
138        }
139    }

Solidity Contracts

JavaScript Tests