Update buy_packs_primary_sale.cdc (#88)

* Update buy_packs_primary_sale.cdc

* Update buy_packs_primary_sale.cdc

* Update buy_packs_primary_sale.cdc

* Update buy_packs_primary_sale.cdc

Author: Deewai

atlas/transactions/user/buy_packs_primary_sale.cdc

@@ -1,7 +1,7 @@
 import FungibleToken from 0x{{.FungibleTokenContractAddress}}
 import NonFungibleToken from 0x{{.NonFungibleTokenContractAddress}}
 import DapperUtilityCoin from 0x{{.DapperUtilityCoinContractAddress}}
-import PackNFT from 0x{{.NFTContractAddress}}
+import PackNFT, {{.NFTProductName}} from 0x{{.NFTContractAddress}}
 import NFTStorefront from 0x{{.NFTStorefrontV1ContractAddress}}
 
 /// Transaction facilitates the purchase of listed NFT.
@@ -14,17 +14,43 @@ import NFTStorefront from 0x{{.NFTStorefrontV1ContractAddress}}
 
 transaction() {
     let paymentVault: @{FungibleToken.Vault}
-    let PackNFTCollection: &{NonFungibleToken.CollectionPublic}
+    let PackNFTCollection: &PackNFT.Collection
     let storefront: &NFTStorefront.Storefront
     let listings: [&{NFTStorefront.ListingPublic}]
 
-    prepare(universalDucPayer: auth(Storage, Capabilities) &Account) {
+    prepare(universalDucPayer: auth(Storage, Capabilities) &Account, user: auth(Storage, Capabilities) &Account) {
         let sellerAddress: Address = 0x{{.StorefrontAddress}}
         let buyerAddress: Address = 0x{{.RecipientAddress}}
         let listingIds: [UInt64] = [{{.ListingIds}}]
         let buyer = getAccount(buyerAddress)
 
         assert(sellerAddress != buyerAddress, message : "Buyer and seller can not be same")
+
+        // Ensure the user has an {{.NFTProductName}} collection set up
+        if user.storage.borrow<&{{.NFTProductName}}.Collection>(from: {{.NFTProductName}}.CollectionStoragePath) == nil {
+            // Create a new collection and save it to the account storage
+            user.storage.save(<- {{.NFTProductName}}.createEmptyCollection(nftType: Type<@{{.NFTProductName}}.NFT>()), to: {{.NFTProductName}}.CollectionStoragePath)
+
+            // Create a public capability for the collection
+            user.capabilities.unpublish({{.NFTProductName}}.CollectionPublicPath)
+            user.capabilities.publish(
+                user.capabilities.storage.issue<&{{.NFTProductName}}.Collection>({{.NFTProductName}}.CollectionStoragePath),
+                at: {{.NFTProductName}}.CollectionPublicPath
+            )
+        }
+        // Ensure the user has a PackNFT collection set up
+        if user.storage.borrow<&PackNFT.Collection>(from: PackNFT.CollectionStoragePath) == nil {
+            // Create a new collection and save it to the account storage
+            user.storage.save(<- PackNFT.createEmptyCollection(nftType: Type<@PackNFT.NFT>()), to: PackNFT.CollectionStoragePath)
+
+            // Create a public capability for the collection
+            user.capabilities.unpublish(PackNFT.CollectionPublicPath)
+            user.capabilities.publish(
+                user.capabilities.storage.issue<&PackNFT.Collection>(PackNFT.CollectionStoragePath),
+                at: PackNFT.CollectionPublicPath
+            )
+        }
+
         // Access the storefront public resource of the seller to purchase the listing.
         self.storefront = getAccount(sellerAddress)
             .capabilities.borrow<&NFTStorefront.Storefront>(
@@ -51,7 +77,7 @@ transaction() {
         self.paymentVault <- provider.withdraw(amount: salePrice)
 
         // Access the buyer's NFT collection to store the purchased NFT.
-        self.PackNFTCollection = buyer.capabilities.borrow<&{NonFungibleToken.CollectionPublic}>(PackNFT.CollectionPublicPath)
+        self.PackNFTCollection = buyer.capabilities.borrow<&PackNFT.Collection>(PackNFT.CollectionPublicPath)
         ?? panic("Could not borrow Storefront from provided address")
 
     }