allow listing more than 2 filters
This commit is contained in:
parent
a857cfcc1e
commit
1b1b731227
1 changed files with 32 additions and 33 deletions
65
src/main.rs
65
src/main.rs
|
|
@ -10,7 +10,6 @@
|
||||||
extern crate clap;
|
extern crate clap;
|
||||||
extern crate glob;
|
extern crate glob;
|
||||||
extern crate oxipng;
|
extern crate oxipng;
|
||||||
extern crate regex;
|
|
||||||
|
|
||||||
use clap::{App, Arg, ArgMatches};
|
use clap::{App, Arg, ArgMatches};
|
||||||
use glob::glob;
|
use glob::glob;
|
||||||
|
|
@ -18,7 +17,6 @@ use oxipng::AlphaOptim;
|
||||||
use oxipng::Deflaters;
|
use oxipng::Deflaters;
|
||||||
use oxipng::Headers;
|
use oxipng::Headers;
|
||||||
use oxipng::{Options, PngError};
|
use oxipng::{Options, PngError};
|
||||||
use regex::Regex;
|
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
use std::fs::DirBuilder;
|
use std::fs::DirBuilder;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
|
|
@ -441,40 +439,41 @@ fn parse_numeric_range_opts(
|
||||||
min_value: u8,
|
min_value: u8,
|
||||||
max_value: u8,
|
max_value: u8,
|
||||||
) -> Result<HashSet<u8>, String> {
|
) -> Result<HashSet<u8>, String> {
|
||||||
let one_item = Regex::new(format!(r"^[{}-{}]$", min_value, max_value).as_ref()).unwrap();
|
let error_message = "Not a valid input";
|
||||||
let multiple_items = Regex::new(
|
|
||||||
format!(
|
|
||||||
r"^([{}-{}])(,|-)([{}-{}])$",
|
|
||||||
min_value, max_value, min_value, max_value
|
|
||||||
).as_ref(),
|
|
||||||
).unwrap();
|
|
||||||
let mut items = HashSet::new();
|
let mut items = HashSet::new();
|
||||||
|
|
||||||
if one_item.is_match(input) {
|
// one value
|
||||||
items.insert(input.parse::<u8>().unwrap());
|
if let Some(one_value) = input.parse::<u8>().ok() {
|
||||||
return Ok(items);
|
if (min_value <= one_value) && (one_value <= max_value) {
|
||||||
}
|
items.insert(one_value);
|
||||||
|
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());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
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);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue