import FlowStakingCollection from 0x8d0e87b65159ae63 import FlowIDTableStaking from 0x8624b52f9ddcd04a import FungibleToken from 0xf233dcee88fe0abe import FlowToken from 0x1654653399040a61 access(all) contract StakingRewardDelegatee { access(all) let StakingRewardDelegateePath: StoragePath access(all) resource interface Withdrawer { pub fun withdrawAllRewards(): @FlowToken.Vault pub fun withdrawSingleReward(nodeID: String, delegatorID: UInt32): @FlowToken.Vault } pub resource interface Restaker { pub fun restakeAllRewards(): Void pub fun restakeSingleReward(nodeID: String, delegatorID: UInt32): Void } pub resource Delegatee: Withdrawer, Restaker { access(self) let collectionCap: Capability<&FlowStakingCollection.StakingCollection> access(self) let unlockedVault: Capability<&FlowToken.Vault> init(collectionCap: Capability<&FlowStakingCollection.StakingCollection>, unlockedVault: Capability<&FlowToken.Vault>) { self.collectionCap = collectionCap self.unlockedVault = unlockedVault } // Withdraw all rewards from the staking collection and return the vault pub fun withdrawAllRewards(): @FlowToken.Vault { let collection = self.collectionCap.borrow() ?? panic("Could not borrow collection capability") var totalAmountWithdrawn = 0.0 for nodeID in collection.getDelegatorIDs() { let delegatorInfo = FlowIDTableStaking.DelegatorInfo(nodeID: nodeID.delegatorNodeID, delegatorID: nodeID.delegatorID) if delegatorInfo.tokensRewarded == 0.0 { continue } totalAmountWithdrawn = totalAmountWithdrawn + delegatorInfo.tokensRewarded collection.withdrawRewardedTokens(nodeID: nodeID.delegatorNodeID, delegatorID: nodeID.delegatorID, amount: delegatorInfo.tokensRewarded) } let vault = self.unlockedVault.borrow() ?? panic("Could not borrow unlocked vault capability") return <- (vault.withdraw(amount: totalAmountWithdrawn) as! @FlowToken.Vault) } // Withdraw a single reward from the staking collection and return the vault access(all) fun withdrawSingleReward(nodeID: String, delegatorID: UInt32): @FlowToken.Vault { let collection = self.collectionCap.borrow() ?? panic("Could not borrow collection capability") let delegatorInfo = FlowIDTableStaking.DelegatorInfo(nodeID: nodeID, delegatorID: delegatorID) if delegatorInfo.tokensRewarded == 0.0 { return <- (FlowToken.createEmptyVault() as! @FlowToken.Vault) } collection.withdrawRewardedTokens(nodeID: nodeID, delegatorID: delegatorID, amount: delegatorInfo.tokensRewarded) let vault = self.unlockedVault.borrow() ?? panic("Could not borrow unlocked vault capability") return <- (vault.withdraw(amount: delegatorInfo.tokensRewarded) as! @FlowToken.Vault) } // Restake all rewards in the staking collection access(all) fun restakeAllRewards(): Void { let collection = self.collectionCap.borrow() ?? panic("Could not borrow collection capability") for delegatorID in collection.getDelegatorIDs() { let delegatorInfo = FlowIDTableStaking.DelegatorInfo(nodeID: delegatorID.delegatorNodeID, delegatorID: delegatorID.delegatorID) if delegatorInfo.tokensRewarded == 0.0 { continue } collection.withdrawRewardedTokens(nodeID: delegatorID.delegatorNodeID, delegatorID: delegatorID.delegatorID, amount: delegatorInfo.tokensRewarded) } } // Restake a single reward in the staking collection access(all) fun restakeSingleReward(nodeID: String, delegatorID: UInt32): Void { let collection = self.collectionCap.borrow() ?? panic("Could not borrow collection capability") let delegatorInfo = FlowIDTableStaking.DelegatorInfo(nodeID: nodeID, delegatorID: delegatorID) if delegatorInfo.tokensRewarded == 0.0 { return } collection.stakeRewardedTokens(nodeID: nodeID, delegatorID: delegatorID, amount: delegatorInfo.tokensRewarded) } } // This creates a new delegatee resource which can be used to withdraw and restake rewards // You can restrict this type to allow either withdrawing or restaking only // You MUST use the correct unlocked vault for this to work access(all) fun createDelegatee( collectionCap: Capability<&FlowStakingCollection.StakingCollection>, unlockedVault: Capability<&FlowToken.Vault> ): @Delegatee { return <- create Delegatee(collectionCap: collectionCap, unlockedVault: unlockedVault) } init() { self.StakingRewardDelegateePath = /storage/StakingRewardWithdrawDelegatee } }

🎉Happy diving into the flow

Transaction

🎉Happy diving into the flow