@etherisc/gif-next 0.0.2-ec436bf-674 → 0.0.2-ed1536c-164

package/contracts/staking/StakeManagerLib.sol

@@ -0,0 +1,156 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.20;
+
+import {IERC20Metadata} from "@openzeppelin/contracts/token/ERC20/extensions/IERC20Metadata.sol";
+
+import {Amount, AmountLib} from "../type/Amount.sol";
+import {Component} from "../shared/Component.sol";
+import {IRegistry} from "../registry/IRegistry.sol";
+import {IRegistryService} from "../registry/IRegistryService.sol";
+import {IStaking} from "./IStaking.sol";
+import {Key32} from "../type/Key32.sol";
+import {NftId} from "../type/NftId.sol";
+import {ObjectType, INSTANCE, PROTOCOL, TARGET} from "../type/ObjectType.sol";
+import {Seconds, SecondsLib} from "../type/Seconds.sol";
+import {StakingReader} from "./StakingReader.sol";
+import {Timestamp, TimestampLib} from "../type/Timestamp.sol";
+import {UFixed, UFixedLib} from "../type/UFixed.sol";
+
+
+library StakeManagerLib {
+
+    function checkActiveTarget(
+        StakingReader stakingReader,
+        NftId targetNftId
+    )
+        public
+        view
+    {
+        // target nft id must be registered
+        if (!stakingReader.isTarget(targetNftId)) {
+            revert IStaking.ErrorStakingNotTarget(targetNftId);
+        }
+
+        // only accept stakes for active targets
+        if (!stakingReader.isActive(targetNftId)) {
+            revert IStaking.ErrorStakingTargetNotActive(targetNftId);
+        }
+    }
+
+
+    function checkStakeParameters(
+        StakingReader stakingReader,
+        NftId targetNftId, 
+        Amount dipAmount
+    )
+        external
+        view
+        returns (
+            Timestamp lockedUntil
+        )
+    {
+        // target nft id must be registered
+        if (!stakingReader.isTarget(targetNftId)) {
+            revert IStaking.ErrorStakingNotTarget(targetNftId);
+        }
+
+        // only accept stakes for active targets
+        if (!stakingReader.isActive(targetNftId)) {
+            revert IStaking.ErrorStakingTargetNotActive(targetNftId);
+        }
+
+        checkDipAmount(targetNftId, dipAmount);
+
+        Timestamp currentTime = TimestampLib.blockTimestamp();
+        lockedUntil = currentTime.addSeconds(
+            stakingReader.getTargetInfo(targetNftId).lockingPeriod);
+    }
+
+
+    function checkDipAmount(
+        NftId targetNftId, 
+        Amount dipAmount
+    )
+        public 
+        pure 
+    {
+        // check stake amount > 0
+        if (dipAmount.eqz()) {
+            revert IStaking.ErrorStakingStakeAmountZero(targetNftId);
+        }
+
+        // TODO add check for target specific max dip amount (min stake + tvl * stake rate + buffer)
+    }
+
+
+    function checkDipBalanceAndAllowance(
+        IERC20Metadata dip, 
+        address owner, 
+        address tokenHandlerAddress, 
+        Amount dipAmount
+    )
+        public
+        view
+    {
+        // check balance
+        uint256 amount = dipAmount.toInt();
+        uint256 dipBalance = dip.balanceOf(owner);
+        if (dipBalance < amount) {
+            revert IStaking.ErrorStakingDipBalanceInsufficient(owner, amount, dipBalance);
+        }
+
+        // check allowance
+        uint256 dipAllowance = dip.allowance(owner, tokenHandlerAddress);
+        if (dipAllowance < amount) {
+            revert IStaking.ErrorStakingDipAllowanceInsufficient(owner, tokenHandlerAddress, amount, dipAllowance);
+        }
+    }
+
+
+    function calculateRewardIncrease(
+        StakingReader stakingReader,
+        NftId stakeNftId
+    )
+        public
+        view
+        returns (Amount rewardIncreaseAmount)
+    {
+        (
+            UFixed rewardRate,
+            Amount stakeAmount,
+            Timestamp lastUpdatedAt
+        ) = stakingReader.getRewardCalculationInput(stakeNftId);
+
+        Seconds duration = SecondsLib.toSeconds(
+            block.timestamp - lastUpdatedAt.toInt());
+        
+        rewardIncreaseAmount = calculateRewardAmount(
+            rewardRate,
+            duration,
+            stakeAmount);
+    }
+
+
+    function calculateRewardAmount(
+        UFixed rewardRate,
+        Seconds duration, 
+        Amount stakeAmount
+    )
+        public 
+        pure 
+        returns(
+            Amount rewardAmount
+        )
+    {
+        UFixed yearFraction = getYearFraction(duration);
+        UFixed rewardRateFraction = yearFraction * rewardRate;
+        UFixed amountUFixed = rewardRateFraction * stakeAmount.toUFixed();
+        rewardAmount = AmountLib.toAmount(amountUFixed.toInt());
+    }
+
+
+    function getYearFraction(Seconds duration) public pure returns (UFixed yearFraction) {
+        return UFixedLib.toUFixed(duration.toInt()) / UFixedLib.toUFixed(SecondsLib.oneYear().toInt());
+    }
+
+}

package/contracts/staking/Staking.sol

@@ -0,0 +1,502 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.20;
+
+import {Amount, AmountLib} from "../type/Amount.sol";
+import {ChainNft} from "../registry/ChainNft.sol";
+import {Component} from "../shared/Component.sol";
+import {IRegistry} from "../registry/IRegistry.sol";
+import {IRegistryService} from "../registry/IRegistryService.sol";
+import {IStaking} from "./IStaking.sol";
+import {IVersionable} from "../shared/IVersionable.sol";
+import {Key32} from "../type/Key32.sol";
+import {KEEP_STATE} from "../type/StateId.sol";
+import {LibNftIdSet} from "../type/NftIdSet.sol";
+import {NftId, NftIdLib} from "../type/NftId.sol";
+import {NftIdSetManager} from "../shared/NftIdSetManager.sol";
+import {ObjectType, INSTANCE, PROTOCOL, STAKE, STAKING, TARGET} from "../type/ObjectType.sol";
+import {Seconds, SecondsLib} from "../type/Seconds.sol";
+import {StakeManagerLib} from "./StakeManagerLib.sol";
+import {StakingReader} from "./StakingReader.sol";
+import {StakingStore} from "./StakingStore.sol";
+import {TargetManagerLib} from "./TargetManagerLib.sol";
+import {Timestamp, TimestampLib} from "../type/Timestamp.sol";
+import {TokenHandler} from "../shared/TokenHandler.sol";
+import {TokenRegistry} from "../registry/TokenRegistry.sol";
+import {UFixed, UFixedLib} from "../type/UFixed.sol";
+import {Version, VersionLib} from "../type/Version.sol";
+import {Versionable} from "../shared/Versionable.sol";
+
+contract Staking is 
+    Component,
+    Versionable,
+    IStaking
+{
+    string public constant CONTRACT_NAME = "Staking";
+    uint8 private constant GIF_MAJOR_VERSION = 3;
+
+    // keccak256(abi.encode(uint256(keccak256("gif-next.contracts.component.Staking.sol")) - 1)) & ~bytes32(uint256(0xff));
+    bytes32 public constant STAKING_LOCATION_V1 = 0xafe8d4462b2ed26a47154f4b8f6d1497d2f772496965791d25bd456e342b7f00;
+
+    struct StakingStorage {
+        IRegistryService _registryService;
+        TokenRegistry _tokenRegistry;
+        StakingStore _store;
+        StakingReader _reader;
+
+        mapping(uint256 chainId => mapping(address token => UFixed stakingRate)) _stakingRate;
+
+        mapping(NftId targetNftId => Amount stakedAmount) _stakedAmount;
+        mapping(NftId targetNftId => mapping(address token => Amount tvlAmount)) _tvlAmount;
+
+        NftId _protocolNftId;
+    }
+
+
+    modifier onlyStake(NftId stakeNftId) {
+        if (!_getStakingStorage()._store.exists(stakeNftId)) {
+            revert ErrorStakingNotStake(stakeNftId);
+        }
+        _;
+    }
+
+
+    modifier onlyTarget(NftId targetNftId) {
+        if (!_getStakingStorage()._store.getTargetManager().exists(targetNftId)) {
+            revert ErrorStakingNotTarget(targetNftId);
+        }
+        _;
+    }
+
+
+    // from Versionable
+    function getVersion()
+        public 
+        pure 
+        virtual override (IVersionable, Versionable)
+        returns(Version)
+    {
+        return VersionLib.toVersion(GIF_MAJOR_VERSION,0,0);
+    }
+
+    // set/update staking reader
+    function setStakingReader(StakingReader stakingReader)
+        external
+        virtual
+        onlyOwner
+    {
+        if(stakingReader.getStaking() != IStaking(this)) {
+            revert ErrorStakingStakingReaderStakingMismatch(address(stakingReader.getStaking()));
+        }
+
+        _getStakingStorage()._reader = stakingReader;
+    }
+
+
+    // rate management 
+    function setStakingRate(uint256 chainId, address token, UFixed stakingRate)
+        external
+        virtual
+        onlyOwner
+    {
+
+    }
+
+    // reward management 
+
+    // target management
+
+    function registerTarget(
+        NftId targetNftId,
+        ObjectType expectedObjectType,
+        uint256 chainId,
+        Seconds initialLockingPeriod,
+        UFixed initialRewardRate
+    )
+        external
+        virtual
+        // restricted // staking service access
+    {
+        TargetManagerLib.checkTargetParameters(
+            getRegistry(), 
+            _getStakingStorage()._reader,
+            targetNftId, 
+            expectedObjectType, 
+            initialLockingPeriod, 
+            initialRewardRate);
+
+        _getStakingStorage()._store.createTarget(
+            targetNftId,
+            TargetInfo({
+                objectType: expectedObjectType,
+                chainId: chainId,
+                lockingPeriod: initialLockingPeriod,
+                rewardRate: initialRewardRate}));
+    }
+
+
+    function setLockingPeriod(
+        NftId targetNftId, 
+        Seconds lockingPeriod
+    )
+        external
+        virtual
+        // TODO add restricted // only staking service
+        onlyTarget(targetNftId) // maybe duplicate check
+    {
+        (
+            Seconds oldLockingPeriod,
+            TargetInfo memory targetInfo
+        ) = TargetManagerLib.updateLockingPeriod(
+            this,
+            targetNftId,
+            lockingPeriod);
+        
+        _getStakingStorage()._store.updateTarget(targetNftId, targetInfo);
+
+        emit LogStakingLockingPeriodSet(targetNftId, oldLockingPeriod, lockingPeriod);
+    }
+
+    // TODO add function to set protocol reward rate: onlyOwner
+    // get protocol nft id (from where)
+
+    function setRewardRate(NftId targetNftId, UFixed rewardRate)
+        external
+        virtual
+        onlyTarget(targetNftId) // maybe duplicate check
+    {
+        (
+            UFixed oldRewardRate,
+            TargetInfo memory targetInfo
+        ) = TargetManagerLib.updateRewardRate(
+            this,
+            targetNftId,
+            rewardRate);
+        
+        _getStakingStorage()._store.updateTarget(targetNftId, targetInfo);
+
+        emit LogStakingRewardRateSet(targetNftId, oldRewardRate, rewardRate);
+    }
+
+
+    function refillRewardReserves(NftId targetNftId, Amount dipAmount)
+        external
+        virtual
+    {
+
+    }
+
+    function withdrawRewardReserves(NftId targetNftId, Amount dipAmount)
+        external
+        virtual
+        // onlyNftOwner(targetNftId)
+    {
+        
+    }
+
+
+    function increaseTvl(NftId targetNftId, address token, Amount amount)
+        external
+        virtual
+        // restricted // service to service access
+    {
+
+    }
+
+    function decreaseTvl(NftId targetNftId, address token, Amount amount)
+        external
+        virtual
+        // restricted // service to service access
+    {
+
+    }
+
+
+    function registerRemoteTarget(NftId targetNftId, TargetInfo memory targetInfo)
+        external
+        virtual
+        onlyOwner // or CCIP
+    {
+        
+    }
+
+    function updateRemoteTvl(NftId targetNftId, address token, Amount amount)
+        external
+        virtual
+        onlyOwner // or CCIP
+    {
+        
+    }
+
+    //--- staking functions -------------------------------------------------//
+
+    function registerStake(
+        NftId stakeNftId, 
+        NftId targetNftId, 
+        Amount dipAmount
+    )
+        external
+        virtual
+        // TODO add restricted() // only staking service
+    {
+        Timestamp lockedUntil = StakeManagerLib.checkStakeParameters(
+            _getStakingStorage()._reader,
+            targetNftId,
+            dipAmount);
+
+        _getStakingStorage()._store.create(
+            stakeNftId, 
+            StakeInfo({
+                lockedUntil: lockedUntil}),
+            dipAmount);
+    }
+
+
+    function stake(NftId stakeNftId, Amount dipAmount)
+        external
+        virtual
+        // TODO add restricted() // only staking service
+        onlyStake(stakeNftId)
+    {
+        StakingStorage storage $ = _getStakingStorage();
+        Amount rewardIncrement = StakeManagerLib.calculateRewardIncrease(
+            $._reader, 
+            stakeNftId);
+
+        $._store.increaseBalance(
+            stakeNftId, 
+            dipAmount,
+            rewardIncrement);
+    }
+
+
+    function restake(NftId stakeNftId)
+        external
+        virtual
+        // TODO add restricted() // only staking service
+        onlyStake(stakeNftId)
+    {
+        // TODO add check that target is active and allows additional staking amount
+        StakingStorage storage $ = _getStakingStorage();
+        Amount rewardIncrement = StakeManagerLib.calculateRewardIncrease(
+            $._reader, 
+            stakeNftId);
+
+        $._store.restakeRewards(
+            stakeNftId, 
+            rewardIncrement);
+    }
+
+
+    function updateRewards(NftId stakeNftId)
+        external
+        virtual
+        // TODO add restricted() // only staking service
+        onlyStake(stakeNftId)
+    {
+        StakingStorage storage $ = _getStakingStorage();
+        _updateRewards($._reader, $._store, stakeNftId);
+    }
+
+
+    function claimRewards(NftId stakeNftId)
+        external
+        virtual
+        // TODO add restricted() // only staking service
+        onlyStake(stakeNftId)
+        returns (
+            Amount rewardsClaimedAmount
+        )
+    {
+        StakingStorage storage $ = _getStakingStorage();
+
+        // update rewards since last update
+        _updateRewards($._reader, $._store, stakeNftId);
+
+        // unstake all available rewards
+        rewardsClaimedAmount = $._store.claimUpTo(
+            stakeNftId,
+            AmountLib.max());
+    }
+
+
+    function unstake(NftId stakeNftId)
+        external
+        virtual
+        // TODO add restricted() // only staking service
+        onlyStake(stakeNftId)
+        returns (
+            Amount unstakedAmount,
+            Amount rewardsClaimedAmount
+        )
+    {
+        // TODO add check that stake locking is in the past
+        StakingStorage storage $ = _getStakingStorage();
+
+        // update rewards since last update
+        _updateRewards($._reader, $._store, stakeNftId);
+
+        // unstake all available dips
+        (
+            unstakedAmount, 
+            rewardsClaimedAmount
+        ) = $._store.unstakeUpTo(
+            stakeNftId,
+            AmountLib.max(), // unstake all stakes
+            AmountLib.max()); // claim all rewards
+    }
+
+
+
+    function _updateRewards(
+        StakingReader reader,
+        StakingStore store,
+        NftId stakeNftId
+    )
+        internal
+        virtual
+    {
+        Amount rewardIncrement = StakeManagerLib.calculateRewardIncrease(
+            reader, 
+            stakeNftId);
+
+        store.updateRewards(
+            stakeNftId, 
+            rewardIncrement);
+    }
+
+
+    //--- other functions ---------------------------------------------------//
+
+    function collectDipAmount(address from, Amount dipAmount)
+        external
+        // TODO add restricted() // only staking service
+    {
+        TokenHandler tokenHandler = getTokenHandler();
+        address stakingWallet = getWallet();
+
+        StakeManagerLib.checkDipBalanceAndAllowance(
+            getToken(), 
+            from, 
+            address(tokenHandler), 
+            dipAmount);
+
+        tokenHandler.transfer(from, stakingWallet, dipAmount);
+    }
+
+
+    function transferDipAmount(address to, Amount dipAmount)
+        external
+        // TODO add restricted() // only staking service
+    {
+        TokenHandler tokenHandler = getTokenHandler();
+        address stakingWallet = getWallet();
+
+        StakeManagerLib.checkDipBalanceAndAllowance(
+            getToken(), 
+            stakingWallet, 
+            address(tokenHandler), 
+            dipAmount);
+
+        tokenHandler.transfer(stakingWallet, to, dipAmount);
+    }
+
+
+    //--- view functions ----------------------------------------------------//
+
+    function getStakingReader() public view returns (StakingReader reader) {
+        return _getStakingStorage()._reader;
+    }
+
+    function getStakingStore() external view returns (StakingStore stakingStore) {
+        return _getStakingStorage()._store;
+    }
+
+    function getStakingRate(uint256 chainId, address token) external view returns (UFixed stakingRate) {
+        return _getStakingStorage()._stakingRate[chainId][token];
+    }
+
+    function getTvlAmount(NftId targetNftId, address token) external view returns (Amount tvlAmount) {
+        return _getStakingStorage()._tvlAmount[targetNftId][token];
+    }
+
+    function getStakedAmount(NftId targetNftId) external view returns (Amount stakeAmount) {
+        return _getStakingStorage()._stakedAmount[targetNftId];
+    }
+
+    function getTokenRegistryAddress() external view returns (address tokenRegistry) {
+        return address(_getStakingStorage()._tokenRegistry);
+    }
+
+    function calculateRewardIncrementAmount(
+        NftId targetNftId,
+        Timestamp rewardsLastUpdatedAt
+    )
+        public 
+        virtual
+        view 
+        returns (Amount rewardIncrementAmount)
+    {
+
+    }
+
+    //--- internal functions ------------------------------------------------//
+
+    function _initialize(
+        address owner, 
+        bytes memory data
+    )
+        internal
+        virtual override
+        initializer
+    {
+        (
+            address initialAuthority,
+            address registryAddress,
+            address stakingReaderAddress,
+            address stakingStoreAddress,
+            address initialOwner
+        ) = abi.decode(data, (address, address, address, address, address));
+
+        IRegistry registry = IRegistry(registryAddress);
+        TokenRegistry tokenRegistry = TokenRegistry(registry.getTokenRegistryAddress());
+        address dipTokenAddress = address(tokenRegistry.getDipToken());
+
+        initializeComponent(
+            initialAuthority,
+            registryAddress, 
+            registry.getNftId(), 
+            CONTRACT_NAME,
+            dipTokenAddress,
+            STAKING(), 
+            false, // is interceptor
+            initialOwner, 
+            "", // registry data
+            ""); // component data
+
+        _createAndSetTokenHandler();
+
+        // wiring to external contracts
+        StakingStorage storage $ = _getStakingStorage();
+        $._protocolNftId = getRegistry().getProtocolNftId();
+        $._store = StakingStore(stakingStoreAddress);
+        $._reader = StakingReader(stakingReaderAddress);
+        $._tokenRegistry = TokenRegistry(
+            address(tokenRegistry));
+
+        // wiring to staking
+        $._reader.setStakingDependencies(
+            registryAddress,
+            address(this),
+            stakingStoreAddress);
+
+        registerInterface(type(IStaking).interfaceId);
+    }
+
+
+    function _getStakingStorage() private pure returns (StakingStorage storage $) {
+        assembly {
+            $.slot := STAKING_LOCATION_V1
+        }
+    }
+
+}