107 lines
3.4 KiB
Solidity
107 lines
3.4 KiB
Solidity
/*
|
|
|
|
Copyright 2020 DODO ZOO.
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
*/
|
|
|
|
pragma solidity 0.6.9;
|
|
pragma experimental ABIEncoderV2;
|
|
|
|
import {InitializableOwnable} from "./InitializableOwnable.sol";
|
|
import {IERC20} from "../intf/IERC20.sol";
|
|
import {SafeMath} from "./SafeMath.sol";
|
|
|
|
interface ICrowdPooling {
|
|
function _QUOTE_RESERVE_() external view returns (uint256);
|
|
function getShares(address user) external view returns (uint256);
|
|
}
|
|
|
|
interface IFee {
|
|
function getUserFee(address user) external view returns (uint256);
|
|
}
|
|
|
|
interface IQuota {
|
|
function getUserQuota(address user) external view returns (uint256);
|
|
}
|
|
|
|
contract FeeRateImpl is InitializableOwnable {
|
|
using SafeMath for uint256;
|
|
|
|
struct CPPoolInfo {
|
|
address quoteToken;
|
|
uint256 globalQuota;
|
|
address feeAddr;
|
|
address quotaAddr;
|
|
}
|
|
|
|
mapping(address => CPPoolInfo) cpPools;
|
|
|
|
function addCpPoolInfo(address cpPool, address quoteToken, uint256 globalQuota, address feeAddr, address quotaAddr) external onlyOwner {
|
|
CPPoolInfo memory cpPoolInfo = CPPoolInfo({
|
|
quoteToken: quoteToken,
|
|
globalQuota: globalQuota,
|
|
feeAddr: feeAddr,
|
|
quotaAddr: quotaAddr
|
|
});
|
|
cpPools[cpPool] = cpPoolInfo;
|
|
}
|
|
|
|
function setCpPoolInfo(address cpPool, address quoteToken, uint256 globalQuota, address feeAddr, address quotaAddr) external onlyOwner {
|
|
cpPools[cpPool].quoteToken = quoteToken;
|
|
cpPools[cpPool].globalQuota = globalQuota;
|
|
cpPools[cpPool].feeAddr = feeAddr;
|
|
cpPools[cpPool].quotaAddr = quotaAddr;
|
|
}
|
|
|
|
function getFeeRate(address pool, address user) external view returns (uint256) {
|
|
CPPoolInfo memory cpPoolInfo = cpPools[pool];
|
|
address quoteToken = cpPoolInfo.quoteToken;
|
|
if(quoteToken == address(0)) {
|
|
return 0;
|
|
}else {
|
|
uint256 userInput = IERC20(quoteToken).balanceOf(pool).sub(ICrowdPooling(pool)._QUOTE_RESERVE_());
|
|
uint256 userStake = ICrowdPooling(pool).getShares(user);
|
|
address feeAddr = cpPoolInfo.feeAddr;
|
|
address quotaAddr = cpPoolInfo.quotaAddr;
|
|
uint256 curQuota = cpPoolInfo.globalQuota;
|
|
if(quotaAddr != address(0))
|
|
curQuota = IQuota(quotaAddr).getUserQuota(user);
|
|
|
|
if(curQuota != 0 && userInput.add(userStake) > curQuota) {
|
|
return 2 * 10**18; //over hardcap, will revert
|
|
} else {
|
|
if(feeAddr == address(0)) {
|
|
return 0;
|
|
}else {
|
|
return IFee(feeAddr).getUserFee(user);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
function getCPInfoByUser(address pool, address user) external view returns(bool isHaveCap, uint256 curQuota, uint256 userFee) {
|
|
CPPoolInfo memory cpPoolInfo = cpPools[pool];
|
|
address quotaAddr = cpPoolInfo.quotaAddr;
|
|
curQuota = cpPoolInfo.globalQuota;
|
|
if(quotaAddr != address(0))
|
|
curQuota = IQuota(quotaAddr).getUserQuota(user);
|
|
|
|
if(curQuota == 0) {
|
|
isHaveCap = false;
|
|
}else {
|
|
isHaveCap = true;
|
|
uint256 userStake = ICrowdPooling(pool).getShares(user);
|
|
curQuota = curQuota.sub(userStake);
|
|
}
|
|
|
|
address feeAddr = cpPoolInfo.feeAddr;
|
|
if(feeAddr == address(0)) {
|
|
userFee = 0;
|
|
} else {
|
|
userFee = IFee(feeAddr).getUserFee(user);
|
|
}
|
|
}
|
|
}
|