greentic-dev 0.6.0-dev.25051861356

use std::ffi::OsString;

use crate::config::{self, GreenticConfig};
use crate::passthrough::resolve_binary;
use crate::util::process::{self, CommandOutput, CommandSpec, StreamMode};
use anyhow::{Context, Result, anyhow, bail};

const TOOL_NAME: &str = "greentic-component";

pub struct ComponentDelegate {
    program: OsString,
}

impl ComponentDelegate {
    pub fn from_config(config: &GreenticConfig) -> Result<Self> {
        let resolved = resolve_program(config)?;
        Ok(Self {
            program: resolved.program,
        })
    }

    pub fn run_passthrough(&self, args: &[String]) -> Result<()> {
        let argv: Vec<OsString> = args.iter().map(OsString::from).collect();
        let label = args.first().map(|s| s.as_str()).unwrap_or("<component>");
        let output = self.exec(argv, false)?;
        self.ensure_success(label, false, &output)
    }

    fn exec(&self, args: Vec<OsString>, capture: bool) -> Result<CommandOutput> {
        let mut spec = CommandSpec::new(self.program.clone());
        spec.args = args;
        if capture {
            spec.stdout = StreamMode::Capture;
            spec.stderr = StreamMode::Capture;
        } else {
            spec.stdout = StreamMode::Inherit;
            spec.stderr = StreamMode::Inherit;
        }
        process::run(spec)
            .with_context(|| format!("failed to spawn `{}`", self.program.to_string_lossy()))
    }

    fn ensure_success(&self, label: &str, capture: bool, output: &CommandOutput) -> Result<()> {
        if output.status.success() {
            return Ok(());
        }

        if capture
            && let Some(stderr) = output.stderr.as_ref()
            && !stderr.is_empty()
        {
            eprintln!("{}", String::from_utf8_lossy(stderr));
        }
        let code = output.status.code().unwrap_or_default();
        bail!(
            "`{}` {label} failed with exit code {code}",
            self.program.to_string_lossy()
        );
    }
}

struct ResolvedProgram {
    program: OsString,
}

fn resolve_program(config: &GreenticConfig) -> Result<ResolvedProgram> {
    if let Some(custom) = config.tools.greentic_component.path.as_ref() {
        if !custom.exists() {
            bail!(
                "configured greentic-component path `{}` does not exist",
                custom.display()
            );
        }
        return Ok(ResolvedProgram {
            program: custom.as_os_str().to_os_string(),
        });
    }

    match resolve_binary(TOOL_NAME) {
        Ok(path) => Ok(ResolvedProgram {
            program: path.into_os_string(),
        }),
        Err(error) => {
            let config_hint = config::config_path()
                .map(|path| path.display().to_string())
                .unwrap_or_else(|| "$XDG_CONFIG_HOME/greentic-dev/config.toml".to_string());
            Err(anyhow!(
                "failed to locate `{TOOL_NAME}` on PATH ({error}). Install it via `cargo install \
                 greentic-component` or set [tools.greentic-component].path in {config_hint}."
            ))
        }
    }
}

#[cfg(test)]
mod tests {
    use super::{ComponentDelegate, resolve_program};
    use crate::config::GreenticConfig;
    use crate::util::process::CommandOutput;
    use std::ffi::OsString;
    use std::process::ExitStatus;
    use tempfile::tempdir;

    #[cfg(unix)]
    fn success_status() -> ExitStatus {
        use std::os::unix::process::ExitStatusExt;
        ExitStatus::from_raw(0)
    }

    #[cfg(unix)]
    fn failure_status(code: i32) -> ExitStatus {
        use std::os::unix::process::ExitStatusExt;
        ExitStatus::from_raw(code << 8)
    }

    #[test]
    fn resolve_program_uses_existing_configured_path() {
        let dir = tempdir().unwrap();
        let custom = dir.path().join("greentic-component");
        std::fs::write(&custom, "stub").unwrap();

        let mut config = GreenticConfig::default();
        config.tools.greentic_component.path = Some(custom.clone());

        let resolved = resolve_program(&config).unwrap();
        assert_eq!(resolved.program, custom.into_os_string());
    }

    #[test]
    fn resolve_program_rejects_missing_configured_path() {
        let mut config = GreenticConfig::default();
        config.tools.greentic_component.path =
            Some(tempdir().unwrap().path().join("missing-component"));

        let err = resolve_program(&config)
            .err()
            .expect("expected missing path");
        assert!(err.to_string().contains("does not exist"));
    }

    #[test]
    fn ensure_success_accepts_successful_status() {
        let delegate = ComponentDelegate {
            program: OsString::from("greentic-component"),
        };
        let output = CommandOutput {
            status: success_status(),
            stdout: None,
            stderr: None,
        };

        delegate.ensure_success("doctor", false, &output).unwrap();
    }

    #[test]
    fn ensure_success_reports_failure_code() {
        let delegate = ComponentDelegate {
            program: OsString::from("greentic-component"),
        };
        let output = CommandOutput {
            status: failure_status(7),
            stdout: None,
            stderr: Some(b"boom".to_vec()),
        };

        let err = delegate
            .ensure_success("doctor", true, &output)
            .unwrap_err();
        assert!(err.to_string().contains("exit code 7"));
    }
}