diff --git a/Makefile b/Makefile index 1d679ecf6..3f8f6111d 100644 --- a/Makefile +++ b/Makefile @@ -19,6 +19,7 @@ PROGS := \ rm \ rmdir \ sleep \ + tee \ true \ wc \ whoami \ diff --git a/README.md b/README.md index b26bbf0cd..2e2b248e1 100644 --- a/README.md +++ b/README.md @@ -154,7 +154,6 @@ To do - tac-pipe - tac - tail -- tee - test - timeout - touch diff --git a/tee/tee.rs b/tee/tee.rs new file mode 100644 index 000000000..543c5aabc --- /dev/null +++ b/tee/tee.rs @@ -0,0 +1,155 @@ +#[crate_id(name="tee", vers="1.0.0", author="Aleksander Bielawski")]; +#[license="MIT"]; + +/* + * This file is part of the uutils coreutils package. + * + * (c) Aleksander Bielawski + * + * 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::{stdin, stdout, Append, File, Truncate, Write}; +use std::io::{io_error, EndOfFile}; +use std::io::signal::{Interrupt, Listener}; +use std::io::util::{copy, NullWriter, MultiWriter}; +use std::os::{args, set_exit_status}; +use extra::getopts::groups::{getopts, optflag, usage}; + +static NAME: &'static str = "tee"; +static VERSION: &'static str = "1.0.0"; + +fn main() { + match options(args()).and_then(exec) { + Ok(_) => set_exit_status(0), + Err(_) => set_exit_status(1) + } +} + +struct Options { + program: ~str, + append: bool, + ignore_interrupts: bool, + print_and_exit: Option<~str>, + files: ~[Path] +} + +fn options(args: &[~str]) -> Result { + let opts = ~[ + optflag("a", "append", "append to the given FILEs, do not overwrite"), + optflag("i", "ignore-interrupts", "ignore interrupt signals"), + optflag("", "help", "display this help and exit"), + optflag("", "version", "output version information and exit")]; + getopts(args.tail(), opts).map_err(|e| e.to_err_msg()).and_then(|m| { + let version = format!("{} {}", NAME, VERSION); + let program = args[0].clone(); + let arguments = "[OPTION]... [FILE]..."; + let brief = "Copy standard input to each FILE, and also to standard " + + "output."; + 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); + let names = m.free + ~[~"-"]; + let to_print = if m.opt_present("help") { Some(help) } + else if m.opt_present("version") { Some(version) } + else { None }; + Ok(Options { + program: program, + append: m.opt_present("append"), + ignore_interrupts: m.opt_present("ignore-interrupts"), + print_and_exit: to_print, + files: names.map(|name| Path::new(name.clone())) + }) + }).map_err(|message| warn(message)) +} + +fn exec(options: Options) -> Result<(), ()> { + match options.print_and_exit { + Some(text) => Ok(println(text)), + None => tee(options) + } +} + +fn tee(options: Options) -> Result<(), ()> { + let mut handler = Listener::new(); + if options.ignore_interrupts { + handler.register(Interrupt); + } + let mut ok = true; + io_error::cond.trap(|_| ok = false).inside(|| { + let writers = options.files.map(|path| open(path, options.append)); + let output = &mut MultiWriter::new(writers); + let input = &mut NamedReader { inner: ~stdin() as ~Reader }; + copy(input, output); + output.flush(); + }); + if ok { Ok(()) } else { Err(()) } +} + +fn open(path: &Path, append: bool) -> ~Writer { + let inner = with_path(path, || if *path == Path::new("-") { + ~stdout() as ~Writer + } else { + let mode = if append { Append } else { Truncate }; + match File::open_mode(path, mode, Write) { + Some(file) => ~file as ~Writer, + None => ~NullWriter as ~Writer + } + }); + ~NamedWriter { inner: inner, path: ~path.clone() } as ~Writer +} + +struct NamedWriter { + priv inner: ~Writer, + priv path: ~Path +} + +impl Writer for NamedWriter { + fn write(&mut self, buf: &[u8]) { + with_path(self.path, || io_error::cond.trap(|e| { + self.inner = ~NullWriter as ~Writer; + io_error::cond.raise(e); + }).inside(|| self.inner.write(buf))) + } + + fn flush(&mut self) { + with_path(self.path, || io_error::cond.trap(|e| { + self.inner = ~NullWriter as ~Writer; + io_error::cond.raise(e); + }).inside(|| self.inner.flush())) + } +} + +struct NamedReader { + priv inner: ~Reader +} + +impl Reader for NamedReader { + fn read(&mut self, buf: &mut [u8]) -> Option { + with_path(&Path::new("stdin"), || io_error::cond.trap(|e| { + if e.kind != EndOfFile { + io_error::cond.raise(e) + } + }).inside(|| self.inner.read(buf))) + + } + + fn eof(&mut self) -> bool { + self.inner.eof() + } +} + +fn with_path(path: &Path, cb: || -> T) -> T { + io_error::cond.trap(|e| { + warn(format!("{}: {}", path.display(), e.desc)); + io_error::cond.raise(e); + }).inside(cb) +} + +fn warn(message: &str) { + error!("{}: {}", args()[0], message); +}