use std::{
    ffi::{OsStr, OsString},
    fmt::{self, Display},
    iter::once,
    process::{Command, Output},
};

use format::lazy_format;
use miette::{ensure, Context, IntoDiagnostic};

pub(crate) fn which(name: &'static str, desc: &str) -> miette::Result<OsString> {
    let found = ::which::which(name)
        .into_diagnostic()
        .wrap_err(lazy_format!("failed to find `{name}` executable"))?;
    log::debug!("using {desc} from {}", found.display());
    Ok(found.file_name().unwrap().to_owned())
}

pub(crate) struct EasyCommand {
    inner: Command,
}

impl EasyCommand {
    pub(crate) fn new<C>(cmd: C, f: impl FnOnce(&mut Command) -> &e='color:red'>mut Command) -> Self
    where
        C: AsRef<OsStr>,
    {
        let mut cmd = Command::new(cmd);
        f(&mut cmd);
        Self { inner: cmd }
    }

    pub(crate) fn spawn(&mut self) -> miette::Result<()> {
        log::debug!("spawning {self}…");
        let status = self
            .inner
            .spawn()
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to spawn {self}"))?
            .wait()
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to wait for exit code from {self}"))?;
        log::debug!("{self} returned {:?}", status.code());
        ensure!(status.success(), "{self} returned {:?}", status.code());
        Ok(())
    }

    fn just_stdout(&mut self) -> miette::Result<Vec<u8>> {
        log::debug!("getting `stdout` output of {self}");
        let output = self
            .inner
            .output()
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to execute `{self}`"))?;
        let Output {
            status,
            stdout: _,
            stderr,
        } = &output;
        log::debug!("{self} returned {:?}", status.code());
        ensure!(
            status.success(),
            "{self} returned {:?}; full output: {output:#?}",
            status.code(),
        );
        assert!(stderr.is_empty());
        Ok(output.stdout)
    }

    pub(crate) fn just_stdout_utf8(&mut self) -> miette::Result<String> {
        String::from_utf8(self.just_stdout()?)
            .into_diagnostic()
            .wrap_err_with(|| format!("output of {self} was not UTF-8 (!?)"))
    }
}

impl Display for EasyCommand {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let Self { inner } = self;
        let prog = inner.get_program().to_string_lossy();
        let args = inner.get_args().map(|a| a.to_string_lossy());
        let shell_words = ::shell_words::join(once(prog).chain(args));
        write!(f, "`{shell_words}`")
    }
}