forked from jmug/stoatchat
feat: add roles migration
This commit is contained in:
@@ -4,8 +4,7 @@ use crate::{
|
||||
mongodb::{
|
||||
bson::{doc, from_bson, from_document, to_document, Bson, DateTime, Document},
|
||||
options::FindOptions,
|
||||
},
|
||||
AbstractChannels, AbstractServers, Channel, Invite, MongoDb, User, DISCRIMINATOR_SEARCH_SPACE,
|
||||
}, AbstractChannels, AbstractServers, Channel, Invite, MongoDb, Server, User, DISCRIMINATOR_SEARCH_SPACE
|
||||
};
|
||||
use bson::{oid::ObjectId, to_bson};
|
||||
use futures::StreamExt;
|
||||
@@ -22,7 +21,7 @@ struct MigrationInfo {
|
||||
revision: i32,
|
||||
}
|
||||
|
||||
pub const LATEST_REVISION: i32 = 41; // MUST BE +1 to last migration
|
||||
pub const LATEST_REVISION: i32 = 42; // MUST BE +1 to last migration
|
||||
|
||||
pub async fn migrate_database(db: &MongoDb) {
|
||||
let migrations = db.col::<Document>("migrations");
|
||||
@@ -1165,6 +1164,43 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
|
||||
.expect("failed to update users");
|
||||
}
|
||||
|
||||
if revision <= 41 {
|
||||
info!("Running migration [revision 41 / 05-06-2025]: convert role ranks to uniform numbers.");
|
||||
|
||||
let mut servers = db.db()
|
||||
.collection::<Server>("servers")
|
||||
.find(doc! {
|
||||
"roles": {
|
||||
"$exists": true,
|
||||
"$ne": []
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap()
|
||||
.filter_map(|s| async { s.ok() })
|
||||
.boxed();
|
||||
|
||||
while let Some(mut server) = servers.next().await {
|
||||
let mut ordered_roles = server.roles.clone().into_iter().collect::<Vec<_>>();
|
||||
ordered_roles.sort_by(|(_, role_a), (_, role_b)| role_a.rank.cmp(&role_b.rank));
|
||||
let ordered_roles = ordered_roles.into_iter().map(|(id, _)| id).collect::<Vec<_>>();
|
||||
|
||||
for (id, role) in server.roles.iter_mut() {
|
||||
role.rank = ordered_roles.iter().position(|x| id == x).unwrap() as i64;
|
||||
}
|
||||
|
||||
db.db().collection::<Server>("servers")
|
||||
.update_one(
|
||||
doc! { "_id": &server.id },
|
||||
doc! { "$set": {
|
||||
"roles": bson::to_bson(&server.roles).unwrap()
|
||||
} }
|
||||
)
|
||||
.await
|
||||
.unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
// Reminder to update LATEST_REVISION when adding new migrations.
|
||||
LATEST_REVISION.max(revision)
|
||||
}
|
||||
|
||||
@@ -49,6 +49,6 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
|
||||
permissions_set::set_role_permission,
|
||||
permissions_set_default::set_default_permissions,
|
||||
emoji_list::list_emoji,
|
||||
roles_edit_positions::edit_role_positions
|
||||
roles_edit_positions::edit_role_ranks
|
||||
]
|
||||
}
|
||||
|
||||
@@ -8,7 +8,7 @@ use revolt_models::v0;
|
||||
///
|
||||
/// Edit's server role's ranks.
|
||||
#[openapi(tag = "Server Permissions")]
|
||||
#[patch("/<target>/roles/ranks", data = "<data>")]
|
||||
#[patch("/<target>/roles/ranks", data = "<data>", rank = 1)]
|
||||
pub async fn edit_role_ranks(
|
||||
db: &State<Database>,
|
||||
user: User,
|
||||
@@ -36,7 +36,7 @@ pub async fn edit_role_ranks(
|
||||
// find all roles above the member which we should not be able to reorder
|
||||
let cant_modify = server.roles.clone()
|
||||
.into_iter()
|
||||
.filter(|(_, role)| if let Some(top_rank) = member_top_rank { role.rank >= top_rank } else { true })
|
||||
.filter(|(_, role)| if let Some(top_rank) = member_top_rank { role.rank <= top_rank } else { true })
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
// check if any roles which we cant reorder have tried to been reordered
|
||||
|
||||
Reference in New Issue
Block a user