coreutils/basename/basename.rs

#[crate_id(name="basename", vers="1.0.0", author="Jimmy Lu")];

/*
 * This file is part of the uutils coreutils package.
 *
 * (c) Jimmy Lu <jimmy.lu.2011@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

extern mod extra;

use std::io::stderr;
use std::os;
use std::str;
use std::str::StrSlice;
use extra::getopts::groups;

static VERSION: &'static str = "1.0.0";

fn main() {
    let args = os::args();

    let program = strip_dir(&args[ 0 ].clone());

    //
    // Argument parsing
    //

    let opts = ~[
        groups::optflag("h", "help", "display this help and exit"),
        groups::optflag("V", "version", "output version information and exit"),
    ];

    let matches = match groups::getopts(args.tail(), opts) {
        Ok(m) => m,
        Err(f) => {
            writeln!(&mut stderr() as &mut Writer,
                   "Invalid options\n{}", f.to_err_msg());
            os::set_exit_status(1);
            return;
        }
    };

    if matches.opt_present("help") {
        println!("Usage: {0:s} NAME [SUFFIX]", program);
        println!("  or: {0:s} OPTION", program);
        println("Print NAME with any leading directory components removed.");
        println("If specified, also remove a trailing SUFFIX.");

        print(groups::usage("", opts));

        return;
    }

    if matches.opt_present("version") {
        println(program + " " + VERSION);
        return;
    }

    // too few arguments
    if args.len() < 2 {
        println(program + ": missing operand");
        println("Try `" + program + " --help' for more information.");
        return;
    }
    // too many arguments
    else if args.len() > 3 {
        println(program + ": extra operand `" + args[ 3 ] + "'");
        println("Try `" + program + " --help' for more information.");
        return;
    }

    //
    // Main Program Processing
    //

    let fullname = args[ 1 ].clone();

    let mut name = strip_dir(&fullname);

    if args.len() > 2 {
        let suffix = args[ 2 ].clone();
        name = strip_suffix(&name, &suffix);
    }

    println(name);
}

fn strip_dir(fullname :&~str) -> ~str {
    let mut name = ~"";

    for c in fullname.chars_rev() {
        if c == '/' || c == '\\' {
            return name;
        }

        name = str::from_char(c) + name;
    }

    return fullname.clone();
}

fn strip_suffix(name: &~str, suffix: &~str) -> ~str {
    if name == suffix {
        return name.clone();
    }

    if name.ends_with(*suffix) {
        return name.slice_to(name.len() - suffix.len()).into_owned();
    }

    return name.clone();
}