use std::str::FromStr;
use regex::Regex;
use crate::variant::Variant;
#[derive(Debug, Clone, PartialEq)]
pub enum NumericRange {
Index(u32),
Range(u32, u32),
MultipleRanges(Vec<NumericRange>),
}
#[test]
fn valid_numeric_ranges() {
let valid_ranges = vec![
("0", NumericRange::Index(0), "0"),
("0000", NumericRange::Index(0), "0"),
("1", NumericRange::Index(1), "1"),
("0123456789", NumericRange::Index(123456789), "123456789"),
("4294967295", NumericRange::Index(4294967295), "4294967295"),
("1:2", NumericRange::Range(1, 2), "1:2"),
("2:3", NumericRange::Range(2, 3), "2:3"),
("0:1,0:2,0:3,0:4,0:5", NumericRange::MultipleRanges(vec![
NumericRange::Range(0, 1),
NumericRange::Range(0, 2),
NumericRange::Range(0, 3),
NumericRange::Range(0, 4),
NumericRange::Range(0, 5)
]), "0:1,0:2,0:3,0:4,0:5"),
("0:1,2,3,0:4,5,6,7,8,0:9", NumericRange::MultipleRanges(vec![
NumericRange::Range(0, 1),
NumericRange::Index(2),
NumericRange::Index(3),
NumericRange::Range(0, 4),
NumericRange::Index(5),
NumericRange::Index(6),
NumericRange::Index(7),
NumericRange::Index(8),
NumericRange::Range(0, 9)
]), "0:1,2,3,0:4,5,6,7,8,0:9")
];
for vr in valid_ranges {
let range = vr.0.parse::<NumericRange>();
if range.is_err() {
println!("Range {} is in error when it should be ok", vr.0);
}
assert!(range.is_ok());
assert_eq!(range.unwrap(), vr.1);
assert_eq!(vr.2, &vr.1.as_string());
}
}
#[test]
fn invalid_numeric_ranges() {
let invalid_ranges = vec![
"", " ", " 1", "1 ", ":", ":1", "1:1", "2:1", "1:", "1:1:2", ",", ":,", ",:",
",1", "1,", "1,2,", "1,,2", "01234567890", "0,1,2,3,4,5,6,7,8,9,10",
"4294967296", "0:4294967296", "4294967296:0"
];
for vr in invalid_ranges {
println!("vr = {}", vr);
let range = vr.parse::<NumericRange>();
if range.is_ok() {
println!("Range {} is ok when it should be in error", vr);
}
assert!(range.is_err());
}
}
const MAX_INDICES: usize = 10;
impl FromStr for NumericRange {
type Err = ();
fn from_str(s: &str) -> Result<Self, Self::Err> {
let parts: Vec<_> = s.split(',').collect();
match parts.len() {
1 => Self::parse_range(&parts[0]),
2..=MAX_INDICES => {
let mut ranges = Vec::with_capacity(parts.len());
for p in &parts {
if let Ok(range) = Self::parse_range(&p) {
ranges.push(range);
} else {
return Err(());
}
}
Ok(NumericRange::MultipleRanges(ranges))
}
_ => Err(()),
}
}
}
impl NumericRange {
pub fn new<T>(s: T) -> Result<Self, ()> where T: Into<String> {
Self::from_str(s.into().as_ref())
}
pub fn as_string(&self) -> String {
match *self {
NumericRange::Index(idx) => {
format!("{}", idx)
}
NumericRange::Range(min, max) => {
format!("{}:{}", min, max)
}
NumericRange::MultipleRanges(ref ranges) => {
let ranges: Vec<String> = ranges.iter().map(|r| r.as_string()).collect();
ranges.join(",")
}
}
}
fn parse_range(s: &str) -> Result<NumericRange, ()> {
if s.is_empty() {
Err(())
} else {
lazy_static! {
static ref RE: Regex = Regex::new("^(?P<min>[0-9]{1,10})(:(?P<max>[0-9]{1,10}))?$").unwrap();
}
if let Some(captures) = RE.captures(s) {
let min = captures.name("min");
let max = captures.name("max");
if min.is_none() && max.is_none() {
Err(())
} else if min.is_some() && max.is_none() {
if let Ok(min) = min.unwrap().as_str().parse::<u32>() {
Ok(NumericRange::Index(min))
} else {
Err(())
}
} else if let Ok(min) = min.unwrap().as_str().parse::<u32>() {
if let Ok(max) = max.unwrap().as_str().parse::<u32>() {
if min >= max {
Err(())
} else {
Ok(NumericRange::Range(min, max))
}
} else {
Err(())
}
} else {
Err(())
}
} else {
Err(())
}
}
}
pub fn is_valid(&self) -> bool {
match *self {
NumericRange::Index(_) => true,
NumericRange::Range(min, max) => { min < max }
NumericRange::MultipleRanges(ref ranges) => {
let mut valid = true;
for r in ranges {
match *r {
NumericRange::MultipleRanges(_) => {
valid = false;
break;
}
_ => {
if !r.is_valid() {
valid = false;
break;
}
}
}
}
valid
}
}
}
pub fn is_valid_for_array(&self, array: &Variant) -> bool {
match *array {
Variant::Array(_) => {
unimplemented!();
}
Variant::MultiDimensionArray(_) => {
unimplemented!();
}
_ => {
panic!("The array is not an array")
}
}
}
}