d-bis/dodo-contractV2
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add helper functions

Browse Source
This commit is contained in:
mingda
2021-02-07 15:39:55 +08:00
committed by owen05
parent 029b4951e7
commit e1e24f8007
1 changed files with 23 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
contracts/DODOToken/vDODOToken.sol
View File

@@ -134,7 +134,7 @@ contract vDODOToken is InitializableOwnable {
IERC20(_DODO_TOKEN_).transfer(_OWNER_, dodoBalance); IERC20(_DODO_TOKEN_).transfer(_OWNER_, dodoBalance);
} }
// ============ Functions ============ // ============ Mint & Redeem & Donate ============
function mint(uint256 dodoAmount, address superiorAddress) public { function mint(uint256 dodoAmount, address superiorAddress) public {
require( require(
@@ -215,7 +215,7 @@ contract vDODOToken is InitializableOwnable {
); );
} }
// ============ Functions(ERC20) ============ // ============ ERC20 Functions ============
function totalSupply() public view returns (uint256 vDODOSupply) { function totalSupply() public view returns (uint256 vDODOSupply) {
vDODOSupply = IERC20(_DODO_TOKEN_).balanceOf(address(this)) / _DODO_RATIO_; vDODOSupply = IERC20(_DODO_TOKEN_).balanceOf(address(this)) / _DODO_RATIO_;
@@ -225,22 +225,6 @@ contract vDODOToken is InitializableOwnable {
vDODOAmount = dodoBalanceOf(account) / _DODO_RATIO_; vDODOAmount = dodoBalanceOf(account) / _DODO_RATIO_;
} }
function availableBalanceOf(address account) public view returns (uint256 balance) {
if (_DOOD_GOV_ == address(0)) {
balance = balanceOf(account);
} else {
uint256 lockedBalance = IGovernance(_DOOD_GOV_).getLockedDODO(account);
balance = balanceOf(account).sub(lockedBalance);
}
}
function dodoBalanceOf(address account) public view returns (uint256 dodoAmount) {
UserInfo memory user = userInfo[account];
dodoAmount = DecimalMath.mulFloor(uint256(user.stakingPower), getLatestAlpha()).sub(
user.credit
);
}
function transfer(address to, uint256 vDODOAmount) public returns (bool) { function transfer(address to, uint256 vDODOAmount) public returns (bool) {
_updateAlpha(); _updateAlpha();
_transfer(msg.sender, to, vDODOAmount); _transfer(msg.sender, to, vDODOAmount);
@@ -269,7 +253,7 @@ contract vDODOToken is InitializableOwnable {
return _ALLOWED_[owner][spender]; return _ALLOWED_[owner][spender];
} }
// ============ View Functions ============ // ============ Helper Functions ============
function getLatestAlpha() public view returns (uint256) { function getLatestAlpha() public view returns (uint256) {
uint256 accuDODO = _DODO_PER_BLOCK_ * (block.number - lastRewardBlock); uint256 accuDODO = _DODO_PER_BLOCK_ * (block.number - lastRewardBlock);
@@ -280,6 +264,22 @@ contract vDODOToken is InitializableOwnable {
} }
} }
function availableBalanceOf(address account) public view returns (uint256 balance) {
if (_DOOD_GOV_ == address(0)) {
balance = balanceOf(account);
} else {
uint256 lockedBalance = IGovernance(_DOOD_GOV_).getLockedDODO(account);
balance = balanceOf(account).sub(lockedBalance);
}
}
function dodoBalanceOf(address account) public view returns (uint256 dodoAmount) {
UserInfo memory user = userInfo[account];
dodoAmount = DecimalMath.mulFloor(uint256(user.stakingPower), getLatestAlpha()).sub(
user.credit
);
}
function getWithdrawResult(uint256 dodoAmount) function getWithdrawResult(uint256 dodoAmount)
public public
view view
@@ -303,6 +303,10 @@ contract vDODOToken is InitializableOwnable {
feeRatio = IDODOCirculationHelper(_DODO_CIRCULATION_HELPER_).getDodoWithdrawFeeRatio(); feeRatio = IDODOCirculationHelper(_DODO_CIRCULATION_HELPER_).getDodoWithdrawFeeRatio();
} }
function getSuperior(address account) public view returns (address superior) {
return userInfo[account].superior;
}
// ============ Internal Functions ============ // ============ Internal Functions ============
function _updateAlpha() internal { function _updateAlpha() internal {