Migrate to Autumn 1.0.0

src/database/entities/autumn.rs

@@ -4,6 +4,7 @@ use serde::{Deserialize, Serialize};
 #[serde(tag = "type")]
 enum Metadata {
     File,
+    Text,
     Image { width: isize, height: isize },
     Video { width: isize, height: isize },
     Audio,
@@ -13,9 +14,18 @@ enum Metadata {
 pub struct File {
     #[serde(rename = "_id")]
     pub id: String,
+    tag: String,
     filename: String,
     metadata: Metadata,
     content_type: String,
+    size: isize,
 
+    #[serde(skip_serializing_if = "Option::is_none")]
     message_id: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    user_id: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    server_id: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    object_id: Option<String>,
 }

src/database/migrations/init.rs

@@ -29,6 +29,10 @@ pub async fn create_database() {
         .await
         .expect("Failed to create migrations collection.");
 
+    db.create_collection("attachments", None)
+        .await
+        .expect("Failed to create attachments collection.");
+
     db.create_collection(
         "pubsub",
         CreateCollectionOptions::builder()

src/database/migrations/scripts.rs

@@ -1,8 +1,8 @@
 use crate::database::get_collection;
 
 use log::info;
-use mongodb::bson::{doc, from_document};
 use serde::{Deserialize, Serialize};
+use mongodb::bson::{doc, from_document};
 
 #[derive(Serialize, Deserialize)]
 struct MigrationInfo {
@@ -10,7 +10,7 @@ struct MigrationInfo {
     revision: i32,
 }
 
-pub const LATEST_REVISION: i32 = 0;
+pub const LATEST_REVISION: i32 = 1;
 
 pub async fn migrate_database() {
     let migrations = get_collection("migrations");
@@ -53,6 +53,29 @@ pub async fn run_migrations(revision: i32) -> i32 {
         info!("Running migration [revision 0]: Test migration system.");
     }
 
+    if revision <= 1 {
+        info!("Running migration [revision 1 / 2021-04-24]: Migrate to Autumn v1.0.0.");
+
+        let messages = get_collection("messages");
+        let attachments = get_collection("attachments");
+
+        messages.update_many(
+            doc! { "attachment": { "$exists": 1 } },
+            doc! { "$set": { "attachment.tag": "attachments", "attachment.size": 0 } },
+            None
+        )
+        .await
+        .expect("Failed to update messages.");
+
+        attachments.update_many(
+            doc! { },
+            doc! { "$set": { "tag": "attachments", "size": 0 } },
+            None
+        )
+        .await
+        .expect("Failed to update attachments.");
+    }
+
     // Reminder to update LATEST_REVISION when adding new migrations.
     LATEST_REVISION
 }