@etherisc/gif-next 0.0.2-fd4931b-974 → 0.0.2-fd5f17a-625

package/contracts/staking/StakingService.sol

@@ -1,6 +1,8 @@
 // SPDX-License-Identifier: Apache-2.0
 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";
@@ -13,9 +15,11 @@ 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";
 
 
@@ -30,31 +34,17 @@ contract StakingService is
     struct StakingServiceStorage {
         RegistryService _registryService;
         IStaking _staking;
+        IERC20Metadata _dip;
+        TokenHandler _tokenHandler;
     }
 
-    function getDomain() public pure override returns(ObjectType) {
-        return STAKING();
-    }
-
-
-    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());
+    modifier onlyNftOwner(NftId nftId) {
+        if(msg.sender != getRegistry().ownerOf(nftId)) {
+            revert ErrorStakingServiceNotNftOwner(nftId, getRegistry().ownerOf(nftId), msg.sender);
+        }
+        _;
     }
 
-    // TODO cleanup
-    event LogDebug(uint256 id, string message);
 
     function createInstanceTarget(
         NftId targetNftId,
@@ -63,10 +53,8 @@ contract StakingService is
     )
         external
         virtual
-        // restricted // TODO re-enable once services have stable roles
+        restricted()
     {
-        emit LogDebug(1, "before chainid");
-
         uint256 chainId = block.chainid;
         _getStakingServiceStorage()._staking.registerTarget(
             targetNftId,
@@ -74,6 +62,70 @@ contract StakingService is
             chainId,
             initialLockingPeriod,
             initialRewardRate);
+
+        emit LogStakingServiceInstanceTargetRegistered(targetNftId, chainId);
+    }
+
+
+    function setInstanceLockingPeriod(NftId instanceNftId, Seconds lockingPeriod)
+        external
+        virtual
+        restricted()
+    {
+        _getStakingServiceStorage()._staking.setLockingPeriod(
+            instanceNftId, 
+            lockingPeriod);
+    }
+
+
+    function setInstanceRewardRate(NftId instanceNftId, UFixed rewardRate)
+        external
+        virtual
+        restricted()
+    {
+        _getStakingServiceStorage()._staking.setRewardRate(
+            instanceNftId, 
+            rewardRate);
+    }
+
+
+    function refillInstanceRewardReserves(NftId instanceNftId, address rewardProvider, Amount dipAmount)
+        external
+        virtual
+        restricted()
+        returns (Amount newBalance)
+    {
+        return _refillRewardReserves(instanceNftId, rewardProvider, dipAmount);
+    }
+
+
+    function refillRewardReservesBySender(NftId targetNftId, Amount dipAmount)
+        external
+        virtual
+        restricted()
+        returns (Amount newBalance)
+    {
+        address rewardProvider = msg.sender;
+        return _refillRewardReserves(targetNftId, rewardProvider, dipAmount);
+    }
+
+
+    function withdrawInstanceRewardReserves(NftId instanceNftId, Amount dipAmount)
+        external
+        virtual
+        restricted()
+        returns (Amount newBalance)
+    {
+        // update reward reserve book keeping
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        newBalance = $._staking.withdrawRewardReserves(instanceNftId, dipAmount);
+
+        // transfer withdrawal amount to target owner
+        address instanceOwner = getRegistry().ownerOf(instanceNftId);
+        emit LogStakingServiceRewardReservesDecreased(instanceNftId, instanceOwner, dipAmount, newBalance);
+        $._staking.transferDipAmount(
+            instanceOwner,
+            dipAmount);
     }
 
 
@@ -89,33 +141,15 @@ contract StakingService is
     )
         external
         virtual
+        restricted()
         returns (
             NftId stakeNftId
         )
     {
         StakingServiceStorage storage $ = _getStakingServiceStorage();
-        StakingReader stakingReader = $._staking.getStakingReader();
         address stakeOwner = msg.sender;
 
-        // check target nft id
-        if (targetNftId.eqz()) {
-            revert ErrorStakingServiceZeroTargetNftId();
-        }
-
-        if (!stakingReader.isTarget(targetNftId)) {
-            revert ErrorStakingServiceNotTargetNftId(targetNftId);
-        }
-
-        if (!stakingReader.isActive(targetNftId)) {
-            revert ErrorStakingServiceNotActiveTargetNftId(targetNftId);
-        }
-
-        // check balance and allowance
-        //TODO implement
-
-        // collect staked dip amount
-        //TODO implement
-
+        // target nft id checks are performed in $._staking.createStake() below
         // register new stake object with registry
         stakeNftId = $._registryService.registerStake(
             IRegistry.ObjectInfo({
@@ -127,97 +161,149 @@ contract StakingService is
                 initialOwner: stakeOwner,
                 data: ""
             }));
-        
-        // create stake info in staking
-        $._staking.create(
+
+        // create stake info with staking
+        $._staking.createStake(
             stakeNftId, 
             targetNftId,
             dipAmount);
 
-        emit LogStakingServiceNewStakeCreated(stakeNftId, stakeOwner, targetNftId, dipAmount);
+        emit LogStakingServiceStakeCreated(stakeNftId, targetNftId, stakeOwner, dipAmount);
+
+        // collect staked dip by staking
+        $._staking.collectDipAmount(
+            stakeOwner,
+            dipAmount);
     }
 
 
     function stake(
         NftId stakeNftId,
-        Amount amount
+        Amount dipAmount
     )
         external
         virtual
+        restricted()
+        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()) {
+            emit LogStakingServiceStakeIncreased(stakeNftId, stakeOwner, dipAmount, stakeBalance);
+            $._staking.collectDipAmount(
+                stakeOwner,
+                dipAmount);
+        }
     }
 
 
-    function unstake(
+    function restakeToNewTarget(
         NftId stakeNftId,
-        Amount amount
+        NftId newTargetNftId
     )
         external
         virtual
+        restricted()
+        onlyNftOwner(stakeNftId)
+        returns (
+            NftId newStakeNftId
+        )
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        // TODO implement
+    } 
 
-    }
 
-    function close(
+    function updateRewards(
         NftId stakeNftId
     )
         external
         virtual
+        restricted()
     {
+        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()
+        onlyNftOwner(stakeNftId)
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
 
+        Amount rewardsClaimedAmount = $._staking.claimRewards(stakeNftId);
+        emit LogStakingServiceRewardsClaimed(stakeNftId, stakeOwner, rewardsClaimedAmount);
+        $._staking.transferDipAmount(
+            stakeOwner,
+            rewardsClaimedAmount);
     }
 
 
-    function increaseTotalValueLocked(
-        NftId targetNftId,
-        address token,
-        Amount amount
-    )
+    function unstake(NftId stakeNftId)
         external
         virtual
-        returns (Amount totalValueLocked)
+        restricted()
+        onlyNftOwner(stakeNftId)
     {
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        address stakeOwner = msg.sender;
 
+        (
+            Amount unstakedAmount,
+            Amount rewardsClaimedAmount
+        ) = $._staking.unstake(stakeNftId);
+
+        Amount totalAmount = unstakedAmount + rewardsClaimedAmount;
+        emit LogStakingServiceUnstaked(stakeNftId, stakeOwner, totalAmount);
+        $._staking.transferDipAmount(
+            stakeOwner,
+            totalAmount);
     }
 
 
-    function decreaseTotalValueLocked(
+    function setTotalValueLocked(
         NftId targetNftId,
         address token,
         Amount amount
     )
         external
         virtual
-        returns (Amount totalValueLocked)
+        restricted()
     {
+        // TODO implement
 
     }
 
+    //--- view functions ----------------------------------------------------//
 
-    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;
     }
 
 
@@ -229,6 +315,7 @@ contract StakingService is
         return _getStakingServiceStorage()._staking;
     }
 
+    //--- internal functions ------------------------------------------------//
 
     function _initialize(
         address owner, 
@@ -239,17 +326,20 @@ 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);
+        _registerInterface(type(IStakingService).interfaceId);
     }
 
 
@@ -286,9 +376,32 @@ contract StakingService is
     }
 
 
+    function _refillRewardReserves(NftId targetNftId, address rewardProvider, Amount dipAmount)
+        internal
+        virtual
+        returns (Amount newBalance)
+    {
+        // update reward reserve book keeping
+        StakingServiceStorage storage $ = _getStakingServiceStorage();
+        newBalance = $._staking.refillRewardReserves(targetNftId, dipAmount);
+
+        emit LogStakingServiceRewardReservesIncreased(targetNftId, rewardProvider, dipAmount, newBalance);
+
+        // collect reward dip from provider
+        $._staking.collectDipAmount(
+            rewardProvider,
+            dipAmount);
+    }
+
+
     function _getStakingServiceStorage() private pure returns (StakingServiceStorage storage $) {
         assembly {
             $.slot := STAKING_SERVICE_LOCATION_V1
         }
     }
+
+
+    function _getDomain() internal pure override returns(ObjectType) {
+        return STAKING();
+    }
 }

package/contracts/staking/StakingServiceManager.sol

@@ -2,8 +2,8 @@
 pragma solidity ^0.8.20;
 
 import {IRegistry} from "../registry/IRegistry.sol";
-import {IVersionable} from "../shared/IVersionable.sol";
-import {ProxyManager} from "../shared/ProxyManager.sol";
+import {IVersionable} from "../upgradeability/IVersionable.sol";
+import {ProxyManager} from "../upgradeability/ProxyManager.sol";
 import {StakingService} from "./StakingService.sol";
 
 contract StakingServiceManager is
@@ -14,6 +14,7 @@ contract StakingServiceManager is
 
     /// @dev initializes proxy manager with service implementation 
     constructor(
+        address authority,
         address registryAddress,
         bytes32 salt
     )
@@ -21,16 +22,15 @@ contract StakingServiceManager is
     {
         StakingService svc = new StakingService();
         bytes memory data = abi.encode(
+            authority, 
             registryAddress, 
             IRegistry(registryAddress).getStakingAddress());
-
         IVersionable versionable = deployDetermenistic(
             address(svc), 
             data,
             salt);
 
         _stakingService = StakingService(address(versionable));
-        _stakingService.registerProtocolTarget();
     }
 
     //--- view functions ----------------------------------------------------//