🐛 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

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, FruniqueRole::Admin)?;
+					Ok(())
+				},
 			}
 		}
 

pallets/afloat/src/types.rs

@@ -197,6 +197,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: AfloatRole },
 }
 
 // ! Transaction structures