core-agent-ide/codex-rs/debug-client/src/commands.rs

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum InputAction {
    Message(String),
    Command(UserCommand),
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum UserCommand {
    Help,
    Quit,
    NewThread,
    Resume(String),
    Use(String),
    RefreshThread,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ParseError {
    EmptyCommand,
    MissingArgument { name: &'static str },
    UnknownCommand { command: String },
}

impl ParseError {
    pub fn message(&self) -> String {
        match self {
            Self::EmptyCommand => "empty command after ':'".to_string(),
            Self::MissingArgument { name } => {
                format!("missing required argument: {name}")
            }
            Self::UnknownCommand { command } => format!("unknown command: {command}"),
        }
    }
}

pub fn parse_input(line: &str) -> Result<Option<InputAction>, ParseError> {
    let trimmed = line.trim();
    if trimmed.is_empty() {
        return Ok(None);
    }

    let Some(command_line) = trimmed.strip_prefix(':') else {
        return Ok(Some(InputAction::Message(trimmed.to_string())));
    };

    let mut parts = command_line.split_whitespace();
    let Some(command) = parts.next() else {
        return Err(ParseError::EmptyCommand);
    };

    match command {
        "help" | "h" => Ok(Some(InputAction::Command(UserCommand::Help))),
        "quit" | "q" | "exit" => Ok(Some(InputAction::Command(UserCommand::Quit))),
        "new" => Ok(Some(InputAction::Command(UserCommand::NewThread))),
        "resume" => {
            let thread_id = parts
                .next()
                .ok_or(ParseError::MissingArgument { name: "thread-id" })?;
            Ok(Some(InputAction::Command(UserCommand::Resume(
                thread_id.to_string(),
            ))))
        }
        "use" => {
            let thread_id = parts
                .next()
                .ok_or(ParseError::MissingArgument { name: "thread-id" })?;
            Ok(Some(InputAction::Command(UserCommand::Use(
                thread_id.to_string(),
            ))))
        }
        "refresh-thread" => Ok(Some(InputAction::Command(UserCommand::RefreshThread))),
        _ => Err(ParseError::UnknownCommand {
            command: command.to_string(),
        }),
    }
}

#[cfg(test)]
mod tests {
    use pretty_assertions::assert_eq;

    use super::InputAction;
    use super::ParseError;
    use super::UserCommand;
    use super::parse_input;

    #[test]
    fn parses_message() {
        let result = parse_input("hello there").unwrap();
        assert_eq!(
            result,
            Some(InputAction::Message("hello there".to_string()))
        );
    }

    #[test]
    fn parses_help_command() {
        let result = parse_input(":help").unwrap();
        assert_eq!(result, Some(InputAction::Command(UserCommand::Help)));
    }

    #[test]
    fn parses_new_thread() {
        let result = parse_input(":new").unwrap();
        assert_eq!(result, Some(InputAction::Command(UserCommand::NewThread)));
    }

    #[test]
    fn parses_resume() {
        let result = parse_input(":resume thr_123").unwrap();
        assert_eq!(
            result,
            Some(InputAction::Command(UserCommand::Resume(
                "thr_123".to_string()
            )))
        );
    }

    #[test]
    fn parses_use() {
        let result = parse_input(":use thr_456").unwrap();
        assert_eq!(
            result,
            Some(InputAction::Command(UserCommand::Use(
                "thr_456".to_string()
            )))
        );
    }

    #[test]
    fn parses_refresh_thread() {
        let result = parse_input(":refresh-thread").unwrap();
        assert_eq!(
            result,
            Some(InputAction::Command(UserCommand::RefreshThread))
        );
    }

    #[test]
    fn rejects_missing_resume_arg() {
        let result = parse_input(":resume");
        assert_eq!(
            result,
            Err(ParseError::MissingArgument { name: "thread-id" })
        );
    }

    #[test]
    fn rejects_missing_use_arg() {
        let result = parse_input(":use");
        assert_eq!(
            result,
            Err(ParseError::MissingArgument { name: "thread-id" })
        );
    }
}