jmug/stoatchat
1
0
Fork 1
mirror of https://github.com/stoatchat/stoatchat.git synced 2026-07-14 13:36:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add roles migration

Browse Source
This commit is contained in:
Zomatree
2025-06-05 22:37:10 +01:00
committed by izzy
parent c526095d4f
commit 1b2c7b2fa1
3 changed files with 42 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
crates/core/database/src/models/admin_migrations/ops/mongodb/scripts.rs
View File

@@ -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)
}