rsmultigit 0.1.10

#![allow(dead_code)]

use std::fs;
use std::io::Write;
use std::path::{Path, PathBuf};
use std::process::{Command, Output, Stdio};
use tempfile::TempDir;

/// Write a minimal config file at <dir>/config.toml with `repos = ["<dir>/*"]`
/// and the given `extra` body appended (for `[[check]]` blocks, etc).
/// Returns the config path, to be fed into `RSMULTIGIT_CONFIG`.
pub fn write_config(dir: &Path, extra: &str) -> PathBuf {
    let path = dir.join("config.toml");
    let body = format!("repos = [\"{}/*\"]\n\n{}", dir.display(), extra);
    fs::write(&path, body).unwrap();
    path
}

/// Run rsmultigit against a tempdir of git repos, pointing the tool at
/// a config written by `write_config`.
pub fn run_rsmultigit(dir: &Path, args: &[&str]) -> Output {
    let cfg = write_config(dir, "");
    let cfg_str = cfg.to_string_lossy().into_owned();
    run_rsmultigit_with_env(dir, args, &[("RSMULTIGIT_CONFIG", &cfg_str)])
}

/// Run the rsmultigit binary with extra env vars layered on top of the parent env.
pub fn run_rsmultigit_with_env(dir: &Path, args: &[&str], env: &[(&str, &str)]) -> Output {
    let bin_path = env!("CARGO_BIN_EXE_rsmultigit");
    let mut cmd = Command::new(bin_path);
    cmd.current_dir(dir).args(args);
    for (k, v) in env {
        cmd.env(k, v);
    }
    cmd.output().expect("Failed to execute rsmultigit")
}

/// Run the rsmultigit binary with piped stdin so tests can feed interactive input.
/// Writes `stdin_bytes` to the child's stdin, then reads the output.
pub fn run_rsmultigit_with_stdin(
    dir: &Path,
    args: &[&str],
    env: &[(&str, &str)],
    stdin_bytes: &[u8],
) -> Output {
    let bin_path = env!("CARGO_BIN_EXE_rsmultigit");
    let mut cmd = Command::new(bin_path);
    cmd.current_dir(dir)
        .args(args)
        .stdin(Stdio::piped())
        .stdout(Stdio::piped())
        .stderr(Stdio::piped());
    for (k, v) in env {
        cmd.env(k, v);
    }
    let mut child = cmd.spawn().expect("Failed to spawn rsmultigit");
    {
        let stdin = child.stdin.as_mut().expect("stdin must be piped");
        stdin.write_all(stdin_bytes).expect("failed writing stdin");
    }
    child
        .wait_with_output()
        .expect("Failed to wait on rsmultigit")
}

/// Get stdout from an Output as a trimmed String.
pub fn stdout_str(output: &Output) -> String {
    String::from_utf8_lossy(&output.stdout).trim().to_string()
}

/// Get stderr from an Output as a trimmed String.
pub fn stderr_str(output: &Output) -> String {
    String::from_utf8_lossy(&output.stderr).trim().to_string()
}

/// Create a temp directory containing `n` fake git repos as immediate subdirectories.
/// Returns the TempDir (caller must hold it to keep the directory alive).
pub fn setup_git_repos(names: &[&str]) -> TempDir {
    let tmp = TempDir::new().expect("Failed to create temp dir");
    for name in names {
        let repo_path = tmp.path().join(name);
        init_git_repo(&repo_path);
    }
    tmp
}

/// Initialise a minimal git repo at `path` with one commit.
pub fn init_git_repo(path: &Path) {
    fs::create_dir_all(path).unwrap();
    let status = Command::new("git")
        .args(["init", "-q"])
        .current_dir(path)
        .status()
        .unwrap();
    assert!(status.success(), "git init failed");

    // Configure user for the repo so commits work
    Command::new("git")
        .args(["config", "user.email", "test@test.com"])
        .current_dir(path)
        .status()
        .unwrap();
    Command::new("git")
        .args(["config", "user.name", "Test"])
        .current_dir(path)
        .status()
        .unwrap();

    // Create an initial commit so HEAD exists
    Command::new("git")
        .args(["commit", "--allow-empty", "-m", "initial"])
        .current_dir(path)
        .status()
        .unwrap();
}