use crate::proxy_manager::PROXY_MANAGER; use directories::BaseDirs; use serde::{Deserialize, Serialize}; use std::fs::{self, create_dir_all}; use std::path::{Path, PathBuf}; use std::process::Command; use std::time::{SystemTime, UNIX_EPOCH}; use sysinfo::{Pid, System}; use tauri::Emitter; use crate::browser::{create_browser, BrowserType, ProxySettings}; use crate::browser_version_service::{ BrowserVersionInfo, BrowserVersionService, BrowserVersionsResult, }; use crate::download::{DownloadProgress, Downloader}; use crate::downloaded_browsers::DownloadedBrowsersRegistry; use crate::extraction::Extractor; #[derive(Debug, Serialize, Deserialize, Clone)] pub struct BrowserProfile { pub name: String, pub browser: String, pub version: String, pub profile_path: String, #[serde(default)] pub proxy: Option, #[serde(default)] pub process_id: Option, #[serde(default)] pub last_launch: Option, } // Platform-specific modules #[cfg(target_os = "macos")] mod macos { use super::*; use std::ffi::OsString; pub fn is_tor_or_mullvad_browser(exe_name: &str, cmd: &[OsString], browser_type: &str) -> bool { match browser_type { "mullvad-browser" => { let has_mullvad_in_exe = exe_name.contains("mullvad"); let has_firefox_exe = exe_name == "firefox" || exe_name.contains("firefox-bin"); let has_mullvad_in_cmd = cmd.iter().any(|arg| { let arg_str = arg.to_str().unwrap_or(""); arg_str.contains("Mullvad Browser.app") || arg_str.contains("mullvad") || arg_str.contains("Mullvad") || arg_str.contains("/Applications/Mullvad Browser.app/") || arg_str.contains("MullvadBrowser") }); has_mullvad_in_exe || (has_firefox_exe && has_mullvad_in_cmd) } "tor-browser" => { let has_tor_in_exe = exe_name.contains("tor"); let has_firefox_exe = exe_name == "firefox" || exe_name.contains("firefox-bin"); let has_tor_in_cmd = cmd.iter().any(|arg| { let arg_str = arg.to_str().unwrap_or(""); arg_str.contains("Tor Browser.app") || arg_str.contains("tor-browser") || arg_str.contains("TorBrowser") || arg_str.contains("/Applications/Tor Browser.app/") || arg_str.contains("TorBrowser-Data") }); has_tor_in_exe || (has_firefox_exe && has_tor_in_cmd) } _ => false, } } pub async fn launch_browser_process( executable_path: &std::path::Path, args: &[String], ) -> Result> { println!("Launching browser on macOS: {executable_path:?} with args: {args:?}"); Ok(Command::new(executable_path).args(args).spawn()?) } pub async fn open_url_in_existing_browser_firefox_like( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let pid = profile.process_id.unwrap(); // First try: Use Firefox remote command println!("Trying Firefox remote command for PID: {pid}"); let browser = create_browser(browser_type); if let Ok(executable_path) = browser.get_executable_path(browser_dir) { let remote_args = vec![ "-profile".to_string(), profile.profile_path.clone(), "-new-tab".to_string(), url.to_string(), ]; let remote_output = Command::new(executable_path).args(&remote_args).output(); match remote_output { Ok(output) if output.status.success() => { println!("Firefox remote command succeeded"); return Ok(()); } Ok(output) => { let stderr = String::from_utf8_lossy(&output.stderr); println!( "Firefox remote command failed with stderr: {stderr}, trying AppleScript fallback" ); } Err(e) => { println!("Firefox remote command error: {e}, trying AppleScript fallback"); } } } // Fallback: Use AppleScript let escaped_url = url .replace("\"", "\\\"") .replace("\\", "\\\\") .replace("'", "\\'"); let script = format!( r#" try tell application "System Events" -- Find the exact process by PID set targetProcess to (first application process whose unix id is {pid}) -- Verify the process exists if not (exists targetProcess) then error "No process found with PID {pid}" end if -- Get the process name for verification set processName to name of targetProcess -- Bring the process to the front first set frontmost of targetProcess to true delay 1.0 -- Check if the process has any visible windows set windowList to windows of targetProcess set hasVisibleWindow to false repeat with w in windowList if visible of w is true then set hasVisibleWindow to true exit repeat end if end repeat if not hasVisibleWindow then -- No visible windows, create a new one tell targetProcess keystroke "n" using command down delay 2.0 end tell end if -- Ensure the process is frontmost again set frontmost of targetProcess to true delay 0.5 -- Focus on the address bar and open URL tell targetProcess -- Open a new tab keystroke "t" using command down delay 1.5 -- Focus address bar (Cmd+L) keystroke "l" using command down delay 0.5 -- Type the URL keystroke "{escaped_url}" delay 0.5 -- Press Enter to navigate keystroke return end tell return "Successfully opened URL in " & processName & " (PID: {pid})" end tell on error errMsg number errNum return "AppleScript failed: " & errMsg & " (Error " & errNum & ")" end try "# ); println!("Executing AppleScript fallback for Firefox-based browser (PID: {pid})..."); let output = Command::new("osascript").args(["-e", &script]).output()?; if !output.status.success() { let error_msg = String::from_utf8_lossy(&output.stderr); println!("AppleScript failed: {error_msg}"); return Err( format!( "Both Firefox remote command and AppleScript failed. AppleScript error: {error_msg}" ) .into(), ); } else { println!("AppleScript succeeded"); } Ok(()) } pub async fn open_url_in_existing_browser_tor_mullvad( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let pid = profile.process_id.unwrap(); println!("Opening URL in TOR/Mullvad browser using file-based approach (PID: {pid})"); // Method 1: Try using a temporary HTML file approach println!("Attempting file-based URL opening for TOR/Mullvad browser"); let temp_dir = std::env::temp_dir(); let temp_file_name = format!("donut_browser_url_{}.html", std::process::id()); let temp_file_path = temp_dir.join(&temp_file_name); let html_content = format!( r#" Redirecting...

Redirecting to {url}...

"# ); match std::fs::write(&temp_file_path, html_content) { Ok(()) => { println!("Created temporary HTML file: {temp_file_path:?}"); let browser = create_browser(browser_type.clone()); if let Ok(executable_path) = browser.get_executable_path(browser_dir) { let open_result = Command::new("open") .args([ "-a", executable_path.to_str().unwrap(), temp_file_path.to_str().unwrap(), ]) .output(); // Clean up the temporary file after a short delay let temp_file_path_clone = temp_file_path.clone(); tokio::spawn(async move { tokio::time::sleep(tokio::time::Duration::from_secs(5)).await; let _ = std::fs::remove_file(temp_file_path_clone); }); match open_result { Ok(output) if output.status.success() => { println!("Successfully opened URL using file-based approach"); return Ok(()); } Ok(output) => { let stderr = String::from_utf8_lossy(&output.stderr); println!("File-based approach failed: {stderr}"); } Err(e) => { println!("File-based approach error: {e}"); } } } let _ = std::fs::remove_file(&temp_file_path); } Err(e) => { println!("Failed to create temporary HTML file: {e}"); } } // Method 2: Try using the 'open' command directly with the URL println!("Attempting direct URL opening with 'open' command"); let browser = create_browser(browser_type.clone()); if let Ok(executable_path) = browser.get_executable_path(browser_dir) { let direct_open_result = Command::new("open") .args(["-a", executable_path.to_str().unwrap(), url]) .output(); match direct_open_result { Ok(output) if output.status.success() => { println!("Successfully opened URL using direct 'open' command"); return Ok(()); } Ok(output) => { let stderr = String::from_utf8_lossy(&output.stderr); println!("Direct 'open' command failed: {stderr}"); } Err(e) => { println!("Direct 'open' command error: {e}"); } } } // If all methods fail, return a helpful error message Err( format!( "Failed to open URL in existing TOR/Mullvad browser (PID: {pid}). All methods failed:\n\ 1. File-based approach failed\n\ 2. Direct 'open' command failed\n\ \n\ This may be due to browser security restrictions or the browser process may have changed.\n\ Try closing and reopening the browser, or manually paste the URL: {url}" ) .into(), ) } pub async fn open_url_in_existing_browser_chromium( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let pid = profile.process_id.unwrap(); // First, try using the browser's built-in URL opening capability println!("Trying Chromium URL opening for PID: {pid}"); let browser = create_browser(browser_type); if let Ok(executable_path) = browser.get_executable_path(browser_dir) { let remote_output = Command::new(executable_path) .args([&format!("--user-data-dir={}", profile.profile_path), url]) .output(); match remote_output { Ok(output) if output.status.success() => { println!("Chromium URL opening succeeded"); return Ok(()); } Ok(output) => { let stderr = String::from_utf8_lossy(&output.stderr); println!("Chromium URL opening failed: {stderr}, trying AppleScript"); } Err(e) => { println!("Chromium URL opening error: {e}, trying AppleScript"); } } } // Fallback to AppleScript let escaped_url = url .replace("\"", "\\\"") .replace("\\", "\\\\") .replace("'", "\\'"); let script = format!( r#" try tell application "System Events" -- Find the exact process by PID set targetProcess to (first application process whose unix id is {pid}) -- Verify the process exists if not (exists targetProcess) then error "No process found with PID {pid}" end if -- Get the process name for verification set processName to name of targetProcess -- Bring the process to the front first set frontmost of targetProcess to true delay 1.0 -- Check if the process has any visible windows set windowList to windows of targetProcess set hasVisibleWindow to false repeat with w in windowList if visible of w is true then set hasVisibleWindow to true exit repeat end if end repeat if not hasVisibleWindow then -- No visible windows, create a new one tell targetProcess keystroke "n" using command down delay 2.0 end tell end if -- Ensure the process is frontmost again set frontmost of targetProcess to true delay 0.5 -- Focus on the address bar and open URL tell targetProcess -- Open a new tab keystroke "t" using command down delay 1.5 -- Focus address bar (Cmd+L) keystroke "l" using command down delay 0.5 -- Type the URL keystroke "{escaped_url}" delay 0.5 -- Press Enter to navigate keystroke return end tell return "Successfully opened URL in " & processName & " (PID: {pid})" end tell on error errMsg number errNum return "AppleScript failed: " & errMsg & " (Error " & errNum & ")" end try "# ); println!("Executing AppleScript for Chromium-based browser (PID: {pid})..."); let output = Command::new("osascript").args(["-e", &script]).output()?; if !output.status.success() { let error_msg = String::from_utf8_lossy(&output.stderr); println!("AppleScript failed: {error_msg}"); return Err( format!("Failed to open URL in existing Chromium-based browser: {error_msg}").into(), ); } else { println!("AppleScript succeeded"); } Ok(()) } pub async fn kill_browser_process_impl( pid: u32, ) -> Result<(), Box> { println!("Attempting to kill browser process with PID: {pid}"); // First try SIGTERM (graceful shutdown) let output = Command::new("kill") .args(["-TERM", &pid.to_string()]) .output() .map_err(|e| format!("Failed to execute kill command: {e}"))?; if !output.status.success() { // If SIGTERM fails, try SIGKILL (force kill) let output = Command::new("kill") .args(["-KILL", &pid.to_string()]) .output()?; if !output.status.success() { return Err( format!( "Failed to kill process {}: {}", pid, String::from_utf8_lossy(&output.stderr) ) .into(), ); } } println!("Successfully killed browser process with PID: {pid}"); Ok(()) } } #[cfg(target_os = "windows")] mod windows { use super::*; use std::ffi::OsString; pub fn is_tor_or_mullvad_browser( _exe_name: &str, _cmd: &[OsString], _browser_type: &str, ) -> bool { // Windows implementation would go here false } pub async fn launch_browser_process( executable_path: &std::path::Path, args: &[String], ) -> Result> { println!( "Launching browser on Windows: {:?} with args: {:?}", executable_path, args ); Ok(Command::new(executable_path).args(args).spawn()?) } pub async fn open_url_in_existing_browser_firefox_like( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let browser = create_browser(browser_type); let executable_path = browser .get_executable_path(browser_dir) .map_err(|e| format!("Failed to get executable path: {}", e))?; let output = Command::new(executable_path) .args(["-profile", &profile.profile_path, "-new-tab", url]) .output()?; if !output.status.success() { return Err( format!( "Failed to open URL in existing browser: {}", String::from_utf8_lossy(&output.stderr) ) .into(), ); } Ok(()) } pub async fn open_url_in_existing_browser_tor_mullvad( _profile: &BrowserProfile, _url: &str, _browser_type: BrowserType, _browser_dir: &Path, ) -> Result<(), Box> { Err("Opening URLs in existing Firefox-based browsers is not supported on Windows when using -no-remote".into()) } pub async fn open_url_in_existing_browser_chromium( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let browser = create_browser(browser_type); let executable_path = browser .get_executable_path(browser_dir) .map_err(|e| format!("Failed to get executable path: {}", e))?; let output = Command::new(executable_path) .args([&format!("--user-data-dir={}", profile.profile_path), url]) .output()?; if !output.status.success() { return Err( format!( "Failed to open URL in existing Chromium-based browser: {}", String::from_utf8_lossy(&output.stderr) ) .into(), ); } Ok(()) } pub async fn kill_browser_process_impl( pid: u32, ) -> Result<(), Box> { let system = System::new_all(); if let Some(process) = system.process(Pid::from(pid as usize)) { if !process.kill() { return Err(format!("Failed to kill process {}", pid).into()); } } else { return Err(format!("Process {} not found", pid).into()); } println!("Successfully killed browser process with PID: {pid}"); Ok(()) } } #[cfg(target_os = "linux")] mod linux { use super::*; use std::ffi::OsString; pub fn is_tor_or_mullvad_browser( _exe_name: &str, _cmd: &[OsString], _browser_type: &str, ) -> bool { // Linux implementation would go here false } pub async fn launch_browser_process( executable_path: &std::path::Path, args: &[String], ) -> Result> { println!( "Launching browser on Linux: {:?} with args: {:?}", executable_path, args ); // Check if the executable exists and is executable if !executable_path.exists() { return Err(format!("Browser executable not found: {:?}", executable_path).into()); } // Check if we can read the executable to detect architecture issues early if let Err(e) = std::fs::File::open(executable_path) { return Err(format!("Cannot access browser executable: {}", e).into()); } // Ensure the executable has proper permissions if let Err(e) = std::fs::metadata(executable_path) { return Err(format!("Cannot get executable metadata: {}", e).into()); } // On Linux, we might need to set LD_LIBRARY_PATH for some browsers let mut cmd = Command::new(executable_path); cmd.args(args); // For Firefox-based browsers, ensure library path includes the installation directory if let Some(install_dir) = executable_path.parent() { let mut ld_library_path = Vec::new(); // Add multiple potential library directories let lib_dirs = [ install_dir.join("lib"), install_dir.join("../lib"), // Parent directory lib install_dir.join("../../lib"), // Grandparent directory lib install_dir.to_path_buf(), // Installation directory itself ]; for lib_dir in &lib_dirs { if lib_dir.exists() { ld_library_path.push(lib_dir.to_string_lossy().to_string()); } } // For Firefox specifically, add common system library paths that might be needed let firefox_lib_paths = [ "/usr/lib/firefox", "/usr/lib/x86_64-linux-gnu", "/usr/lib/aarch64-linux-gnu", "/lib/x86_64-linux-gnu", "/lib/aarch64-linux-gnu", ]; for lib_path in &firefox_lib_paths { let path = std::path::Path::new(lib_path); if path.exists() { ld_library_path.push(lib_path.to_string()); } } // Preserve existing LD_LIBRARY_PATH if let Ok(existing_path) = std::env::var("LD_LIBRARY_PATH") { ld_library_path.push(existing_path); } // Set the combined LD_LIBRARY_PATH if !ld_library_path.is_empty() { cmd.env("LD_LIBRARY_PATH", ld_library_path.join(":")); println!("Set LD_LIBRARY_PATH to: {}", ld_library_path.join(":")); } } // Additional Linux-specific environment variables for better compatibility cmd.env( "DISPLAY", std::env::var("DISPLAY").unwrap_or(":0".to_string()), ); // Set MOZ_ENABLE_WAYLAND for better Wayland support if std::env::var("WAYLAND_DISPLAY").is_ok() { cmd.env("MOZ_ENABLE_WAYLAND", "1"); } // Disable GPU acceleration if running in headless environments if std::env::var("DISPLAY").is_err() || std::env::var("WAYLAND_DISPLAY").is_err() { println!("No display detected, browser may fail to start"); } // Attempt to spawn with better error handling for architecture issues match cmd.spawn() { Ok(child) => Ok(child), Err(e) => { // Detect architecture mismatch errors if e.kind() == std::io::ErrorKind::Other { let error_msg = e.to_string(); if error_msg.contains("Exec format error") { return Err(format!( "Architecture mismatch: The browser executable is not compatible with your system architecture ({}). \ This typically happens when trying to run x86_64 binaries on ARM64 systems. \ Please use a browser that supports your architecture, such as Zen Browser or Brave. \ Executable: {:?}", std::env::consts::ARCH, executable_path ).into()); } else if error_msg.contains("No such file or directory") { return Err(format!( "Executable or required library not found. This might be due to missing dependencies or incorrect executable path. \ Try installing missing libraries or verify the browser installation. \ Executable: {:?}, Error: {}", executable_path, error_msg ).into()); } } Err(format!("Failed to launch browser: {}", e).into()) } } } pub async fn open_url_in_existing_browser_firefox_like( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let browser = create_browser(browser_type); let executable_path = browser .get_executable_path(browser_dir) .map_err(|e| format!("Failed to get executable path: {}", e))?; let output = Command::new(executable_path) .args(["-profile", &profile.profile_path, "-new-tab", url]) .output()?; if !output.status.success() { return Err( format!( "Failed to open URL in existing browser: {}", String::from_utf8_lossy(&output.stderr) ) .into(), ); } Ok(()) } pub async fn open_url_in_existing_browser_tor_mullvad( _profile: &BrowserProfile, _url: &str, _browser_type: BrowserType, _browser_dir: &Path, ) -> Result<(), Box> { Err("Opening URLs in existing Firefox-based browsers is not supported on Linux when using -no-remote".into()) } pub async fn open_url_in_existing_browser_chromium( profile: &BrowserProfile, url: &str, browser_type: BrowserType, browser_dir: &Path, ) -> Result<(), Box> { let browser = create_browser(browser_type); let executable_path = browser .get_executable_path(browser_dir) .map_err(|e| format!("Failed to get executable path: {}", e))?; let output = Command::new(executable_path) .args([&format!("--user-data-dir={}", profile.profile_path), url]) .output()?; if !output.status.success() { return Err( format!( "Failed to open URL in existing Chromium-based browser: {}", String::from_utf8_lossy(&output.stderr) ) .into(), ); } Ok(()) } pub async fn kill_browser_process_impl( pid: u32, ) -> Result<(), Box> { let system = System::new_all(); if let Some(process) = system.process(Pid::from(pid as usize)) { if !process.kill() { return Err(format!("Failed to kill process {}", pid).into()); } } else { return Err(format!("Process {} not found", pid).into()); } println!("Successfully killed browser process with PID: {pid}"); Ok(()) } } pub struct BrowserRunner { base_dirs: BaseDirs, } impl BrowserRunner { pub fn new() -> Self { Self { base_dirs: BaseDirs::new().expect("Failed to get base directories"), } } // Helper function to check if a process matches TOR/Mullvad browser fn is_tor_or_mullvad_browser( &self, exe_name: &str, cmd: &[std::ffi::OsString], browser_type: &str, ) -> bool { #[cfg(target_os = "macos")] return macos::is_tor_or_mullvad_browser(exe_name, cmd, browser_type); #[cfg(target_os = "windows")] return windows::is_tor_or_mullvad_browser(exe_name, cmd, browser_type); #[cfg(target_os = "linux")] return linux::is_tor_or_mullvad_browser(exe_name, cmd, browser_type); #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] { let _ = (exe_name, cmd, browser_type); false } } pub fn get_binaries_dir(&self) -> PathBuf { let mut path = self.base_dirs.data_local_dir().to_path_buf(); path.push(if cfg!(debug_assertions) { "DonutBrowserDev" } else { "DonutBrowser" }); path.push("binaries"); path } pub fn get_profiles_dir(&self) -> PathBuf { let mut path = self.base_dirs.data_local_dir().to_path_buf(); path.push(if cfg!(debug_assertions) { "DonutBrowserDev" } else { "DonutBrowser" }); path.push("profiles"); path } pub fn create_profile( &self, name: &str, browser: &str, version: &str, proxy: Option, ) -> Result> { // Check if a profile with this name already exists (case insensitive) let existing_profiles = self.list_profiles()?; if existing_profiles .iter() .any(|p| p.name.to_lowercase() == name.to_lowercase()) { return Err(format!("Profile with name '{name}' already exists").into()); } let snake_case_name = name.to_lowercase().replace(" ", "_"); // Create profile directory let mut profile_path = self.get_profiles_dir(); profile_path.push(&snake_case_name); create_dir_all(&profile_path)?; let profile = BrowserProfile { name: name.to_string(), browser: browser.to_string(), version: version.to_string(), profile_path: profile_path.to_string_lossy().to_string(), proxy: proxy.clone(), process_id: None, last_launch: None, }; // Save profile info self.save_profile(&profile)?; // Create user.js with common Firefox preferences and apply proxy settings if provided if let Some(proxy_settings) = &proxy { self.apply_proxy_settings_to_profile(&profile_path, proxy_settings, None)?; } else { // Create user.js with common Firefox preferences but no proxy self.disable_proxy_settings_in_profile(&profile_path)?; } Ok(profile) } pub fn update_profile_proxy( &self, profile_name: &str, proxy: Option, ) -> Result> { let profiles_dir = self.get_profiles_dir(); let profile_file = profiles_dir.join(format!( "{}.json", profile_name.to_lowercase().replace(" ", "_") )); let profile_path = profiles_dir.join(profile_name.to_lowercase().replace(" ", "_")); if !profile_file.exists() { return Err(format!("Profile {profile_name} not found").into()); } // Read the profile let content = fs::read_to_string(&profile_file)?; let mut profile: BrowserProfile = serde_json::from_str(&content)?; // Update proxy settings profile.proxy = proxy.clone(); // Save the updated profile self.save_profile(&profile)?; // Get internal proxy if the browser is running let internal_proxy = if let Some(pid) = profile.process_id { PROXY_MANAGER.get_proxy_settings(pid) } else { None }; // Apply proxy settings if provided if let Some(proxy_settings) = &proxy { self.apply_proxy_settings_to_profile( &profile_path, proxy_settings, internal_proxy.as_ref(), )?; } else { self.disable_proxy_settings_in_profile(&profile_path)?; } Ok(profile) } pub fn update_profile_version( &self, profile_name: &str, version: &str, ) -> Result> { let profiles_dir = self.get_profiles_dir(); let profile_file = profiles_dir.join(format!( "{}.json", profile_name.to_lowercase().replace(" ", "_") )); if !profile_file.exists() { return Err(format!("Profile {profile_name} not found").into()); } // Read the profile let content = fs::read_to_string(&profile_file)?; let mut profile: BrowserProfile = serde_json::from_str(&content)?; // Check if the browser is currently running if profile.process_id.is_some() { return Err( "Cannot update version while browser is running. Please stop the browser first.".into(), ); } // Verify the new version is downloaded let browser_type = BrowserType::from_str(&profile.browser) .map_err(|_| format!("Invalid browser type: {}", profile.browser))?; let browser = create_browser(browser_type.clone()); let binaries_dir = self.get_binaries_dir(); if !browser.is_version_downloaded(version, &binaries_dir) { return Err(format!("Browser version {version} is not downloaded").into()); } // Update version profile.version = version.to_string(); // Save the updated profile self.save_profile(&profile)?; // Check if auto-delete of unused binaries is enabled let settings_manager = crate::settings_manager::SettingsManager::new(); if let Ok(settings) = settings_manager.load_settings() { if settings.auto_delete_unused_binaries { // Perform cleanup in the background let _ = self.cleanup_unused_binaries_internal(); } } Ok(profile) } /// Internal method to cleanup unused binaries (used by auto-cleanup) fn cleanup_unused_binaries_internal(&self) -> Result, Box> { // Load current profiles let profiles = self.list_profiles()?; // Load registry let mut registry = crate::downloaded_browsers::DownloadedBrowsersRegistry::load()?; // Get active browser versions let active_versions = registry.get_active_browser_versions(&profiles); // Cleanup unused binaries let cleaned_up = registry.cleanup_unused_binaries(&active_versions)?; // Save updated registry registry.save()?; Ok(cleaned_up) } fn get_common_firefox_preferences(&self) -> Vec { vec![ // Disable default browser check "user_pref(\"browser.shell.checkDefaultBrowser\", false);".to_string(), // Disable automatic updates "user_pref(\"app.update.enabled\", false);".to_string(), "user_pref(\"app.update.auto\", false);".to_string(), "user_pref(\"app.update.mode\", 0);".to_string(), "user_pref(\"app.update.service.enabled\", false);".to_string(), "user_pref(\"app.update.silent\", false);".to_string(), // Disable update checking entirely "user_pref(\"app.update.checkInstallTime\", false);".to_string(), "user_pref(\"app.update.url\", \"\");".to_string(), "user_pref(\"app.update.url.manual\", \"\");".to_string(), "user_pref(\"app.update.url.details\", \"\");".to_string(), // Disable background update downloads "user_pref(\"app.update.download.attemptOnce\", false);".to_string(), "user_pref(\"app.update.idletime\", -1);".to_string(), // Additional update-related preferences for completeness "user_pref(\"security.tls.insecure_fallback_hosts\", \"\");".to_string(), "user_pref(\"app.update.staging.enabled\", false);".to_string(), ] } fn apply_proxy_settings_to_profile( &self, profile_path: &Path, proxy: &ProxySettings, internal_proxy: Option<&ProxySettings>, ) -> Result<(), Box> { let user_js_path = profile_path.join("user.js"); let mut preferences = Vec::new(); // Add common Firefox preferences (like disabling default browser check) preferences.extend(self.get_common_firefox_preferences()); if proxy.enabled { // Create PAC file from template let template_path = Path::new("assets/template.pac"); let pac_content = fs::read_to_string(template_path)?; // Format proxy URL based on type and whether we have an internal proxy let proxy_url = if let Some(internal) = internal_proxy { // Use internal proxy as the primary proxy format!("HTTP {}:{}", internal.host, internal.port) } else { // Use user-configured proxy directly match proxy.proxy_type.as_str() { "http" => format!("HTTP {}:{}", proxy.host, proxy.port), "https" => format!("HTTPS {}:{}", proxy.host, proxy.port), "socks4" => format!("SOCKS4 {}:{}", proxy.host, proxy.port), "socks5" => format!("SOCKS5 {}:{}", proxy.host, proxy.port), _ => return Err(format!("Unsupported proxy type: {}", proxy.proxy_type).into()), } }; // Replace placeholders in PAC file let pac_content = pac_content .replace("{{proxy_url}}", &proxy_url) .replace("{{proxy_credentials}}", ""); // Credentials are now handled by the PAC file // Save PAC file in profile directory let pac_path = profile_path.join("proxy.pac"); fs::write(&pac_path, pac_content)?; // Configure Firefox to use the PAC file preferences.extend([ "user_pref(\"network.proxy.type\", 2);".to_string(), format!( "user_pref(\"network.proxy.autoconfig_url\", \"file://{}\");", pac_path.to_string_lossy() ), "user_pref(\"network.proxy.failover_direct\", false);".to_string(), "user_pref(\"network.proxy.socks_remote_dns\", true);".to_string(), "user_pref(\"network.proxy.no_proxies_on\", \"\");".to_string(), "user_pref(\"signon.autologin.proxy\", true);".to_string(), "user_pref(\"network.proxy.share_proxy_settings\", false);".to_string(), "user_pref(\"network.automatic-ntlm-auth.allow-proxies\", false);".to_string(), "user_pref(\"network.auth-use-sspi\", false);".to_string(), ]); } else { preferences.push("user_pref(\"network.proxy.type\", 0);".to_string()); preferences.push("user_pref(\"network.proxy.failover_direct\", true);".to_string()); let pac_content = "function FindProxyForURL(url, host) { return 'DIRECT'; }"; let pac_path = profile_path.join("proxy.pac"); fs::write(&pac_path, pac_content)?; preferences.push(format!( "user_pref(\"network.proxy.autoconfig_url\", \"file://{}\");", pac_path.to_string_lossy() )); } // Write settings to user.js file fs::write(user_js_path, preferences.join("\n"))?; Ok(()) } pub fn disable_proxy_settings_in_profile( &self, profile_path: &Path, ) -> Result<(), Box> { let user_js_path = profile_path.join("user.js"); let mut preferences = Vec::new(); // Add common Firefox preferences (like disabling default browser check) preferences.extend(self.get_common_firefox_preferences()); preferences.push("user_pref(\"network.proxy.type\", 0);".to_string()); preferences.push("user_pref(\"network.proxy.failover_direct\", true);".to_string()); fs::write(user_js_path, preferences.join("\n"))?; Ok(()) } pub fn save_profile(&self, profile: &BrowserProfile) -> Result<(), Box> { let profiles_dir = self.get_profiles_dir(); let profile_file = profiles_dir.join(format!( "{}.json", profile.name.to_lowercase().replace(" ", "_") )); let json = serde_json::to_string_pretty(profile)?; fs::write(profile_file, json)?; Ok(()) } pub fn list_profiles(&self) -> Result, Box> { let profiles_dir = self.get_profiles_dir(); if !profiles_dir.exists() { return Ok(vec![]); } let mut profiles = Vec::new(); for entry in fs::read_dir(profiles_dir)? { let entry = entry?; let path = entry.path(); if path.extension().is_some_and(|ext| ext == "json") { let content = fs::read_to_string(path)?; let profile: BrowserProfile = serde_json::from_str(&content)?; profiles.push(profile); } } Ok(profiles) } pub async fn launch_browser( &self, profile: &BrowserProfile, url: Option, ) -> Result> { // Create browser instance let browser_type = BrowserType::from_str(&profile.browser) .map_err(|_| format!("Invalid browser type: {}", profile.browser))?; let browser = create_browser(browser_type.clone()); // Get executable path - path structure: binaries/// let mut browser_dir = self.get_binaries_dir(); browser_dir.push(&profile.browser); browser_dir.push(&profile.version); let executable_path = browser .get_executable_path(&browser_dir) .expect("Failed to get executable path"); // Prepare the executable (set permissions, etc.) if let Err(e) = browser.prepare_executable(&executable_path) { println!("Warning: Failed to prepare executable: {e}"); // Continue anyway, the error might not be critical } // Get launch arguments let browser_args = browser .create_launch_args(&profile.profile_path, profile.proxy.as_ref(), url) .expect("Failed to create launch arguments"); // Launch browser using platform-specific method let child = { #[cfg(target_os = "macos")] { macos::launch_browser_process(&executable_path, &browser_args).await? } #[cfg(target_os = "windows")] { windows::launch_browser_process(&executable_path, &browser_args).await? } #[cfg(target_os = "linux")] { linux::launch_browser_process(&executable_path, &browser_args).await? } #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] { return Err("Unsupported platform for browser launching".into()); } }; let launcher_pid = child.id(); println!( "Launched browser with launcher PID: {} for profile: {}", launcher_pid, profile.name ); // For TOR and Mullvad browsers, we need to find the actual browser process // because they use launcher scripts that spawn the real browser process let actual_pid = if matches!( browser_type, BrowserType::TorBrowser | BrowserType::MullvadBrowser ) { println!("Waiting for TOR/Mullvad browser to fully start..."); // Wait a bit for the browser to fully start tokio::time::sleep(tokio::time::Duration::from_millis(3000)).await; // Search for the actual browser process let system = System::new_all(); let mut found_pid: Option = None; // Try multiple times to find the process as it might take time to start for attempt in 1..=5 { println!("Attempt {attempt} to find actual browser process..."); for (pid, process) in system.processes() { let cmd = process.cmd(); if cmd.len() >= 2 { // Check if this is the right browser executable let exe_name = process.name().to_string_lossy().to_lowercase(); let is_correct_browser = match profile.browser.as_str() { "mullvad-browser" => { self.is_tor_or_mullvad_browser(&exe_name, cmd, "mullvad-browser") } "tor-browser" => self.is_tor_or_mullvad_browser(&exe_name, cmd, "tor-browser"), _ => false, }; if !is_correct_browser { continue; } // Check for profile path match let profile_path_match = cmd.iter().any(|s| { let arg = s.to_str().unwrap_or(""); arg == profile.profile_path || arg == format!("-profile={}", profile.profile_path) || (arg == "-profile" && cmd .iter() .any(|s2| s2.to_str().unwrap_or("") == profile.profile_path)) }); if profile_path_match { found_pid = Some(pid.as_u32()); println!( "Found actual browser process with PID: {} for profile: {}", pid.as_u32(), profile.name ); break; } } } if found_pid.is_some() { break; } // Wait before next attempt tokio::time::sleep(tokio::time::Duration::from_millis(1000)).await; } found_pid.unwrap_or(launcher_pid) } else { // For other browsers, the launcher PID is usually the actual browser PID launcher_pid }; // Update profile with process info let mut updated_profile = profile.clone(); updated_profile.process_id = Some(actual_pid); updated_profile.last_launch = Some(SystemTime::now().duration_since(UNIX_EPOCH)?.as_secs()); // Save the updated profile self .save_process_info(&updated_profile) .expect("Failed to save process info"); println!( "Browser launched successfully with PID: {} for profile: {}", actual_pid, profile.name ); Ok(updated_profile) } pub async fn open_url_in_existing_browser( &self, app_handle: tauri::AppHandle, profile: &BrowserProfile, url: &str, ) -> Result<(), Box> { // Use the comprehensive browser status check let is_running = self.check_browser_status(app_handle, profile).await?; if !is_running { return Err("Browser is not running".into()); } // Get the updated profile with current PID let profiles = self.list_profiles().expect("Failed to list profiles"); let updated_profile = profiles .into_iter() .find(|p| p.name == profile.name) .unwrap_or_else(|| profile.clone()); // Ensure we have a valid process ID if updated_profile.process_id.is_none() { return Err("No valid process ID found for the browser".into()); } let browser_type = BrowserType::from_str(&updated_profile.browser) .map_err(|_| format!("Invalid browser type: {}", updated_profile.browser))?; // Get browser directory for all platforms - path structure: binaries/// let mut browser_dir = self.get_binaries_dir(); browser_dir.push(&updated_profile.browser); browser_dir.push(&updated_profile.version); match browser_type { BrowserType::Firefox | BrowserType::FirefoxDeveloper | BrowserType::Zen => { #[cfg(target_os = "macos")] return macos::open_url_in_existing_browser_firefox_like( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "windows")] return windows::open_url_in_existing_browser_firefox_like( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "linux")] return linux::open_url_in_existing_browser_firefox_like( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] return Err("Unsupported platform".into()); } BrowserType::MullvadBrowser | BrowserType::TorBrowser => { #[cfg(target_os = "macos")] return macos::open_url_in_existing_browser_tor_mullvad( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "windows")] return windows::open_url_in_existing_browser_tor_mullvad( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "linux")] return linux::open_url_in_existing_browser_tor_mullvad( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] return Err("Unsupported platform".into()); } BrowserType::Chromium | BrowserType::Brave => { #[cfg(target_os = "macos")] return macos::open_url_in_existing_browser_chromium( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "windows")] return windows::open_url_in_existing_browser_chromium( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(target_os = "linux")] return linux::open_url_in_existing_browser_chromium( &updated_profile, url, browser_type, &browser_dir, ) .await; #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] return Err("Unsupported platform".into()); } } } pub async fn launch_or_open_url( &self, app_handle: tauri::AppHandle, profile: &BrowserProfile, url: Option, ) -> Result> { // Get the most up-to-date profile data let profiles = self.list_profiles().expect("Failed to list profiles"); let updated_profile = profiles .into_iter() .find(|p| p.name == profile.name) .unwrap_or_else(|| profile.clone()); // Check if browser is already running let is_running = self .check_browser_status(app_handle.clone(), &updated_profile) .await?; // Get the updated profile again after status check (PID might have been updated) let profiles = self.list_profiles().expect("Failed to list profiles"); let final_profile = profiles .into_iter() .find(|p| p.name == profile.name) .unwrap_or_else(|| updated_profile.clone()); println!( "Browser status check - Profile: {}, Running: {}, URL: {:?}, PID: {:?}", final_profile.name, is_running, url, final_profile.process_id ); if is_running && url.is_some() { // Browser is running and we have a URL to open if let Some(url_ref) = url.as_ref() { println!("Opening URL in existing browser: {url_ref}"); // For TOR/Mullvad browsers, add extra verification if matches!( final_profile.browser.as_str(), "tor-browser" | "mullvad-browser" ) { println!("TOR/Mullvad browser detected - ensuring we have correct PID"); if final_profile.process_id.is_none() { println!( "ERROR: No PID found for running TOR/Mullvad browser - this should not happen" ); return Err("No PID found for running browser".into()); } } match self .open_url_in_existing_browser(app_handle, &final_profile, url_ref) .await { Ok(()) => { println!("Successfully opened URL in existing browser"); Ok(final_profile) } Err(e) => { println!("Failed to open URL in existing browser: {e}"); // For Mullvad and Tor browsers, don't fall back to new instance since they use -no-remote // and can't have multiple instances with the same profile match final_profile.browser.as_str() { "mullvad-browser" | "tor-browser" => { Err(format!( "Failed to open URL in existing {} browser. Cannot launch new instance due to profile conflict: {}", final_profile.browser, e ).into()) } _ => { println!( "Falling back to new instance for browser: {}", final_profile.browser ); // Fallback to launching a new instance for other browsers self.launch_browser(&final_profile, url).await } } } } } else { // This case shouldn't happen since we checked is_some() above, but handle it gracefully println!("URL was unexpectedly None, launching new browser instance"); self.launch_browser(&final_profile, url).await } } else { // Browser is not running or no URL provided, launch new instance if !is_running { println!("Launching new browser instance - browser not running"); } else { println!("Launching new browser instance - no URL provided"); } self.launch_browser(&final_profile, url).await } } pub fn rename_profile( &self, old_name: &str, new_name: &str, ) -> Result> { let profiles_dir = self.get_profiles_dir(); let old_profile_file = profiles_dir.join(format!( "{}.json", old_name.to_lowercase().replace(" ", "_") )); let old_profile_path = profiles_dir.join(old_name.to_lowercase().replace(" ", "_")); // Check if new name already exists (case insensitive) let existing_profiles = self.list_profiles()?; if existing_profiles .iter() .any(|p| p.name.to_lowercase() == new_name.to_lowercase()) { return Err(format!("Profile with name '{new_name}' already exists").into()); } // Read the profile let content = fs::read_to_string(&old_profile_file)?; let mut profile: BrowserProfile = serde_json::from_str(&content)?; // Update profile name profile.name = new_name.to_string(); // Create new paths let _ = profiles_dir.join(format!( "{}.json", new_name.to_lowercase().replace(" ", "_") )); let new_profile_path = profiles_dir.join(new_name.to_lowercase().replace(" ", "_")); // Rename directory if old_profile_path.exists() { fs::rename(&old_profile_path, &new_profile_path)?; } // Update profile path profile.profile_path = new_profile_path.to_string_lossy().to_string(); // Save profile with new name self.save_profile(&profile)?; // Delete old profile file if old_profile_file.exists() { fs::remove_file(old_profile_file)?; } Ok(profile) } fn save_process_info(&self, profile: &BrowserProfile) -> Result<(), Box> { let profiles_dir = self.get_profiles_dir(); let profile_file = profiles_dir.join(format!( "{}.json", profile.name.to_lowercase().replace(" ", "_") )); let json = serde_json::to_string_pretty(&profile)?; fs::write(profile_file, json)?; Ok(()) } pub fn delete_profile(&self, profile_name: &str) -> Result<(), Box> { let profiles_dir = self.get_profiles_dir(); let profile_file = profiles_dir.join(format!( "{}.json", profile_name.to_lowercase().replace(" ", "_") )); let profile_path = profiles_dir.join(profile_name.to_lowercase().replace(" ", "_")); // Delete profile directory if profile_path.exists() { fs::remove_dir_all(profile_path)? } // Delete profile JSON file if profile_file.exists() { fs::remove_file(profile_file)? } // Check if auto-delete of unused binaries is enabled let settings_manager = crate::settings_manager::SettingsManager::new(); if let Ok(settings) = settings_manager.load_settings() { if settings.auto_delete_unused_binaries { // Perform cleanup in the background after profile deletion let _ = self.cleanup_unused_binaries_internal(); } } Ok(()) } pub async fn check_browser_status( &self, app_handle: tauri::AppHandle, profile: &BrowserProfile, ) -> Result> { let mut inner_profile = profile.clone(); let system = System::new_all(); let mut is_running = false; let mut found_pid: Option = None; // First check if the stored PID is still valid if let Some(pid) = profile.process_id { if let Some(process) = system.process(Pid::from(pid as usize)) { let cmd = process.cmd(); // Verify this process is actually our browser with the correct profile let profile_path_match = cmd.iter().any(|s| { let arg = s.to_str().unwrap_or(""); // For Firefox-based browsers, check for exact profile path match if profile.browser == "tor-browser" || profile.browser == "firefox" || profile.browser == "firefox-developer" || profile.browser == "mullvad-browser" || profile.browser == "zen" { arg == profile.profile_path || arg == format!("-profile={}", profile.profile_path) || (arg == "-profile" && cmd .iter() .any(|s2| s2.to_str().unwrap_or("") == profile.profile_path)) } else { // For Chromium-based browsers, check for user-data-dir arg.contains(&format!("--user-data-dir={}", profile.profile_path)) || arg == profile.profile_path } }); if profile_path_match { is_running = true; found_pid = Some(pid); println!( "Found existing browser process with PID: {} for profile: {}", pid, profile.name ); } else { println!("PID {pid} exists but doesn't match our profile path exactly, searching for correct process..."); } } else { println!("Stored PID {pid} no longer exists, searching for browser process..."); } } // If we didn't find the browser with the stored PID, search all processes if !is_running { for (pid, process) in system.processes() { let cmd = process.cmd(); if cmd.len() >= 2 { // Check if this is the right browser executable first let exe_name = process.name().to_string_lossy().to_lowercase(); let is_correct_browser = match profile.browser.as_str() { "firefox" => { exe_name.contains("firefox") && !exe_name.contains("developer") && !exe_name.contains("tor") && !exe_name.contains("mullvad") } "firefox-developer" => exe_name.contains("firefox") && exe_name.contains("developer"), "mullvad-browser" => self.is_tor_or_mullvad_browser(&exe_name, cmd, "mullvad-browser"), "tor-browser" => self.is_tor_or_mullvad_browser(&exe_name, cmd, "tor-browser"), "zen" => exe_name.contains("zen"), "chromium" => exe_name.contains("chromium"), "brave" => exe_name.contains("brave"), _ => false, }; if !is_correct_browser { continue; } // Check for profile path match let profile_path_match = cmd.iter().any(|s| { let arg = s.to_str().unwrap_or(""); // For Firefox-based browsers, check for exact profile path match if profile.browser == "tor-browser" || profile.browser == "firefox" || profile.browser == "firefox-developer" || profile.browser == "mullvad-browser" || profile.browser == "zen" { arg == profile.profile_path || arg == format!("-profile={}", profile.profile_path) || (arg == "-profile" && cmd .iter() .any(|s2| s2.to_str().unwrap_or("") == profile.profile_path)) } else { // For Chromium-based browsers, check for user-data-dir arg.contains(&format!("--user-data-dir={}", profile.profile_path)) || arg == profile.profile_path } }); if profile_path_match { // Found a matching process found_pid = Some(pid.as_u32()); is_running = true; println!( "Found browser process with PID: {} for profile: {}", pid.as_u32(), profile.name ); break; } } } } // Update the process ID if we found a different one if let Some(pid) = found_pid { if inner_profile.process_id != Some(pid) { inner_profile.process_id = Some(pid); if let Err(e) = self.save_process_info(&inner_profile) { println!("Warning: Failed to update process info: {e}"); } else { println!( "Updated process ID for profile '{}' to: {}", inner_profile.name, pid ); } } } else if is_running { println!("Browser is running but no PID found - this shouldn't happen"); } else { // Browser is not running, clear the PID if it was set if inner_profile.process_id.is_some() { inner_profile.process_id = None; if let Err(e) = self.save_process_info(&inner_profile) { println!("Warning: Failed to clear process info: {e}"); } else { println!("Cleared process ID for profile '{}'", inner_profile.name); } } } // Handle proxy management based on browser status if let Some(proxy) = &inner_profile.proxy { if proxy.enabled { if is_running { // Browser is running, check if proxy is active let proxy_active = PROXY_MANAGER .get_proxy_settings(inner_profile.process_id.unwrap_or(0)) .is_some(); if !proxy_active { // Browser is running but proxy is not - restart the proxy if let Some((upstream_url, _preferred_port)) = PROXY_MANAGER.get_profile_proxy_info(&inner_profile.name) { // Restart the proxy with the same configuration match PROXY_MANAGER .start_proxy( app_handle, &upstream_url, inner_profile.process_id.unwrap(), Some(&inner_profile.name), ) .await { Ok(_) => { println!("Restarted proxy for profile {}", inner_profile.name); } Err(e) => { eprintln!( "Failed to restart proxy for profile {}: {}", inner_profile.name, e ); } } } } } else { // Browser is not running, stop the proxy if it exists if let Some(pid) = profile.process_id { let _ = PROXY_MANAGER.stop_proxy(app_handle, pid).await; } } } } Ok(is_running) } pub async fn kill_browser_process( &self, app_handle: tauri::AppHandle, profile: &BrowserProfile, ) -> Result<(), Box> { // Get the current process ID let pid = if let Some(pid) = profile.process_id { pid } else { // Try to find the process by searching all processes let system = System::new_all(); let mut found_pid: Option = None; for (pid, process) in system.processes() { let cmd = process.cmd(); if cmd.len() >= 2 { // Check if this is the right browser executable first let exe_name = process.name().to_string_lossy().to_lowercase(); let is_correct_browser = match profile.browser.as_str() { "firefox" => { exe_name.contains("firefox") && !exe_name.contains("developer") && !exe_name.contains("tor") && !exe_name.contains("mullvad") } "firefox-developer" => exe_name.contains("firefox") && exe_name.contains("developer"), "mullvad-browser" => self.is_tor_or_mullvad_browser(&exe_name, cmd, "mullvad-browser"), "tor-browser" => self.is_tor_or_mullvad_browser(&exe_name, cmd, "tor-browser"), "zen" => exe_name.contains("zen"), "chromium" => exe_name.contains("chromium"), "brave" => exe_name.contains("brave"), _ => false, }; if !is_correct_browser { continue; } // Check for profile path match let profile_path_match = cmd.iter().any(|s| { let arg = s.to_str().unwrap_or(""); // For Firefox-based browsers, check for exact profile path match if profile.browser == "tor-browser" || profile.browser == "firefox" || profile.browser == "firefox-developer" || profile.browser == "mullvad-browser" || profile.browser == "zen" { arg == profile.profile_path || arg == format!("-profile={}", profile.profile_path) } else { // For Chromium-based browsers, check for user-data-dir arg.contains(&format!("--user-data-dir={}", profile.profile_path)) || arg == profile.profile_path } }); if profile_path_match { found_pid = Some(pid.as_u32()); break; } } } found_pid.ok_or("Browser process not found")? }; println!("Attempting to kill browser process with PID: {pid}"); // Stop any associated proxy first if let Err(e) = PROXY_MANAGER.stop_proxy(app_handle, pid).await { println!("Warning: Failed to stop proxy for PID {pid}: {e}"); } // Kill the process using platform-specific implementation #[cfg(target_os = "macos")] macos::kill_browser_process_impl(pid).await?; #[cfg(target_os = "windows")] windows::kill_browser_process_impl(pid).await?; #[cfg(target_os = "linux")] linux::kill_browser_process_impl(pid).await?; #[cfg(not(any(target_os = "macos", target_os = "windows", target_os = "linux")))] return Err("Unsupported platform".into()); // Clear the process ID from the profile let mut updated_profile = profile.clone(); updated_profile.process_id = None; self .save_process_info(&updated_profile) .map_err(|e| format!("Failed to update profile: {e}"))?; Ok(()) } } #[tauri::command] pub fn create_browser_profile( name: String, browser: String, version: String, proxy: Option, ) -> Result { let browser_runner = BrowserRunner::new(); browser_runner .create_profile(&name, &browser, &version, proxy) .map_err(|e| format!("Failed to create profile: {e}")) } #[tauri::command] pub fn list_browser_profiles() -> Result, String> { let browser_runner = BrowserRunner::new(); browser_runner .list_profiles() .map_err(|e| format!("Failed to list profiles: {e}")) } #[tauri::command] pub async fn launch_browser_profile( app_handle: tauri::AppHandle, profile: BrowserProfile, url: Option, ) -> Result { let browser_runner = BrowserRunner::new(); // Launch browser or open URL in existing instance let updated_profile = browser_runner .launch_or_open_url(app_handle.clone(), &profile, url) .await .map_err(|e| { // Check if this is an architecture compatibility issue if let Some(io_error) = e.downcast_ref::() { if io_error.kind() == std::io::ErrorKind::Other && io_error.to_string().contains("Exec format error") { return format!("Failed to launch browser: Executable format error. This browser version is not compatible with your system architecture ({}). Please try a different browser or version that supports your platform.", std::env::consts::ARCH); } } format!("Failed to launch browser or open URL: {e}") })?; // If the profile has proxy settings, start a proxy for it if let Some(proxy) = &profile.proxy { if proxy.enabled { // Get the process ID if let Some(pid) = updated_profile.process_id { // Start a proxy for the upstream URL let upstream_url = format!("{}://{}:{}", proxy.proxy_type, proxy.host, proxy.port); // Start the proxy match PROXY_MANAGER .start_proxy(app_handle.clone(), &upstream_url, pid, Some(&profile.name)) .await { Ok(internal_proxy_settings) => { let browser_runner = BrowserRunner::new(); let profiles_dir = browser_runner.get_profiles_dir(); let profile_path = profiles_dir.join(profile.name.to_lowercase().replace(" ", "_")); // Apply the proxy settings with the internal proxy browser_runner .apply_proxy_settings_to_profile(&profile_path, proxy, Some(&internal_proxy_settings)) .map_err(|e| format!("Failed to update profile proxy: {e}"))?; } Err(e) => { eprintln!("Failed to start proxy: {e}"); // Continue without proxy } } } } } Ok(updated_profile) } #[tauri::command] pub fn update_profile_proxy( profile_name: String, proxy: Option, ) -> Result { let browser_runner = BrowserRunner::new(); browser_runner .update_profile_proxy(&profile_name, proxy) .map_err(|e| format!("Failed to update profile: {e}")) } #[tauri::command] pub fn update_profile_version( profile_name: String, version: String, ) -> Result { let browser_runner = BrowserRunner::new(); browser_runner .update_profile_version(&profile_name, &version) .map_err(|e| format!("Failed to update profile version: {e}")) } #[tauri::command] pub async fn check_browser_status( app_handle: tauri::AppHandle, profile: BrowserProfile, ) -> Result { let browser_runner = BrowserRunner::new(); browser_runner .check_browser_status(app_handle, &profile) .await .map_err(|e| format!("Failed to check browser status: {e}")) } #[tauri::command] pub fn rename_profile( _app_handle: tauri::AppHandle, old_name: &str, new_name: &str, ) -> Result { let browser_runner = BrowserRunner::new(); browser_runner .rename_profile(old_name, new_name) .map_err(|e| format!("Failed to delete profile: {e}")) } #[tauri::command] pub fn delete_profile(_app_handle: tauri::AppHandle, profile_name: String) -> Result<(), String> { let browser_runner = BrowserRunner::new(); browser_runner .delete_profile(profile_name.as_str()) .map_err(|e| format!("Failed to delete profile: {e}")) } #[tauri::command] pub fn get_supported_browsers() -> Result, String> { let service = BrowserVersionService::new(); Ok(service.get_supported_browsers()) } #[tauri::command] pub fn is_browser_supported_on_platform(browser_str: String) -> Result { let service = BrowserVersionService::new(); service .is_browser_supported(&browser_str) .map_err(|e| format!("Failed to check browser support: {e}")) } #[tauri::command] pub async fn fetch_browser_versions_cached_first( browser_str: String, ) -> Result, String> { let service = BrowserVersionService::new(); // Get cached versions immediately if available if let Some(cached_versions) = service.get_cached_browser_versions_detailed(&browser_str) { // Check if we should update cache in background if service.should_update_cache(&browser_str) { // Start background update but return cached data immediately let service_clone = BrowserVersionService::new(); let browser_str_clone = browser_str.clone(); tokio::spawn(async move { if let Err(e) = service_clone .fetch_browser_versions_detailed(&browser_str_clone, false) .await { eprintln!("Background version update failed for {browser_str_clone}: {e}"); } }); } Ok(cached_versions) } else { // No cache available, fetch fresh service .fetch_browser_versions_detailed(&browser_str, false) .await .map_err(|e| format!("Failed to fetch detailed browser versions: {e}")) } } #[tauri::command] pub async fn fetch_browser_versions_with_count_cached_first( browser_str: String, ) -> Result { let service = BrowserVersionService::new(); // Get cached versions immediately if available if let Some(cached_versions) = service.get_cached_browser_versions(&browser_str) { // Check if we should update cache in background if service.should_update_cache(&browser_str) { // Start background update but return cached data immediately let service_clone = BrowserVersionService::new(); let browser_str_clone = browser_str.clone(); tokio::spawn(async move { if let Err(e) = service_clone .fetch_browser_versions_with_count(&browser_str_clone, false) .await { eprintln!("Background version update failed for {browser_str_clone}: {e}"); } }); } // Return cached data in the expected format Ok(BrowserVersionsResult { versions: cached_versions.clone(), new_versions_count: None, // No new versions when returning cached data total_versions_count: cached_versions.len(), }) } else { // No cache available, fetch fresh service .fetch_browser_versions_with_count(&browser_str, false) .await .map_err(|e| format!("Failed to fetch browser versions: {e}")) } } #[tauri::command] pub async fn download_browser( app_handle: tauri::AppHandle, browser_str: String, version: String, ) -> Result { let browser_runner = BrowserRunner::new(); let browser_type = BrowserType::from_str(&browser_str).map_err(|e| format!("Invalid browser type: {e}"))?; let browser = create_browser(browser_type.clone()); // Load registry and check if already downloaded let mut registry = DownloadedBrowsersRegistry::load() .map_err(|e| format!("Failed to load browser registry: {e}"))?; if registry.is_browser_downloaded(&browser_str, &version) { return Ok(version); } // Check if browser is supported on current platform before attempting download let version_service = BrowserVersionService::new(); if !version_service .is_browser_supported(&browser_str) .unwrap_or(false) { return Err(format!( "Browser '{}' is not supported on your platform ({} {}). Supported browsers: {}", browser_str, std::env::consts::OS, std::env::consts::ARCH, version_service.get_supported_browsers().join(", ") )); } let download_info = version_service .get_download_info(&browser_str, &version) .map_err(|e| format!("Failed to get download info: {e}"))?; // Create browser directory let mut browser_dir = browser_runner.get_binaries_dir(); browser_dir.push(browser_type.as_str()); browser_dir.push(&version); // Clean up any failed previous download if let Err(e) = registry.cleanup_failed_download(&browser_str, &version) { println!("Warning: Failed to cleanup previous download: {e}"); } create_dir_all(&browser_dir).map_err(|e| format!("Failed to create browser directory: {e}"))?; // Mark download as started in registry registry.mark_download_started(&browser_str, &version, browser_dir.clone()); registry .save() .map_err(|e| format!("Failed to save registry: {e}"))?; // Use the new download module let downloader = Downloader::new(); let download_path = match downloader .download_browser( &app_handle, browser_type.clone(), &version, &download_info, &browser_dir, ) .await { Ok(path) => path, Err(e) => { // Clean up failed download let _ = registry.cleanup_failed_download(&browser_str, &version); let _ = registry.save(); return Err(format!("Failed to download browser: {e}")); } }; // Use the new extraction module if download_info.is_archive { let extractor = Extractor::new(); match extractor .extract_browser( &app_handle, browser_type.clone(), &version, &download_path, &browser_dir, ) .await { Ok(_) => { // Clean up the downloaded archive if let Err(e) = std::fs::remove_file(&download_path) { println!("Warning: Could not delete archive file: {e}"); } } Err(e) => { // Clean up failed download let _ = registry.cleanup_failed_download(&browser_str, &version); let _ = registry.save(); return Err(format!("Failed to extract browser: {e}")); } } // Give filesystem a moment to settle after extraction tokio::time::sleep(tokio::time::Duration::from_millis(500)).await; } // Emit verification progress let progress = DownloadProgress { browser: browser_str.clone(), version: version.clone(), downloaded_bytes: 0, total_bytes: None, percentage: 100.0, speed_bytes_per_sec: 0.0, eta_seconds: None, stage: "verifying".to_string(), }; let _ = app_handle.emit("download-progress", &progress); // Verify the browser was downloaded correctly println!("Verifying download for browser: {browser_str}, version: {version}"); // Use the browser's own verification method let binaries_dir = browser_runner.get_binaries_dir(); if !browser.is_version_downloaded(&version, &binaries_dir) { let _ = registry.cleanup_failed_download(&browser_str, &version); let _ = registry.save(); return Err("Browser download completed but verification failed".to_string()); } // Mark download as completed in registry let actual_version = if browser_str == "chromium" { Some(version.clone()) } else { None }; registry .mark_download_completed_with_actual_version(&browser_str, &version, actual_version) .map_err(|e| format!("Failed to mark download as completed: {e}"))?; registry .save() .map_err(|e| format!("Failed to save registry: {e}"))?; // Emit completion let progress = DownloadProgress { browser: browser_str.clone(), version: version.clone(), downloaded_bytes: 0, total_bytes: None, percentage: 100.0, speed_bytes_per_sec: 0.0, eta_seconds: Some(0.0), stage: "completed".to_string(), }; let _ = app_handle.emit("download-progress", &progress); Ok(version) } #[tauri::command] pub fn is_browser_downloaded(browser_str: String, version: String) -> bool { if let Ok(registry) = DownloadedBrowsersRegistry::load() { if registry.is_browser_downloaded(&browser_str, &version) { return true; } } let browser_type = BrowserType::from_str(&browser_str).expect("Invalid browser type"); let browser_runner = BrowserRunner::new(); let browser = create_browser(browser_type.clone()); let binaries_dir = browser_runner.get_binaries_dir(); browser.is_version_downloaded(&version, &binaries_dir) } #[tauri::command] pub fn check_browser_exists(browser_str: String, version: String) -> bool { // This is an alias for is_browser_downloaded to provide clearer semantics for auto-updates is_browser_downloaded(browser_str, version) } #[tauri::command] pub async fn kill_browser_profile( app_handle: tauri::AppHandle, profile: BrowserProfile, ) -> Result<(), String> { let browser_runner = BrowserRunner::new(); browser_runner .kill_browser_process(app_handle, &profile) .await .map_err(|e| format!("Failed to kill browser: {e}")) } #[tauri::command] pub fn create_browser_profile_new( name: String, browser_str: String, version: String, proxy: Option, ) -> Result { let browser_type = BrowserType::from_str(&browser_str).map_err(|e| format!("Invalid browser type: {e}"))?; create_browser_profile(name, browser_type.as_str().to_string(), version, proxy) } #[tauri::command] pub async fn fetch_browser_versions_with_count( browser_str: String, ) -> Result { let service = BrowserVersionService::new(); service .fetch_browser_versions_with_count(&browser_str, false) .await .map_err(|e| format!("Failed to fetch browser versions: {e}")) } #[tauri::command] pub fn get_downloaded_browser_versions(browser_str: String) -> Result, String> { let registry = DownloadedBrowsersRegistry::load() .map_err(|e| format!("Failed to load browser registry: {e}"))?; Ok(registry.get_downloaded_versions(&browser_str)) } #[tauri::command] pub fn cleanup_unused_binaries() -> Result, String> { let browser_runner = BrowserRunner::new(); browser_runner .cleanup_unused_binaries_internal() .map_err(|e| format!("Failed to cleanup unused binaries: {e}")) } #[cfg(test)] mod tests { use super::*; use crate::browser::ProxySettings; use tempfile::TempDir; fn create_test_browser_runner() -> (BrowserRunner, TempDir) { let temp_dir = TempDir::new().unwrap(); // Mock the base directories by setting environment variables std::env::set_var("HOME", temp_dir.path()); let browser_runner = BrowserRunner::new(); (browser_runner, temp_dir) } #[test] fn test_browser_runner_creation() { let (_runner, _temp_dir) = create_test_browser_runner(); // If we get here without panicking, the test passes } #[test] fn test_get_binaries_dir() { let (runner, _temp_dir) = create_test_browser_runner(); let binaries_dir = runner.get_binaries_dir(); assert!(binaries_dir.to_string_lossy().contains("DonutBrowser")); assert!(binaries_dir.to_string_lossy().contains("binaries")); } #[test] fn test_get_profiles_dir() { let (runner, _temp_dir) = create_test_browser_runner(); let profiles_dir = runner.get_profiles_dir(); assert!(profiles_dir.to_string_lossy().contains("DonutBrowser")); assert!(profiles_dir.to_string_lossy().contains("profiles")); } #[test] fn test_create_profile() { let (runner, _temp_dir) = create_test_browser_runner(); let profile = runner .create_profile("Test Profile", "firefox", "139.0", None) .unwrap(); assert_eq!(profile.name, "Test Profile"); assert_eq!(profile.browser, "firefox"); assert_eq!(profile.version, "139.0"); assert!(profile.proxy.is_none()); assert!(profile.process_id.is_none()); } #[test] fn test_create_profile_with_proxy() { let (runner, _temp_dir) = create_test_browser_runner(); let proxy = ProxySettings { enabled: true, proxy_type: "http".to_string(), host: "127.0.0.1".to_string(), port: 8080, }; let profile = runner .create_profile( "Test Profile with Proxy", "firefox", "139.0", Some(proxy.clone()), ) .unwrap(); assert_eq!(profile.name, "Test Profile with Proxy"); assert!(profile.proxy.is_some()); let profile_proxy = profile.proxy.unwrap(); assert_eq!(profile_proxy.proxy_type, "http"); assert_eq!(profile_proxy.host, "127.0.0.1"); assert_eq!(profile_proxy.port, 8080); } #[test] fn test_save_and_load_profile() { let (runner, _temp_dir) = create_test_browser_runner(); let profile = runner .create_profile("Test Save Load", "firefox", "139.0", None) .unwrap(); // Save the profile runner.save_profile(&profile).unwrap(); // Load profiles and verify let profiles = runner.list_profiles().unwrap(); assert_eq!(profiles.len(), 1); assert_eq!(profiles[0].name, "Test Save Load"); assert_eq!(profiles[0].browser, "firefox"); assert_eq!(profiles[0].version, "139.0"); } #[test] fn test_update_profile_proxy() { let (runner, _temp_dir) = create_test_browser_runner(); // Create profile without proxy let profile = runner .create_profile("Test Update Proxy", "firefox", "139.0", None) .unwrap(); assert!(profile.proxy.is_none()); // Update with proxy let proxy = ProxySettings { enabled: true, proxy_type: "socks5".to_string(), host: "192.168.1.1".to_string(), port: 1080, }; let updated_profile = runner .update_profile_proxy("Test Update Proxy", Some(proxy.clone())) .unwrap(); assert!(updated_profile.proxy.is_some()); let profile_proxy = updated_profile.proxy.unwrap(); assert_eq!(profile_proxy.proxy_type, "socks5"); assert_eq!(profile_proxy.host, "192.168.1.1"); assert_eq!(profile_proxy.port, 1080); } #[test] fn test_rename_profile() { let (runner, _temp_dir) = create_test_browser_runner(); // Create profile let _ = runner .create_profile("Original Name", "firefox", "139.0", None) .unwrap(); // Rename profile let renamed_profile = runner.rename_profile("Original Name", "New Name").unwrap(); assert_eq!(renamed_profile.name, "New Name"); // Verify old profile is gone and new one exists let profiles = runner.list_profiles().unwrap(); assert_eq!(profiles.len(), 1); assert_eq!(profiles[0].name, "New Name"); } #[test] fn test_delete_profile() { let (runner, _temp_dir) = create_test_browser_runner(); // Create profile let _ = runner .create_profile("To Delete", "firefox", "139.0", None) .unwrap(); // Verify profile exists let profiles = runner.list_profiles().unwrap(); assert_eq!(profiles.len(), 1); // Delete profile runner.delete_profile("To Delete").unwrap(); // Verify profile is gone let profiles = runner.list_profiles().unwrap(); assert_eq!(profiles.len(), 0); } #[test] fn test_profile_name_sanitization() { let (runner, _temp_dir) = create_test_browser_runner(); // Create profile with spaces and special characters let profile = runner .create_profile("Test Profile With Spaces", "firefox", "139.0", None) .unwrap(); // Profile path should use snake_case assert!(profile.profile_path.contains("test_profile_with_spaces")); } #[test] fn test_multiple_profiles() { let (runner, _temp_dir) = create_test_browser_runner(); // Create multiple profiles let _ = runner .create_profile("Profile 1", "firefox", "139.0", None) .unwrap(); let _ = runner .create_profile("Profile 2", "chromium", "1465660", None) .unwrap(); let _ = runner .create_profile("Profile 3", "brave", "v1.81.9", None) .unwrap(); // List profiles let profiles = runner.list_profiles().unwrap(); assert_eq!(profiles.len(), 3); let profile_names: Vec<&str> = profiles.iter().map(|p| p.name.as_str()).collect(); assert!(profile_names.contains(&"Profile 1")); assert!(profile_names.contains(&"Profile 2")); assert!(profile_names.contains(&"Profile 3")); } #[test] fn test_profile_validation() { let (runner, _temp_dir) = create_test_browser_runner(); // Test that we can't rename to an existing profile name let _ = runner .create_profile("Profile 1", "firefox", "139.0", None) .unwrap(); let _ = runner .create_profile("Profile 2", "firefox", "139.0", None) .unwrap(); // Try to rename profile2 to profile1's name (should fail) let result = runner.rename_profile("Profile 2", "Profile 1"); assert!(result.is_err()); assert!(result.unwrap_err().to_string().contains("already exists")); } #[test] fn test_error_handling() { let (runner, _temp_dir) = create_test_browser_runner(); // Test updating non-existent profile let result = runner.update_profile_proxy("Non Existent", None); assert!(result.is_err()); assert!(result.unwrap_err().to_string().contains("not found")); // Test deleting non-existent profile let result = runner.delete_profile("Non Existent"); assert!(result.is_ok()); // Delete should be idempotent // Test renaming non-existent profile let result = runner.rename_profile("Non Existent", "New Name"); assert!(result.is_err()); } #[test] fn test_firefox_default_browser_preferences() { let (runner, _temp_dir) = create_test_browser_runner(); // Create profile without proxy let profile = runner .create_profile("Test Firefox Prefs", "firefox", "139.0", None) .unwrap(); // Check that user.js file was created with default browser preference let user_js_path = std::path::Path::new(&profile.profile_path).join("user.js"); assert!(user_js_path.exists()); let user_js_content = std::fs::read_to_string(user_js_path).unwrap(); assert!(user_js_content.contains("browser.shell.checkDefaultBrowser")); assert!(user_js_content.contains("false")); // Verify automatic update disabling preferences are present assert!(user_js_content.contains("app.update.enabled")); assert!(user_js_content.contains("app.update.auto")); // Create profile with proxy let proxy = ProxySettings { enabled: true, proxy_type: "http".to_string(), host: "127.0.0.1".to_string(), port: 8080, }; let profile_with_proxy = runner .create_profile("Test Firefox Prefs Proxy", "firefox", "139.0", Some(proxy)) .unwrap(); // Check that user.js file contains both proxy settings and default browser preference let user_js_path_proxy = std::path::Path::new(&profile_with_proxy.profile_path).join("user.js"); assert!(user_js_path_proxy.exists()); let user_js_content_proxy = std::fs::read_to_string(user_js_path_proxy).unwrap(); assert!(user_js_content_proxy.contains("browser.shell.checkDefaultBrowser")); assert!(user_js_content_proxy.contains("network.proxy.type")); // Verify automatic update disabling preferences are present even with proxy assert!(user_js_content_proxy.contains("app.update.enabled")); assert!(user_js_content_proxy.contains("app.update.auto")); } }