From f10d92d6230069375c147b319258fca54235898b Mon Sep 17 00:00:00 2001
From: quexeky <git@quexeky.dev>
Date: Sat, 9 Nov 2024 20:15:04 +1100
Subject: [PATCH] feat(downloads): Generic function to set download state

Signed-off-by: quexeky <git@quexeky.dev>
---
 pages/store/index.vue                        | 19 ++++++++++++++++++-
 src-tauri/src/downloads/download_commands.rs | 10 ++++++++++
 src-tauri/src/downloads/download_logic.rs    |  1 -
 src-tauri/src/lib.rs                         |  3 ++-
 4 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/pages/store/index.vue b/pages/store/index.vue
index c695039..4f349bf 100644
--- a/pages/store/index.vue
+++ b/pages/store/index.vue
@@ -37,6 +37,13 @@
   >
     Resume game download
   </button>
+  <button
+    class="w-full rounded-md p-4 bg-blue-600 text-white"
+    @click="setGameDownloadWrapper"
+  >
+    Set game download
+  </button>
+  
 </template>
 <script setup lang="ts">
 import { invoke } from "@tauri-apps/api/core";
@@ -118,6 +125,16 @@ function resumeGameDownloadWrapper() {
     .catch((e) => {
       console.log(e)
     })
-
+}
+async function setGameDownload() {
+  console.log("Setting game download status");
+  await invoke("set_download_state", { gameId: gameId.value, status: "Paused" })
+}
+function setGameDownloadWrapper() {
+  setGameDownload()
+    .then(() => {})
+    .catch((e) => {
+      console.log(e)
+    })
 }
 </script>
diff --git a/src-tauri/src/downloads/download_commands.rs b/src-tauri/src/downloads/download_commands.rs
index 5a1f0cd..24131ea 100644
--- a/src-tauri/src/downloads/download_commands.rs
+++ b/src-tauri/src/downloads/download_commands.rs
@@ -135,6 +135,16 @@ pub async fn resume_game_download(
 
     Ok(())
 }
+#[tauri::command]
+pub async fn set_download_state(
+    state: tauri::State<'_, Mutex<AppState>>,
+    game_id: String,
+    status: GameDownloadState
+) -> Result<(), String> {
+    get_game_download(state, game_id).change_state(status);
+
+    Ok(())
+}
 
 fn get_game_download(
     state: tauri::State<'_, Mutex<AppState>>,
diff --git a/src-tauri/src/downloads/download_logic.rs b/src-tauri/src/downloads/download_logic.rs
index c6c0fd0..1f6ac90 100644
--- a/src-tauri/src/downloads/download_logic.rs
+++ b/src-tauri/src/downloads/download_logic.rs
@@ -12,7 +12,6 @@ use std::{
     path::PathBuf,
     sync::{Arc, RwLock},
     thread::sleep,
-    time::Duration,
 };
 use urlencoding::encode;
 
diff --git a/src-tauri/src/lib.rs b/src-tauri/src/lib.rs
index 41ec199..3158e2e 100644
--- a/src-tauri/src/lib.rs
+++ b/src-tauri/src/lib.rs
@@ -125,7 +125,8 @@ pub fn run() {
             cancel_specific_game_download,
             get_game_download_progress,
             resume_game_download,
-            pause_game_download
+            pause_game_download,
+            set_download_state
         ])
         .plugin(tauri_plugin_shell::init())
         .setup(|app| {