simplify return type of `parse()` and rename ..

`with_prefix()`
This commit is contained in:
Ali MJ Al-Nasrawy 2021-06-27 15:33:14 +03:00
parent d1438cda48
commit a29a1a6dca
1 changed files with 36 additions and 52 deletions

View File

@ -48,29 +48,19 @@ impl ResourceDef {
/// Panics if path pattern is malformed. /// Panics if path pattern is malformed.
pub fn new<T: IntoPattern>(path: T) -> Self { pub fn new<T: IntoPattern>(path: T) -> Self {
if path.is_single() { if path.is_single() {
let patterns = path.patterns(); ResourceDef::from_single_pattern(&path.patterns()[0], false)
ResourceDef::with_prefix(&patterns[0], false)
} else { } else {
let set = path.patterns();
let mut data = Vec::new(); let mut data = Vec::new();
let mut re_set = Vec::new(); let mut re_set = Vec::new();
for path in set { for pattern in path.patterns() {
let (pattern, _, _) = ResourceDef::parse(&path, false, true); match ResourceDef::parse(&pattern, false, true) {
(PatternType::Dynamic(re, names), _) => {
let re = match Regex::new(&pattern) { re_set.push(re.as_str().to_owned());
Ok(re) => re, data.push((re, names));
Err(err) => panic!("Wrong path pattern: \"{}\" {}", path, err), }
}; _ => unreachable!(),
// actix creates one router per thread }
let names: Vec<_> = re
.capture_names()
.filter_map(|name| {
name.map(|name| Box::leak(Box::new(name.to_owned())).as_str())
})
.collect();
data.push((re, names));
re_set.push(pattern);
} }
ResourceDef { ResourceDef {
@ -89,7 +79,7 @@ impl ResourceDef {
/// ///
/// Panics if path regex pattern is malformed. /// Panics if path regex pattern is malformed.
pub fn prefix(path: &str) -> Self { pub fn prefix(path: &str) -> Self {
ResourceDef::with_prefix(path, true) ResourceDef::from_single_pattern(path, true)
} }
/// Parse path pattern and create new `Pattern` instance. /// Parse path pattern and create new `Pattern` instance.
@ -100,7 +90,7 @@ impl ResourceDef {
/// ///
/// Panics if path regex pattern is malformed. /// Panics if path regex pattern is malformed.
pub fn root_prefix(path: &str) -> Self { pub fn root_prefix(path: &str) -> Self {
ResourceDef::with_prefix(&insert_slash(path), true) ResourceDef::from_single_pattern(&insert_slash(path), true)
} }
/// Resource id /// Resource id
@ -113,36 +103,17 @@ impl ResourceDef {
self.id = id; self.id = id;
} }
/// Parse path pattern and create new `Pattern` instance with custom prefix /// Parse path pattern and create a new instance
fn with_prefix(path: &str, for_prefix: bool) -> Self { fn from_single_pattern(pattern: &str, for_prefix: bool) -> Self {
let path = path.to_owned(); let pattern = pattern.to_owned();
let (pattern, elements, is_dynamic) = ResourceDef::parse(&path, for_prefix, false); let (tp, elements) = ResourceDef::parse(&pattern, for_prefix, false);
let tp = if is_dynamic {
let re = match Regex::new(&pattern) {
Ok(re) => re,
Err(err) => panic!("Wrong path pattern: \"{}\" {}", path, err),
};
// actix creates one router per thread
let names = re
.capture_names()
.filter_map(|name| {
name.map(|name| Box::leak(Box::new(name.to_owned())).as_str())
})
.collect();
PatternType::Dynamic(re, names)
} else if for_prefix {
PatternType::Prefix(pattern)
} else {
PatternType::Static(pattern)
};
ResourceDef { ResourceDef {
tp, tp,
elements: Some(elements), elements: Some(elements),
id: 0, id: 0,
name: String::new(), name: String::new(),
pattern: path, pattern: pattern.to_string(),
} }
} }
@ -404,15 +375,17 @@ impl ResourceDef {
mut pattern: &str, mut pattern: &str,
mut for_prefix: bool, mut for_prefix: bool,
force_dynamic: bool, force_dynamic: bool,
) -> (String, Vec<PatternElement>, bool) { ) -> (PatternType, Vec<PatternElement>) {
if !force_dynamic && pattern.find('{').is_none() && !pattern.ends_with('*') { if !force_dynamic && pattern.find('{').is_none() && !pattern.ends_with('*') {
return ( let tp = if for_prefix {
String::from(pattern), PatternType::Prefix(String::from(pattern))
vec![PatternElement::Const(String::from(pattern))], } else {
false, PatternType::Static(String::from(pattern))
); };
return (tp, vec![PatternElement::Const(String::from(pattern))]);
} }
let pattern_orig = pattern.to_owned();
let mut elements = Vec::new(); let mut elements = Vec::new();
let mut re = format!("{}^", REGEX_FLAGS); let mut re = format!("{}^", REGEX_FLAGS);
let mut dyn_elements = 0; let mut dyn_elements = 0;
@ -452,7 +425,18 @@ impl ResourceDef {
if !for_prefix { if !for_prefix {
re.push('$'); re.push('$');
} }
(re, elements, true)
let re = match Regex::new(&re) {
Ok(re) => re,
Err(err) => panic!("Wrong path pattern: \"{}\" {}", pattern_orig, err),
};
// actix creates one router per thread
let names = re
.capture_names()
.filter_map(|name| name.map(|name| Box::leak(Box::new(name.to_owned())).as_str()))
.collect();
(PatternType::Dynamic(re, names), elements)
} }
} }