diff --git a/cli/src/connection.rs b/cli/src/connection.rs index 38819f8..82cb722 100644 --- a/cli/src/connection.rs +++ b/cli/src/connection.rs @@ -234,6 +234,7 @@ pub struct DaemonOptions<'a> { pub action_policy: Option<&'a str>, pub confirm_actions: Option<&'a str>, pub engine: Option<&'a str>, + pub auto_connect: bool, pub idle_timeout: Option<&'a str>, pub cdp: Option<&'a str>, } @@ -302,6 +303,9 @@ fn apply_daemon_env(cmd: &mut Command, session: &str, opts: &DaemonOptions) { if let Some(engine) = opts.engine { cmd.env("AGENT_BROWSER_ENGINE", engine); } + if opts.auto_connect { + cmd.env("AGENT_BROWSER_AUTO_CONNECT", "1"); + } if let Some(idle) = opts.idle_timeout { cmd.env("AGENT_BROWSER_IDLE_TIMEOUT_MS", idle); } @@ -533,6 +537,8 @@ fn is_transient_error(error: &str) -> bool { || error.contains("os error 2") // No such file or directory (socket gone) || error.contains("os error 61") // Connection refused (macOS) || error.contains("os error 111") // Connection refused (Linux) + || error.contains("os error 10061") // Connection refused (Windows) + || error.contains("os error 10054") // Connection reset by peer (Windows) } fn send_command_once(cmd: &Value, session: &str) -> Result { @@ -708,6 +714,20 @@ mod tests { )); } + #[test] + fn test_is_transient_error_connection_refused_windows() { + assert!(is_transient_error( + "Failed to connect: No connection could be made because the target machine actively refused it. (os error 10061)" + )); + } + + #[test] + fn test_is_transient_error_connection_reset_windows() { + assert!(is_transient_error( + "Failed to send: An existing connection was forcibly closed by the remote host. (os error 10054)" + )); + } + #[test] fn test_is_transient_error_non_transient() { // These should NOT be considered transient diff --git a/cli/src/main.rs b/cli/src/main.rs index cb9f36f..18dafe8 100644 --- a/cli/src/main.rs +++ b/cli/src/main.rs @@ -316,6 +316,7 @@ fn main() { action_policy: flags.action_policy.as_deref(), confirm_actions: flags.confirm_actions.as_deref(), engine: flags.engine.as_deref(), + auto_connect: flags.auto_connect, idle_timeout: flags.idle_timeout.as_deref(), cdp: flags.cdp.as_deref(), }; @@ -617,6 +618,7 @@ fn main() { || !flags.extensions.is_empty()) && flags.cdp.is_none() && flags.provider.is_none() + && !flags.auto_connect { let mut launch_cmd = json!({ "id": gen_id(), diff --git a/cli/src/native/actions.rs b/cli/src/native/actions.rs index 7e0c2cc..b752dd7 100644 --- a/cli/src/native/actions.rs +++ b/cli/src/native/actions.rs @@ -792,6 +792,19 @@ pub async fn execute_command(cmd: &Value, state: &mut DaemonState) -> Value { // Auto-launch // --------------------------------------------------------------------------- +/// Connect to a running Chrome via auto-discovery and open a fresh tab so +/// subsequent navigations don't hijack the user's existing tabs. +async fn connect_auto_with_fresh_tab() -> Result { + let mut mgr = BrowserManager::connect_auto().await?; + mgr.tab_new(None).await?; + let session_id = mgr.active_session_id()?.to_string(); + let _ = mgr + .client + .send_command("Page.bringToFront", None, Some(&session_id)) + .await; + Ok(mgr) +} + async fn auto_launch(state: &mut DaemonState) -> Result<(), String> { let options = launch_options_from_env(); let engine = env::var("AGENT_BROWSER_ENGINE").ok(); @@ -806,8 +819,7 @@ async fn auto_launch(state: &mut DaemonState) -> Result<(), String> { } if env::var("AGENT_BROWSER_AUTO_CONNECT").is_ok() { - let mgr = BrowserManager::connect_auto().await?; - state.browser = Some(mgr); + state.browser = Some(connect_auto_with_fresh_tab().await?); state.subscribe_to_browser_events(); state.update_stream_client().await; try_auto_restore_state(state).await; @@ -971,7 +983,7 @@ async fn handle_launch(cmd: &Value, state: &mut DaemonState) -> Result Result { if let Ok(ws_url) = discover_cdp_url("127.0.0.1", port).await { return Ok(ws_url); } - // M144+: direct WebSocket - let ws_url = format!("ws://127.0.0.1:{}{}", port, ws_path); - return Ok(ws_url); + // M144+: direct WebSocket — verify the port is actually listening + // before returning, otherwise a stale DevToolsActivePort file + // (left behind after Chrome exits/crashes) produces a confusing + // "connection refused" error instead of falling through. + if is_port_reachable(port) { + let ws_url = format!("ws://127.0.0.1:{}{}", port, ws_path); + return Ok(ws_url); + } + // Port is dead — remove the stale file so future runs skip it. + let stale = dir.join("DevToolsActivePort"); + let _ = std::fs::remove_file(&stale); } } @@ -469,6 +477,12 @@ pub async fn auto_connect_cdp() -> Result { Err("No running Chrome instance found. Launch Chrome with --remote-debugging-port or use --cdp.".to_string()) } +fn is_port_reachable(port: u16) -> bool { + use std::net::TcpStream; + let addr = format!("127.0.0.1:{}", port); + TcpStream::connect_timeout(&addr.parse().unwrap(), Duration::from_millis(500)).is_ok() +} + fn get_chrome_user_data_dirs() -> Vec { let mut dirs = Vec::new();