close #50

freestrings · Sep 7, 2021 · 1d7bada · 1d7bada
2 parents 33a9f99 + a1b323a
commit 1d7bada
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 18 deletions.
diff --git a/src/selector/terms.rs b/src/selector/terms.rs
@@ -450,6 +450,27 @@ impl<'a> FilterTerms<'a> {
             return current;
         }
 
+        if let Some(Some(e)) = self.pop_term() {
+            match e {
+                ExprTerm::Json(rel, _, vec) => {
+                    return if vec.is_empty() {
+                        Some(Vec::new())
+                    } else if let Some(vec) = rel {
+                        let index = utils::abs_index(index as isize, vec.len());
+                        let ret = vec.get(index).map_or(Vec::new(), |v| vec![*v]);
+                        Some(ret)
+                    } else {
+                        let index = utils::abs_index(index as isize, vec.len());
+                        let ret = vec.get(index).map_or(Vec::new(), |v| vec![*v]);
+                        Some(ret)
+                    };
+                }
+                _ => {
+                    self.push_term(Some(e));
+                }
+            }
+        }
+
         let acc = ValueWalker::next_with_num(&current.unwrap(), index);
 
         if acc.is_empty() {
@@ -459,15 +480,6 @@ impl<'a> FilterTerms<'a> {
         Some(acc)
     }
 
-    pub fn collect_next_all(&mut self, current: Option<Vec<&'a Value>>) -> Option<Vec<&'a Value>> {
-        if current.is_none() {
-            debug!("collect_next_all : {:?}", &current);
-            return current;
-        }
-
-        Some(ValueWalker::next_all(&current.unwrap()))
-    }
-
     pub fn collect_next_with_str(&mut self, current: Option<Vec<&'a Value>>, keys: &[&'a str]) -> Option<Vec<&'a Value>> {
         if current.is_none() {
             debug!(
@@ -486,6 +498,15 @@ impl<'a> FilterTerms<'a> {
         Some(acc)
     }
 
+    pub fn collect_next_all(&mut self, current: Option<Vec<&'a Value>>) -> Option<Vec<&'a Value>> {
+        if current.is_none() {
+            debug!("collect_next_all : {:?}", &current);
+            return current;
+        }
+
+        Some(ValueWalker::next_all(&current.unwrap()))
+    }
+
     pub fn collect_all(&mut self, current: Option<Vec<&'a Value>>) -> Option<Vec<&'a Value>> {
         if current.is_none() {
             debug!("collect_all: {:?}", &current);

diff --git a/src/selector/value_walker.rs b/src/selector/value_walker.rs
@@ -32,15 +32,6 @@ impl<'a> ValueWalker {
     pub fn next_with_num(vec: &Vec<&'a Value>, index: f64) -> Vec<&'a Value> {
         vec.iter().fold(Vec::new(), |mut acc, v| {
             match v {
-                Value::Object(map) => {
-                    for k in map.keys() {
-                        if let Some(Value::Array(vec)) = map.get(k) {
-                            if let Some(v) = vec.get(utils::abs_index(index as isize, vec.len())) {
-                                acc.push(v);
-                            }
-                        }
-                    }
-                }
                 Value::Array(vec) => {
                     if let Some(v) = vec.get(utils::abs_index(index as isize, vec.len())) {
                         acc.push(v);

diff --git a/tests/array_filter.rs b/tests/array_filter.rs
@@ -250,4 +250,15 @@ fn bugs40_bracket_notation_after_recursive_descent() {
            "more"
         ]),
     );
+}
+
+#[test]
+fn bugs50() {
+    setup();
+
+    select_and_then_compare(
+        "$[0]",
+        json!({"f": [1,2,3]}),
+        json!([])
+    );
 }