ShibaSwap V3

1​
2    / function to claim all the tokens locked by user, after the locking period
3    function claimAll(uint256 r) external {
4        claimAllForUser(r, msg.sender);
5    }
6​
7    / function to get claimable amount for any user
8    function getClaimableAmount(address _user) external view returns(uint256) {
9        LockInfo[] memory lockInfoArrayForUser = lockInfoByUser[_user];
10        uint256 totalTransferableAmount = 0;
11        uint i;
12        for (i=latestCounterByUser[_user]; i<lockInfoArrayForUser.length; i++){
13            uint256 lockingPeriodHere = lockingPeriod;
14            if(lockInfoArrayForUser[i]._isDev){
15                lockingPeriodHere = devLockingPeriod;
16            }
17            if(now >= (lockInfoArrayForUser[i]._timestamp.add(lockingPeriodHere))){
18                totalTransferableAmount = totalTransferableAmount.add(lockInfoArrayForUser[i]._amount);
19            } else {
20                break;
21            }
22        }
23        return totalTransferableAmount;
24    }
25​
26    // get the left and right headers for a user, left header is the index counter till which we have already iterated, right header is basically the length of user's lockInfo array
27    function getLeftRightCounters(address _user) external view returns(uint256, uint256){
28        return(latestCounterByUser[_user], lockInfoByUser[_user].length);
29    }
30​
31    // in cases of emergency, emergency address can set this to true, which will enable emergencyWithdraw function
32    function setEmergencyFlag(bool _emergencyFlag) external {
33        require(msg.sender == emergencyAddress, "This function can only be called by emergencyAddress");
34        emergencyFlag = _emergencyFlag;
35    }
36​
37    // function for owner to transfer all tokens to another address
38    function emergencyWithdrawOwner(address _to) external onlyOwner{
39        uint256 amount = boneToken.balanceOf(address(this));
40        require(boneToken.transfer(_to, amount), 'MerkleDistributor: Transfer failed.');
41    }
42​
43    // emergency address can be updated from here
44    function setEmergencyAddr(address _newAddr) external {
45        require(msg.sender == emergencyAddress, "This function can only be called by emergencyAddress");
46        require(_newAddr != address(0), "_newAddr is a zero address");
47        emergencyAddress = _newAddr;
48    }
49​
50    // function to update/change the normal & dev locking period
51    function setLockingPeriod(uint256 _newLockingPeriod, uint256 _newDevLockingPeriod) external onlyOwner {
52        lockingPeriod = _newLockingPeriod;
53        devLockingPeriod = _newDevLockingPeriod;
54        emit LockingPeriod(msg.sender, _newLockingPeriod, _newDevLockingPeriod);
55    }
56}
57​

Solidity Contracts

JavaScript Tests