| mod adblock; |
| mod browser; |
| mod credentials; |
| mod library; |
| mod board; |
| mod color_tools; |
| mod persistence; |
| mod sessions; |
| 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_plugin_sql::{Migration, MigrationKind}; |
|
|
| #[cfg_attr(mobile, tauri::mobile_entry_point)] |
| pub fn run() { |
| tauri::Builder::default() |
| .plugin(tauri_plugin_opener::init()) |
| .plugin(tauri_plugin_store::Builder::default().build()) |
| .plugin(tauri_plugin_fs::init()) |
| .plugin(tauri_plugin_clipboard_manager::init()) |
| .plugin(tauri_plugin_sql::Builder::default().add_migrations("sqlite:muse_alpha_v2.db", migrations()).build()) |
| .plugin(tauri_plugin_stronghold::Builder::new(|password| { let salt = b"muse-vault-kdf-salt-2026-v1-prod"; let mut key = [0u8; 32]; 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("").trim_end_matches('/').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() { |
| "board" | "library" => {} |
| "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()) |
| .manage(board::BoardState::default()) |
| .manage(downloads::DownloadState::default()) |
| .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::commands::browser_init, browser::commands::browser_set_visible, browser::commands::browser_hide_all, browser::commands::tab_create, browser::commands::tab_activate, browser::commands::tab_close, browser::commands::tab_restore, browser::commands::tab_navigate, browser::commands::tab_reload, browser::commands::tab_back, browser::commands::tab_forward, browser::commands::tab_zoom, browser::commands::tab_resize, browser::commands::tab_get_all, browser::commands::tab_pin, browser::commands::tab_find, browser::commands::tab_find_clear, |
| browser::context_menu::browser_context_menu, |
| adblock::commands::shield_get_report, adblock::commands::shield_check_url, adblock::commands::shield_cosmetic_css, adblock::commands::shield_toggle_domain, adblock::commands::shield_is_allowed, adblock::commands::shield_update_lists, adblock::commands::shield_add_user_rule, adblock::commands::shield_list_subscriptions, |
| library::library_add_item, library::library_import_local, library::library_import_data_url, library::library_update_metadata, library::library_remove_tag, library::library_load, library::library_items, library::library_search, library::library_remove_item, library::library_add_tag, |
| 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, |
| persistence::storage_info, persistence::storage_clear_library, persistence::storage_clear_projects, persistence::storage_reveal_folder, |
| sessions::sessions_save, sessions::sessions_load, sessions::sessions_list, sessions::sessions_auto_save, sessions::sessions_delete, sessions::sessions_rename, |
| downloads::downloads_list, downloads::downloads_clear_completed, downloads::download_to_library, downloads::web_clip_page, |
| credentials::credentials_list, credentials::credentials_generate_password, |
| study::study_start, study::study_complete, study::study_list, |
| color_tools::color_export, color_tools::color_search_library, |
| ]) |
| .setup(|app| { |
| #[cfg(desktop)] |
| app.handle().plugin(tauri_plugin_global_shortcut::Builder::new().build())?; |
| if let Ok(zoom_mem) = crate::persistence::load_json::<std::collections::HashMap<String, f64>>(app.handle(), "zoom_memory.json") { if !zoom_mem.is_empty() { let app_state = app.state::<AppState>(); let mut tabs = app_state.tabs.lock().expect("tabs lock"); tabs.zoom_memory = zoom_mem; } } |
| adblock::updater::spawn_updater(app.handle().clone()); |
| Ok(()) |
| }) |
| .run(tauri::generate_context!()) |
| .expect("error while running Refstudio"); |
| } |
|
|
| fn migrations() -> Vec<Migration> { vec![ Migration { version: 1, description: "phase0_init", sql: include_str!("../migrations/001_phase0_init.sql"), kind: MigrationKind::Up }, 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() } |
|
|
