diff --git a/Makefile b/Makefile index 188c81552..5e78649df 100644 --- a/Makefile +++ b/Makefile @@ -7,7 +7,7 @@ RUSTCFLAGS := --opt-level=3 RMFLAGS := # Output names -EXES := false printenv true yes cat whoami env wc +EXES := false printenv true yes cat whoami env wc echo TESTS := cat # Utils stuff diff --git a/echo/echo.rs b/echo/echo.rs new file mode 100644 index 000000000..17ed41e8c --- /dev/null +++ b/echo/echo.rs @@ -0,0 +1,229 @@ +#[link(name="echo", vers="1.0.0", author="Derek Chiang")]; + +/* + * This file is part of the uutils coreutils package. + * + * (c) Derek Chiang + * + * For the full copyright and license information, please view the LICENSE + * file that was distributed with this source code. + */ + +extern mod extra; + +use std::os; +use std::io::stderr; +use std::uint; +use extra::getopts::groups; + +static VERSION: &'static str = "1.0.0"; + +fn print_char(c: char) { + print!("{}", c); +} + +fn to_char(bytes: &[u8], base: uint) -> char { + uint::parse_bytes(bytes, base).unwrap() as u8 as char +} + +fn isxdigit(c: u8) -> bool { + match c as char { + '0' | '1' | '2' | '3' | '4' | '5' | '6' | '7' | + '8' | '9' | 'A' | 'B' | 'C' | 'D' | 'E' | 'F' => true, + _ => false + } +} + +fn isodigit(c: u8) -> bool { + match c as char { + '0' | '1' | '2' | '3' | '4' | '5' | '6' | '7' => true, + _ => false + } +} + +fn main() { + let args = os::args(); + let program = args[0].clone(); + let opts = ~[ + groups::optflag("n", "", "do not output the trailing newline"), + groups::optflag("e", "", "enable interpretation of backslash escapes"), + groups::optflag("E", "", "disable interpretation of backslash escapes (default)"), + 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("echo " + VERSION + " - display a line of text"); + println(""); + println("Usage:"); + println!(" {0:s} [SHORT-OPTION]... [STRING]...", program); + println!(" {0:s} LONG-OPTION", program); + println(""); + print(groups::usage("Echo the STRING(s) to standard output.", opts)); + print("\nIf -e is in effect, the following sequences are recognized:\n\ +\n\ + \\\\ backslash\n\ + \\a alert (BEL)\n\ + \\b backspace\n\ + \\c produce no further output\n\ + \\e escape\n\ + \\f form feed\n\ + \\n new line\n\ + \\r carriage return\n\ + \\t horizontal tab\n\ + \\v vertical tab\n\ + \\0NNN byte with octal value NNN (1 to 3 digits)\n\ + \\xHH byte with hexadecimal value HH (1 to 2 digits)\n\ +"); + return; + } + + if matches.opt_present("version") { + return println("echo version: " + VERSION); + } + + if matches.free.is_empty() { + print(""); + } else { + let string = matches.free.connect(" "); + if matches.opt_present("e") { + let mut prev_was_slash = false; + let mut iter = string.chars().enumerate(); + loop { + match iter.next() { + Some((index, c)) => { + if !prev_was_slash { + if c != '\\' { + print_char(c); + } else { + prev_was_slash = true; + } + } else { + prev_was_slash = false; + match c { + '\\' => print_char('\\'), + 'a' => print_char('\x07'), + 'b' => print_char('\x08'), + 'c' => break, + 'e' => print_char('\x1B'), + 'f' => print_char('\x0C'), + 'n' => print_char('\n'), + 'r' => print_char('\r'), + 't' => print_char('\t'), + 'v' => print_char('\x0B'), + 'x' => { + if index == string.len() - 1 { + print_char('\\'); + print_char('x'); + } else if index == string.len() - 2 { + let next_char = string[index + 1]; + if isxdigit(next_char) { + print_char(to_char([next_char as u8], 16u)); + iter.next(); + } else { + print_char('\\'); + print_char('x'); + } + } else { + let next_char = string[index + 1]; + let next_next_char = string[index + 2]; + match (isxdigit(next_char), isxdigit(next_next_char)) { + (true, true) => { + print_char(to_char([next_char as u8, next_next_char as u8], 16u)); + iter.next(); iter.next(); + } + (true, false) => { + print_char(to_char([next_char as u8], 16u)); + iter.next(); + } + _ => { + print_char('\\'); + print_char('x'); + } + }; + } + }, + '0' => { + if index == string.len() - 1 { + print_char('\\'); + print_char('0'); + } else if index == string.len() - 2 { + let next_char = string[index + 1]; + if isodigit(next_char) { + print_char(to_char([next_char as u8], 8u)); + iter.next(); + } else { + print_char('\\'); + print_char('0'); + } + } else if index == string.len() - 3 { + let next_char = string[index + 1]; + let next_next_char = string[index + 2]; + match (isodigit(next_char), isodigit(next_next_char)) { + (true, true) => { + print_char(to_char([next_char as u8, next_next_char as u8], 8u)); + iter.next(); iter.next(); + } + (true, false) => { + print_char(to_char([next_char as u8], 8u)); + iter.next(); + } + _ => { + print_char('\\'); + print_char('x'); + } + }; + } else { + let next_char = string[index + 1]; + let next_next_char = string[index + 2]; + let next_next_next_char = string[index + 3]; + match (isodigit(next_char), isodigit(next_next_char), + isodigit(next_next_next_char)) { + (true, true, true) => { + print_char(to_char([next_char as u8, next_next_char as u8, next_next_next_char as u8], 8u)); + iter.next(); iter.next(); iter.next(); + } + (true, true, false) => { + print_char(to_char([next_char as u8, next_next_char as u8], 8u)); + iter.next(); iter.next(); + } + (true, false, false) => { + print_char(to_char([next_char as u8], 8u)); + iter.next(); + } + _ => { + print_char('\\'); + print_char('x'); + } + }; + } + } + _ => { + print_char('\\'); + print_char(c); + } + } + } + } + None => break + } + } + } else { + print(string); + } + } + + if !matches.opt_present("n") { + println("") + } +}