From aa1f70d3449dd346152af21a7d7d819e6d9c9e74 Mon Sep 17 00:00:00 2001
From: DecDuck <declanahofmeyr@gmail.com>
Date: Mon, 25 Nov 2024 16:09:29 +1100
Subject: [PATCH] feat(settings): finish download dir CRUD interface

---
 desktop/pages/settings/downloads.vue | 33 +++++++++++++++++++++++-----
 desktop/src-tauri/src/db.rs          | 15 ++++++++++++-
 desktop/src-tauri/src/lib.rs         |  5 +++--
 3 files changed, 44 insertions(+), 9 deletions(-)

diff --git a/desktop/pages/settings/downloads.vue b/desktop/pages/settings/downloads.vue
index 0878be79..237ea72c 100644
--- a/desktop/pages/settings/downloads.vue
+++ b/desktop/pages/settings/downloads.vue
@@ -27,7 +27,7 @@
     </div>
     <ul role="list" class="divide-y divide-gray-800">
       <li
-        v-for="dir in dirs"
+        v-for="(dir, dirIdx) in dirs"
         :key="dir"
         class="flex justify-between gap-x-6 py-5"
       >
@@ -43,7 +43,16 @@
           </div>
         </div>
         <div class="flex shrink-0 items-center gap-x-6">
-          <button class="-m-2.5 block p-2.5 text-zinc-400 hover:text-zinc-100">
+          <button
+            @click="() => deleteDirectory(dirIdx)"
+            :disabled="dirs.length <= 1"
+            :class="[
+              dirs.length <= 1
+                ? 'text-zinc-700'
+                : 'text-zinc-400 hover:text-zinc-100',
+              '-m-2.5 block p-2.5',
+            ]"
+          >
             <span class="sr-only">Open options</span>
             <TrashIcon class="size-5" aria-hidden="true" />
           </button>
@@ -169,7 +178,14 @@ const currentDirectory = ref<string | undefined>(undefined);
 const error = ref<string | undefined>(undefined);
 const createDirectoryLoading = ref(false);
 
-const dirs = ref(await invoke<Array<string>>("fetch_download_dir_stats"));
+const dirs = ref<Array<string>>([]);
+
+async function updateDirs() {
+  const newDirs = await invoke<Array<string>>("fetch_download_dir_stats");
+  dirs.value = newDirs;
+}
+
+await updateDirs();
 
 async function selectDirectoryDialog(): Promise<string> {
   const res = await invoke("plugin:dialog|open", {
@@ -201,12 +217,12 @@ async function submitDirectory() {
     createDirectoryLoading.value = true;
 
     // Add directory
-    await invoke("add_new_download_dir", { newDir: currentDirectory.value });
+    await invoke("add_download_dir", { newDir: currentDirectory.value });
 
     // Update list
-    const newDirs = await invoke<Array<string>>("fetch_download_dir_stats");
-    dirs.value = newDirs;
+    await updateDirs();
 
+    currentDirectory.value = undefined;
     createDirectoryLoading.value = false;
     open.value = false;
   } catch (e) {
@@ -214,4 +230,9 @@ async function submitDirectory() {
     createDirectoryLoading.value = false;
   }
 }
+
+async function deleteDirectory(index: number) {
+  await invoke("delete_download_dir", { index });
+  await updateDirs();
+}
 </script>
diff --git a/desktop/src-tauri/src/db.rs b/desktop/src-tauri/src/db.rs
index c6d32e7e..9ebd141d 100644
--- a/desktop/src-tauri/src/db.rs
+++ b/desktop/src-tauri/src/db.rs
@@ -101,7 +101,7 @@ impl DatabaseImpls for DatabaseInterface {
 }
 
 #[tauri::command]
-pub fn add_new_download_dir(new_dir: String) -> Result<(), String> {
+pub fn add_download_dir(new_dir: String) -> Result<(), String> {
     // Check the new directory is all good
     let new_dir_path = Path::new(&new_dir);
     if new_dir_path.exists() {
@@ -124,6 +124,9 @@ pub fn add_new_download_dir(new_dir: String) -> Result<(), String> {
 
     // Add it to the dictionary
     let mut lock = DB.borrow_data_mut().unwrap();
+    if lock.games.install_dirs.contains(&new_dir) {
+        return Err("Download directory already used".to_string());
+    }
     lock.games.install_dirs.push(new_dir);
     drop(lock);
     DB.save().unwrap();
@@ -131,6 +134,16 @@ pub fn add_new_download_dir(new_dir: String) -> Result<(), String> {
     Ok(())
 }
 
+#[tauri::command]
+pub fn delete_download_dir(index: usize) -> Result<(), String> {
+    let mut lock = DB.borrow_data_mut().unwrap();
+    lock.games.install_dirs.remove(index);
+    drop(lock);
+    DB.save().unwrap();
+
+    Ok(())
+}
+
 // Will, in future, return disk/remaining size
 // Just returns the directories that have been set up
 #[tauri::command]
diff --git a/desktop/src-tauri/src/lib.rs b/desktop/src-tauri/src/lib.rs
index 707b787a..06eef38c 100644
--- a/desktop/src-tauri/src/lib.rs
+++ b/desktop/src-tauri/src/lib.rs
@@ -10,7 +10,7 @@ mod tests;
 
 use crate::db::DatabaseImpls;
 use auth::{auth_initiate, generate_authorization_header, recieve_handshake};
-use db::{add_new_download_dir, fetch_download_dir_stats, DatabaseInterface, DATA_ROOT_DIR};
+use db::{add_download_dir, delete_download_dir, fetch_download_dir_stats, DatabaseInterface, DATA_ROOT_DIR};
 use downloads::download_commands::*;
 use downloads::download_manager::DownloadManagerBuilder;
 use downloads::download_manager_interface::DownloadManager;
@@ -126,7 +126,8 @@ pub fn run() {
             // Library
             fetch_library,
             fetch_game,
-            add_new_download_dir,
+            add_download_dir,
+            delete_download_dir,
             fetch_download_dir_stats,
             // Downloads
             download_game,