feat: integrate Google semantic search and branded desktop launcher

apps/desktop/src-tauri/src/main.rs

@@ -1,21 +1,75 @@
 #![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
 
+mod model_download;
+mod semantic_commands;
+mod settings;
+
 use searchi_core::{IndexReport, IndexStats, SearchIndex, SearchOptions, SearchResult};
-use serde::Deserialize;
+use semantic_commands::{
+    hybrid_search, index_embeddings, semantic_status, setup_model, SemanticState,
+};
+use serde::{Deserialize, Serialize};
+use settings::{Settings, SettingsStore};
 use std::{
     path::PathBuf,
     sync::{
         atomic::{AtomicBool, Ordering},
-        Arc,
+        Arc, Mutex,
     },
 };
-use tauri::{Manager, State};
+use tauri::{Emitter, Manager, State};
 use tauri_plugin_dialog::DialogExt;
+use tauri_plugin_global_shortcut::{GlobalShortcutExt, ShortcutState};
 use tauri_plugin_opener::OpenerExt;
 
 struct AppState {
     database: PathBuf,
+    lexical: Arc<Mutex<SearchIndex>>,
     indexing: Arc<AtomicBool>,
+    settings: Arc<Mutex<SettingsStore>>,
+    shortcut_status: String,
+    semantic: Arc<SemanticState>,
+}
+
+#[derive(Serialize)]
+struct DesktopStatus {
+    settings: Settings,
+    indexing: bool,
+    shortcut_status: String,
+}
+
+#[tauri::command]
+fn desktop_status(state: State<'_, AppState>) -> Result<DesktopStatus, String> {
+    Ok(DesktopStatus {
+        settings: state
+            .settings
+            .lock()
+            .map_err(|_| "Settings are unavailable.")?
+            .value
+            .clone(),
+        indexing: state.indexing.load(Ordering::Acquire),
+        shortcut_status: state.shortcut_status.clone(),
+    })
+}
+
+#[tauri::command]
+fn set_cpu_threads(threads: usize, state: State<'_, AppState>) -> Result<Settings, String> {
+    if state.indexing.load(Ordering::Acquire) {
+        return Err("Wait for indexing to finish before changing resource settings.".into());
+    }
+    let mut store = state
+        .settings
+        .lock()
+        .map_err(|_| "Settings are unavailable.")?;
+    let mut next = store.value.clone();
+    next.cpu_threads = threads;
+    next.validate()?;
+    state.semantic.reset_session()?;
+    store.save(next)?;
+    state
+        .semantic
+        .begin_automatic_setup(state.database.clone(), threads);
+    Ok(store.value.clone())
 }
 
 #[derive(Default, Deserialize)]
@@ -48,12 +102,12 @@ async fn search(
     filters: Option<Filters>,
     state: State<'_, AppState>,
 ) -> Result<Vec<SearchResult>, String> {
-    let database = state.database.clone();
+    let lexical = state.lexical.clone();
     tauri::async_runtime::spawn_blocking(move || {
-        SearchIndex::open(database)
-            .and_then(|index| {
-                index.search_with_options(&query, 50, &filters.unwrap_or_default().into())
-            })
+        lexical
+            .try_lock()
+            .map_err(|_| "Keyword worker is busy; try the latest query again.".to_owned())?
+            .search_with_options(&query, 50, &filters.unwrap_or_default().into())
             .map_err(|e| e.to_string())
     })
     .await
@@ -150,11 +204,23 @@ async fn index_folder(root: String, state: State<'_, AppState>) -> Result<IndexR
     }
     let guard = IndexGuard(state.indexing.clone());
     let database = state.database.clone();
+    let settings = state.settings.clone();
+    let lexical = state.lexical.clone();
     tauri::async_runtime::spawn_blocking(move || {
         let _guard = guard;
-        SearchIndex::open(database)
-            .and_then(|mut index| index.index_folder(root))
-            .map_err(|e| e.to_string())
+        settings
+            .lock()
+            .map_err(|_| "Settings are unavailable.")?
+            .add_root(std::path::Path::new(&root))?;
+        let mut index = SearchIndex::open(database).map_err(|error| error.to_string())?;
+        let report = index
+            .index_folder(root)
+            .map_err(|error| error.to_string())?;
+        index.prewarm().map_err(|error| error.to_string())?;
+        *lexical
+            .lock()
+            .map_err(|_| "Keyword engine is unavailable.")? = index;
+        Ok(report)
     })
     .await
     .map_err(|e| e.to_string())?
@@ -164,14 +230,78 @@ fn main() {
     tauri::Builder::default()
         .plugin(tauri_plugin_dialog::init())
         .plugin(tauri_plugin_opener::init())
+        .plugin(tauri_plugin_global_shortcut::Builder::new().with_handler(|app, _, event| {
+            if event.state == ShortcutState::Pressed {
+                if let Some(window) = app.get_webview_window("main") {
+                    let result = window.show().and_then(|_| window.unminimize()).and_then(|_| window.set_focus());
+                    if let Err(error) = result { eprintln!("Could not focus Searchi: {error}"); }
+                    let _ = window.emit("focus-search", ());
+                }
+            }
+        }).build())
         .setup(|app| {
             let directory = app.path().app_data_dir()?;
             std::fs::create_dir_all(&directory)?;
             let database = directory.join("searchi.sqlite");
-            SearchIndex::open(&database)?;
+            let lexical = Arc::new(Mutex::new(SearchIndex::open(&database)?));
+            let warm_database = database.clone();
+            let warm_target = lexical.clone();
+            tauri::async_runtime::spawn_blocking(move || {
+                // Prepare a separate reader and swap it in briefly; foreground
+                // queries never wait on the long statistics-preparation lock.
+                if let Ok(mut index) = SearchIndex::open(warm_database) {
+                    if index.prewarm().is_ok() {
+                        if let Ok(mut current) = warm_target.lock() { *current = index; }
+                    }
+                }
+            });
+            let settings = SettingsStore::load(directory.join("settings.json")).map_err(std::io::Error::other)?;
+            let development_root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../../..");
+            let development_models = development_root.join(".searchi/models/embeddinggemma2");
+            let bundled_models = app.path().resource_dir()?.join("embeddinggemma2");
+            let model_directory = settings.value.model_directory.clone().unwrap_or_else(|| {
+                if cfg!(debug_assertions) && development_models.is_dir() {
+                    development_models.canonicalize().unwrap_or(development_models)
+                } else if bundled_models.is_dir() {
+                    bundled_models
+                } else {
+                    searchi_assets::default_model_directory().unwrap_or_else(|_| directory.join("Models"))
+                }
+            });
+            let worker_name = if cfg!(windows) { "searchi-gemma-worker.exe" } else { "searchi-gemma-worker" };
+            let development_worker = development_root.join(".searchi/runtime/cpu")
+                .join(if cfg!(windows) { "Release" } else { "" }).join(worker_name);
+            let worker = std::env::var_os("SEARCHI_EMBEDDING_WORKER").map(PathBuf::from)
+                .unwrap_or_else(|| {
+                    if cfg!(debug_assertions) && development_worker.is_file() {
+                        development_worker.canonicalize().unwrap_or(development_worker)
+                    } else {
+                        app.path().resource_dir().unwrap_or_else(|_| directory.clone()).join(worker_name)
+                    }
+                });
+            let semantic = Arc::new(SemanticState::new(model_directory, worker));
+            semantic.begin_automatic_setup(database.clone(), settings.value.cpu_threads);
+            #[cfg(target_os = "macos")]
+            let shortcut = "Shift+Super+Space";
+            #[cfg(not(target_os = "macos"))]
+            let shortcut = "Control+Shift+Space";
+            let wayland = cfg!(target_os = "linux") &&
+                (std::env::var("XDG_SESSION_TYPE").is_ok_and(|value| value == "wayland") || std::env::var_os("WAYLAND_DISPLAY").is_some());
+            let shortcut_status = if wayland {
+                "Global shortcut unavailable on Wayland. Open Searchi from your application launcher.".into()
+            } else {
+                match app.global_shortcut().register(shortcut) {
+                    Ok(()) => format!("Global shortcut: {shortcut}"),
+                    Err(error) => format!("Global shortcut unavailable: {error}"),
+                }
+            };
             app.manage(AppState {
                 database,
+                lexical,
                 indexing: Arc::new(AtomicBool::new(false)),
+                settings: Arc::new(Mutex::new(settings)),
+                shortcut_status,
+                semantic,
             });
             Ok(())
         })
@@ -180,7 +310,13 @@ fn main() {
             stats,
             index_folder,
             pick_folder,
-            open_result
+            open_result,
+            desktop_status,
+            set_cpu_threads,
+            semantic_status,
+            setup_model,
+            index_embeddings,
+            hybrid_search
         ])
         .run(tauri::generate_context!())
         .expect("Searchi desktop could not start");