diff --git a/codex-rs/cli/src/main.rs b/codex-rs/cli/src/main.rs index c9fb1855564..235096f37ce 100644 --- a/codex-rs/cli/src/main.rs +++ b/codex-rs/cli/src/main.rs @@ -36,10 +36,10 @@ use codex_utils_absolute_path::AbsolutePathBuf; use codex_utils_cli::CliConfigOverrides; use codex_utils_cli::ProfileV2Name; use codex_utils_cli::SharedCliOptions; -use codex_utils_cli::resume_hint; use owo_colors::OwoColorize; use std::collections::HashSet; use std::io::IsTerminal; +use std::io::Write; use std::path::PathBuf; use supports_color::Stream; @@ -698,10 +698,11 @@ fn parse_socket_path(raw: &str) -> Result { } fn format_exit_messages(exit_info: AppExitInfo, color_enabled: bool) -> Vec { + let is_fatal = matches!(&exit_info.exit_reason, ExitReason::Fatal(_)); let AppExitInfo { token_usage, thread_id: conversation_id, - thread_name, + resume_hint, .. } = exit_info; @@ -710,13 +711,15 @@ fn format_exit_messages(exit_info: AppExitInfo, color_enabled: bool) -> Vec Vec anyhow::Result<()> { - match exit_info.exit_reason { + let is_fatal = match &exit_info.exit_reason { ExitReason::Fatal(message) => { eprintln!("ERROR: {message}"); - std::process::exit(1); + true } - ExitReason::UserRequested => { /* normal exit */ } - } + ExitReason::UserRequested => false, + }; let update_action = exit_info.update_action; let color_enabled = supports_color::on(Stream::Stdout).is_some(); for line in format_exit_messages(exit_info, color_enabled) { println!("{line}"); } + if is_fatal { + std::io::stdout().flush()?; + std::process::exit(1); + } if let Some(action) = update_action { run_update_action(action)?; } @@ -3037,12 +3044,13 @@ mod tests { total_tokens: 2, ..Default::default() }; + let thread_id = conversation_id + .map(ThreadId::from_string) + .map(Result::unwrap); AppExitInfo { token_usage, - thread_id: conversation_id - .map(ThreadId::from_string) - .map(Result::unwrap), - thread_name: thread_name.map(str::to_string), + thread_id, + resume_hint: codex_utils_cli::resume_hint(thread_name, thread_id), update_action: None, exit_reason: ExitReason::UserRequested, } @@ -3053,7 +3061,7 @@ mod tests { let exit_info = AppExitInfo { token_usage: TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }; @@ -3061,6 +3069,40 @@ mod tests { assert!(lines.is_empty()); } + #[test] + fn format_exit_messages_includes_session_id_for_fatal_exit_without_resume_hint() { + let exit_info = AppExitInfo { + token_usage: TokenUsage::default(), + thread_id: Some(ThreadId::from_string("123e4567-e89b-12d3-a456-426614174000").unwrap()), + resume_hint: None, + update_action: None, + exit_reason: ExitReason::Fatal("boom".to_string()), + }; + let lines = format_exit_messages(exit_info, /*color_enabled*/ false); + assert_eq!( + lines, + vec!["Session ID: 123e4567-e89b-12d3-a456-426614174000".to_string()] + ); + } + + #[test] + fn format_exit_messages_includes_resume_hint_for_fatal_exit() { + let mut exit_info = sample_exit_info( + Some("123e4567-e89b-12d3-a456-426614174000"), + /*thread_name*/ None, + ); + exit_info.exit_reason = ExitReason::Fatal("boom".to_string()); + let lines = format_exit_messages(exit_info, /*color_enabled*/ false); + assert_eq!( + lines, + vec![ + "Token usage: total=2 input=0 output=2".to_string(), + "To continue this session, run codex resume 123e4567-e89b-12d3-a456-426614174000" + .to_string(), + ] + ); + } + #[test] fn format_exit_messages_includes_resume_hint_without_color() { let exit_info = sample_exit_info( diff --git a/codex-rs/tui/src/app.rs b/codex-rs/tui/src/app.rs index 10d86de54bb..0df5b81d5eb 100644 --- a/codex-rs/tui/src/app.rs +++ b/codex-rs/tui/src/app.rs @@ -397,7 +397,7 @@ const COMMIT_ANIMATION_TICK: Duration = tui::TARGET_FRAME_INTERVAL; pub struct AppExitInfo { pub token_usage: TokenUsage, pub thread_id: Option, - pub thread_name: Option, + pub resume_hint: Option, pub update_action: Option, pub exit_reason: ExitReason, } @@ -407,7 +407,7 @@ impl AppExitInfo { Self { token_usage: TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::Fatal(message.into()), } @@ -433,10 +433,7 @@ fn session_summary( rollout_path: Option<&Path>, ) -> Option { let usage_line = (!token_usage.is_zero()).then(|| token_usage.to_string()); - let resumable_thread = resumable_thread(thread_id, thread_name, rollout_path); - let resume_hint = resumable_thread.as_ref().and_then(|thread| { - codex_utils_cli::resume_hint(thread.thread_name.as_deref(), Some(thread.thread_id)) - }); + let resume_hint = resume_hint_for_resumable_thread(thread_id, thread_name, rollout_path); if usage_line.is_none() && resume_hint.is_none() { return None; @@ -467,6 +464,15 @@ fn resumable_thread( }) } +fn resume_hint_for_resumable_thread( + thread_id: Option, + thread_name: Option, + rollout_path: Option<&Path>, +) -> Option { + let thread = resumable_thread(thread_id, thread_name, rollout_path)?; + codex_utils_cli::resume_hint(thread.thread_name.as_deref(), Some(thread.thread_id)) +} + fn rollout_path_is_resumable(rollout_path: &Path) -> bool { std::fs::metadata(rollout_path).is_ok_and(|metadata| metadata.is_file() && metadata.len() > 0) } @@ -1216,15 +1222,16 @@ See the Codex keymap documentation for supported actions and examples." return Err(err); } }; - let resumable_thread = resumable_thread( - app.chat_widget.thread_id(), + let thread_id = app.chat_widget.thread_id().or(app.primary_thread_id); + let resume_hint = resume_hint_for_resumable_thread( + thread_id, app.chat_widget.thread_name(), app.chat_widget.rollout_path().as_deref(), ); Ok(AppExitInfo { token_usage: app.token_usage(), - thread_id: resumable_thread.as_ref().map(|thread| thread.thread_id), - thread_name: resumable_thread.and_then(|thread| thread.thread_name), + thread_id, + resume_hint, update_action: app.pending_update_action, exit_reason, }) diff --git a/codex-rs/tui/src/app/startup_prompts.rs b/codex-rs/tui/src/app/startup_prompts.rs index 4834656617e..57f70a559e0 100644 --- a/codex-rs/tui/src/app/startup_prompts.rs +++ b/codex-rs/tui/src/app/startup_prompts.rs @@ -343,7 +343,7 @@ pub(super) async fn handle_model_migration_prompt_if_needed( return Some(AppExitInfo { token_usage: TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }); diff --git a/codex-rs/tui/src/lib.rs b/codex-rs/tui/src/lib.rs index 13aecf217bb..4a82da9332f 100644 --- a/codex-rs/tui/src/lib.rs +++ b/codex-rs/tui/src/lib.rs @@ -1381,7 +1381,7 @@ async fn run_ratatui_app( return Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: Some(action), exit_reason: ExitReason::UserRequested, }); @@ -1474,7 +1474,7 @@ async fn run_ratatui_app( return Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }); @@ -1524,7 +1524,7 @@ async fn run_ratatui_app( Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::Fatal(format!( "No saved session found with ID {id_str}. Run `codex {action}` without an ID to choose from existing sessions." @@ -1581,7 +1581,7 @@ async fn run_ratatui_app( return Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }); @@ -1642,7 +1642,7 @@ async fn run_ratatui_app( return Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }); @@ -1687,7 +1687,7 @@ async fn run_ratatui_app( return Ok(AppExitInfo { token_usage: crate::token_usage::TokenUsage::default(), thread_id: None, - thread_name: None, + resume_hint: None, update_action: None, exit_reason: ExitReason::UserRequested, }); diff --git a/codex-rs/tui/src/main.rs b/codex-rs/tui/src/main.rs index 9c69c7c81af..486dbe6499d 100644 --- a/codex-rs/tui/src/main.rs +++ b/codex-rs/tui/src/main.rs @@ -7,14 +7,15 @@ use codex_tui::Cli; use codex_tui::ExitReason; use codex_tui::run_main; use codex_utils_cli::CliConfigOverrides; -use codex_utils_cli::resume_hint; +use std::io::Write; use supports_color::Stream; fn format_exit_messages(exit_info: AppExitInfo, color_enabled: bool) -> Vec { + let is_fatal = matches!(&exit_info.exit_reason, ExitReason::Fatal(_)); let AppExitInfo { token_usage, thread_id, - thread_name, + resume_hint, .. } = exit_info; @@ -23,13 +24,15 @@ fn format_exit_messages(exit_info: AppExitInfo, color_enabled: bool) -> Vec anyhow::Result<()> { /*explicit_remote_endpoint*/ None, ) .await?; - match exit_info.exit_reason { + let is_fatal = match &exit_info.exit_reason { ExitReason::Fatal(message) => { eprintln!("ERROR: {message}"); - std::process::exit(1); + true } - ExitReason::UserRequested => {} - } + ExitReason::UserRequested => false, + }; let color_enabled = supports_color::on(Stream::Stdout).is_some(); for line in format_exit_messages(exit_info, color_enabled) { println!("{line}"); } + if is_fatal { + std::io::stdout().flush()?; + std::process::exit(1); + } Ok(()) }) }