coreutils/tee/tee.rs

155 lines
4.6 KiB
Rust
Raw Normal View History

2014-07-06 08:13:36 +00:00
#![crate_name = "tee"]
2014-03-31 16:40:21 +00:00
#![license="MIT"]
#![feature(phase)]
#![feature(macro_rules)]
2013-12-23 17:12:26 +00:00
/*
* This file is part of the uutils coreutils package.
*
* (c) Aleksander Bielawski <pabzdzdzwiagief@gmail.com>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
extern crate getopts;
2014-06-12 04:43:47 +00:00
#[phase(plugin, link)] extern crate log;
2013-12-23 17:12:26 +00:00
use std::io::{println, stdin, stdout, Append, File, Truncate, Write};
use std::io::{IoResult};
2013-12-27 20:04:11 +00:00
use std::io::util::{copy, NullWriter, MultiWriter};
use std::os;
use getopts::{getopts, optflag, usage};
2013-12-23 17:12:26 +00:00
static NAME: &'static str = "tee";
static VERSION: &'static str = "1.0.0";
pub fn uumain(args: Vec<String>) -> int {
2014-05-28 11:43:37 +00:00
match options(args.as_slice()).and_then(exec) {
Ok(_) => 0,
Err(_) => 1
2013-12-23 17:12:26 +00:00
}
}
struct Options {
2014-05-25 09:20:52 +00:00
program: String,
2013-12-23 17:12:26 +00:00
append: bool,
ignore_interrupts: bool,
2014-05-25 09:20:52 +00:00
print_and_exit: Option<String>,
2014-05-09 00:12:57 +00:00
files: Box<Vec<Path>>
2013-12-23 17:12:26 +00:00
}
2014-05-25 09:20:52 +00:00
fn options(args: &[String]) -> Result<Options, ()> {
2014-05-30 08:35:54 +00:00
let opts = [
2013-12-23 17:12:26 +00:00
optflag("a", "append", "append to the given FILEs, do not overwrite"),
optflag("i", "ignore-interrupts", "ignore interrupt signals"),
optflag("h", "help", "display this help and exit"),
optflag("V", "version", "output version information and exit"),
];
2014-05-28 06:33:39 +00:00
let args: Vec<String> = args.iter().map(|x| x.to_string()).collect();
2014-05-17 10:32:14 +00:00
2014-06-15 10:50:40 +00:00
getopts(args.tail(), opts).map_err(|e| format!("{}", e)).and_then(|m| {
2013-12-23 17:12:26 +00:00
let version = format!("{} {}", NAME, VERSION);
2014-05-17 10:32:14 +00:00
let program = args.get(0).as_slice();
2013-12-23 17:12:26 +00:00
let arguments = "[OPTION]... [FILE]...";
2014-05-23 12:28:40 +00:00
let brief = "Copy standard input to each FILE, and also to standard output.";
2013-12-23 17:12:26 +00:00
let comment = "If a FILE is -, copy again to standard output.";
let help = format!("{}\n\nUsage:\n {} {}\n\n{}\n{}",
version, program, arguments, usage(brief, opts),
comment);
2014-05-28 06:33:39 +00:00
let names = m.free.clone().move_iter().collect::<Vec<String>>().append_one("-".to_string());
2013-12-23 17:12:26 +00:00
let to_print = if m.opt_present("help") { Some(help) }
else if m.opt_present("version") { Some(version) }
else { None };
Ok(Options {
2014-05-28 06:33:39 +00:00
program: program.to_string(),
2013-12-23 17:12:26 +00:00
append: m.opt_present("append"),
ignore_interrupts: m.opt_present("ignore-interrupts"),
print_and_exit: to_print,
2014-05-09 00:12:57 +00:00
files: box names.iter().map(|name| Path::new(name.clone())).collect()
2013-12-23 17:12:26 +00:00
})
2014-05-17 10:32:14 +00:00
}).map_err(|message| warn(message.as_slice()))
2013-12-23 17:12:26 +00:00
}
2013-12-27 20:04:11 +00:00
fn exec(options: Options) -> Result<(), ()> {
2013-12-23 17:12:26 +00:00
match options.print_and_exit {
2014-05-23 12:28:40 +00:00
Some(text) => Ok(println(text.as_slice())),
2013-12-23 17:12:26 +00:00
None => tee(options)
}
}
2013-12-27 20:04:11 +00:00
fn tee(options: Options) -> Result<(), ()> {
2014-03-22 08:18:52 +00:00
let writers = options.files.iter().map(|path| open(path, options.append)).collect();
let output = &mut MultiWriter::new(writers);
2014-05-09 00:12:57 +00:00
let input = &mut NamedReader { inner: box stdin() as Box<Reader> };
if copy(input, output).is_err() || output.flush().is_err() {
Err(())
} else {
Ok(())
2013-12-23 17:12:26 +00:00
}
}
2013-12-28 00:49:44 +00:00
2014-05-09 00:12:57 +00:00
fn open(path: &Path, append: bool) -> Box<Writer> {
let inner = if *path == Path::new("-") {
2014-05-09 00:12:57 +00:00
box stdout() as Box<Writer>
2013-12-28 01:23:13 +00:00
} else {
let mode = if append { Append } else { Truncate };
2013-12-27 20:04:11 +00:00
match File::open_mode(path, mode, Write) {
2014-05-09 00:12:57 +00:00
Ok(file) => box file as Box<Writer>,
Err(_) => box NullWriter as Box<Writer>
2013-12-27 20:04:11 +00:00
}
};
2014-05-09 00:12:57 +00:00
box NamedWriter { inner: inner, path: box path.clone() } as Box<Writer>
2013-12-27 20:04:11 +00:00
}
struct NamedWriter {
2014-05-09 00:12:57 +00:00
inner: Box<Writer>,
path: Box<Path>
2013-12-27 20:04:11 +00:00
}
impl Writer for NamedWriter {
fn write(&mut self, buf: &[u8]) -> IoResult<()> {
2014-02-15 04:50:03 +00:00
with_path(self.path.clone(), || {
let val = self.inner.write(buf);
if val.is_err() {
2014-05-09 00:12:57 +00:00
self.inner = box NullWriter as Box<Writer>;
}
val
})
2013-12-27 20:04:11 +00:00
}
fn flush(&mut self) -> IoResult<()> {
2014-02-15 04:50:03 +00:00
with_path(self.path.clone(), || {
let val = self.inner.flush();
if val.is_err() {
2014-05-09 00:12:57 +00:00
self.inner = box NullWriter as Box<Writer>;
}
val
})
2013-12-28 01:23:13 +00:00
}
}
2013-12-27 20:04:11 +00:00
struct NamedReader {
2014-05-09 00:12:57 +00:00
inner: Box<Reader>
2013-12-27 20:04:11 +00:00
}
impl Reader for NamedReader {
fn read(&mut self, buf: &mut [u8]) -> IoResult<uint> {
with_path(&Path::new("stdin"), || {
self.inner.read(buf)
})
2013-12-27 20:04:11 +00:00
}
}
fn with_path<T>(path: &Path, cb: || -> IoResult<T>) -> IoResult<T> {
match cb() {
2014-05-23 12:28:40 +00:00
Err(f) => { warn(format!("{}: {}", path.display(), f.to_str()).as_slice()); Err(f) }
okay => okay
}
2013-12-27 20:04:11 +00:00
}
2013-12-28 00:49:44 +00:00
fn warn(message: &str) {
error!("{}: {}", os::args().get(0), message);
2013-12-28 00:49:44 +00:00
}