coreutils/mkuutils.rs

#![allow(unstable)]

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

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

    let mut crates = String::new();
    let mut util_map = String::new();
    let mut hashsum = false;
    for prog in args[2..].iter() {
        match prog.as_slice() {
            "hashsum" | "md5sum" | "sha1sum" | "sha224sum" | "sha256sum" | "sha384sum" | "sha512sum" => {
                if !hashsum {
                    crates.push_str("extern crate hashsum;\n");
                    util_map.push_str("map.insert(\"hashsum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"md5sum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"sha1sum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"sha224sum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"sha256sum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"sha384sum\", hashsum::uumain);\n");
                    util_map.push_str("map.insert(\"sha512sum\", hashsum::uumain);\n");
                    hashsum = true;
                }
            }
            "true" => util_map.push_str("fn uutrue(_: Vec<String>) -> isize { 0 }\nmap.insert(\"true\", uutrue);\n"),
            "false" => util_map.push_str("fn uufalse(_: Vec<String>) -> isize { 1 }\nmap.insert(\"false\", uufalse);\n"),
            _ => {
                crates.push_str(format!("extern crate \"{0}\" as uu{0};\n", prog).as_slice());
                util_map.push_str(format!("map.insert(\"{prog}\", uu{prog}::uumain as fn(Vec<String>) -> isize);\n", prog = prog).as_slice());
            }
        }
    }
    let outfile = args[1].as_slice();

    // XXX: this all just assumes that the IO works correctly
    let mut out = File::open_mode(&Path::new(outfile), Truncate, Write).unwrap();
    let mut input = File::open(&Path::new("src/uutils/uutils.rs")).unwrap();
    let main = input.read_to_string().unwrap().replace("@CRATES@", crates.as_slice()).replace("@UTIL_MAP@", util_map.as_slice());

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