1
0
mirror of https://github.com/fafhrd91/actix-net synced 2024-12-19 12:12:37 +01:00
actix-net/router/src/resource.rs

944 lines
32 KiB
Rust
Raw Normal View History

2019-01-05 22:19:25 +01:00
use std::cmp::min;
use std::collections::HashMap;
2019-01-05 22:19:25 +01:00
use std::hash::{Hash, Hasher};
2019-12-25 12:10:01 +01:00
use regex::{escape, Regex, RegexSet};
2019-01-05 22:19:25 +01:00
use crate::path::{Path, PathItem};
use crate::{IntoPattern, Resource, ResourcePath};
2019-01-05 22:19:25 +01:00
const MAX_DYNAMIC_SEGMENTS: usize = 16;
2019-02-09 16:24:35 +01:00
/// ResourceDef describes an entry in resources table
2019-01-05 22:19:25 +01:00
///
2019-02-09 16:24:35 +01:00
/// Resource definition can contain only 16 dynamic segments
2019-01-05 22:19:25 +01:00
#[derive(Clone, Debug)]
2019-02-09 16:24:35 +01:00
pub struct ResourceDef {
2019-03-08 21:33:44 +01:00
id: u16,
2019-01-05 22:19:25 +01:00
tp: PatternType,
2019-02-09 16:24:35 +01:00
name: String,
2019-01-05 22:19:25 +01:00
pattern: String,
elements: Vec<PatternElement>,
}
#[derive(Debug, Clone, PartialEq)]
enum PatternElement {
Str(String),
Var(String),
}
#[derive(Clone, Debug)]
enum PatternType {
Static(String),
Prefix(String),
2019-12-31 07:04:35 +01:00
Dynamic(Regex, Vec<&'static str>, usize),
DynamicSet(RegexSet, Vec<(Regex, Vec<&'static str>, usize)>),
2019-01-05 22:19:25 +01:00
}
2019-02-09 16:24:35 +01:00
impl ResourceDef {
2019-01-05 22:19:25 +01:00
/// Parse path pattern and create new `Pattern` instance.
///
2019-12-25 12:10:01 +01:00
/// Panics if path pattern is malformed.
2019-12-25 16:54:20 +01:00
pub fn new<T: IntoPattern>(path: T) -> Self {
2019-12-25 12:34:21 +01:00
if path.is_single() {
let patterns = path.patterns();
ResourceDef::with_prefix(&patterns[0], false)
} else {
2019-12-25 16:54:20 +01:00
let set = path.patterns();
2019-12-25 12:34:21 +01:00
let mut data = Vec::new();
let mut re_set = Vec::new();
2019-12-25 12:10:01 +01:00
2019-12-25 12:34:21 +01:00
for path in set {
let (pattern, _, _, len) = ResourceDef::parse(&path, false);
2019-12-25 12:10:01 +01:00
2019-12-25 12:34:21 +01:00
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: Vec<_> = re
.capture_names()
2019-12-31 07:04:35 +01:00
.filter_map(|name| {
name.map(|name| Box::leak(Box::new(name.to_owned())).as_str())
})
2019-12-25 12:34:21 +01:00
.collect();
data.push((re, names, len));
re_set.push(pattern);
}
2019-12-25 12:10:01 +01:00
2019-12-25 12:34:21 +01:00
ResourceDef {
id: 0,
tp: PatternType::DynamicSet(RegexSet::new(re_set).unwrap(), data),
elements: Vec::new(),
name: String::new(),
pattern: "".to_owned(),
}
2019-12-25 12:10:01 +01:00
}
}
2019-01-05 22:19:25 +01:00
/// Parse path pattern and create new `Pattern` instance.
///
/// Use `prefix` type instead of `static`.
///
2019-12-25 12:10:01 +01:00
/// Panics if path regex pattern is malformed.
2019-01-05 22:19:25 +01:00
pub fn prefix(path: &str) -> Self {
2019-02-09 16:24:35 +01:00
ResourceDef::with_prefix(path, true)
}
/// Parse path pattern and create new `Pattern` instance.
/// Inserts `/` to begging of the pattern.
///
///
/// Use `prefix` type instead of `static`.
///
2019-12-25 12:10:01 +01:00
/// Panics if path regex pattern is malformed.
pub fn root_prefix(path: &str) -> Self {
ResourceDef::with_prefix(&insert_slash(path), true)
}
2019-03-08 21:33:44 +01:00
/// Resource id
pub fn id(&self) -> u16 {
self.id
}
/// Set resource id
pub fn set_id(&mut self, id: u16) {
self.id = id;
2019-01-05 22:19:25 +01:00
}
/// Parse path pattern and create new `Pattern` instance with custom prefix
fn with_prefix(path: &str, for_prefix: bool) -> Self {
let path = path.to_owned();
2019-02-09 16:24:35 +01:00
let (pattern, elements, is_dynamic, len) = ResourceDef::parse(&path, for_prefix);
2019-01-05 22:19:25 +01:00
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()
2019-12-31 07:04:35 +01:00
.filter_map(|name| {
name.map(|name| Box::leak(Box::new(name.to_owned())).as_str())
})
2019-01-05 22:19:25 +01:00
.collect();
PatternType::Dynamic(re, names, len)
} else if for_prefix {
PatternType::Prefix(pattern)
2019-01-05 22:19:25 +01:00
} else {
PatternType::Static(pattern)
2019-01-05 22:19:25 +01:00
};
2019-02-09 16:24:35 +01:00
ResourceDef {
2019-01-05 22:19:25 +01:00
tp,
elements,
2019-03-08 21:33:44 +01:00
id: 0,
2019-02-09 16:24:35 +01:00
name: String::new(),
pattern: path,
2019-01-05 22:19:25 +01:00
}
}
2019-03-08 21:33:44 +01:00
/// Resource pattern name
pub fn name(&self) -> &str {
&self.name
}
/// Mutable reference to a name of a resource definition.
pub fn name_mut(&mut self) -> &mut String {
&mut self.name
}
2019-01-05 22:19:25 +01:00
/// Path pattern of the resource
pub fn pattern(&self) -> &str {
&self.pattern
}
#[inline]
2019-01-05 22:19:25 +01:00
/// Check if path matchs this pattern?
pub fn is_match(&self, path: &str) -> bool {
match self.tp {
PatternType::Static(ref s) => s == path,
PatternType::Prefix(ref s) => path.starts_with(s),
2019-12-25 12:10:01 +01:00
PatternType::Dynamic(ref re, _, _) => re.is_match(path),
PatternType::DynamicSet(ref re, _) => re.is_match(path),
2019-01-05 22:19:25 +01:00
}
}
/// Is prefix path a match against this resource?
pub fn is_prefix_match(&self, path: &str) -> Option<usize> {
let plen = path.len();
let path = if path.is_empty() { "/" } else { path };
match self.tp {
PatternType::Static(ref s) => {
if s == path {
Some(plen)
} else {
None
}
}
PatternType::Dynamic(ref re, _, len) => {
if let Some(captures) = re.captures(path) {
let mut pos = 0;
let mut passed = false;
for capture in captures.iter() {
if let Some(ref m) = capture {
if !passed {
passed = true;
continue;
}
pos = m.end();
}
}
Some(pos + len)
} else {
None
}
}
PatternType::Prefix(ref s) => {
let len = if path == s {
s.len()
} else if path.starts_with(s)
&& (s.ends_with('/') || path.split_at(s.len()).1.starts_with('/'))
{
if s.ends_with('/') {
s.len() - 1
} else {
s.len()
}
} else {
return None;
};
Some(min(plen, len))
}
2019-12-25 12:10:01 +01:00
PatternType::DynamicSet(ref re, ref params) => {
if let Some(idx) = re.matches(path).into_iter().next() {
let (ref pattern, _, len) = params[idx];
if let Some(captures) = pattern.captures(path) {
let mut pos = 0;
let mut passed = false;
for capture in captures.iter() {
if let Some(ref m) = capture {
if !passed {
passed = true;
continue;
}
pos = m.end();
}
}
Some(pos + len)
} else {
None
}
} else {
None
}
}
}
}
2019-01-05 22:19:25 +01:00
/// Is the given path and parameters a match against this pattern?
pub fn match_path<T: ResourcePath>(&self, path: &mut Path<T>) -> bool {
2019-01-05 22:19:25 +01:00
match self.tp {
PatternType::Static(ref s) => {
if s == path.path() {
path.skip(path.len() as u16);
true
} else {
false
}
}
2019-12-25 12:10:01 +01:00
PatternType::Prefix(ref s) => {
let rpath = path.path();
let len = if s == rpath {
s.len()
} else if rpath.starts_with(s)
&& (s.ends_with('/') || rpath.split_at(s.len()).1.starts_with('/'))
{
if s.ends_with('/') {
s.len() - 1
} else {
s.len()
}
} else {
return false;
};
let rpath_len = rpath.len();
path.skip(min(rpath_len, len) as u16);
true
}
2019-01-05 22:19:25 +01:00
PatternType::Dynamic(ref re, ref names, len) => {
let mut idx = 0;
let mut pos = 0;
let mut segments: [PathItem; MAX_DYNAMIC_SEGMENTS] =
[PathItem::Static(""); MAX_DYNAMIC_SEGMENTS];
if let Some(captures) = re.captures(path.path()) {
for (no, name) in names.iter().enumerate() {
if let Some(m) = captures.name(&name) {
2019-01-05 22:19:25 +01:00
idx += 1;
pos = m.end();
segments[no] = PathItem::Segment(m.start() as u16, m.end() as u16);
} else {
log::error!(
"Dynamic path match but not all segments found: {}",
name
);
return false;
2019-01-05 22:19:25 +01:00
}
}
} else {
return false;
}
for idx in 0..idx {
path.add(names[idx].clone(), segments[idx]);
}
path.skip((pos + len) as u16);
true
}
2019-12-25 12:10:01 +01:00
PatternType::DynamicSet(ref re, ref params) => {
if let Some(idx) = re.matches(path.path()).into_iter().next() {
let (ref pattern, ref names, len) = params[idx];
let mut idx = 0;
let mut pos = 0;
let mut segments: [PathItem; MAX_DYNAMIC_SEGMENTS] =
[PathItem::Static(""); MAX_DYNAMIC_SEGMENTS];
if let Some(captures) = pattern.captures(path.path()) {
for (no, name) in names.iter().enumerate() {
if let Some(m) = captures.name(&name) {
idx += 1;
pos = m.end();
segments[no] =
PathItem::Segment(m.start() as u16, m.end() as u16);
} else {
log::error!(
"Dynamic path match but not all segments found: {}",
name
);
return false;
}
}
2019-01-05 22:19:25 +01:00
} else {
2019-12-25 12:10:01 +01:00
return false;
}
for idx in 0..idx {
path.add(names[idx].clone(), segments[idx]);
2019-01-05 22:19:25 +01:00
}
2019-12-25 12:10:01 +01:00
path.skip((pos + len) as u16);
true
2019-01-05 22:19:25 +01:00
} else {
2019-12-25 12:10:01 +01:00
false
}
2019-01-05 22:19:25 +01:00
}
}
}
2019-05-15 19:21:29 +02:00
/// Is the given path and parameters a match against this pattern?
pub fn match_path_checked<R, T, F, U>(
&self,
res: &mut R,
check: &F,
user_data: &Option<U>,
) -> bool
where
T: ResourcePath,
R: Resource<T>,
F: Fn(&R, &Option<U>) -> bool,
{
match self.tp {
PatternType::Static(ref s) => {
if s == res.resource_path().path() && check(res, user_data) {
let path = res.resource_path();
path.skip(path.len() as u16);
true
} else {
false
}
}
2019-12-25 12:10:01 +01:00
PatternType::Prefix(ref s) => {
let len = {
let rpath = res.resource_path().path();
if s == rpath {
s.len()
} else if rpath.starts_with(s)
&& (s.ends_with('/') || rpath.split_at(s.len()).1.starts_with('/'))
{
if s.ends_with('/') {
s.len() - 1
} else {
s.len()
}
} else {
return false;
}
};
if !check(res, user_data) {
return false;
}
let path = res.resource_path();
path.skip(min(path.path().len(), len) as u16);
true
}
2019-05-15 19:21:29 +02:00
PatternType::Dynamic(ref re, ref names, len) => {
let mut idx = 0;
let mut pos = 0;
let mut segments: [PathItem; MAX_DYNAMIC_SEGMENTS] =
[PathItem::Static(""); MAX_DYNAMIC_SEGMENTS];
if let Some(captures) = re.captures(res.resource_path().path()) {
for (no, name) in names.iter().enumerate() {
if let Some(m) = captures.name(&name) {
idx += 1;
pos = m.end();
segments[no] = PathItem::Segment(m.start() as u16, m.end() as u16);
} else {
log::error!(
"Dynamic path match but not all segments found: {}",
name
);
return false;
2019-05-15 19:21:29 +02:00
}
}
} else {
return false;
}
if !check(res, user_data) {
return false;
}
let path = res.resource_path();
for idx in 0..idx {
path.add(names[idx].clone(), segments[idx]);
}
path.skip((pos + len) as u16);
true
}
2019-12-25 12:10:01 +01:00
PatternType::DynamicSet(ref re, ref params) => {
let path = res.resource_path().path();
if let Some(idx) = re.matches(path).into_iter().next() {
let (ref pattern, ref names, len) = params[idx];
let mut idx = 0;
let mut pos = 0;
let mut segments: [PathItem; MAX_DYNAMIC_SEGMENTS] =
[PathItem::Static(""); MAX_DYNAMIC_SEGMENTS];
if let Some(captures) = pattern.captures(path) {
for (no, name) in names.iter().enumerate() {
if let Some(m) = captures.name(&name) {
idx += 1;
pos = m.end();
segments[no] =
PathItem::Segment(m.start() as u16, m.end() as u16);
} else {
log::error!(
"Dynamic path match but not all segments found: {}",
name
);
return false;
}
2019-05-15 19:21:29 +02:00
}
} else {
return false;
}
2019-12-25 12:10:01 +01:00
if !check(res, user_data) {
return false;
}
let path = res.resource_path();
for idx in 0..idx {
path.add(names[idx].clone(), segments[idx]);
}
path.skip((pos + len) as u16);
true
} else {
false
2019-05-15 19:21:29 +02:00
}
}
}
}
/// Build resource path from elements. Returns `true` on success.
pub fn resource_path<U, I>(&self, path: &mut String, elements: &mut U) -> bool
where
U: Iterator<Item = I>,
I: AsRef<str>,
{
match self.tp {
PatternType::Prefix(ref p) => path.push_str(p),
PatternType::Static(ref p) => path.push_str(p),
PatternType::Dynamic(..) => {
for el in &self.elements {
match *el {
PatternElement::Str(ref s) => path.push_str(s),
PatternElement::Var(_) => {
if let Some(val) = elements.next() {
path.push_str(val.as_ref())
} else {
return false;
}
}
}
}
}
2019-12-25 12:10:01 +01:00
PatternType::DynamicSet(..) => {
return false;
}
}
true
}
2019-01-05 22:19:25 +01:00
/// Build resource path from elements. Returns `true` on success.
pub fn resource_path_named<K, V, S>(
&self,
path: &mut String,
elements: &HashMap<K, V, S>,
) -> bool
where
K: std::borrow::Borrow<str> + Eq + Hash,
V: AsRef<str>,
S: std::hash::BuildHasher,
{
match self.tp {
PatternType::Prefix(ref p) => path.push_str(p),
PatternType::Static(ref p) => path.push_str(p),
PatternType::Dynamic(..) => {
for el in &self.elements {
match *el {
PatternElement::Str(ref s) => path.push_str(s),
PatternElement::Var(ref name) => {
if let Some(val) = elements.get(name) {
path.push_str(val.as_ref())
} else {
return false;
}
}
}
}
}
PatternType::DynamicSet(..) => {
return false;
}
}
true
}
2019-04-23 06:19:22 +02:00
fn parse_param(pattern: &str) -> (PatternElement, String, &str, bool) {
2019-01-05 22:19:25 +01:00
const DEFAULT_PATTERN: &str = "[^/]+";
2019-04-23 06:19:22 +02:00
const DEFAULT_PATTERN_TAIL: &str = ".*";
2019-01-05 22:19:25 +01:00
let mut params_nesting = 0usize;
let close_idx = pattern
.find(|c| match c {
'{' => {
params_nesting += 1;
false
}
'}' => {
params_nesting -= 1;
params_nesting == 0
}
_ => false,
})
.expect("malformed dynamic segment");
2019-04-23 06:19:22 +02:00
let (mut param, mut rem) = pattern.split_at(close_idx + 1);
2019-01-05 22:19:25 +01:00
param = &param[1..param.len() - 1]; // Remove outer brackets
2019-04-23 06:19:22 +02:00
let tail = rem == "*";
2019-01-05 22:19:25 +01:00
let (name, pattern) = match param.find(':') {
Some(idx) => {
2019-04-23 06:19:22 +02:00
if tail {
panic!("Custom regex is not supported for remainder match");
}
2019-01-05 22:19:25 +01:00
let (name, pattern) = param.split_at(idx);
(name, &pattern[1..])
}
2019-04-23 06:19:22 +02:00
None => (
param,
if tail {
rem = &rem[1..];
DEFAULT_PATTERN_TAIL
} else {
DEFAULT_PATTERN
},
),
2019-01-05 22:19:25 +01:00
};
(
PatternElement::Var(name.to_string()),
format!(r"(?P<{}>{})", &name, &pattern),
rem,
2019-04-23 06:19:22 +02:00
tail,
2019-01-05 22:19:25 +01:00
)
}
fn parse(
mut pattern: &str,
2019-04-23 06:19:22 +02:00
mut for_prefix: bool,
2019-01-05 22:19:25 +01:00
) -> (String, Vec<PatternElement>, bool, usize) {
if pattern.find('{').is_none() {
2019-04-23 06:19:22 +02:00
return if pattern.ends_with('*') {
let path = &pattern[..pattern.len() - 1];
let re = String::from("^") + path + "(.*)";
(re, vec![PatternElement::Str(String::from(path))], true, 0)
} else {
(
String::from(pattern),
vec![PatternElement::Str(String::from(pattern))],
false,
pattern.chars().count(),
)
};
}
2019-01-05 22:19:25 +01:00
let mut elems = Vec::new();
let mut re = String::from("^");
let mut dyn_elems = 0;
while let Some(idx) = pattern.find('{') {
let (prefix, rem) = pattern.split_at(idx);
elems.push(PatternElement::Str(String::from(prefix)));
re.push_str(&escape(prefix));
2019-04-23 06:19:22 +02:00
let (param_pattern, re_part, rem, tail) = Self::parse_param(rem);
if tail {
for_prefix = true;
}
2019-01-05 22:19:25 +01:00
elems.push(param_pattern);
re.push_str(&re_part);
pattern = rem;
dyn_elems += 1;
}
elems.push(PatternElement::Str(String::from(pattern)));
re.push_str(&escape(pattern));
if dyn_elems > MAX_DYNAMIC_SEGMENTS {
panic!(
"Only {} dynanic segments are allowed, provided: {}",
MAX_DYNAMIC_SEGMENTS, dyn_elems
);
}
if !for_prefix {
re.push_str("$");
}
(re, elems, true, pattern.chars().count())
}
}
2019-02-09 16:24:35 +01:00
impl Eq for ResourceDef {}
impl PartialEq for ResourceDef {
fn eq(&self, other: &ResourceDef) -> bool {
2019-01-05 22:19:25 +01:00
self.pattern == other.pattern
}
}
2019-02-09 16:24:35 +01:00
impl Hash for ResourceDef {
2019-01-05 22:19:25 +01:00
fn hash<H: Hasher>(&self, state: &mut H) {
self.pattern.hash(state);
}
}
2019-02-09 16:24:35 +01:00
impl<'a> From<&'a str> for ResourceDef {
fn from(path: &'a str) -> ResourceDef {
ResourceDef::new(path)
}
}
impl From<String> for ResourceDef {
fn from(path: String) -> ResourceDef {
2019-12-25 12:34:21 +01:00
ResourceDef::new(path)
2019-02-09 16:24:35 +01:00
}
}
pub(crate) fn insert_slash(path: &str) -> String {
let mut path = path.to_owned();
if !path.is_empty() && !path.starts_with('/') {
path.insert(0, '/');
};
path
}
2019-01-05 22:19:25 +01:00
#[cfg(test)]
mod tests {
use super::*;
2019-12-05 11:40:24 +01:00
use http::Uri;
use std::convert::TryFrom;
2019-01-05 22:19:25 +01:00
#[test]
fn test_parse_static() {
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/"));
assert!(!re.is_match("/a"));
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/name");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/name"));
assert!(!re.is_match("/name1"));
assert!(!re.is_match("/name/"));
assert!(!re.is_match("/name~"));
assert_eq!(re.is_prefix_match("/name"), Some(5));
assert_eq!(re.is_prefix_match("/name1"), None);
assert_eq!(re.is_prefix_match("/name/"), None);
assert_eq!(re.is_prefix_match("/name~"), None);
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/name/");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/name/"));
assert!(!re.is_match("/name"));
assert!(!re.is_match("/name/gs"));
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/user/profile");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/user/profile"));
assert!(!re.is_match("/user/profile/profile"));
}
#[test]
fn test_parse_param() {
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/user/{id}");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(!re.is_match("/user/2345/"));
assert!(!re.is_match("/user/2345/sdg"));
let mut path = Path::new("/user/profile");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "profile");
let mut path = Path::new("/user/1245125");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "1245125");
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/v{version}/resource/{id}");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/v1/resource/320120"));
assert!(!re.is_match("/v/resource/1"));
assert!(!re.is_match("/resource"));
let mut path = Path::new("/v151/resource/adahg32");
assert!(re.match_path(&mut path));
assert_eq!(path.get("version").unwrap(), "151");
assert_eq!(path.get("id").unwrap(), "adahg32");
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/{id:[[:digit:]]{6}}");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/012345"));
assert!(!re.is_match("/012"));
assert!(!re.is_match("/01234567"));
assert!(!re.is_match("/XXXXXX"));
let mut path = Path::new("/012345");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "012345");
}
2019-12-25 12:10:01 +01:00
#[test]
fn test_dynamic_set() {
2019-12-25 16:54:20 +01:00
let re = ResourceDef::new(vec![
2019-12-25 12:34:21 +01:00
"/user/{id}",
"/v{version}/resource/{id}",
"/{id:[[:digit:]]{6}}",
]);
2019-12-25 12:10:01 +01:00
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(!re.is_match("/user/2345/"));
assert!(!re.is_match("/user/2345/sdg"));
let mut path = Path::new("/user/profile");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "profile");
let mut path = Path::new("/user/1245125");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "1245125");
assert!(re.is_match("/v1/resource/320120"));
assert!(!re.is_match("/v/resource/1"));
assert!(!re.is_match("/resource"));
let mut path = Path::new("/v151/resource/adahg32");
assert!(re.match_path(&mut path));
assert_eq!(path.get("version").unwrap(), "151");
assert_eq!(path.get("id").unwrap(), "adahg32");
assert!(re.is_match("/012345"));
assert!(!re.is_match("/012"));
assert!(!re.is_match("/01234567"));
assert!(!re.is_match("/XXXXXX"));
let mut path = Path::new("/012345");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "012345");
2019-12-25 16:54:20 +01:00
let re = ResourceDef::new([
"/user/{id}",
"/v{version}/resource/{id}",
"/{id:[[:digit:]]{6}}",
]);
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(!re.is_match("/user/2345/"));
assert!(!re.is_match("/user/2345/sdg"));
let re = ResourceDef::new([
"/user/{id}".to_string(),
"/v{version}/resource/{id}".to_string(),
"/{id:[[:digit:]]{6}}".to_string(),
]);
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(!re.is_match("/user/2345/"));
assert!(!re.is_match("/user/2345/sdg"));
2019-12-25 12:10:01 +01:00
}
2019-04-23 06:19:22 +02:00
#[test]
fn test_parse_tail() {
let re = ResourceDef::new("/user/-{id}*");
let mut path = Path::new("/user/-profile");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "profile");
let mut path = Path::new("/user/-2345");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "2345");
let mut path = Path::new("/user/-2345/");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "2345/");
let mut path = Path::new("/user/-2345/sdg");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "2345/sdg");
}
#[test]
fn test_static_tail() {
let re = ResourceDef::new("/user*");
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(re.is_match("/user/2345/"));
assert!(re.is_match("/user/2345/sdg"));
let re = ResourceDef::new("/user/*");
assert!(re.is_match("/user/profile"));
assert!(re.is_match("/user/2345"));
assert!(re.is_match("/user/2345/"));
assert!(re.is_match("/user/2345/sdg"));
}
2019-02-01 22:25:56 +01:00
#[test]
fn test_parse_urlencoded_param() {
2019-02-09 16:24:35 +01:00
let re = ResourceDef::new("/user/{id}/test");
2019-02-01 22:25:56 +01:00
let mut path = Path::new("/user/2345/test");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "2345");
let mut path = Path::new("/user/qwe%25/test");
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "qwe%25");
let uri = Uri::try_from("/user/qwe%25/test").unwrap();
let mut path = Path::new(uri);
assert!(re.match_path(&mut path));
assert_eq!(path.get("id").unwrap(), "qwe%25");
}
2019-01-05 22:19:25 +01:00
#[test]
fn test_resource_prefix() {
2019-02-09 16:24:35 +01:00
let re = ResourceDef::prefix("/name");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/name"));
assert!(re.is_match("/name/"));
assert!(re.is_match("/name/test/test"));
assert!(re.is_match("/name1"));
assert!(re.is_match("/name~"));
assert_eq!(re.is_prefix_match("/name"), Some(5));
assert_eq!(re.is_prefix_match("/name/"), Some(5));
assert_eq!(re.is_prefix_match("/name/test/test"), Some(5));
assert_eq!(re.is_prefix_match("/name1"), None);
assert_eq!(re.is_prefix_match("/name~"), None);
2019-02-09 16:24:35 +01:00
let re = ResourceDef::prefix("/name/");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/name/"));
assert!(re.is_match("/name/gs"));
assert!(!re.is_match("/name"));
let re = ResourceDef::root_prefix("name/");
assert!(re.is_match("/name/"));
assert!(re.is_match("/name/gs"));
assert!(!re.is_match("/name"));
2019-01-05 22:19:25 +01:00
}
#[test]
fn test_reousrce_prefix_dynamic() {
2019-02-09 16:24:35 +01:00
let re = ResourceDef::prefix("/{name}/");
2019-01-05 22:19:25 +01:00
assert!(re.is_match("/name/"));
assert!(re.is_match("/name/gs"));
assert!(!re.is_match("/name"));
assert_eq!(re.is_prefix_match("/name/"), Some(6));
assert_eq!(re.is_prefix_match("/name/gs"), Some(6));
assert_eq!(re.is_prefix_match("/name"), None);
2019-01-05 22:19:25 +01:00
let mut path = Path::new("/test2/");
assert!(re.match_path(&mut path));
assert_eq!(&path["name"], "test2");
assert_eq!(&path[0], "test2");
let mut path = Path::new("/test2/subpath1/subpath2/index.html");
assert!(re.match_path(&mut path));
assert_eq!(&path["name"], "test2");
assert_eq!(&path[0], "test2");
}
#[test]
fn test_resource_path() {
let mut s = String::new();
let resource = ResourceDef::new("/user/{item1}/test");
assert!(resource.resource_path(&mut s, &mut (&["user1"]).into_iter()));
assert_eq!(s, "/user/user1/test");
let mut s = String::new();
let resource = ResourceDef::new("/user/{item1}/{item2}/test");
assert!(resource.resource_path(&mut s, &mut (&["item", "item2"]).into_iter()));
assert_eq!(s, "/user/item/item2/test");
let mut s = String::new();
let resource = ResourceDef::new("/user/{item1}/{item2}");
assert!(resource.resource_path(&mut s, &mut (&["item", "item2"]).into_iter()));
assert_eq!(s, "/user/item/item2");
let mut s = String::new();
let resource = ResourceDef::new("/user/{item1}/{item2}/");
assert!(resource.resource_path(&mut s, &mut (&["item", "item2"]).into_iter()));
assert_eq!(s, "/user/item/item2/");
let mut s = String::new();
assert!(!resource.resource_path(&mut s, &mut (&["item"]).into_iter()));
let mut s = String::new();
assert!(resource.resource_path(&mut s, &mut (&["item", "item2"]).into_iter()));
assert_eq!(s, "/user/item/item2/");
assert!(!resource.resource_path(&mut s, &mut (&["item"]).into_iter()));
let mut s = String::new();
assert!(resource.resource_path(&mut s, &mut vec!["item", "item2"].into_iter()));
assert_eq!(s, "/user/item/item2/");
let mut map = HashMap::new();
map.insert("item1", "item");
let mut s = String::new();
assert!(!resource.resource_path_named(&mut s, &map));
let mut s = String::new();
map.insert("item2", "item2");
assert!(resource.resource_path_named(&mut s, &map));
assert_eq!(s, "/user/item/item2/");
}
2019-01-05 22:19:25 +01:00
}