Implement array access

Author: Marko Mikulicic

src/grammar.pest

@@ -18,8 +18,9 @@ char = {
 }
 
 union =  { "[" ~ unionElement ~ ("," ~ unionElement)* ~ "]" }
-unionElement = _{ unionChild }
+unionElement = _{ unionChild | unionArrayIndex }
 unionChild = { doubleQuotedString | singleQuotedString }
+unionArrayIndex = { "-" ? ~ ASCII_DIGIT+ }
 
 doubleQuotedString = _{ "\"" ~ doubleInner ~ "\"" }
 doubleInner = @{ doubleChar* }

src/matchers.rs

@@ -82,7 +82,7 @@ impl Matcher for ArrayIndex {
             self.index as usize
         } else {
             let abs = (-self.index) as usize;
-            if abs < len {
+            if abs <= len {
                 len - abs
             } else {
                 return Box::new(iter::empty());
@@ -163,6 +163,14 @@ mod tests {
         assert_eq!(format!("{:?}", r), "[Number(2)]");
     }
 
+    #[test]
+    fn array_index_negative_extreme() {
+        let s = ArrayIndex::new(-2);
+        let j = json!([1, 2]);
+        let r: Vec<&Value> = s.select(&j).collect();
+        assert_eq!(format!("{:?}", r), "[Number(1)]");
+    }
+
     #[test]
     fn array_index_negative_oob() {
         let s = ArrayIndex::new(-10);

src/parser.rs

@@ -75,10 +75,18 @@ fn parse_dot_child_matcher(
 fn parse_union(matcher_rule: pest::iterators::Pair<Rule>) -> Vec<Box<dyn matchers::Matcher>> {
     let mut ms: Vec<Box<dyn matchers::Matcher>> = Vec::new();
     for r in matcher_rule.into_inner() {
-        if let Rule::unionChild = r.as_rule() {
-            for m in parse_union_child(r) {
-                ms.push(m)
+        match r.as_rule() {
+            Rule::unionChild => {
+                for m in parse_union_child(r) {
+                    ms.push(m)
+                }
+            }
+            Rule::unionArrayIndex => {
+                for m in parse_union_array_index(r) {
+                    ms.push(m)
+                }
             }
+            _ => {}
         }
     }
     vec![Box::new(matchers::Union::new(ms))]
@@ -102,6 +110,15 @@ fn parse_union_child(matcher_rule: pest::iterators::Pair<Rule>) -> Vec<Box<dyn m
     ms
 }
 
+fn parse_union_array_index(
+    matcher_rule: pest::iterators::Pair<Rule>,
+) -> Vec<Box<dyn matchers::Matcher>> {
+    let mut ms: Vec<Box<dyn matchers::Matcher>> = Vec::new();
+    let i = matcher_rule.as_str().parse().unwrap();
+    ms.push(Box::new(matchers::ArrayIndex::new(i)));
+    ms
+}
+
 const ESCAPED: &str = "\"'\\/bfnrt";
 const UNESCAPED: &str = "\"'\\/\u{0008}\u{000C}\u{000A}\u{000D}\u{0009}";
 

tests/cts.json

@@ -461,5 +461,35 @@
     "name": "union child, single quotes, incomplete escape",
     "selector": "$['\\']",
     "invalid_selector": true
+  }, {
+    "name": "union array access",
+    "selector": "$[0]",
+    "document": ["first", "second"],
+    "result": ["first"]
+  }, {
+    "name": "union array access, 1",
+    "selector": "$[1]",
+    "document": ["first", "second"],
+    "result": ["second"]
+  }, {
+    "name": "union array access, out of bound",
+    "selector": "$[2]",
+    "document": ["first", "second"],
+    "result": []
+  }, {
+    "name": "union array access, negative",
+    "selector": "$[-1]",
+    "document": ["first", "second"],
+    "result": ["second"]
+  }, {
+    "name": "union array access, more negative",
+    "selector": "$[-2]",
+    "document": ["first", "second"],
+    "result": ["first"]
+  }, {
+    "name": "union array access, negative out of bound",
+    "selector": "$[-3]",
+    "document": ["first", "second"],
+    "result": []
   }
-]}
+]}