homebrew 0.2.6

use std::{collections::HashMap, process::{Child, Command, Stdio}};
use std::io::{self, BufRead};

use anyhow::{anyhow, Result};

use crate::config::get_brew_bin;

/// `brew` 命令构造器
///
/// 有一个快速执行方式见 [`brew`]
///
/// Examples
///
/// ```
/// extern crate homebrew;
///
/// let out = homebrew::Brew::new("--caskroom")
///     // .set_env("HOMEBREW_NO_AUTO_UPDATE", "1")
///     // use set_env_no_auto_update replace
///     .set_env_no_auto_update()
///     .output().unwrap();
///
/// let out2 = homebrew::brew("--caskroom").unwrap();
///
/// assert_eq!(out, out2);
/// ```
///
/// Examples
///
/// use `output_vec()`
///
/// ```
/// extern crate homebrew;
///
/// let out = homebrew::Brew::new("search wget")
///     .set_env_no_auto_update()
///     .output_vec().unwrap();
///
/// assert_eq!(out, ["wget", "wget2", "wgetpaste"]);
/// ```
///
/// Examples
///
/// use `default()`
///
/// ```
/// extern crate homebrew;
///
/// let out = homebrew::Brew::default()
///     .set_cmd("search wget")
///     .output_vec().unwrap();
///
/// assert_eq!(out, ["wget", "wget2", "wgetpaste"]);
/// ```
#[derive(Debug, Clone)]
pub struct Brew {
    cmd_: String,
    env_: HashMap<String, String>,
}

impl Default for Brew {
    /// 给一个默认的 `Brew` 会自动设置环境变量等信息
    fn default() -> Self{
        let mut e = HashMap::new();
        e.insert("HOMEBREW_NO_AUTO_UPDATE".to_string(), "1".to_string());
        Self {
            cmd_: String::new(),
            env_: e,
        }
    }
}

impl Brew {
    /// 新建一个 `brew` 命令构造器
    pub fn new<T: AsRef<str>>(cmd: T) -> Self {
        Self {
            cmd_: cmd.as_ref().to_string(),
            env_: HashMap::new(),
        }
    }

    /// 设置命令
    pub fn set_cmd<S: AsRef<str>>(&mut self, cmd: S) -> &mut Self{
        self.cmd_ = cmd.as_ref().to_string();
        self
    }

    /// 添加环境变量
    pub fn set_env<K: AsRef<str>, V: AsRef<str>>(&mut self, key: K, value: V) -> &mut Self{
        self.env_.insert(key.as_ref().to_string(), value.as_ref().to_string());
        self
    }

    /// 添加环境变量 HOMEBREW_NO_AUTO_UPDATE=1
    ///
    /// 更多环境变量可以通过 [`Brew::set_env`] 设置
    pub fn set_env_no_auto_update(&mut self) -> &mut Self {
        self.set_env("HOMEBREW_NO_AUTO_UPDATE", "1")
    }

    /// 返回 `brew` 命令并得到 [`String`] 类型数据
    pub fn output(&self) -> Result<String> {
        let bin = get_brew_bin()?;
        let cmds = self.cmd_.split(' ');
        let output = Command::new(bin)
            .args(cmds)
            .envs(&self.env_)
            .output()?;

        let outerr = String::from_utf8_lossy(&output.stderr);
        if !outerr.is_empty() {
            if let Some(e) = outerr.strip_suffix("\n") {
                return Err(anyhow!("{e}"));
            } else {
                return Err(anyhow!("{outerr}"));
            }
        }
        let out = String::from_utf8_lossy(&output.stdout);
        if let Some(o) = out.strip_suffix("\n") {
            return Ok(o.to_string());
        }
        Ok(out.to_string())
    }

    /// 返回 `brew` 命令并得到 [`Vec<String>`] 数据
    pub fn output_vec(&self) -> Result<Vec<String>> {
        let out = self.output()?;
        let res: Vec<String> = out.split('\n')
            .filter(|s| !s.is_empty())
            .map(|s| s.to_string())
            .collect();
        Ok(res)
    }

    /// 实时打印命令信息，但是不会返回结果
    ///
    /// Examples
    ///
    /// ```ignore
    /// extern crate homebrew;
    ///
    /// homebrew::Brew::new("update").spawn.unwrap();
    /// ```
    ///
    /// ```bash
    /// HOMEBREW_BREW_GIT_REMOTE set: using https://mirrors.tuna.tsinghua.edu.cn/git/homebrew/brew.git as the Homebrew/brew Git remote.
    /// ==> Updating Homebrew...
    /// Already up-to-date.
    /// ```
    pub fn spawn(&self) -> Result<()> {
        let bin = get_brew_bin()?;
        let cmds = self.cmd_.split(' ');
        let mut child: Child = Command::new(bin)
            .args(cmds)
            .stdout(Stdio::piped()) // 将标准输出设置为管道
            .spawn()?; // 启动命令

        // 获取标准输出的句柄
        let stdout = child.stdout.take().ok_or_else(|| anyhow!("Could not capture standard output"))?;

        // 创建一个 BufReader 来逐行读取输出
        let reader = io::BufReader::new(stdout);

        // 逐行读取输出并打印
        for line in reader.lines() {
            let line = line?;
            println!("{}", line);
        }

        // 等待命令完成
        let _ = child.wait()?;
        Ok(())
    }
}

/// 执行 `brew` 命令
///
/// 默认添加环境变量 HOMEBREW_NO_AUTO_UPDATE=1
///
/// 如果想要更复杂的构造执行器，请使用 [`Brew`]
///
/// Examples
///
/// ```
/// extern crate homebrew;
///
/// let out = homebrew::brew("--prefix").unwrap();
///
/// // arm64 架构的可以运行通过，intel 芯片请使用 `/usr/local/homebrew`
/// assert_eq!(out, "/opt/homebrew");
/// ```
pub fn brew(cmd: &str) -> Result<String> {
    Brew::new(cmd)
        .set_env_no_auto_update()
        .output()
}

/// 执行 `brew` 命令并实时输出信息
///
/// 默认添加环境变量 HOMEBREW_NO_AUTO_UPDATE=1
///
/// 如果想要更复杂的构造执行器，请使用 [`Brew`]
///
/// Examples
///
/// ```ignore
/// extern crate homebrew;
///
/// homebrew::brew_spawn("update")?;
/// ```
///
/// ```bash
/// HOMEBREW_BREW_GIT_REMOTE set: using https://mirrors.tuna.tsinghua.edu.cn/git/homebrew/brew.git as the Homebrew/brew Git remote.
/// ==> Updating Homebrew...
/// Already up-to-date.
/// ```
pub fn brew_spawn(cmd: &str) -> Result<()> {
    Brew::new(cmd)
        .set_env_no_auto_update()
        .spawn()
}