feat:auto download essential modules

Author: 0xbrayo

src-tauri/Cargo.lock

@@ -21,7 +21,7 @@ tauri-build = { version = "2.0.5", features = [] }
 
 [dependencies]
 shell-words = "1.1.0"
-tauri = { version = "2.2.2", features = ["tray-icon"] }
+tauri = { version = "2.2.5", features = ["tray-icon"] }
 tauri-plugin-shell = "2.2.0"
 tauri-plugin-dialog = "2.2.0"
 tauri-plugin-notification = "2.2.1"
@@ -30,16 +30,17 @@ tauri-plugin-single-instance = "2.2.1"
 notify = "8.0.0"
 directories = "6.0.0"
 serde = { version = "1.0.217", features = ["derive"] }
-serde_json = "1.0.135"
-toml = "0.8.19"
+serde_json = "1.0.138"
+toml = "0.8.20"
+csv="1.3.1"
 lazy_static = "1.5.0"
 log = "0.4.25"
 fern = { version = "0.7.1", features = ["colored"] }
 chrono = "0.4.39"
 aw-server = { git = "https://github.com/ActivityWatch/aw-server-rust.git", branch = "master" }
 aw-datastore = { git = "https://github.com/ActivityWatch/aw-server-rust.git", branch = "master" }
-tauri-plugin-opener = "2"
-tauri-plugin-http = "2"
+tauri-plugin-opener = "2.2.5"
+tauri-plugin-http = "2.3.0"
 [target.'cfg(unix)'.dependencies]
 nix = { version = "0.29.0", features = ["process", "signal"] }
 [target.'cfg(windows)'.dependencies]

src-tauri/Cargo.toml

@@ -18,6 +18,7 @@ use tauri_plugin_opener::OpenerExt;
 
 mod logging;
 mod manager;
+mod modules_dl;
 
 use log::info;
 use tauri::{

src-tauri/src/lib.rs

@@ -7,7 +7,6 @@
 /// their state.
 ///
 /// If a module crashes, the manager will notify the user and ask if they want to restart it.
-
 #[cfg(unix)]
 use {
     nix::sys::signal::{self, Signal},
@@ -33,6 +32,7 @@ use std::{env, fs, thread};
 use tauri::menu::{CheckMenuItem, Menu, MenuItem, SubmenuBuilder};
 use tauri_plugin_dialog::{DialogExt, MessageDialogKind};
 
+use crate::modules_dl::has_essential_modules;
 use crate::{get_app_handle, get_config, get_tray_id, HANDLE_CONDVAR};
 
 #[derive(Debug)]
@@ -53,6 +53,8 @@ pub enum ModuleMessage {
 pub struct ManagerState {
     tx: Sender<ModuleMessage>,
     pub modules_running: BTreeMap<String, bool>,
+    // TODO: the next four could be merged into one
+    // modules_metadata hashmap? worse for readability
     pub modules_discovered: BTreeMap<String, PathBuf>,
     pub modules_pid: HashMap<String, u32>,
     pub modules_restart_count: HashMap<String, u32>,
@@ -103,6 +105,15 @@ impl ManagerState {
         let quit = MenuItem::with_id(app, "quit", "Quit", true, None::<&str>)
             .expect("failed to create quit menu item");
 
+        if !has_essential_modules(self.modules_discovered.keys().cloned().collect()) {
+            // todo!()
+            thread::spawn(|| {
+                tauri::async_runtime::block_on(async {
+                    crate::modules_dl::download_modules().await.unwrap();
+                });
+            });
+        }
+
         let mut modules_submenu_builder = SubmenuBuilder::new(app, "Modules");
         for (module, running) in self.modules_running.iter() {
             let label = module;