From da5357d23f8450d4609f72d079876d7d439ef5a0 Mon Sep 17 00:00:00 2001 From: akbash Date: Tue, 25 Aug 2026 22:38:14 +0300 Subject: fix(cli): accept reordered agent report arguments (#2928) refs #2926 Co-authored-by: akbash-bot <300245827+akbash-bot@users.noreply.github.com> --- src/cli/pane.rs | 73 ++++++++++++++++++++++++++++++++++++++++++++------------- 1 file changed, 57 insertions(+), 16 deletions(-) (limited to 'src') diff --git a/src/cli/pane.rs b/src/cli/pane.rs index 363a2e05..17d3c144 100644 --- a/src/cli/pane.rs +++ b/src/cli/pane.rs @@ -1164,12 +1164,21 @@ fn parse_pane_wait_output_args(args: &[String]) -> Result std::io::Result { - let Some(raw_pane_id) = args.first() else { - eprintln!("usage: herdr pane report-agent --source ID --agent LABEL --state idle|working|blocked|unknown [--message TEXT] [--seq N] [--agent-session-id ID] [--agent-session-path PATH]"); - return Ok(2); - }; + const USAGE: &str = "usage: herdr pane report-agent --source ID --agent LABEL --state idle|working|blocked|unknown [--message TEXT] [--seq N] [--agent-session-id ID] [--agent-session-path PATH]"; - let pane_id = super::normalize_pane_id(raw_pane_id); + let args = super::expand_equals_args( + args, + &[ + "--source", + "--agent", + "--state", + "--message", + "--seq", + "--agent-session-id", + "--agent-session-path", + ], + ); + let mut pane_id = None; let mut source = None; let mut agent = None; let mut state = None; @@ -1178,7 +1187,7 @@ fn pane_report_agent(args: &[String]) -> std::io::Result { let mut agent_session_id = None; let mut agent_session_path = None; - let mut index = 1; + let mut index = 0; while index < args.len() { match args[index].as_str() { "--source" => { @@ -1237,13 +1246,25 @@ fn pane_report_agent(args: &[String]) -> std::io::Result { agent_session_path = Some(value.clone()); index += 2; } - other => { - eprintln!("unknown option: {other}"); + option if option.starts_with('-') => { + eprintln!("unknown option: {option}"); return Ok(2); } + positional => { + if pane_id.is_some() { + eprintln!("unexpected argument: {positional}"); + return Ok(2); + } + pane_id = Some(super::normalize_pane_id(positional)); + index += 1; + } } } + let Some(pane_id) = pane_id else { + eprintln!("{USAGE}"); + return Ok(2); + }; let Some(source) = source.and_then(|source| { let source = source.trim().to_string(); (!source.is_empty()).then_some(source) @@ -1273,12 +1294,20 @@ fn pane_report_agent(args: &[String]) -> std::io::Result { } fn pane_report_agent_session(args: &[String]) -> std::io::Result { - let Some(raw_pane_id) = args.first() else { - eprintln!("usage: herdr pane report-agent-session --source ID --agent LABEL [--seq N] [--agent-session-id ID] [--agent-session-path PATH] [--session-start-source SOURCE]"); - return Ok(2); - }; + const USAGE: &str = "usage: herdr pane report-agent-session --source ID --agent LABEL [--seq N] [--agent-session-id ID] [--agent-session-path PATH] [--session-start-source SOURCE]"; - let pane_id = super::normalize_pane_id(raw_pane_id); + let args = super::expand_equals_args( + args, + &[ + "--source", + "--agent", + "--seq", + "--agent-session-id", + "--agent-session-path", + "--session-start-source", + ], + ); + let mut pane_id = None; let mut source = None; let mut agent = None; let mut seq = None; @@ -1286,7 +1315,7 @@ fn pane_report_agent_session(args: &[String]) -> std::io::Result { let mut agent_session_path = None; let mut session_start_source = None; - let mut index = 1; + let mut index = 0; while index < args.len() { match args[index].as_str() { "--source" => { @@ -1337,13 +1366,25 @@ fn pane_report_agent_session(args: &[String]) -> std::io::Result { session_start_source = Some(value.clone()); index += 2; } - other => { - eprintln!("unknown option: {other}"); + option if option.starts_with('-') => { + eprintln!("unknown option: {option}"); return Ok(2); } + positional => { + if pane_id.is_some() { + eprintln!("unexpected argument: {positional}"); + return Ok(2); + } + pane_id = Some(super::normalize_pane_id(positional)); + index += 1; + } } } + let Some(pane_id) = pane_id else { + eprintln!("{USAGE}"); + return Ok(2); + }; let Some(source) = source.and_then(|source| { let source = source.trim().to_string(); (!source.is_empty()).then_some(source) -- cgit v1.2.3