refactor: minor formatting changes

This commit is contained in:
izzy
2025-06-08 10:37:38 +01:00
parent 77daf82b94
commit 4e4e598daf
4 changed files with 44 additions and 28 deletions

View File

@@ -254,12 +254,12 @@ impl Server {
} }
} }
/// reorders the server's roles rankings /// Reorders the server's roles rankings
pub async fn update_role_rankings(&mut self, db: &Database, new_order: Vec<String>) -> Result<()> { pub async fn set_role_ordering(&mut self, db: &Database, new_order: Vec<String>) -> Result<()> {
// new_order must always contain every role // New order must always contain every role
debug_assert_eq!(self.roles.len(), new_order.len()); debug_assert_eq!(self.roles.len(), new_order.len());
// set the role's ranks to the positions in the vec // Set the role's ranks to the positions in the vec
for (rank, id) in new_order.iter().enumerate() { for (rank, id) in new_order.iter().enumerate() {
self.roles.get_mut(id).unwrap().rank = rank as i64; self.roles.get_mut(id).unwrap().rank = rank as i64;
} }
@@ -270,13 +270,14 @@ impl Server {
roles: Some(self.roles.clone()), roles: Some(self.roles.clone()),
..Default::default() ..Default::default()
}, },
Vec::new() Vec::new(),
).await?; )
.await?;
// publish bulk update event // Publish bulk update event
EventV1::ServerRoleRanksUpdate { EventV1::ServerRoleRanksUpdate { ranks: new_order }
ranks: new_order .p(self.id.clone())
}.p(self.id.clone()).await; .await;
Ok(()) Ok(())
} }

View File

@@ -267,8 +267,7 @@ auto_derived!(
pub hoist: Option<bool>, pub hoist: Option<bool>,
/// Ranking position /// Ranking position
/// ///
/// **Deprecated** - use the edit server role positions route. /// **Removed** - no effect, use the edit server role positions route
/// Smaller values take priority.
pub rank: Option<i64>, pub rank: Option<i64>,
/// Fields to remove from role object /// Fields to remove from role object
#[cfg_attr(feature = "validator", validate(length(min = 1)))] #[cfg_attr(feature = "validator", validate(length(min = 1)))]

View File

@@ -46,7 +46,6 @@ pub async fn edit(
colour, colour,
hoist, hoist,
remove, remove,
// rank is deprecated
.. ..
} = data; } = data;

View File

@@ -1,8 +1,11 @@
use revolt_database::{events::client::EventV1, util::{permissions::DatabasePermissionQuery, reference::Reference}, Database, PartialRole, PartialServer, User}; use revolt_database::{
use revolt_permissions::{calculate_server_permissions, ChannelPermission}; util::{permissions::DatabasePermissionQuery, reference::Reference},
use rocket::{serde::json::Json, State}; Database, User,
use revolt_result::{create_error, Result}; };
use revolt_models::v0; use revolt_models::v0;
use revolt_permissions::{calculate_server_permissions, ChannelPermission};
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State};
/// # Edits server roles ranks /// # Edits server roles ranks
/// ///
@@ -26,27 +29,41 @@ pub async fn edit_role_ranks(
let existing_order = server.roles.keys().cloned().collect::<Vec<_>>(); let existing_order = server.roles.keys().cloned().collect::<Vec<_>>();
let new_order = data.ranks.clone().into_iter().collect::<Vec<_>>(); let new_order = data.ranks.clone().into_iter().collect::<Vec<_>>();
// verify all roles are in the new ordering // Verify all roles are in the new ordering
if data.ranks.len() != server.roles.len() && server.roles.iter().all(|(id, _)| data.ranks.contains(id)) { if data.ranks.len() != server.roles.len()
return Err(create_error!(InvalidOperation)) && server.roles.iter().all(|(id, _)| data.ranks.contains(id))
{
return Err(create_error!(InvalidOperation));
} }
// dont have to check what the user cant modify if they are the server owner // Don't have to check what the user cant modify if they are the server owner
if server.owner != user.id { if server.owner != user.id {
let member_top_rank = query.get_member_rank(); let member_top_rank = query.get_member_rank();
// find all roles above the member which we should not be able to reorder then if server
// check if any roles which we cant reorder have tried to been reordered .roles
if server.roles
.iter() .iter()
.filter(|(_, role)| if let Some(top_rank) = member_top_rank { role.rank <= top_rank } else { true }) // Find all roles above the member which we should not be able to reorder
.any(|(id, _)| existing_order.iter().position(|existing_id| id == existing_id) != new_order.iter().position(|new_id| id == new_id)) .filter(|(_, role)| {
if let Some(top_rank) = member_top_rank {
role.rank <= top_rank
} else {
true
}
})
// Check if user is trying to reorder roles they can't reorder (as found previously)
.any(|(id, _)| {
existing_order
.iter()
.position(|existing_id| id == existing_id)
!= new_order.iter().position(|new_id| id == new_id)
})
{ {
return Err(create_error!(NotElevated)) return Err(create_error!(NotElevated));
} }
} }
server.update_role_rankings(db, new_order).await?; server.set_role_ordering(db, new_order).await?;
Ok(Json(server.into())) Ok(Json(server.into()))
} }