donutbrowser/src-tauri/src/browser_runner.rs

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<ProxySettings>,
  #[serde(default)]
  pub process_id: Option<u32>,
  #[serde(default)]
  pub last_launch: Option<u64>,
}

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 {
    match browser_type {
      "mullvad-browser" => {
        // More specific detection for 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" => {
        // More specific detection for 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,
    }
  }

  // Helper function to validate PID for TOR/Mullvad browsers
  // TODO: make available for other platforms once other functionality is implemented
  #[cfg(target_os = "macos")]
  fn validate_tor_mullvad_pid(&self, profile: &BrowserProfile, pid: u32) -> bool {
    let system = System::new_all();

    if let Some(process) = system.process(Pid::from(pid as usize)) {
      let exe_name = process.name().to_string_lossy().to_lowercase();
      let cmd = process.cmd();

      // Check if this is the correct browser type
      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"),
        _ => return false,
      };

      if !is_correct_browser {
        println!(
          "PID {} is not the correct browser type for {}",
          pid, profile.browser
        );
        return false;
      }

      // Check 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 {
        println!(
          "PID {} does not match profile path for {}",
          pid, profile.name
        );
        return false;
      }

      println!(
        "PID {} validated successfully for {} profile {}",
        pid, profile.browser, profile.name
      );
      true
    } else {
      println!("PID {pid} does not exist");
      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<ProxySettings>,
  ) -> Result<BrowserProfile, Box<dyn std::error::Error>> {
    // 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<ProxySettings>,
  ) -> Result<BrowserProfile, Box<dyn std::error::Error>> {
    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<BrowserProfile, Box<dyn std::error::Error>> {
    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)?;

    Ok(profile)
  }

  fn get_common_firefox_preferences(&self) -> Vec<String> {
    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<dyn std::error::Error>> {
    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<dyn std::error::Error>> {
    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<dyn std::error::Error>> {
    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<Vec<BrowserProfile>, Box<dyn std::error::Error>> {
    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<String>,
  ) -> Result<BrowserProfile, Box<dyn std::error::Error + Send + Sync>> {
    // 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
    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");

    // 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
    let child = Command::new(executable_path).args(&browser_args).spawn()?;
    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<u32> = 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<dyn std::error::Error + Send + Sync>> {
    // 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))?;

    match browser_type {
      BrowserType::Firefox | BrowserType::FirefoxDeveloper | BrowserType::Zen => {
        // These browsers don't use -no-remote, so we can use Firefox remote commands
        #[cfg(target_os = "macos")]
        {
          let pid = updated_profile.process_id.unwrap();

          // First try: Use Firefox remote command (most reliable for these browsers)
          println!("Trying Firefox remote command for PID: {pid}");
          let mut browser_dir = self.get_binaries_dir();
          browser_dir.push(&updated_profile.browser);
          browser_dir.push(&updated_profile.version);

          let browser = create_browser(browser_type);
          if let Ok(executable_path) = browser.get_executable_path(&browser_dir) {
            let remote_args = vec![
              "-profile".to_string(),
              updated_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 if remote command fails
          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");
          }
        }

        #[cfg(not(target_os = "macos"))]
        {
          // For non-macOS platforms, use Firefox remote command
          let mut browser_dir = self.get_binaries_dir();
          browser_dir.push(&updated_profile.browser);
          browser_dir.push(&updated_profile.version);

          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", &updated_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(),
            );
          }
        }
      }
      BrowserType::MullvadBrowser | BrowserType::TorBrowser => {
        // These browsers use -no-remote, so we need a different approach that doesn't require accessibility permissions
        #[cfg(target_os = "macos")]
        {
          let pid = updated_profile.process_id.unwrap();

          println!("Opening URL in TOR/Mullvad browser using file-based approach (PID: {pid})");

          // Validate that we have the correct PID for this TOR/Mullvad browser
          if !self.validate_tor_mullvad_pid(&updated_profile, pid) {
            return Err(
              format!(
                "PID {} is not valid for {} profile {}. The browser process may have changed.",
                pid, updated_profile.browser, updated_profile.name
              )
              .into(),
            );
          }

          // Method 1: Try using a temporary HTML file approach that doesn't require accessibility permissions
          println!("Attempting file-based URL opening for TOR/Mullvad browser");

          // Create a temporary HTML file that redirects to the target URL
          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#"<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <meta http-equiv="refresh" content="0; url={url}">
    <title>Redirecting...</title>
    <script>
        window.location.href = "{url}";
    </script>
</head>
<body>
    <p>Redirecting to <a href="{url}">{url}</a>...</p>
</body>
</html>"#
          );

          // Write the HTML file
          match std::fs::write(&temp_file_path, html_content) {
            Ok(()) => {
              println!("Created temporary HTML file: {temp_file_path:?}");

              // Get the browser executable path to use with 'open'
              let mut browser_dir = self.get_binaries_dir();
              browser_dir.push(&updated_profile.browser);
              browser_dir.push(&updated_profile.version);

              let browser = create_browser(browser_type.clone());
              if let Ok(executable_path) = browser.get_executable_path(&browser_dir) {
                // Use 'open' command to open the HTML file with the specific browser instance
                // This approach works because it uses the existing browser process
                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}");
                  }
                }
              }

              // Clean up temp file if the approach failed
              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");

          // Get the browser executable path
          let mut browser_dir = self.get_binaries_dir();
          browser_dir.push(&updated_profile.browser);
          browser_dir.push(&updated_profile.version);

          let browser = create_browser(browser_type.clone());
          if let Ok(executable_path) = browser.get_executable_path(&browser_dir) {
            // Try to open the URL directly with the browser
            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}");
              }
            }
          }

          // Method 3: Try using osascript without accessibility features (just to bring window to front)
          println!("Attempting minimal AppleScript approach without accessibility features");

          let minimal_script = format!(
            r#"
try
  tell application "System Events"
    set targetProcess to (first application process whose unix id is {pid})
    if not (exists targetProcess) then
      error "No process found with PID {pid}"
    end if
    
    -- Just bring the process to front without trying to control it
    set frontmost of targetProcess to true
    
    return "Process brought to front successfully"
  end tell
on error errMsg
  return "Minimal AppleScript failed: " & errMsg
end try
            "#
          );

          let minimal_output = Command::new("osascript")
            .args(["-e", &minimal_script])
            .output();

          match minimal_output {
            Ok(output) => {
              let result = String::from_utf8_lossy(&output.stdout).trim().to_string();
              if output.status.success() && result.contains("successfully") {
                println!("Successfully brought browser to front: {result}");

                // Now try to use the system's default URL opening mechanism
                let system_open_result = Command::new("open").args([url]).output();

                match system_open_result {
                  Ok(output) if output.status.success() => {
                    println!("Successfully opened URL using system default handler");
                    return Ok(());
                  }
                  Ok(output) => {
                    let stderr = String::from_utf8_lossy(&output.stderr);
                    println!("System default URL opening failed: {stderr}");
                  }
                  Err(e) => {
                    println!("System default URL opening error: {e}");
                  }
                }
              } else {
                println!("Minimal AppleScript failed: {result}");
              }
            }
            Err(e) => {
              println!("Minimal AppleScript execution error: {e}");
            }
          }

          // If all methods fail, return a more helpful error message
          return 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\
            3. Minimal AppleScript approach 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());
        }

        #[cfg(not(target_os = "macos"))]
        {
          // For non-macOS platforms, we can't use AppleScript, so try a different approach
          // This is a limitation - Firefox with -no-remote can't receive remote commands
          return Err("Opening URLs in existing Firefox-based browsers is not supported on this platform when using -no-remote".into());
        }
      }
      BrowserType::Chromium | BrowserType::Brave => {
        // For Chromium-based browsers, use a more targeted approach
        #[cfg(target_os = "macos")]
        {
          let pid = updated_profile.process_id.unwrap();

          // First, try using the browser's built-in URL opening capability
          println!("Trying Chromium URL opening for PID: {pid}");
          let mut browser_dir = self.get_binaries_dir();
          browser_dir.push(&updated_profile.browser);
          browser_dir.push(&updated_profile.version);

          let browser = create_browser(browser_type);
          if let Ok(executable_path) = browser.get_executable_path(&browser_dir) {
            // Try to open URL in existing instance using the same user data dir
            let remote_output = Command::new(executable_path)
              .args([
                &format!("--user-data-dir={}", updated_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 with more precise targeting
          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");
          }
        }

        #[cfg(not(target_os = "macos"))]
        {
          // For non-macOS platforms, try using the browser's remote opening capability
          let mut browser_dir = self.get_binaries_dir();
          browser_dir.push(&updated_profile.browser);
          browser_dir.push(&updated_profile.version);

          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))?;

          // Try to open in existing instance
          let output = Command::new(executable_path)
            .args([
              &format!("--user-data-dir={}", updated_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 launch_or_open_url(
    &self,
    app_handle: tauri::AppHandle,
    profile: &BrowserProfile,
    url: Option<String>,
  ) -> Result<BrowserProfile, Box<dyn std::error::Error + Send + Sync>> {
    // 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<BrowserProfile, Box<dyn std::error::Error>> {
    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 _new_profile_file = 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)
  }

  pub fn get_saved_mullvad_releases(&self) -> Result<Vec<String>, Box<dyn std::error::Error>> {
    let mut data_path = self.base_dirs.data_local_dir().to_path_buf();
    data_path.push(if cfg!(debug_assertions) {
      "DonutBrowserDev"
    } else {
      "DonutBrowser"
    });
    data_path.push("data");
    let releases_file = data_path.join("mullvad_releases.json");

    if !releases_file.exists() {
      return Ok(vec![]);
    }

    let mut versions = Vec::new();
    let mut browser_dir = self.base_dirs.data_local_dir().to_path_buf();
    browser_dir.push(if cfg!(debug_assertions) {
      "DonutBrowserDev"
    } else {
      "DonutBrowser"
    });
    browser_dir.push("binaries");
    browser_dir.push("mullvad-browser");
    for entry in fs::read_dir(browser_dir)? {
      let entry = entry?;
      if entry.path().is_dir() {
        if let Some(version_str) = entry.file_name().to_str() {
          versions.push(version_str.to_string());
        }
      }
    }

    // Sort versions in descending order (newest first)
    versions.sort_by(|a, b| b.cmp(a));

    Ok(versions)
  }

  fn save_process_info(&self, profile: &BrowserProfile) -> Result<(), Box<dyn std::error::Error>> {
    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<dyn std::error::Error>> {
    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)?
    }

    Ok(())
  }

  pub async fn check_browser_status(
    &self,
    app_handle: tauri::AppHandle,
    profile: &BrowserProfile,
  ) -> Result<bool, Box<dyn std::error::Error + Send + Sync>> {
    let mut inner_profile = profile.clone();
    let system = System::new_all();
    let mut is_running = false;
    let mut found_pid: Option<u32> = 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<dyn std::error::Error + Send + Sync>> {
    // 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<u32> = 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
    #[cfg(target_os = "macos")]
    {
      use std::process::Command;

      // 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(),
          );
        }
      }
    }

    #[cfg(not(target_os = "macos"))]
    {
      // For other platforms, use the sysinfo crate
      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());
      }
    }

    // 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}"))?;

    println!("Successfully killed browser process with PID: {pid}");
    Ok(())
  }
}

#[tauri::command]
pub fn create_browser_profile(
  name: String,
  browser: String,
  version: String,
  proxy: Option<ProxySettings>,
) -> Result<BrowserProfile, String> {
  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<Vec<BrowserProfile>, 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<String>,
) -> Result<BrowserProfile, String> {
  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
    .expect("Failed to launch browser or open URL");

  // 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)
}

// Add Tauri command to get saved releases
#[tauri::command]
pub fn get_saved_mullvad_releases() -> Result<Vec<String>, String> {
  let browser_runner = BrowserRunner::new();
  browser_runner
    .get_saved_mullvad_releases()
    .map_err(|e| e.to_string())
}

#[tauri::command]
pub fn update_profile_proxy(
  profile_name: String,
  proxy: Option<ProxySettings>,
) -> Result<BrowserProfile, String> {
  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<BrowserProfile, String> {
  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<bool, String> {
  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<BrowserProfile, String> {
  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<Vec<&'static str>, String> {
  Ok(vec![
    BrowserType::MullvadBrowser.as_str(),
    BrowserType::Firefox.as_str(),
    BrowserType::FirefoxDeveloper.as_str(),
    BrowserType::Chromium.as_str(),
    BrowserType::Brave.as_str(),
    BrowserType::Zen.as_str(),
    BrowserType::TorBrowser.as_str(),
  ])
}

#[tauri::command]
pub async fn fetch_browser_versions_detailed(
  browser_str: String,
) -> Result<Vec<BrowserVersionInfo>, String> {
  let service = BrowserVersionService::new();
  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_cached_first(
  browser_str: String,
) -> Result<Vec<BrowserVersionInfo>, 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<BrowserVersionsResult, String> {
  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 fn get_cached_browser_versions_detailed(
  browser_str: String,
) -> Result<Option<Vec<BrowserVersionInfo>>, String> {
  let service = BrowserVersionService::new();
  Ok(service.get_cached_browser_versions_detailed(&browser_str))
}

#[tauri::command]
pub fn should_update_browser_cache(browser_str: String) -> Result<bool, String> {
  let service = BrowserVersionService::new();
  Ok(service.should_update_cache(&browser_str))
}

#[tauri::command]
pub async fn download_browser(
  app_handle: tauri::AppHandle,
  browser_str: String,
  version: String,
) -> Result<String, String> {
  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);
  }

  // Use the centralized browser version service for download info
  let version_service = BrowserVersionService::new();
  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<ProxySettings>,
) -> Result<BrowserProfile, String> {
  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(browser_str: String) -> Result<Vec<String>, String> {
  let service = BrowserVersionService::new();
  service
    .fetch_browser_versions(&browser_str, false)
    .await
    .map_err(|e| format!("Failed to fetch browser versions: {e}"))
}

#[tauri::command]
pub async fn fetch_browser_versions_with_count(
  browser_str: String,
) -> Result<BrowserVersionsResult, String> {
  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<Vec<String>, String> {
  let registry = DownloadedBrowsersRegistry::load()
    .map_err(|e| format!("Failed to load browser registry: {e}"))?;
  Ok(registry.get_downloaded_versions(&browser_str))
}

#[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 _profile = 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 _profile = 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 _profile1 = runner
      .create_profile("Profile 1", "firefox", "139.0", None)
      .unwrap();
    let _profile2 = runner
      .create_profile("Profile 2", "chromium", "1465660", None)
      .unwrap();
    let _profile3 = 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 _profile1 = runner
      .create_profile("Profile 1", "firefox", "139.0", None)
      .unwrap();
    let _profile2 = 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"));
  }
}