allow listing more than 2 filters

This commit is contained in:
anonymous 2018-06-01 21:15:50 +03:00
parent a857cfcc1e
commit 1b1b731227

View file

@ -10,7 +10,6 @@
extern crate clap;
extern crate glob;
extern crate oxipng;
extern crate regex;
use clap::{App, Arg, ArgMatches};
use glob::glob;
@ -18,7 +17,6 @@ use oxipng::AlphaOptim;
use oxipng::Deflaters;
use oxipng::Headers;
use oxipng::{Options, PngError};
use regex::Regex;
use std::collections::HashSet;
use std::fs::DirBuilder;
use std::path::PathBuf;
@ -441,40 +439,41 @@ fn parse_numeric_range_opts(
min_value: u8,
max_value: u8,
) -> Result<HashSet<u8>, String> {
let one_item = Regex::new(format!(r"^[{}-{}]$", min_value, max_value).as_ref()).unwrap();
let multiple_items = Regex::new(
format!(
r"^([{}-{}])(,|-)([{}-{}])$",
min_value, max_value, min_value, max_value
).as_ref(),
).unwrap();
let error_message = "Not a valid input";
let mut items = HashSet::new();
if one_item.is_match(input) {
items.insert(input.parse::<u8>().unwrap());
return Ok(items);
}
if let Some(captures) = multiple_items.captures(input) {
let first = captures[1].parse::<u8>().unwrap();
let second = captures[3].parse::<u8>().unwrap();
if first >= second {
return Err("Not a valid input".to_owned());
// one value
if let Some(one_value) = input.parse::<u8>().ok() {
if (min_value <= one_value) && (one_value <= max_value) {
items.insert(one_value);
return Ok(items);
}
match &captures[2] {
"," => {
items.insert(first);
items.insert(second);
}
"-" => for i in first..second + 1 {
items.insert(i);
},
_ => unreachable!(),
};
return Ok(items);
}
Err("Not a valid input".to_owned())
// a range ("A-B")
let range_values = input.split('-').collect::<Vec<&str>>();
if range_values.len() == 2 {
let first_opt = range_values[0].parse::<u8>().ok();
let second_opt = range_values[1].parse::<u8>().ok();
if let (Some(first), Some(second)) = (first_opt, second_opt) {
if min_value <= first && first < second && second <= max_value {
for i in first..second + 1 {
items.insert(i);
}
return Ok(items);
}
}
}
// a list ("A,B[,…]")
for value in input.split(',') {
if let Some(value_int) = value.parse::<u8>().ok() {
if (min_value <= value_int) && (value_int <= max_value) && !items.contains(&value_int) {
items.insert(value_int);
continue;
}
}
return Err(error_message.to_owned());
}
return Ok(items);
}