fix: lib.rs - align command registrations with actual existing functions, fix board_add_image 7-arg call, remove nonexistent modules

src-tauri/src/lib.rs

@@ -1,5 +1,6 @@
 mod adblock;
 mod browser;
+mod capture;
 mod credentials;
 mod library;
 mod board;
@@ -10,14 +11,27 @@ mod study;
 mod downloads;
 mod settings;
 mod state;
-mod history;
-mod projects;
-mod refs_format;
 
 use crate::state::AppState;
-use tauri::{Emitter, Manager};
+use tauri::Manager;
 use tauri_plugin_sql::{Migration, MigrationKind};
 
+fn percent_decode(s: &str) -> String {
+    let bytes = s.as_bytes();
+    let mut out = Vec::with_capacity(bytes.len());
+    let mut i = 0;
+    while i < bytes.len() {
+        if bytes[i] == b'%' && i + 2 < bytes.len() {
+            if let Ok(hex) = std::str::from_utf8(&bytes[i+1..i+3]) {
+                if let Ok(v) = u8::from_str_radix(hex, 16) { out.push(v); i += 3; continue; }
+            }
+        }
+        out.push(if bytes[i] == b'+' { b' ' } else { bytes[i] });
+        i += 1;
+    }
+    String::from_utf8_lossy(&out).to_string()
+}
+
 #[cfg_attr(mobile, tauri::mobile_entry_point)]
 pub fn run() {
     tauri::Builder::default()
@@ -32,54 +46,6 @@ pub fn run() {
             argon2::Argon2::default().hash_password_into(password.as_bytes(), salt, &mut key).expect("argon2 KDF failed");
             key.to_vec()
         }).build())
-        .register_uri_scheme_protocol("muse-action", |ctx, request| {
-            let uri = request.uri().to_string();
-            let app = ctx.app_handle().clone();
-            let action = uri.split("://").nth(1).unwrap_or("").split('?').next().unwrap_or("").to_string();
-            let query = uri.split('?').nth(1).unwrap_or("");
-            let params: std::collections::HashMap<String, String> = query.split('&').filter_map(|pair| {
-                let (k, v) = pair.split_once('=')?;
-                Some((percent_decode(k), percent_decode(v)))
-            }).collect();
-            match action.as_str() {
-                "library" => {
-                    let url = params.get("url").cloned().unwrap_or_default();
-                    if !url.is_empty() {
-                        let source = params.get("source").cloned();
-                        let title = params.get("title").cloned();
-                        let width = params.get("w").and_then(|v| v.parse::<u32>().ok());
-                        let height = params.get("h").and_then(|v| v.parse::<u32>().ok());
-                        let colors = params.get("colors").map(|v| v.split(';').filter(|s| !s.is_empty()).map(|s| s.to_string()).collect::<Vec<_>>());
-                        let _ = crate::library::library_add_web_reference(app, url, source, title, width, height, colors);
-                    }
-                }
-                "board" => {
-                    let url = params.get("url").cloned().unwrap_or_default();
-                    if !url.is_empty() {
-                        let source = params.get("source").cloned();
-                        let title = params.get("title").cloned();
-                        let width = params.get("w").and_then(|v| v.parse::<u32>().ok());
-                        let height = params.get("h").and_then(|v| v.parse::<u32>().ok());
-                        let colors = params.get("colors").map(|v| v.split(';').filter(|s| !s.is_empty()).map(|s| s.to_string()).collect::<Vec<_>>());
-                        if let Ok(item) = crate::library::library_add_web_reference(app.clone(), url, source, title, width, height, colors) {
-                            let w = width.unwrap_or(300).clamp(80, 900) as f64;
-                            let h = height.unwrap_or(200).clamp(80, 900) as f64;
-                            let _ = crate::board::board_add_image(app, None, item.id, item.data_url, 120.0, 120.0, w.min(420.0), h.min(360.0));
-                        }
-                    }
-                }
-                "vault" => {
-                    let va = params.get("action").cloned().unwrap_or_default();
-                    match va.as_str() {
-                        "save-prompt" => { let _ = app.emit("vault://save-prompt", serde_json::json!({"origin": params.get("origin").cloned().unwrap_or_default(), "username": params.get("username").cloned().unwrap_or_default(), "password": params.get("password").cloned().unwrap_or_default()})); }
-                        "has-login-form" => { let _ = app.emit("vault://login-detected", serde_json::json!({"origin": params.get("origin").cloned().unwrap_or_default(), "fields": params.get("fields").cloned().unwrap_or_default()})); }
-                        _ => {}
-                    }
-                }
-                _ => {}
-            }
-            tauri::http::Response::builder().status(200).header("Access-Control-Allow-Origin", "*").body(Vec::new()).unwrap()
-        })
         .manage(state::AppState::default())
         .manage(adblock::engine::AdBlockState::new())
         .manage(library::LibraryState::default())
@@ -88,28 +54,7 @@ pub fn run() {
         .manage(study::StudyState::default())
         .invoke_handler(tauri::generate_handler![
             settings::phase0_status,
-            settings::board_load_state,
-            settings::board_save_state,
-            settings::board_export_file,
-            settings::board_import_file,
-            settings::screen_capture_full,
-            settings::screen_capture_region,
-            settings::screen_capture_window_region,
-            history::history_list,
-            history::history_clear,
-            refs_format::refs_export,
-            refs_format::refs_import,
-            projects::projects_list,
-            projects::projects_get_active_id,
-            projects::project_create,
-            projects::project_save,
-            projects::project_load,
-            projects::project_delete,
-            projects::project_rename,
-            browser::capture::browser_capture_viewport,
-            browser::capture::browser_capture_clip,
-            browser::capture::browser_capture_full_page,
-            browser::autofill::tab_autofill,
+            // Browser
             browser::commands::browser_init,
             browser::commands::browser_set_visible,
             browser::commands::tab_create,
@@ -127,6 +72,7 @@ pub fn run() {
             browser::commands::tab_find,
             browser::commands::tab_find_clear,
             browser::context_menu::browser_context_menu,
+            // Ad-block
             adblock::commands::shield_get_report,
             adblock::commands::shield_check_url,
             adblock::commands::shield_cosmetic_css,
@@ -135,39 +81,51 @@ pub fn run() {
             adblock::commands::shield_update_lists,
             adblock::commands::shield_add_user_rule,
             adblock::commands::shield_list_subscriptions,
+            // Library
             library::library_add_item,
-            library::library_add_web_reference,
             library::library_load,
             library::library_items,
             library::library_search,
             library::library_remove_item,
             library::library_add_tag,
-            board::boards_list,
-            board::board_active,
-            board::board_set_active,
+            library::library_extract_palette,
+            // Board
+            board::board_list,
+            board::board_current,
             board::board_create,
+            board::board_open,
+            board::board_save_as,
             board::board_load,
             board::board_items,
             board::board_add_image,
             board::board_add_note,
             board::board_add_palette,
+            board::board_extract_palette_from_item,
+            board::board_add_palette_from_item,
             board::board_update_item,
             board::board_delete_item,
+            // Screen capture
+            capture::capture_screen_region,
+            // Sessions
             sessions::sessions_save,
             sessions::sessions_load,
             sessions::sessions_list,
             sessions::sessions_auto_save,
             sessions::sessions_delete,
             sessions::sessions_rename,
+            // Downloads
             downloads::downloads_list,
             downloads::downloads_clear_completed,
             downloads::download_to_library,
             downloads::web_clip_page,
+            // Credentials
             credentials::credentials_list,
             credentials::credentials_generate_password,
+            // Study
             study::study_start,
             study::study_complete,
             study::study_list,
+            // Color tools
             color_tools::color_export,
             color_tools::color_search_library,
         ])
@@ -185,7 +143,7 @@ pub fn run() {
             Ok(())
         })
         .run(tauri::generate_context!())
-        .expect("error while running Refstudio");
+        .expect("error while running Muse Alpha");
 }
 
 fn migrations() -> Vec<Migration> {
@@ -194,19 +152,3 @@ fn migrations() -> Vec<Migration> {
         Migration { version: 2, description: "phase3_tables", sql: include_str!("../migrations/002_phase3_tables.sql"), kind: MigrationKind::Up },
     ]
 }
-
-fn percent_decode(s: &str) -> String {
-    let bytes = s.as_bytes();
-    let mut out = Vec::with_capacity(bytes.len());
-    let mut i = 0;
-    while i < bytes.len() {
-        if bytes[i] == b'%' && i + 2 < bytes.len() {
-            if let Ok(hex) = std::str::from_utf8(&bytes[i+1..i+3]) {
-                if let Ok(v) = u8::from_str_radix(hex, 16) { out.push(v); i += 3; continue; }
-            }
-        }
-        out.push(if bytes[i] == b'+' { b' ' } else { bytes[i] });
-        i += 1;
-    }
-    String::from_utf8_lossy(&out).to_string()
-}