import LockedTokens from 0x8d0e87b65159ae63 import FlowToken from 0x1654653399040a61 import FlowIDTableStaking from 0x8624b52f9ddcd04a import FlowStakingCollection from 0x8d0e87b65159ae63 transaction(){ var collectionRef: &FlowStakingCollection.StakingCollection? var delegatorRef: &FlowIDTableStaking.NodeDelegator? var stakerRef: &FlowIDTableStaking.NodeStaker? var nodeDelegatorProxy: LockedTokens.LockedNodeDelegatorProxy? var nodeStakerProxy: LockedTokens.LockedNodeStakerProxy? var lockupRef: &LockedTokens.TokenHolder? var flowTokenRef: &FlowToken.Vault? var accountType: Int var nodeID: String? var delegatorID: UInt32? var amount: UFix64 prepare(account: AuthAccount){ self.collectionRef = nil self.delegatorRef = nil self.stakerRef = nil self.nodeDelegatorProxy = nil self.nodeStakerProxy = nil self.lockupRef = nil self.flowTokenRef = nil self.nodeID = nil fun lookup(address: Address): Int{ if FlowStakingCollection.doesAccountHaveStakingCollection(address: address){return 1} let account = getAccount(address) let delegatorCap = getAccount(address).getCapability<&{FlowIDTableStaking.NodeDelegatorPublic}>(/public/flowStakingDelegator) if let delegatorRef = delegatorCap!.borrow(){return 2} let stakingCap = getAccount(address).getCapability<&{FlowIDTableStaking.NodeStakerPublic}>(FlowIDTableStaking.NodeStakerPublicPath) if let stakingRef = stakingCap!.borrow(){return 3} let lockedAccountInfoCap = account.getCapability<&LockedTokens.TokenHolder{LockedTokens.LockedAccountInfo}>(LockedTokens.LockedAccountInfoPublicPath) if let lockedAccountInfoRef = lockedAccountInfoCap!.borrow(){ if let delegatorID = lockedAccountInfoRef.getDelegatorID(){return 4} if let nodeID = lockedAccountInfoRef.getNodeID(){return 5} } return 0 } self.amount = UFix64(00000027913.75695719) self.accountType = lookup(address: account.address) self.delegatorID = nil switch self.accountType{ case 1: // Staking collection self.collectionRef = account.borrow<&FlowStakingCollection.StakingCollection>(from: FlowStakingCollection.StakingCollectionStoragePath) ?? panic("Could not borrow ref to StakingCollection") let nodeIDVector = FlowStakingCollection.getNodeIDs(address: account.address) if nodeIDVector.length == 1 { self.nodeID = FlowStakingCollection.getNodeIDs(address: account.address)[0] } else if nodeIDVector.length > 1 { panic("Expected a single NodeID in the StakingCollection") } else { let delegatorIDVector = FlowStakingCollection.getDelegatorIDs(address: account.address) if delegatorIDVector.length == 0 { panic("No delegation or staking found in StakingCollection for this account") } else if delegatorIDVector.length > 1 { panic("Expected a single DelegatorID in the StakingCollection") } else { self.nodeID = delegatorIDVector[0].delegatorNodeID self.delegatorID = delegatorIDVector[0].delegatorID } } case 2: // unlocked delegation self.delegatorRef = account.borrow<&FlowIDTableStaking.NodeDelegator>(from: FlowIDTableStaking.DelegatorStoragePath) ?? panic("Could not borrow reference to delegator") self.flowTokenRef = account.borrow<&FlowToken.Vault>(from: /storage/flowTokenVault) ?? panic("Could not borrow reference to FLOW Vault") case 3: // unlocked staking self.stakerRef = account.borrow<&FlowIDTableStaking.NodeStaker>(from: FlowIDTableStaking.NodeStakerStoragePath) ?? panic("Could not borrow reference to staking admin") self.flowTokenRef = account.borrow<&FlowToken.Vault>(from: /storage/flowTokenVault) ?? panic("Could not borrow reference to FLOW Vault") case 4: // locked delegation self.lockupRef = account.borrow<&LockedTokens.TokenHolder>(from: LockedTokens.TokenHolderStoragePath) ?? panic("Could not borrow reference to TokenHolder") self.nodeDelegatorProxy = self.lockupRef!.borrowDelegator() self.flowTokenRef = account.borrow<&FlowToken.Vault>(from: /storage/flowTokenVault) ?? panic("Could not borrow reference to FLOW Vault") case 5: // locked staking self.lockupRef = account.borrow<&LockedTokens.TokenHolder>(from: LockedTokens.TokenHolderStoragePath) ?? panic("Could not borrow reference to TokenHolder") self.nodeStakerProxy = self.lockupRef!.borrowStaker() self.flowTokenRef = account.borrow<&FlowToken.Vault>(from: /storage/flowTokenVault) ?? panic("Could not borrow reference to FLOW Vault") default: panic("Could not withdraw rewards for this account") } } execute{ switch self.accountType{ case 1: self.collectionRef!.withdrawRewardedTokens(nodeID: self.nodeID!, delegatorID: self.delegatorID, amount: self.amount) case 2: self.flowTokenRef!.deposit(from: <-self.delegatorRef!.withdrawRewardedTokens(amount: self.amount)) case 3: self.flowTokenRef!.deposit(from: <-self.stakerRef!.withdrawRewardedTokens(amount: self.amount)) case 4: self.nodeDelegatorProxy!.withdrawRewardedTokens(amount: self.amount) self.flowTokenRef!.deposit(from: <-self.lockupRef!.withdraw(amount: self.amount)) case 5: self.nodeStakerProxy!.withdrawRewardedTokens(amount: self.amount) self.flowTokenRef!.deposit(from: <-self.lockupRef!.withdraw(amount: self.amount)) default: panic("Could not withdraw rewards for this account") } } }

Transaction