coreutils/src/nl/helper.rs

122 lines
3.9 KiB
Rust
Raw Normal View History

2014-06-14 14:43:39 +00:00
extern crate getopts;
extern crate regex;
// parse_style parses a style string into a NumberingStyle.
fn parse_style(chars: &[char]) -> Result<::NumberingStyle, String> {
match chars {
2014-11-19 20:50:37 +00:00
['a'] => { Ok(::NumberingStyle::NumberForAll) },
['t'] => { Ok(::NumberingStyle::NumberForNonEmpty) },
['n'] => { Ok(::NumberingStyle::NumberForNone) },
2014-09-10 06:26:42 +00:00
['p', rest..] => {
let s: String = rest.iter().map(|c| *c).collect();
match regex::Regex::new(&s) {
2014-11-19 20:50:37 +00:00
Ok(re) => Ok(::NumberingStyle::NumberForRegularExpression(re)),
Err(_) => Err(String::from("Illegal regular expression")),
2014-06-14 14:43:39 +00:00
}
}
_ => {
Err(String::from("Illegal style encountered"))
2014-06-14 14:43:39 +00:00
},
}
}
// parse_options loads the options into the settings, returning an array of
// error messages.
pub fn parse_options(settings: &mut ::Settings, opts: &getopts::Matches) -> Vec<String> {
// This vector holds error messages encountered.
let mut errs: Vec<String> = vec![];
settings.renumber = !opts.opt_present("p");
match opts.opt_str("s") {
None => {},
Some(val) => { settings.number_separator = val; }
}
match opts.opt_str("n") {
None => {},
Some(val) => match val.as_ref() {
2014-11-19 20:50:37 +00:00
"ln" => { settings.number_format = ::NumberFormat::Left; },
"rn" => { settings.number_format = ::NumberFormat::Right; },
"rz" => { settings.number_format = ::NumberFormat::RightZero; },
_ => { errs.push(String::from("Illegal value for -n")); },
2014-06-14 14:43:39 +00:00
}
}
match opts.opt_str("b") {
None => {},
Some(val) => {
let chars: Vec<char> = val.chars().collect();
match parse_style(&chars) {
2014-06-14 14:43:39 +00:00
Ok(s) => { settings.body_numbering = s; }
Err(message) => { errs.push(message); }
}
}
}
match opts.opt_str("f") {
None => {},
Some(val) => {
let chars: Vec<char> = val.chars().collect();
match parse_style(&chars) {
2014-06-14 14:43:39 +00:00
Ok(s) => { settings.footer_numbering = s; }
Err(message) => { errs.push(message); }
}
}
}
match opts.opt_str("h") {
None => {},
Some(val) => {
let chars: Vec<char> = val.chars().collect();
match parse_style(&chars) {
2014-06-14 14:43:39 +00:00
Ok(s) => { settings.header_numbering = s; }
Err(message) => { errs.push(message); }
}
}
}
match opts.opt_str("i") {
None => {}
Some(val) => {
let conv: Option<u64> = val.parse().ok();
2014-06-14 14:43:39 +00:00
match conv {
None => {
errs.push(String::from("Illegal value for -i"));
2014-06-14 14:43:39 +00:00
}
Some(num) => { settings.line_increment = num }
}
}
}
match opts.opt_str("w") {
None => {}
Some(val) => {
let conv: Option<usize> = val.parse().ok();
2014-06-14 14:43:39 +00:00
match conv {
None => {
errs.push(String::from("Illegal value for -w"));
2014-06-14 14:43:39 +00:00
}
Some(num) => { settings.number_width = num }
}
}
}
match opts.opt_str("v") {
None => {}
Some(val) => {
let conv: Option<u64> = val.parse().ok();
2014-06-14 14:43:39 +00:00
match conv {
None => {
errs.push(String::from("Illegal value for -v"));
2014-06-14 14:43:39 +00:00
}
Some(num) => { settings.starting_line_number = num }
}
}
}
match opts.opt_str("l") {
None => {}
Some(val) => {
let conv: Option<u64> = val.parse().ok();
2014-06-14 14:43:39 +00:00
match conv {
None => {
errs.push(String::from("Illegal value for -l"));
2014-06-14 14:43:39 +00:00
}
Some(num) => { settings.join_blank_lines = num }
}
}
}
errs
}