From 2385bd1cfd41fe14e8a73b9d6b556b5f8bb55ff1 Mon Sep 17 00:00:00 2001
From: hamidreza kalbasi <hamidrezakalbasi@protonmail.com>
Date: Tue, 8 Jun 2021 20:53:24 +0430
Subject: [PATCH] add migrations

---
 src/database.rs       | 29 ++++++++++++++++++++++-------
 src/database/media.rs |  6 +++---
 2 files changed, 25 insertions(+), 10 deletions(-)

diff --git a/src/database.rs b/src/database.rs
index 2870e2d..fa84881 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -20,11 +20,7 @@ use log::error;
 use rocket::futures::{channel::mpsc, stream::FuturesUnordered, StreamExt};
 use ruma::{DeviceId, ServerName, UserId};
 use serde::Deserialize;
-use std::{
-    collections::HashMap,
-    fs::remove_dir_all,
-    sync::{Arc, RwLock},
-};
+use std::{collections::HashMap, fs::{self, remove_dir_all}, io::Write, sync::{Arc, RwLock}};
 use tokio::sync::Semaphore;
 
 #[derive(Clone, Debug, Deserialize)]
@@ -253,9 +249,11 @@ impl Database {
             for (userid, password) in db.users.userid_password.iter() {
                 let password = utils::string_from_bytes(&password);
 
-                if password.map_or(false, |password| {
+                let empty_hashed_password = password.map_or(false, |password| {
                     argon2::verify_encoded(&password, b"").unwrap_or(false)
-                }) {
+                });
+
+                if empty_hashed_password {
                     db.users.userid_password.insert(&userid, b"")?;
                 }
             }
@@ -265,6 +263,23 @@ impl Database {
             println!("Migration: 1 -> 2 finished");
         }
 
+        if db.globals.database_version()? < 3 {
+            // Move media to filesystem
+            for (key, content) in db.media.mediaid_file.iter() {
+                if content.len() == 0 {
+                    continue;
+                }
+
+                let path = db.globals.get_media_file(&key);
+                let mut file = fs::File::create(path)?;
+                file.write_all(&content)?;
+                db.media.mediaid_file.insert(&key, &[])?;
+            }
+
+            db.globals.bump_database_version(3)?;
+
+            println!("Migration: 2 -> 3 finished");
+        }
         // This data is probably outdated
         db.rooms.edus.presenceid_presence.clear()?;
 
diff --git a/src/database/media.rs b/src/database/media.rs
index b81e863..944c5bd 100644
--- a/src/database/media.rs
+++ b/src/database/media.rs
@@ -103,7 +103,7 @@ impl Media {
         let mut iter = self.mediaid_file.scan_prefix(prefix);
         if let Some((key, _)) = iter.next() {
             let path = globals.get_media_file(&key);
-            let mut file = vec![];
+            let mut file = Vec::new();
             File::open(path).await?.read_to_end(&mut file).await?;
             let mut parts = key.rsplit(|&b| b == 0xff);
 
@@ -192,7 +192,7 @@ impl Media {
         if let Some((key, _)) = self.mediaid_file.scan_prefix(thumbnail_prefix).next() {
             // Using saved thumbnail
             let path = globals.get_media_file(&key);
-            let mut file = vec![];
+            let mut file = Vec::new();
             File::open(path).await?.read_to_end(&mut file).await?;
             let mut parts = key.rsplit(|&b| b == 0xff);
 
@@ -227,7 +227,7 @@ impl Media {
         } else if let Some((key, _)) = self.mediaid_file.scan_prefix(original_prefix).next() {
             // Generate a thumbnail
             let path = globals.get_media_file(&key);
-            let mut file = vec![];
+            let mut file = Vec::new();
             File::open(path).await?.read_to_end(&mut file).await?;
 
             let mut parts = key.rsplit(|&b| b == 0xff);