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
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

//! Combinator path matcher

use std::{ops, sync::Arc};

use super::MatchMaker;
use crate::path::Path;

#[derive(Debug, Clone)]
/// Combines several matches together
///
/// It implements normal boolean algebra
/// * `! comb`  will negate the combinator
/// * `comb1 & comb2` both should pass
/// * `comb1 | comb2` at least one should pass
pub enum Combinator {
    /// Represents the actual underlying matcher
    Matcher(Arc<dyn MatchMaker + Sync>),
    /// Negates the expression
    Not(Box<Combinator>),
    /// Both expressions should be valid
    And(Box<Combinator>, Box<Combinator>),
    /// At least one of the expressions should be valid
    Or(Box<Combinator>, Box<Combinator>),
}

impl MatchMaker for Combinator {
    fn match_path(&self, path: &Path) -> bool {
        match self {
            Self::Matcher(matcher) => matcher.match_path(path),
            Self::Not(combinator) => !combinator.match_path(path),
            Self::Or(first, second) => first.match_path(path) || second.match_path(path),
            Self::And(first, second) => first.match_path(path) && second.match_path(path),
        }
    }
}

impl Combinator {
    /// Creates a new matcher combinator
    ///
    /// # Arguments
    /// * `matcher` - matcher to be wrapped
    pub fn new(matcher: impl MatchMaker + 'static + Sync) -> Self {
        Self::Matcher(Arc::new(matcher))
    }
}

impl ops::Not for Combinator {
    type Output = Self;

    fn not(self) -> Self {
        Self::Not(Box::new(self))
    }
}

impl ops::BitAnd for Combinator {
    type Output = Self;
    fn bitand(self, rhs: Self) -> Self {
        Self::And(Box::new(self), Box::new(rhs))
    }
}

impl ops::BitOr for Combinator {
    type Output = Self;
    fn bitor(self, rhs: Self) -> Self {
        Self::Or(Box::new(self), Box::new(rhs))
    }
}

#[cfg(test)]
mod tests {
    use super::{Combinator, MatchMaker};
    use crate::{
        matcher::{Depth, Simple},
        path::Path,
    };
    use std::convert::TryFrom;

    #[test]
    fn wrapper() {
        let comb = Combinator::new(Depth::new(1, Some(1)));
        assert!(comb.match_path(&Path::try_from(r#"{"People"}"#).unwrap()));
        assert!(comb.match_path(&Path::try_from(r#"[0]"#).unwrap()));
    }

    #[test]
    fn not() {
        let comb = !Combinator::new(Depth::new(1, None));
        assert!(!comb.match_path(&Path::try_from(r#"{"People"}"#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#"[0]"#).unwrap()));
        assert!(comb.match_path(&Path::try_from(r#""#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#"{"People"}[0]"#).unwrap()));
    }

    #[test]
    fn and() {
        let comb = Combinator::new(Depth::new(1, Some(1)))
            & Combinator::new(Simple::new(r#"{}"#).unwrap());
        assert!(comb.match_path(&Path::try_from(r#"{"People"}"#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#"[0]"#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#""#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#"{"People"}[0]"#).unwrap()));
    }

    #[test]
    fn or() {
        let comb = Combinator::new(Depth::new(1, Some(1)))
            | Combinator::new(Simple::new(r#"{}[0]"#).unwrap());
        assert!(comb.match_path(&Path::try_from(r#"{"People"}"#).unwrap()));
        assert!(comb.match_path(&Path::try_from(r#"[0]"#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#""#).unwrap()));
        assert!(comb.match_path(&Path::try_from(r#"{"People"}[0]"#).unwrap()));
        assert!(!comb.match_path(&Path::try_from(r#"{"People"}[1]"#).unwrap()));
    }

    #[test]
    fn complex() {
        let comb1 = Combinator::new(Depth::new(1, Some(1)))
            | Combinator::new(Simple::new(r#"{}[0]"#).unwrap());
        let comb2 = Combinator::new(Depth::new(2, Some(2)))
            | Combinator::new(Simple::new(r#"{}"#).unwrap());
        let comb3 = !comb1 & comb2;

        assert!(!comb3.match_path(&Path::try_from(r#"{"People"}"#).unwrap()));
        assert!(!comb3.match_path(&Path::try_from(r#"[0]"#).unwrap()));
        assert!(!comb3.match_path(&Path::try_from(r#""#).unwrap()));
        assert!(!comb3.match_path(&Path::try_from(r#"{"People"}[0]"#).unwrap()));
        assert!(comb3.match_path(&Path::try_from(r#"{"People"}[1]"#).unwrap()));
    }
}