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: auth(Storage, Capabilities) &Account) { if signer.storage.borrow<&{FungibleToken.Vault}>(from: /storage/stFlowTokenVault) == nil { signer.storage.save(<-stFlowToken.createEmptyVault(vaultType:Type<@stFlowToken.Vault>()), to: /storage/stFlowTokenVault) signer.capabilities.publish( signer.capabilities.storage.issue<&{FungibleToken.Receiver}>(/storage/stFlowTokenVault), at: /public/stFlowTokenReceiver ) signer.capabilities.publish( signer.capabilities.storage.issue<&{FungibleToken.Balance}>(/storage/stFlowTokenVault), at: /public/stFlowTokenBalance ) } let tokenVaults: {String: &{FungibleToken.Vault}} = {} if (ifClaimReward == true) { for path in vaultPaths { let vaultRef = signer.storage.borrow<&{FungibleToken.Vault}>(from: StoragePath(identifier: path)!)! let key = SwapConfig.SliceTokenTypeIdentifierFromVaultType(vaultTypeIdentifier: vaultRef.getType().identifier) tokenVaults[key] = vaultRef } } if signer.storage.borrow<&StakingNFT.UserCertificate>(from: StakingNFT.UserCertificateStoragePath) == nil { destroy <-signer.storage.load<@AnyResource>(from: StakingNFT.UserCertificateStoragePath) let userCertificate <- StakingNFT.setupUser() signer.storage.save(<-userCertificate, to: StakingNFT.UserCertificateStoragePath) } let userCertificate = signer.storage.borrow<&StakingNFT.UserCertificate>(from: StakingNFT.UserCertificateStoragePath) ?? panic("cannot borrow reference to StakingNFT.UserCertificate") let userCollectionRef = signer.storage.borrow<&{NonFungibleToken.Collection}>(from: StoragePath(identifier: nftCollectionPath)!) ?? panic("cannot borrow reference to user's staking nft collection") let poolCollectionRef = getAccount(StakingNFT.address).capabilities.borrow<&{StakingNFT.PoolCollectionPublic}>(StakingNFT.CollectionPublicPath) ?? panic("cannot borrow reference to StakingNFT.PoolCollectionPublic") let pool = poolCollectionRef.getPool(pid: pid) let nft <- pool.unstake(userCertificate: userCertificate, tokenId: nftId) userCollectionRef.deposit(token: <- nft) // rewards if (ifClaimReward == true) { let rewards <- pool.claimRewards(userCertificate: userCertificate) let keys = rewards.keys for key in keys { let rewardVaultRef = &rewards[key] as auth(FungibleToken.Withdraw) &{FungibleToken.Vault}? tokenVaults[key]!.deposit(from: <- rewardVaultRef!.withdraw(amount: rewardVaultRef!.balance)) } destroy rewards } } }

Transaction