feat: add roles migration
This commit is contained in:
@@ -4,8 +4,7 @@ use crate::{
|
|||||||
mongodb::{
|
mongodb::{
|
||||||
bson::{doc, from_bson, from_document, to_document, Bson, DateTime, Document},
|
bson::{doc, from_bson, from_document, to_document, Bson, DateTime, Document},
|
||||||
options::FindOptions,
|
options::FindOptions,
|
||||||
},
|
}, AbstractChannels, AbstractServers, Channel, Invite, MongoDb, Server, User, DISCRIMINATOR_SEARCH_SPACE
|
||||||
AbstractChannels, AbstractServers, Channel, Invite, MongoDb, User, DISCRIMINATOR_SEARCH_SPACE,
|
|
||||||
};
|
};
|
||||||
use bson::{oid::ObjectId, to_bson};
|
use bson::{oid::ObjectId, to_bson};
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
@@ -22,7 +21,7 @@ struct MigrationInfo {
|
|||||||
revision: i32,
|
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) {
|
pub async fn migrate_database(db: &MongoDb) {
|
||||||
let migrations = db.col::<Document>("migrations");
|
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");
|
.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.
|
// Reminder to update LATEST_REVISION when adding new migrations.
|
||||||
LATEST_REVISION.max(revision)
|
LATEST_REVISION.max(revision)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -49,6 +49,6 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
|
|||||||
permissions_set::set_role_permission,
|
permissions_set::set_role_permission,
|
||||||
permissions_set_default::set_default_permissions,
|
permissions_set_default::set_default_permissions,
|
||||||
emoji_list::list_emoji,
|
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.
|
/// Edit's server role's ranks.
|
||||||
#[openapi(tag = "Server Permissions")]
|
#[openapi(tag = "Server Permissions")]
|
||||||
#[patch("/<target>/roles/ranks", data = "<data>")]
|
#[patch("/<target>/roles/ranks", data = "<data>", rank = 1)]
|
||||||
pub async fn edit_role_ranks(
|
pub async fn edit_role_ranks(
|
||||||
db: &State<Database>,
|
db: &State<Database>,
|
||||||
user: User,
|
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
|
// find all roles above the member which we should not be able to reorder
|
||||||
let cant_modify = server.roles.clone()
|
let cant_modify = server.roles.clone()
|
||||||
.into_iter()
|
.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<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
// check if any roles which we cant reorder have tried to been reordered
|
// check if any roles which we cant reorder have tried to been reordered
|
||||||
|
|||||||
Reference in New Issue
Block a user