ito-core 0.1.29

use super::types::{Harness, HarnessName, HarnessRunConfig, HarnessRunResult};
use miette::{Result, miette};
use serde::Deserialize;
use std::fs;
use std::path::{Path, PathBuf};
use std::time::{Duration, Instant};

#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "camelCase")]
/// One scripted execution step for the stub harness.
pub struct StubStep {
    /// Captured stdout for this step.
    pub stdout: String,
    #[serde(default)]
    /// Captured stderr for this step.
    pub stderr: String,
    #[serde(default)]
    /// Exit code for this step.
    pub exit_code: i32,
}

#[derive(Debug, Clone)]
/// Harness implementation that returns pre-recorded outputs.
///
/// This is primarily used for tests and offline development workflows.
pub struct StubHarness {
    steps: Vec<StubStep>,
    idx: usize,
}

impl StubHarness {
    /// Create a stub harness with an explicit list of steps.
    pub fn new(steps: Vec<StubStep>) -> Self {
        Self { steps, idx: 0 }
    }

    /// Load stub steps from a JSON file.
    pub fn from_json_path(path: &Path) -> Result<Self> {
        let raw = fs::read_to_string(path)
            .map_err(|e| miette!("Failed to read stub script {p}: {e}", p = path.display()))?;
        let steps: Vec<StubStep> = serde_json::from_str(&raw)
            .map_err(|e| miette!("Invalid stub script JSON in {p}: {e}", p = path.display()))?;
        Ok(Self::new(steps))
    }

    /// Resolve the stub script path from CLI args or `ITO_STUB_SCRIPT`.
    ///
    /// When no script is provided, this returns a single default step that
    /// yields `<promise>COMPLETE</promise>`.
    pub fn from_env_or_default(script_path: Option<PathBuf>) -> Result<Self> {
        let from_env = std::env::var("ITO_STUB_SCRIPT").ok().map(PathBuf::from);
        let path = script_path.or(from_env);
        if let Some(p) = path {
            return Self::from_json_path(&p);
        }

        // Default: single successful completion.
        Ok(Self::new(vec![StubStep {
            stdout: "<promise>COMPLETE</promise>\n".to_string(),
            stderr: String::new(),
            exit_code: 0,
        }]))
    }

    fn next_step(&mut self) -> Option<StubStep> {
        if self.steps.is_empty() {
            return None;
        }
        let step = self
            .steps
            .get(self.idx)
            .cloned()
            .or_else(|| self.steps.last().cloned());
        self.idx = self.idx.saturating_add(1);
        step
    }
}

impl Harness for StubHarness {
    fn name(&self) -> HarnessName {
        HarnessName::Stub
    }

    fn run(&mut self, _config: &HarnessRunConfig) -> Result<HarnessRunResult> {
        let started = Instant::now();
        let step = self
            .next_step()
            .ok_or_else(|| miette!("Stub harness has no steps"))?;

        Ok(HarnessRunResult {
            stdout: step.stdout,
            stderr: step.stderr,
            exit_code: step.exit_code,
            duration: started.elapsed().max(Duration::from_millis(1)),
            timed_out: false,
        })
    }

    fn stop(&mut self) {
        // No-op
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::collections::BTreeMap;

    fn dummy_config() -> HarnessRunConfig {
        HarnessRunConfig {
            prompt: "test".to_string(),
            model: None,
            cwd: std::env::temp_dir(),
            env: BTreeMap::new(),
            interactive: false,
            allow_all: false,
            inactivity_timeout: None,
        }
    }

    #[test]
    fn name_returns_stub() {
        let stub = StubHarness::new(vec![StubStep {
            stdout: "test".to_string(),
            stderr: String::new(),
            exit_code: 0,
        }]);
        assert_eq!(stub.name(), HarnessName::Stub);
    }

    #[test]
    fn streams_output_returns_false() {
        let stub = StubHarness::new(vec![StubStep {
            stdout: "test".to_string(),
            stderr: String::new(),
            exit_code: 0,
        }]);
        assert!(!stub.streams_output());
    }

    #[test]
    fn run_sets_timed_out_false() {
        let mut stub = StubHarness::new(vec![StubStep {
            stdout: "test".to_string(),
            stderr: String::new(),
            exit_code: 0,
        }]);
        let config = dummy_config();
        let result = stub.run(&config).unwrap();
        assert!(!result.timed_out);
    }

    #[test]
    fn run_sets_nonzero_duration() {
        let mut stub = StubHarness::new(vec![StubStep {
            stdout: "test".to_string(),
            stderr: String::new(),
            exit_code: 0,
        }]);
        let config = dummy_config();
        let result = stub.run(&config).unwrap();
        assert!(result.duration > Duration::ZERO);
    }

    #[test]
    fn from_env_or_default_with_explicit_path() {
        use std::io::Write;
        let mut tmpfile = tempfile::NamedTempFile::new().unwrap();
        let json = r#"[{"stdout": "hello", "stderr": "", "exitCode": 0}]"#;
        tmpfile.write_all(json.as_bytes()).unwrap();
        tmpfile.flush().unwrap();

        let mut stub =
            StubHarness::from_env_or_default(Some(tmpfile.path().to_path_buf())).unwrap();
        let config = dummy_config();
        let result = stub.run(&config).unwrap();
        assert_eq!(result.stdout, "hello");
    }
}