batty_cli/agent/
codex.rs

1//! Codex CLI adapter.
2//!
3//! Runs Codex in interactive mode by default, passing the composed task prompt
4//! as the initial user prompt argument.
5#![cfg_attr(not(test), allow(dead_code))]
6
7use std::path::Path;
8
9use anyhow::Context;
10
11use crate::agent::{AgentAdapter, SpawnConfig};
12use crate::prompt::PromptPatterns;
13
14/// Adapter for Codex CLI.
15pub struct CodexCliAdapter {
16    /// Override the codex binary name/path (default: "codex").
17    program: String,
18}
19
20impl CodexCliAdapter {
21    pub fn new(program: Option<String>) -> Self {
22        Self {
23            program: program.unwrap_or_else(|| "codex".to_string()),
24        }
25    }
26}
27
28impl AgentAdapter for CodexCliAdapter {
29    fn name(&self) -> &str {
30        "codex-cli"
31    }
32
33    fn spawn_config(&self, task_description: &str, work_dir: &Path) -> SpawnConfig {
34        SpawnConfig {
35            program: self.program.clone(),
36            args: vec![task_description.to_string()],
37            work_dir: work_dir.to_string_lossy().to_string(),
38            env: vec![],
39        }
40    }
41
42    fn prompt_patterns(&self) -> PromptPatterns {
43        PromptPatterns::codex_cli()
44    }
45
46    fn instruction_candidates(&self) -> &'static [&'static str] {
47        &["AGENTS.md", "CLAUDE.md"]
48    }
49
50    fn wrap_launch_prompt(&self, prompt: &str) -> String {
51        format!(
52            "You are running as Codex under Batty supervision.\n\
53             Treat the launch context below as authoritative session context.\n\n\
54             {prompt}"
55        )
56    }
57
58    fn format_input(&self, response: &str) -> String {
59        format!("{response}\n")
60    }
61
62    fn launch_command(
63        &self,
64        prompt: &str,
65        idle: bool,
66        resume: bool,
67        session_id: Option<&str>,
68    ) -> anyhow::Result<String> {
69        let escaped = prompt.replace('\'', "'\\''");
70        if resume {
71            let sid = session_id.context("missing Codex session ID for resume")?;
72            Ok(format!(
73                "exec codex resume '{sid}' --dangerously-bypass-approvals-and-sandbox"
74            ))
75        } else {
76            let prefix = "exec codex --dangerously-bypass-approvals-and-sandbox";
77            if idle {
78                Ok(prefix.to_string())
79            } else {
80                Ok(format!("{prefix} '{escaped}'"))
81            }
82        }
83    }
84
85    fn supports_resume(&self) -> bool {
86        true
87    }
88
89    fn health_check(&self) -> super::BackendHealth {
90        super::check_binary_available(&self.program)
91    }
92}
93
94#[cfg(test)]
95mod tests {
96    use super::*;
97
98    #[test]
99    fn default_program_is_codex() {
100        let adapter = CodexCliAdapter::new(None);
101        let config = adapter.spawn_config("test", Path::new("/tmp"));
102        assert_eq!(config.program, "codex");
103    }
104
105    #[test]
106    fn custom_program_path() {
107        let adapter = CodexCliAdapter::new(Some("/usr/local/bin/codex".to_string()));
108        let config = adapter.spawn_config("test", Path::new("/tmp"));
109        assert_eq!(config.program, "/usr/local/bin/codex");
110    }
111
112    #[test]
113    fn spawn_sets_work_dir() {
114        let adapter = CodexCliAdapter::new(None);
115        let config = adapter.spawn_config("task", Path::new("/my/worktree"));
116        assert_eq!(config.work_dir, "/my/worktree");
117    }
118
119    #[test]
120    fn prompt_patterns_detect_permission() {
121        let adapter = CodexCliAdapter::new(None);
122        let patterns = adapter.prompt_patterns();
123        let d = patterns.detect("Would you like to run the following command?");
124        assert!(d.is_some());
125        assert!(matches!(
126            d.unwrap().kind,
127            crate::prompt::PromptKind::Permission { .. }
128        ));
129    }
130
131    #[test]
132    fn format_input_appends_newline() {
133        let adapter = CodexCliAdapter::new(None);
134        assert_eq!(adapter.format_input("y"), "y\n");
135        assert_eq!(adapter.format_input("yes"), "yes\n");
136    }
137
138    #[test]
139    fn name_is_codex_cli() {
140        let adapter = CodexCliAdapter::new(None);
141        assert_eq!(adapter.name(), "codex-cli");
142    }
143
144    #[test]
145    fn codex_prefers_agents_md_instruction_order() {
146        let adapter = CodexCliAdapter::new(None);
147        assert_eq!(
148            adapter.instruction_candidates(),
149            &["AGENTS.md", "CLAUDE.md"]
150        );
151    }
152
153    #[test]
154    fn codex_wraps_launch_prompt() {
155        let adapter = CodexCliAdapter::new(None);
156        let wrapped = adapter.wrap_launch_prompt("Launch body");
157        assert!(wrapped.contains("Codex under Batty supervision"));
158        assert!(wrapped.contains("Launch body"));
159    }
160
161    // --- Backend trait method tests ---
162
163    #[test]
164    fn launch_command_active_includes_prompt() {
165        let adapter = CodexCliAdapter::new(None);
166        let cmd = adapter
167            .launch_command("do the thing", false, false, None)
168            .unwrap();
169        assert!(cmd.contains("exec codex --dangerously-bypass-approvals-and-sandbox"));
170        assert!(cmd.contains("'do the thing'"));
171    }
172
173    #[test]
174    fn launch_command_idle_omits_prompt() {
175        let adapter = CodexCliAdapter::new(None);
176        let cmd = adapter
177            .launch_command("ignored", true, false, None)
178            .unwrap();
179        assert_eq!(cmd, "exec codex --dangerously-bypass-approvals-and-sandbox");
180    }
181
182    #[test]
183    fn launch_command_resume_uses_session_id() {
184        let adapter = CodexCliAdapter::new(None);
185        let cmd = adapter
186            .launch_command("ignored", false, true, Some("codex-sess-1"))
187            .unwrap();
188        assert!(cmd.contains("exec codex resume 'codex-sess-1'"));
189        assert!(cmd.contains("--dangerously-bypass-approvals-and-sandbox"));
190    }
191
192    #[test]
193    fn launch_command_resume_without_session_id_errors() {
194        let adapter = CodexCliAdapter::new(None);
195        let result = adapter.launch_command("ignored", false, true, None);
196        assert!(result.is_err());
197    }
198
199    #[test]
200    fn new_session_id_returns_none() {
201        let adapter = CodexCliAdapter::new(None);
202        assert!(adapter.new_session_id().is_none());
203    }
204
205    #[test]
206    fn supports_resume_is_true() {
207        let adapter = CodexCliAdapter::new(None);
208        assert!(adapter.supports_resume());
209    }
210}
batty_cli/agent/codex.rs

batty_cli/agent/
codex.rs
