1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use crate::data_map::{DataMap, SharedDataMap};
use crate::regex_generator::generate_exact_match_regex;
use crate::Error;
use regex::Regex;
use std::fmt::{self, Debug, Formatter};
use std::sync::Arc;

pub(crate) struct ScopedDataMap {
    pub(crate) path: String,
    pub(crate) regex: Regex,
    // Make it an option so that when a router is used to scope in another router,
    // It can be extracted out by 'opt.take()' without taking the whole router's ownership.
    pub(crate) data_map: Option<Arc<DataMap>>,
}

impl ScopedDataMap {
    pub fn new<P: Into<String>>(path: P, data_map: Arc<DataMap>) -> crate::Result<ScopedDataMap> {
        let path = path.into();
        let (re, _) = generate_exact_match_regex(path.as_str()).map_err(|e| {
            Error::new(format!(
                "Could not create an exact match regex for the scoped data map path: {}",
                e
            ))
        })?;

        Ok(ScopedDataMap {
            path,
            regex: re,
            data_map: Some(data_map),
        })
    }

    pub fn clone_data_map(&self) -> SharedDataMap {
        SharedDataMap::new(
            self.data_map
                .as_ref()
                .expect("The data map MUST NOT be `None` in this case")
                .clone(),
        )
    }
}

impl Debug for ScopedDataMap {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        write!(f, "{{ path: {:?}, regex: {:?} }}", self.path, self.regex)
    }
}