fix: continue fix

revoltchat · May 15, 2024 · 878c1a8 · 878c1a8
1 parent 471e0f5
commit 878c1a8
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/crates/core/database/src/models/admin_migrations/ops/mongodb/scripts.rs b/crates/core/database/src/models/admin_migrations/ops/mongodb/scripts.rs
@@ -6,6 +6,7 @@ use crate::{
         options::FindOptions,
     }, Invite, MongoDb, DISCRIMINATOR_SEARCH_SPACE
 };
+use bson::oid::ObjectId;
 use futures::StreamExt;
 use rand::seq::SliceRandom;
 use revolt_permissions::DEFAULT_WEBHOOK_PERMISSIONS;
@@ -1003,16 +1004,23 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
             }
         );
 
+        #[derive(serde::Serialize, serde::Deserialize)]
+        struct Outer {
+            _id: ObjectId,
+            #[serde(flatten)]
+            invite: OldInvite
+        }
+
         let mut invites = db.db()
-            .collection::<OldInvite>("channel_invites")
+            .collection::<Outer>("channel_invites")
             .find(doc! {
                 "type": { "$exists": false }
             }, None)
             .await
             .expect("failed to find invites");
 
         while let Some(Ok(invite)) = invites.next().await {
-            let new_invite = match invite {
+            let new_invite = match invite.invite {
                 OldInvite::Server { code, server, creator, channel } => Invite::Server { code, server, creator, channel },
                 OldInvite::Group { code, creator, channel } => Invite::Group { code, creator, channel }
             };