add missing permissions in afloat (#18)

* 🐛 fix(functions.rs): add call to do_add_account_to_afloat_frunique to ensure consistency with admin role assignment
✨ feat(lib.rs): add support for adding afloat and frunique roles during initial setup
🔧 chore(types.rs): add AddAfloatRole and AddFruniqueRole variants to InitialSetupArgs enum for role assignment

* 🐛 fix(afloat): fix incorrect role parameter in do_add_account_to_afloat_frunique function call
🔧 chore(types): import FruniqueRole from pallet_fruniques to fix type error in InitialSetupArgs

* 🔥 refactor(lib.rs): remove unused `spam_spawning` function from the `pallet` module

The `spam_spawning` function was commented out and not being used. Removing it to improve code cleanliness and readability.

Author: tlacloc

pallets/afloat/src/functions.rs

@@ -856,6 +856,8 @@ impl<T: Config> Pallet<T> {
 
 		Self::give_role_to_user(user_address.clone(), AfloatRole::Admin)?;
 
+		Self::do_add_account_to_afloat_frunique(user_address.clone(), FruniqueRole::Admin)?;
+
 		Self::deposit_event(Event::AdminAdded(authority, user_address));
 		Ok(())
 	}

pallets/afloat/src/lib.rs

@@ -223,6 +223,14 @@ pub mod pallet {
 					Self::do_setup_roles(creator, admin)?;
 					Ok(())
 				},
+				InitialSetupArgs::AddAfloatRole { who, role } => {
+					Self::give_role_to_user(who, role)?;
+					Ok(())
+				},
+				InitialSetupArgs::AddFruniqueRole { who, role } => {
+					Self::do_add_account_to_afloat_frunique(who, role)?;
+					Ok(())
+				},
 			}
 		}
 

pallets/afloat/src/types.rs

@@ -1,5 +1,6 @@
 use super::*;
 use frame_support::{pallet_prelude::*, sp_io::hashing::blake2_256};
+use pallet_fruniques::types::FruniqueRole;
 use sp_runtime::sp_std::vec::Vec;
 
 pub type ShortString = BoundedVec<u8, ConstU32<55>>;
@@ -197,6 +198,8 @@ pub enum CreateAsset<T: Config> {
 pub enum InitialSetupArgs<T: Config> {
 	All { creator: T::AccountId, admin: T::AccountId, asset: CreateAsset<T> },
 	Roles { creator: T::AccountId, admin: T::AccountId },
+	AddAfloatRole { who: T::AccountId, role: AfloatRole },
+	AddFruniqueRole { who: T::AccountId, role: FruniqueRole },
 }
 
 // ! Transaction structures

pallets/fruniques/src/lib.rs

@@ -509,19 +509,5 @@ pub mod pallet {
 			T::Rbac::remove_pallet_storage(Self::pallet_id())?;
 			Ok(())
 		}
-
-		#[pallet::call_index(10)]
-		#[pallet::weight(Weight::from_parts(10_000,0) + T::DbWeight::get().writes(1))]
-		pub fn spam_spawning(
-			origin: OriginFor<T>,
-			class_id: T::CollectionId,
-			instances: u32,
-		) -> DispatchResult {
-			let _ = ensure_signed(origin)?;
-			// for instance in instances {
-			// 	let _ = Self::mint(instance, class_id, None, None, None, None, None, None, None)?;
-			// }
-			Ok(())
-		}
 	}
 }