[desktop] Improve tauri traymenu

safing · May 22, 2024 · 0c99b93 · 0c99b93
1 parent a984032
commit 0c99b93
Show file tree

Hide file tree

Showing 4 changed files with 296 additions and 56 deletions.
diff --git a/desktop/tauri/src-tauri/Cargo.lock b/desktop/tauri/src-tauri/Cargo.lock
diff --git a/desktop/tauri/src-tauri/Cargo.toml b/desktop/tauri/src-tauri/Cargo.toml
@@ -47,6 +47,7 @@ http = "1.0.0"
 url = "2.5.0"
 thiserror = "1.0"
 log = "0.4.21"
+reqwest = { version = "0.12" }
 
 # Linux only
 [target.'cfg(target_os = "linux")'.dependencies]

diff --git a/desktop/tauri/src-tauri/src/portmaster/mod.rs b/desktop/tauri/src-tauri/src/portmaster/mod.rs
@@ -36,6 +36,8 @@ use serde;
 use std::sync::Mutex;
 use tauri::{AppHandle, Manager, Runtime};
 
+const PORTMASTER_BASE_URL: &'static str = "http://127.0.0.1:817/api/v1/";
+
 pub trait Handler {
     fn on_connect(&mut self, cli: PortAPI) -> ();
     fn on_disconnect(&mut self);
@@ -203,6 +205,25 @@ impl<R: Runtime> PortmasterInterface<R> {
         }
     }
 
+    /// Send Shutdown request to portmaster
+    pub fn trigger_shutdown(&self) {
+        tauri::async_runtime::spawn(async move {
+            let client = reqwest::Client::new();
+            match client
+                .post(format!("{}/core/shutdown", PORTMASTER_BASE_URL))
+                .send()
+                .await
+            {
+                Ok(v) => {
+                    debug!("shutdown request sent {:?}", v);
+                }
+                Err(err) => {
+                    error!("failed to send shutdown request {}", err);
+                }
+            }
+        });
+    }
+
     //// Internal functions
     fn start_notification_handler(&self) {
         if let Some(api) = self.get_api() {