coreutils/tac/tac.rs

#![crate_id(name = "tac", vers = "1.0.0", author = "Arcterus")]

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

#![feature(macro_rules)]

extern crate getopts;
extern crate libc;

use std::io;
use std::os;

#[path = "../common/util.rs"]
mod util;

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

fn main() {
    let args: Vec<StrBuf> = os::args().iter().map(|x| x.to_strbuf()).collect();
    let program = args.get(0).clone();

    let opts = ~[
        getopts::optflag("b", "before", "attach the separator before instead of after"),
        getopts::optflag("r", "regex", "interpret the sequence as a regular expression (NOT IMPLEMENTED)"),
        getopts::optopt("s", "separator", "use STRING as the separator instead of newline", "STRING"),
        getopts::optflag("h", "help", "display this help and exit"),
        getopts::optflag("V", "version", "output version information and exit")
    ];
    let matches = match getopts::getopts(args.tail(), opts) {
        Ok(m) => m,
        Err(f) => crash!(1, "{}", f.to_err_msg())
    };
    if matches.opt_present("help") {
        println!("tac {}", VERSION);
        println!("");
        println!("Usage:");
        println!("  {0:s} [OPTION]... [FILE]...", program);
        println!("");
        print!("{}", getopts::usage("Write each file to standard output, last line first.", opts));
    } else if matches.opt_present("version") {
        println!("tac {}", VERSION);
    } else {
        let before = matches.opt_present("b");
        let regex = matches.opt_present("r");
        let separator = match matches.opt_str("s") {
            Some(m) => {
                if m.len() == 0 {
                    crash!(1, "separator cannot be empty")
                } else {
                    m
                }
            }
            None => "\n".to_strbuf()
        };
        let files = if matches.free.is_empty() {
            vec!("-".to_strbuf())
        } else {
            matches.free
        };
        tac(files, before, regex, separator.as_slice());
    }
}

fn tac(filenames: Vec<StrBuf>, before: bool, _: bool, separator: &str) {
    for filename in filenames.move_iter() {
        let mut file = io::BufferedReader::new(
            if filename.as_slice() == "-" {
                box io::stdio::stdin_raw() as Box<Reader>
            } else {
                box crash_if_err!(1, io::File::open(&Path::new(filename))) as Box<Reader>
            }
        );
        let mut data = crash_if_err!(1, file.read_to_str());
        if data.as_slice().ends_with("\n") {
            // removes blank line that is inserted otherwise
            let mut buf = data.into_strbuf();
            let len = buf.len();
            buf.truncate(len - 1);
            data = buf.into_owned();
        }
        let split_vec: Vec<&str> = data.as_slice().split_str(separator).collect();
        let rev: StrBuf = split_vec.iter().rev().fold(StrBuf::new(), |mut a, &b| {
            if before {
               a.push_str(separator);
               a.push_str(b);
            } else {
                a.push_str(b);
                a.push_str(separator);
            }
            a
        });
        print!("{}", rev);
    }
}