add validation to know if a collection exists

Author: tlacloc

pallets/fruniques/src/functions.rs

@@ -60,6 +60,13 @@ impl<T: Config> Pallet<T> {
 		pallet_uniques::Pallet::<T>::owner(*class_id, *instance_id)
 	}
 
+	pub fn collection_exists(class_id: &T::CollectionId) -> bool {
+		if let Some(_owner) = pallet_uniques::Pallet::<T>::collection_owner(*class_id) {
+			return true;
+		}
+		false
+	}
+
 	pub fn set_attribute(
 		origin: OriginFor<T>,
 		class_id: &T::CollectionId,
@@ -178,6 +185,7 @@ impl<T: Config> Pallet<T> {
 		attributes: Option<Vec<(BoundedVec<u8, T::KeyLimit>, BoundedVec<u8, T::ValueLimit>)>>,
 	) -> DispatchResult {
 
+		ensure!(Self::collection_exists(&collection),  <Error<T>>::CollectionNotFound);
 		pallet_uniques::Pallet::<T>::mint(
 			origin.clone(),
 			collection,
@@ -196,12 +204,6 @@ impl<T: Config> Pallet<T> {
 				)?;
 			}
 		}
-		// let instance_id: T::ItemId = Self::u32_to_instance_id(input);
-
-		// pallet_uniques::Pallet::<T>::mint(origin,
-		// 	Self::u32_to_class_id(class_id),
-		// 	Self::u32_to_instance_id(instance_id),
-		// 	owner)?;
 
 		Ok(())
 	}

pallets/fruniques/src/lib.rs

@@ -66,6 +66,8 @@ pub mod pallet {
 		ValueTooLong,
 		// Calling set on a non-existing attributes
 		AttributesEmpty,
+		// The collection doenst exist
+		CollectionNotFound,
 	}
 
 	#[pallet::storage]
@@ -116,6 +118,15 @@ pub mod pallet {
 	where
 		T: pallet_uniques::Config<CollectionId = CollectionId, ItemId = ItemId>,
 	{
+		#[transactional]
+		#[pallet::weight(10_000 + T::DbWeight::get().writes(10))]
+		pub fn initial_setup(
+			origin: OriginFor<T>,
+		) -> DispatchResult {
+			T::RemoveOrigin::ensure_origin(origin.clone())?;
+			// Self::do_initial_setup()?;
+			Ok(())
+		}
 
 		#[pallet::weight(10_000 + T::DbWeight::get().writes(1))]
 		pub fn create_collection(