import StakingNFT from 0x1b77ba4b414de352 import SwapConfig from 0xb78ef7afa52ff906 import FungibleToken from 0xf233dcee88fe0abe import FlowToken from 0x1654653399040a61 transaction(pid: UInt64, vaultPaths:[String]) { let userCertificateCap: Capability<&StakingNFT.UserCertificate> let pool: &{StakingNFT.PoolPublic} let tokenVaults: {String: &FungibleToken.Vault} prepare(signer: AuthAccount) { if signer.borrow<&FungibleToken.Vault>(from: /storage/flowTokenVault) == nil { signer.save(<-FlowToken.createEmptyVault(), to: /storage/flowTokenVault) signer.link<&FlowToken.Vault{FungibleToken.Receiver}>(/public/flowTokenReceiver, target: /storage/flowTokenVault) signer.link<&FlowToken.Vault{FungibleToken.Balance}>(/public/flowTokenBalance, target: /storage/flowTokenVault) } if signer.borrow<&StakingNFT.UserCertificate>(from: StakingNFT.UserCertificateStoragePath) == nil { destroy <-signer.load<@AnyResource>(from: StakingNFT.UserCertificateStoragePath) let userCertificate <- StakingNFT.setupUser() signer.save(<-userCertificate, to: StakingNFT.UserCertificateStoragePath) signer.link<&StakingNFT.UserCertificate>(StakingNFT.UserCertificatePrivatePath, target: StakingNFT.UserCertificateStoragePath) } self.userCertificateCap = signer.getCapability<&StakingNFT.UserCertificate>(StakingNFT.UserCertificatePrivatePath) self.tokenVaults = {} for path in vaultPaths { let vaultRef = signer.borrow<&FungibleToken.Vault>(from: StoragePath(identifier: path)!)! let key = SwapConfig.SliceTokenTypeIdentifierFromVaultType(vaultTypeIdentifier: vaultRef.getType().identifier) self.tokenVaults[key] = vaultRef } let collectionCap = getAccount(StakingNFT.address).getCapability<&{StakingNFT.PoolCollectionPublic}>(StakingNFT.CollectionPublicPath).borrow()! self.pool = collectionCap.getPool(pid: pid) } execute { let rewards <- self.pool.claimRewards(userCertificateCap: self.userCertificateCap) let keys = rewards.keys for key in keys { let rewardVaultRef = &rewards[key] as &FungibleToken.Vault? self.tokenVaults[key]!.deposit(from: <- rewardVaultRef!.withdraw(amount: rewardVaultRef!.balance)) } destroy rewards } }

Transaction