ito_core/harness/
opencode.rs

1use super::streaming_cli::CliHarness;
2use super::types::{HarnessName, HarnessRunConfig};
3
4/// Harness implementation that executes the `opencode` CLI (`opencode run`).
5///
6/// Selected via `ito ralph --harness opencode`; requires the OpenCode CLI on PATH.
7///
8/// # Examples
9///
10/// ```
11/// use ito_core::harness::{Harness, HarnessName, OpencodeHarness};
12///
13/// let h = OpencodeHarness;
14/// assert_eq!(h.name(), HarnessName::Opencode);
15/// assert!(h.streams_output());
16/// ```
17#[derive(Debug, Default)]
18pub struct OpencodeHarness;
19
20impl CliHarness for OpencodeHarness {
21    fn harness_name(&self) -> HarnessName {
22        HarnessName::Opencode
23    }
24
25    fn binary(&self) -> &str {
26        "opencode"
27    }
28
29    fn build_args(&self, config: &HarnessRunConfig) -> Vec<String> {
30        let mut args = vec!["run".to_string()];
31        if let Some(model) = config.model.as_deref() {
32            args.push("-m".to_string());
33            args.push(model.to_string());
34        }
35        if config.allow_all {
36            args.push("--yolo".to_string());
37        }
38        args.push(config.prompt.clone());
39        args
40    }
41}
42
43#[cfg(test)]
44mod tests {
45    use super::*;
46    use std::collections::BTreeMap;
47
48    enum Allow {
49        #[allow(dead_code)]
50        All,
51        None,
52    }
53
54    fn config(allow: Allow, model: Option<&str>) -> HarnessRunConfig {
55        let allow_all = match allow {
56            Allow::All => true,
57            Allow::None => false,
58        };
59        HarnessRunConfig {
60            prompt: "do stuff".to_string(),
61            model: model.map(String::from),
62            cwd: std::env::temp_dir(),
63            env: BTreeMap::new(),
64            interactive: false,
65            allow_all,
66            inactivity_timeout: None,
67        }
68    }
69
70    #[test]
71    fn harness_name_is_opencode() {
72        let harness = OpencodeHarness;
73        assert_eq!(harness.harness_name(), HarnessName::Opencode);
74    }
75
76    #[test]
77    fn binary_is_opencode() {
78        let harness = OpencodeHarness;
79        assert_eq!(harness.binary(), "opencode");
80    }
81
82    #[test]
83    fn build_args_with_model() {
84        let harness = OpencodeHarness;
85        let cfg = config(Allow::None, Some("gpt-4"));
86        let args = harness.build_args(&cfg);
87        assert_eq!(args, vec!["run", "-m", "gpt-4", "do stuff"]);
88    }
89
90    #[test]
91    fn build_args_without_model() {
92        let harness = OpencodeHarness;
93        let cfg = config(Allow::None, None);
94        let args = harness.build_args(&cfg);
95        assert_eq!(args, vec!["run", "do stuff"]);
96    }
97}
ito_core/harness/opencode.rs

ito_core/harness/
opencode.rs
