From 25ba200a5e0be4dcfdb031254d0c268d0542fb7e Mon Sep 17 00:00:00 2001
From: quexeky <git@quexeky.dev>
Date: Sat, 4 Jan 2025 22:29:24 +1100
Subject: [PATCH] feat(database): Added database corruption dialog

Signed-off-by: quexeky <git@quexeky.dev>
---
 app.vue                        | 15 ---------------
 src-tauri/src/db.rs            |  3 ++-
 src-tauri/src/games/library.rs |  2 +-
 src-tauri/src/lib.rs           | 17 +++++++++++++----
 4 files changed, 16 insertions(+), 21 deletions(-)

diff --git a/app.vue b/app.vue
index 00c2190..f81c432 100644
--- a/app.vue
+++ b/app.vue
@@ -29,21 +29,6 @@ router.beforeEach(async () => {
 setupHooks();
 initialNavigation(state);
 
-listen("database_corrupted", (event) => {
-  createModal(
-      ModalType.Notification,
-      {
-        title: "Database corrupted",
-        description: `Drop encountered an error while reading your download. A copy can be found at: "${(
-          event.payload as unknown as string
-        ).toString()}"`,
-        buttonText: "Close"
-      },
-      (e, c) => c()
-    );
-
-})
-
 useHead({
   title: "Drop",
 });
diff --git a/src-tauri/src/db.rs b/src-tauri/src/db.rs
index 12dccb9..2c05261 100644
--- a/src-tauri/src/db.rs
+++ b/src-tauri/src/db.rs
@@ -251,10 +251,11 @@ fn handle_invalid_database(_e: RustbreakError, db_path: PathBuf, games_base_dir:
     let new_path = { 
         let time = Utc::now().timestamp();
         let mut base = db_path.clone().into_os_string();
+        base.push(".");
         base.push(time.to_string());
         base
     };
-    fs::copy(&db_path, &new_path);
+    fs::copy(&db_path, &new_path).unwrap();
 
     let db = Database {
         auth: None,
diff --git a/src-tauri/src/games/library.rs b/src-tauri/src/games/library.rs
index 2ddd573..8d40274 100644
--- a/src-tauri/src/games/library.rs
+++ b/src-tauri/src/games/library.rs
@@ -2,7 +2,7 @@ use std::fs::remove_dir_all;
 use std::sync::Mutex;
 use std::thread::spawn;
 
-use log::{error, info};
+use log::{error, info, warn};
 use serde::{Deserialize, Serialize};
 use tauri::Emitter;
 use tauri::{AppHandle, Manager};
diff --git a/src-tauri/src/lib.rs b/src-tauri/src/lib.rs
index 72e9a9d..079055a 100644
--- a/src-tauri/src/lib.rs
+++ b/src-tauri/src/lib.rs
@@ -42,6 +42,7 @@ use process::process_commands::{kill_game, launch_game};
 use process::process_manager::ProcessManager;
 use remote::{gen_drop_url, use_remote};
 use serde::{Deserialize, Serialize};
+use tauri_plugin_dialog::DialogExt;
 use std::path::Path;
 use std::sync::Arc;
 use std::{
@@ -182,10 +183,6 @@ fn setup(handle: AppHandle) -> AppState<'static> {
             .and_modify(|v| *v = GameDownloadStatus::Remote {});
     }
 
-    if let Some(original) = db_handle.prev_database.take() {
-        warn!("Database corrupted. Original file at {}", original.canonicalize().unwrap().to_string_lossy().to_string());
-        handle.emit("database_corrupted", original.to_string_lossy().to_string()).unwrap();
-    }
     drop(db_handle);
 
 
@@ -334,6 +331,18 @@ pub fn run() {
                 .build(app)
                 .expect("error while setting up tray menu");
 
+            {
+                let mut db_handle = DB.borrow_data_mut().unwrap();
+                if let Some(original) = db_handle.prev_database.take() {
+                    warn!("Database corrupted. Original file at {}", original.canonicalize().unwrap().to_string_lossy().to_string());
+                    app.dialog()
+                        .message("Database corrupted. A copy has been saved at: ".to_string() + original.to_str().unwrap())
+                        .title("Database corrupted")
+                        .show(|_| {});
+                }
+            }
+
+        
             Ok(())
         })
         .register_asynchronous_uri_scheme_protocol("object", move |_ctx, request, responder| {