@etherisc/gif-next 0.0.2-e769e2e-077 → 0.0.2-e802d97-477

package/contracts/staking/StakingReader.sol

@@ -0,0 +1,187 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.20;
+
+import {Amount} from "../type/Amount.sol";
+import {Blocknumber}  from "../type/Blocknumber.sol";
+import {IKeyValueStore} from "../shared/IKeyValueStore.sol";
+import {IComponent} from "../shared/IComponent.sol";
+import {InitializableCustom} from "../shared/InitializableCustom.sol";
+import {IRegistry} from "../registry/IRegistry.sol";
+import {IRegistryLinked} from "../shared/IRegistryLinked.sol";
+import {IStaking} from "../staking/IStaking.sol";
+import {IVersionable} from "../shared/IVersionable.sol";
+import {NftId} from "../type/NftId.sol";
+import {NftIdSetManager} from "../shared/NftIdSetManager.sol";
+import {ObjectType, STAKE, TARGET} from "../type/ObjectType.sol";
+import {Seconds} from "../type/Seconds.sol";
+import {StakingStore} from "./StakingStore.sol";
+import {Timestamp} from "../type/Timestamp.sol";
+import {UFixed, UFixedLib} from "../type/UFixed.sol";
+
+contract StakingReader is
+    IRegistryLinked,
+    InitializableCustom
+{
+
+    error ErrorStakingReaderDependenciesAlreadySet();
+
+    IRegistry private _registry;
+    IStaking private _staking;
+    StakingStore private _store;
+
+    constructor(IRegistry registry) InitializableCustom() {
+        _registry = registry;
+    }
+
+    function initialize(
+        address stakingAddress,
+        address stakingStoreAddress
+    )
+        external
+        initializer
+    {
+        _staking = IStaking(stakingAddress);
+        _store = StakingStore(stakingStoreAddress);
+    }
+
+    // view and pure functions 
+
+    function getRegistry() external view returns (IRegistry registry) {
+        return _registry;
+    }
+
+    function getStaking() external view returns (IStaking staking) {
+        return _staking;
+    }
+
+    function getStakingRate(uint256 chainId, address token) external view returns (UFixed stakingRate) { 
+        return _store.getStakingRate(chainId, token); 
+    }
+
+
+    function isTarget(NftId targetNftId) external view returns (bool) {
+        return _store.getTargetManager().exists(targetNftId);
+    }
+
+
+    function targets() external view returns (uint256) {
+        return _store.getTargetManager().nftIds();
+    }
+
+
+    function getTargetNftId(uint256 idx) external view returns (NftId) {
+        return _store.getTargetManager().getNftId(idx);
+    }
+
+
+    function isActive(NftId targetNftId) external view returns (bool) {
+        return _store.getTargetManager().isActive(targetNftId);
+    }
+
+
+    function activeTargets() external view returns (uint256) {
+        return _store.getTargetManager().activeNftIds();
+    }
+
+
+    function getActiveTargetNftId(uint256 idx) external view returns (NftId) {
+        return _store.getTargetManager().getActiveNftId(idx);
+    }
+
+
+    function getTargetNftId(NftId stakeNftId) public view returns (NftId targetNftId) {
+        return _registry.getObjectInfo(stakeNftId).parentNftId;
+    }
+
+
+    function getTargetInfo(NftId targetNftId) public view returns (IStaking.TargetInfo memory info) {
+        bytes memory data = _store.getData(targetNftId.toKey32(TARGET()));
+        if (data.length > 0) {
+            return abi.decode(data, (IStaking.TargetInfo));
+        }
+    }
+
+
+    function getStakeInfo(NftId stakeNftId) external view returns (IStaking.StakeInfo memory stakeInfo) {
+        bytes memory data = _store.getData(stakeNftId.toKey32(STAKE()));
+        if (data.length > 0) {
+            return abi.decode(data, (IStaking.StakeInfo));
+        }
+    }
+
+
+    /// @dev get the reward rate that applies to the specified stake nft id.
+    function getTargetRewardRate(NftId stakeNftId) external view returns (NftId targetNftId, UFixed rewardRate) {
+        targetNftId = getTargetNftId(stakeNftId);
+        rewardRate = getTargetInfo(targetNftId).rewardRate;
+    }
+
+
+    /// @dev get the reward rate for the specified target nft id.
+    function getRewardRate(NftId targetNftId) external view returns (UFixed rewardRate) {
+        return getTargetInfo(targetNftId).rewardRate;
+    }
+
+    /// @dev returns the current reward reserve balance for the specified target.
+    function getReserveBalance(NftId targetNftId) external view returns (Amount rewardReserveBalance) {
+        return _store.getReserveBalance(targetNftId); 
+    }
+
+    function getStakeBalance(NftId nftId) external view returns (Amount balanceAmount) { 
+        return _store.getStakeBalance(nftId); 
+    }
+
+    function getRewardBalance(NftId nftId) external view returns (Amount rewardAmount) {
+        return _store.getRewardBalance(nftId); 
+    }
+
+    function getBalanceUpdatedAt(NftId nftId) external view returns (Timestamp updatedAt) {
+        return _store.getBalanceUpdatedAt(nftId); 
+    }
+
+    function getBalanceUpdatedIn(NftId nftId) external view returns (Blocknumber blocknumber) {
+        return _store.getBalanceUpdatedIn(nftId); 
+    }
+
+    function getTotalValueLocked(NftId nftId, address token) external view returns (Amount totalValueLocked) {
+        return _store.getTotalValueLocked(nftId, token); 
+    }
+
+    function getRequiredStakeBalance(NftId nftId) external view returns (Amount requiredStakedAmount) {
+        return _store.getRequiredStakeBalance(nftId); 
+    }
+
+    function getTargetBalances(NftId stakeNftId) 
+        public
+        view
+        returns (
+            Amount balanceAmount,
+            Amount stakeAmount,
+            Amount rewardAmount,
+            Blocknumber lastUpdatedIn
+        )
+    {
+        (
+            balanceAmount,
+            stakeAmount,
+            rewardAmount,
+            lastUpdatedIn
+        ) = _store.getTargetBalances(stakeNftId);
+    }
+
+    function getStakeBalances(NftId stakeNftId) 
+        external
+        view
+        returns (
+            Amount stakeAmount,
+            Amount rewardAmount,
+            Timestamp lastUpdatedAt
+        )
+    {
+        (
+            stakeAmount,
+            rewardAmount,
+            lastUpdatedAt
+        ) = _store.getStakeBalances(stakeNftId);
+    }
+}

package/contracts/staking/StakingService.sol

@@ -1,17 +1,26 @@
 // SPDX-License-Identifier: Apache-2.0
-pragma solidity ^0.8.19;
+pragma solidity ^0.8.20;
+
+import {IERC20Metadata} from "@openzeppelin/contracts/token/ERC20/extensions/IERC20Metadata.sol";
 
 import {Amount} from "../type/Amount.sol";
+import {ChainNft} from "../registry/ChainNft.sol";
 import {IPoolService} from "../pool/IPoolService.sol";
 import {IRegisterable} from "../shared/IRegisterable.sol";
 import {IRegistry} from "../registry/IRegistry.sol";
-import {IRegistryService} from "../registry/IRegistryService.sol";
+import {RegistryService} from "../registry/RegistryService.sol";
 import {IStaking} from "./IStaking.sol";
 import {IStakingService} from "./IStakingService.sol";
-import {NftId} from "../type/NftId.sol";
-import {ObjectType, POOL, REGISTRY, STAKE, STAKING} from "../type/ObjectType.sol";
+import {NftId, NftIdLib} from "../type/NftId.sol";
+import {ObjectType, INSTANCE, PROTOCOL, REGISTRY, STAKE, STAKING} from "../type/ObjectType.sol";
+import {Seconds} from "../type/Seconds.sol";
 import {Service} from "../shared/Service.sol";
+import {StakeManagerLib} from "./StakeManagerLib.sol";
+import {StakingReader} from "./StakingReader.sol";
+import {TargetManagerLib} from "./TargetManagerLib.sol";
 import {Timestamp} from "../type/Timestamp.sol";
+import {TokenHandler} from "../shared/TokenHandler.sol";
+import {UFixed} from "../type/UFixed.sol";
 
 
 contract StakingService is 
@@ -23,7 +32,17 @@ contract StakingService is
     bytes32 public constant STAKING_SERVICE_LOCATION_V1 = 0x6548005c3f4340f82f348c576c0ff69f7f529cadd5ad41f96aae61abceeaa300;
 
     struct StakingServiceStorage {
+        RegistryService _registryService;
         IStaking _staking;
+        IERC20Metadata _dip;
+        TokenHandler _tokenHandler;
+    }
+
+    modifier onlyNftOwner(NftId nftId) {
+        if(msg.sender != getRegistry().ownerOf(nftId)) {
+            revert ErrorStakingServiceNotNftOwner(nftId, getRegistry().ownerOf(nftId), msg.sender);
+        }
+        _;
     }
 
     function getDomain() public pure override returns(ObjectType) {
@@ -31,102 +50,286 @@ contract StakingService is
     }
 
 
+    function registerProtocolTarget()
+        external
+        virtual
+    {
+        uint256 protocolId = ChainNft(
+            getRegistry().getChainNftAddress()).PROTOCOL_NFT_ID();
+
+        NftId protocolNftId = NftIdLib.toNftId(protocolId);
+        _getStakingServiceStorage()._staking.registerTarget(
+            protocolNftId,
+            PROTOCOL(),
+            1, // protocol is registered on mainnet
+            TargetManagerLib.getDefaultLockingPeriod(),
+            TargetManagerLib.getDefaultRewardRate());
+
+        emit LogStakingServiceProtocolTargetRegistered(protocolNftId);
+    }
+
+
+    function createInstanceTarget(
+        NftId targetNftId,
+        Seconds initialLockingPeriod,
+        UFixed initialRewardRate
+    )
+        external
+        virtual
+        // restricted // TODO re-enable once services have stable roles
+    {
+        uint256 chainId = block.chainid;
+        _getStakingServiceStorage()._staking.registerTarget(
+            targetNftId,
+            INSTANCE(),
+            chainId,
+            initialLockingPeriod,
+            initialRewardRate);
+
+        emit LogStakingServiceInstanceTargetRegistered(targetNftId, chainId);
+    }
+
+
+    function setLockingPeriod(NftId targetNftId, Seconds lockingPeriod)
+        external
+        virtual
+        onlyNftOwner(targetNftId)
+    {
+        _getStakingServiceStorage()._staking.setLockingPeriod(
+            targetNftId, lockingPeriod);
+    }
+
+
+    function setRewardRate(NftId targetNftId, UFixed rewardRate)
+        external
+        virtual
+        onlyNftOwner(targetNftId)
+    {
+        _getStakingServiceStorage()._staking.setRewardRate(
+            targetNftId, rewardRate);
+
+    }
+
+
+    function refillRewardReserves(NftId targetNftId, Amount dipAmount)
+        external
+        virtual
+        returns (Amount newBalance)
+    {
+        address rewardProvider = msg.sender;
+
+        // update reward reserve book keeping
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        newBalance = $._staking.refillRewardReserves(targetNftId, dipAmount);
+
+        // collect reward dip from provider
+        $._staking.collectDipAmount(
+            rewardProvider,
+            dipAmount);
+
+        emit LogStakingServiceRewardReservesIncreased(targetNftId, rewardProvider, dipAmount, newBalance);
+    }
+
+
+    function withdrawRewardReserves(NftId targetNftId, Amount dipAmount)
+        external
+        virtual
+        onlyNftOwner(targetNftId)
+        returns (Amount newBalance)
+    {
+        // modifyier checks that sender is owner
+        address targetOwner = msg.sender;
+
+        // update reward reserve book keeping
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        newBalance = $._staking.withdrawRewardReserves(targetNftId, dipAmount);
+
+        // transfer withdrawal amount to target owner
+        $._staking.transferDipAmount(
+            targetOwner,
+            dipAmount);
+
+        emit LogStakingServiceRewardReservesDecreased(targetNftId, targetOwner, dipAmount, newBalance);
+    }
+
+
+    /// @dev creates a new stake to the specified target nft id with the provided dip amount
+    /// the target nft id must have been registered as an active staking target prior to this call
+    /// the sender of this transaction becomes the stake owner via the minted nft.
+    /// to create the new stake balance and allowance of the staker need to cover the dip amount
+    /// the allowance needs to be on the token handler of the staking contract (getTokenHandler())
+    /// this is a permissionless function.
     function create(
         NftId targetNftId,
-        Amount amount
+        Amount dipAmount
     )
         external
         virtual
+        // restricted // TODO re-enable once services have stable roles
         returns (
             NftId stakeNftId
         )
     {
-
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
+
+        // target nft id checks are performed in $._staking.createStake() below
+        // register new stake object with registry
+        stakeNftId = $._registryService.registerStake(
+            IRegistry.ObjectInfo({
+                nftId: NftIdLib.zero(),
+                parentNftId: targetNftId,
+                objectType: STAKE(),
+                isInterceptor: false,
+                objectAddress: address(0),
+                initialOwner: stakeOwner,
+                data: ""
+            }));
+
+        // create stake info with staking
+        $._staking.createStake(
+            stakeNftId, 
+            targetNftId,
+            dipAmount);
+
+        // collect staked dip by staking
+        $._staking.collectDipAmount(
+            stakeOwner,
+            dipAmount);
+
+        emit LogStakingServiceStakeCreated(stakeNftId, targetNftId, stakeOwner, dipAmount);
     }
 
 
     function stake(
         NftId stakeNftId,
-        Amount amount
+        Amount dipAmount
     )
         external
         virtual
+        // restricted // TODO re-enable once services have stable roles
+        onlyNftOwner(stakeNftId)
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
+
+        // add additional staked dips by staking 
+        Amount stakeBalance = $._staking.stake(
+            stakeNftId, 
+            dipAmount);
+
+        // collect staked dip by staking
+        if (dipAmount.gtz()) {
+            $._staking.collectDipAmount(
+                stakeOwner,
+                dipAmount);
+        }
 
+        emit LogStakingServiceStakeIncreased(stakeNftId, stakeOwner, dipAmount, stakeBalance);
     }
 
 
-    function unstake(
+    function restakeToNewTarget(
         NftId stakeNftId,
-        Amount amount
+        NftId newTargetNftId
     )
         external
         virtual
+        // restricted // TODO re-enable once services have stable roles
+        onlyNftOwner(stakeNftId)
+        returns (
+            NftId newStakeNftId
+        )
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        // TODO implement
+    } 
 
-    }
 
-    function close(
+    function updateRewards(
         NftId stakeNftId
     )
         external
-        virtual
+        // restricted // TODO re-enable once services have stable roles
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        $._staking.updateRewards(stakeNftId);
 
+        emit LogStakingServiceRewardsUpdated(stakeNftId);
     }
 
-    function reStake(
-        NftId stakeNftId,
-        NftId newTargetNftId
-    )
+
+    function claimRewards(NftId stakeNftId)
         external
         virtual
-        returns (
-            NftId newStakeNftId,
-            Timestamp unlockedAt
-        )
+        // restricted // TODO re-enable once services have stable roles
+        onlyNftOwner(stakeNftId)
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
+
+        Amount rewardsClaimedAmount = $._staking.claimRewards(stakeNftId);
+        $._staking.transferDipAmount(
+            stakeOwner,
+            rewardsClaimedAmount);
 
+        emit LogStakingServiceRewardsClaimed(stakeNftId, stakeOwner, rewardsClaimedAmount);
     }
 
 
-    function increaseTotalValueLocked(
-        NftId targetNftId,
-        address token,
-        Amount amount
-    )
+    function unstake(NftId stakeNftId)
         external
         virtual
-        returns (Amount totalValueLocked)
+        // restricted // TODO re-enable once services have stable roles
+        onlyNftOwner(stakeNftId)
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
+
+        (
+            Amount unstakedAmount,
+            Amount rewardsClaimedAmount
+        ) = $._staking.unstake(stakeNftId);
+
+        Amount totalAmount = unstakedAmount + rewardsClaimedAmount;
+        $._staking.transferDipAmount(
+            stakeOwner,
+            totalAmount);
 
+        emit LogStakingServiceUnstaked(stakeNftId, stakeOwner, totalAmount);
     }
 
 
-    function decreaseTotalValueLocked(
+    function setTotalValueLocked(
         NftId targetNftId,
         address token,
         Amount amount
     )
         external
         virtual
-        returns (Amount totalValueLocked)
     {
 
     }
 
 
-    function setTotalValueLocked(
-        NftId targetNftId,
-        address token,
-        Amount amount
-    )
+    function getDipToken()
         external
         virtual
+        returns (IERC20Metadata dip)
     {
+        return _getStakingServiceStorage()._dip;
+    }
 
+
+    function getTokenHandler()
+        external
+        virtual
+        returns (TokenHandler tokenHandler)
+    {
+        return _getStakingServiceStorage()._tokenHandler;
     }
 
+
     function getStaking()
         external
         virtual
@@ -145,14 +348,18 @@ contract StakingService is
         initializer()
     {
         (
+            address authority,
             address registryAddress,
             address stakingAddress
-        ) = abi.decode(data, (address, address));
+        ) = abi.decode(data, (address, address, address));
 
-        initializeService(registryAddress, address(0), owner);
+        initializeService(registryAddress, authority, owner);
 
         StakingServiceStorage storage $ = _getStakingServiceStorage();
+        $._registryService = RegistryService(_getServiceAddress(REGISTRY()));
         $._staking = _registerStaking(stakingAddress);
+        $._dip = $._staking.getToken();
+        $._tokenHandler = $._staking.getTokenHandler();
 
         registerInterface(type(IStakingService).interfaceId);
     }
@@ -183,10 +390,9 @@ contract StakingService is
         }
 
         address owner = msg.sender;
-        IRegistryService(
-            _getServiceAddress(REGISTRY())).registerStaking(
-                IRegisterable(stakingAddress),
-                owner);
+        _getStakingServiceStorage()._registryService.registerStaking(
+            IRegisterable(stakingAddress),
+            owner);
 
         return IStaking(stakingAddress);
     }

package/contracts/staking/StakingServiceManager.sol

@@ -14,20 +14,24 @@ contract StakingServiceManager is
 
     /// @dev initializes proxy manager with service implementation 
     constructor(
-        address registryAddress
+        address authority,
+        address registryAddress,
+        bytes32 salt
     )
         ProxyManager(registryAddress)
     {
         StakingService svc = new StakingService();
         bytes memory data = abi.encode(
+            authority, 
             registryAddress, 
             IRegistry(registryAddress).getStakingAddress());
-
-        IVersionable versionable = deploy(
+        IVersionable versionable = deployDetermenistic(
             address(svc), 
-            data);
+            data,
+            salt);
 
         _stakingService = StakingService(address(versionable));
+        _stakingService.registerProtocolTarget();
     }
 
     //--- view functions ----------------------------------------------------//