st.rs

#!/usr/bin/env run-cargo-script

use std::process::Command;

fn main() -> std::io::Result<()> {
    let args: Vec<std::ffi::OsString> = std::env::args_os().collect();
    if args.len() > 1 {
        eprintln!("no arguments allowed");
        std::process::exit(1);
    }

    let mut path = std::env::current_dir()?;

    loop {
        let mut hg = path.clone();
        hg.push(".hg");
        if hg.is_dir() {
            std::process::exit(
                Command::new("hg")
                    .arg("status")
                    .spawn()?
                    .wait()?
                    .code()
                    .unwrap(),
            );
        }

        let mut git = path.clone();
        git.push(".git");
        if git.is_dir() {
            std::process::exit(
                Command::new("git")
                    .arg("status")
                    .spawn()?
                    .wait()?
                    .code()
                    .unwrap(),
            );
        }

        if !path.pop() {
            eprintln!("not inside a checkout");
            std::process::exit(1);
        }
    }
}