import StakingNFT from 0x1b77ba4b414de352 import NonFungibleToken from 0x1d7e57aa55817448 import SwapConfig from 0xb78ef7afa52ff906 import FungibleToken from 0xf233dcee88fe0abe import stFlowToken from 0xd6f80565193ad727 // pid + nftCollectionPath from config file // nftId from script transaction(pid: UInt64, nftCollectionPath: String, nftId: UInt64, ifClaimReward: Bool, vaultPaths: [String]) { prepare(signer: AuthAccount) { if signer.borrow<&FungibleToken.Vault>(from: /storage/stFlowTokenVault) == nil { signer.save(<-stFlowToken.createEmptyVault(), to: /storage/stFlowTokenVault) signer.link<&stFlowToken.Vault{FungibleToken.Receiver}>(/public/stFlowTokenReceiver, target: /storage/stFlowTokenVault) signer.link<&stFlowToken.Vault{FungibleToken.Balance}>(/public/stFlowTokenBalance, target: /storage/stFlowTokenVault) } let tokenVaults: {String: &FungibleToken.Vault} = {} if (ifClaimReward == true) { for path in vaultPaths { let vaultRef = signer.borrow<&FungibleToken.Vault>(from: StoragePath(identifier: path)!)! let key = SwapConfig.SliceTokenTypeIdentifierFromVaultType(vaultTypeIdentifier: vaultRef.getType().identifier) tokenVaults[key] = vaultRef } } 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) } let userCertificateCap = signer.getCapability<&StakingNFT.UserCertificate>(StakingNFT.UserCertificatePrivatePath) let userCollectionRef = signer.borrow<&NonFungibleToken.Collection>(from: StoragePath(identifier: nftCollectionPath)!) ?? panic("cannot borrow reference to user's staking nft collection") let poolCollectionRef = getAccount(StakingNFT.address).getCapability<&{StakingNFT.PoolCollectionPublic}>(StakingNFT.CollectionPublicPath).borrow() ?? panic("cannot borrow reference to StakingNFT.PoolCollectionPublic") let pool = poolCollectionRef.getPool(pid: pid) let nft <- pool.unstake(userCertificateCap: userCertificateCap, tokenId: nftId) userCollectionRef.deposit(token: <- nft) // rewards if (ifClaimReward == true) { let rewards <- pool.claimRewards(userCertificateCap: userCertificateCap) let keys = rewards.keys for key in keys { let rewardVaultRef = &rewards[key] as &FungibleToken.Vault? tokenVaults[key]!.deposit(from: <- rewardVaultRef!.withdraw(amount: rewardVaultRef!.balance)) } destroy rewards } } }

Transaction