uutils-coreutils/mkmain.rs

#![allow(unstable)]

use std::io::{File, Truncate, ReadWrite};
use std::os;
use std::path::Path;

static TEMPLATE: &'static str = "\
#![allow(unstable)]
extern crate @UTIL_CRATE@;

use std::os;
use @UTIL_CRATE@::uumain;

fn main() {
    os::set_exit_status(uumain(os::args()));
}
";

fn main() {
    let args = os::args();
    if args.len() != 3 {
        println!("usage: mkbuild <crate> <outfile>");
        os::set_exit_status(1);
        return;
    }

    let crat = match args[1].as_slice() {
        "test" => "uutest",
        "true" => "uutrue",
        "false" => "uufalse",
        "sync" => "uusync",
        s => s.clone(),
    };
    let outfile  = args[2].as_slice();

    let main = TEMPLATE.replace("@UTIL_CRATE@", crat);
    let mut out = File::open_mode(&Path::new(outfile), Truncate, ReadWrite);

    match out.write(main.as_bytes()) {
        Err(e) => panic!("{}", e),
        _ => (),
    }
}