cargo-pipe 0.2.2

use crate::{errors::CmdResult, print::Printer};
use lazy_static::lazy_static;
use regex::{Captures, Regex};
use std::ffi::OsString;
use std::path::Path;
use std::process::Command;

pub(crate) const CARGO_MANIFEST_FILE: &str = "Cargo.toml";
pub(crate) const CARGO_SRC_DIRECTORY: &str = "src";
pub(crate) const CARGO_TARGET_DIRECTORY: &str = "target";
pub(crate) const CARGO_RELEASE_DIRECTORY: &str = "release";
pub(crate) const CARGO_DEBUG_DIRECTORY: &str = "debug";
pub(crate) const CARGO_APP_MAIN: &str = "main.rs";

fn capture(captures: Option<Captures>) -> Option<String> {
    let m = match captures {
        Some(captures) => captures.get(1),
        None => return None,
    };
    m.map(|m| m.as_str().to_owned())
}

fn capture_error_message(line: &str) -> Option<String> {
    lazy_static! {
        static ref ERROR_CODE: Regex = Regex::new(r"error\[E\d{4}\]:\s*(.*)").unwrap();
        static ref ERROR: Regex = Regex::new(r"error:\s*(.*)").unwrap();
    }
    if let Some(error_message) = capture(ERROR_CODE.captures(line)) {
        return Some(error_message);
    }
    if let Some(error_message) = capture(ERROR.captures(line)) {
        return Some(error_message);
    }
    None
}

fn capture_warning_message(line: &str) -> Option<String> {
    lazy_static! {
        static ref WARNING: Regex = Regex::new(r"warning:\s*(.*)").unwrap();
    }
    capture(WARNING.captures(line))
}

// capture error or warning message
fn capture_messages(out: String, printer: &mut Printer) -> CmdResult<()> {
    let lines: Vec<&str> = out.split('\n').collect();
    for line in lines {
        if let Some(error_message) = capture_error_message(line) {
            printer.error(error_message.to_string())?;
            continue;
        }
        if let Some(warning_message) = capture_warning_message(line) {
            printer.warning(warning_message)?;
        }
    }
    Ok(())
}

fn run_cmd(mut cmd: Command) -> CmdResult<(i32, String)> {
    let output = cmd.output()?;
    match output.status.success() {
        true => {
            let stderr = String::from_utf8(output.stderr)?;
            Ok((0, stderr))
        }
        false => {
            let stderr = String::from_utf8(output.stderr)?;
            let err_code = output.status.code().unwrap_or(1);
            Ok((err_code, stderr))
        }
    }
}

fn cargo_binary() -> OsString {
    match std::env::var_os("CARGO") {
        Some(cargo) => cargo,
        None => "cargo".to_owned().into(),
    }
}

pub fn do_cargo_init(path: &Path, printer: &mut Printer) -> CmdResult<i32> {
    printer.status(&"Cargo", "init ...")?;
    let mut cmd = Command::new(cargo_binary());
    cmd.arg("init").arg(path);
    let (err_code, out) = run_cmd(cmd)?;
    match err_code {
        0 => printer.status(&"Cargo", "init succeed")?,
        _ => printer.status(&"Cargo", format!("init failed, stderr: {}", out))?,
    };
    Ok(err_code)
}

pub fn do_cargo_fmt(manifest_path: &Path, printer: &mut Printer) -> CmdResult<i32> {
    printer.status(&"Cargo", "fmt")?;
    let mut cmd = Command::new(cargo_binary());
    cmd.arg("fmt").arg("--manifest-path").arg(manifest_path);
    let (err_code, out) = run_cmd(cmd)?;
    match err_code {
        0 => printer.status(&"Cargo", "fmt succeed")?,
        _ => printer.error(format!("fmt failed, stderr: {}", out))?,
    };
    Ok(err_code)
}

pub fn do_cargo_check(
    manifest_path: &Path,
    verbose: bool,
    debug: bool,
    printer: &mut Printer,
) -> CmdResult<i32> {
    printer.status(&"Cargo", "check ...")?;
    let mut cmd = Command::new(cargo_binary());
    cmd.arg("check").arg("--manifest-path").arg(manifest_path);
    if verbose {
        cmd.arg("--verbose");
    }
    let (status_code, out) = run_cmd(cmd)?;
    if debug {
        if verbose && status_code != 0 {
            printer.error(out)?
        } else {
            capture_messages(out, printer)?
        }
    }
    match status_code {
        0 => printer.status(&"Cargo", "check succeed")?,
        _ => printer.error("cargo check failed")?,
    };
    Ok(status_code)
}

pub fn do_cargo_build(
    manifest_path: &Path,
    release: bool,
    debug: bool,
    verbose: bool,
    printer: &mut Printer,
) -> CmdResult<i32> {
    printer.status(&"Cargo", "build ...")?;
    let mut cmd = Command::new(cargo_binary());
    cmd.arg("build").arg("--manifest-path").arg(manifest_path);
    if release {
        cmd.arg("--release");
    }
    let (status_code, out) = run_cmd(cmd)?;
    if debug {
        if verbose && status_code != 0 {
            printer.error(out)?
        } else {
            capture_messages(out, printer)?
        }
    }
    match status_code {
        0 => printer.status(&"Cargo", "build succeed")?,
        _ => printer.error("cargo build failed")?,
    };
    Ok(status_code)
}