add: exact_search

ynqa · Apr 21, 2024 · 0900de3 · 0900de3
1 parent 4cdf0d3
commit 0900de3
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 2 deletions.
diff --git a/src/jnv.rs b/src/jnv.rs
@@ -98,6 +98,10 @@ impl Jnv {
         indent: usize,
     ) -> Result<Prompt<Self>> {
         let input_stream = deserialize_json(&input)?;
+
+        let mut trie = FilterTrie::default();
+        trie.insert(".", input_stream.clone());
+
         let all_kinds = JsonStream::new(input_stream.clone(), None).flatten_kinds();
         let suggest = Suggest::from_iter(all_kinds.iter().filter_map(|kind| kind.path()).map(
             |segments| {
@@ -157,7 +161,7 @@ impl Jnv {
                         indent,
                     },
                 },
-                trie: FilterTrie::default(),
+                trie,
                 suggest,
                 expand_depth,
                 no_hint,

diff --git a/src/trie.rs b/src/trie.rs
@@ -10,6 +10,10 @@ impl FilterTrie {
         self.0.insert(query.to_string(), json_nodes);
     }
 
+    pub fn exact_search(&self, query: &str) -> Option<&Vec<serde_json::Value>> {
+        self.0.get(query)
+    }
+
     pub fn prefix_search(&self, query: &str) -> Option<&Vec<serde_json::Value>> {
         self.0
             .get_ancestor(query)
@@ -21,12 +25,40 @@ impl FilterTrie {
 mod tests {
     use super::*;
 
+    mod exact_search {
+        use super::*;
+        use serde_json::json;
+
+        #[test]
+        fn test_exact_match() {
+            let mut trie = FilterTrie::default();
+            trie.insert("apple", vec![json!({"type": "fruit"})]);
+            trie.insert("app", vec![json!({"type": "abbreviation"})]);
+
+            let result = trie.exact_search("app");
+            assert!(result.is_some());
+            let values = result.unwrap();
+            assert_eq!(values.len(), 1);
+            assert_eq!(values[0], json!({"type": "abbreviation"}));
+        }
+
+        #[test]
+        fn test_no_match() {
+            let mut trie = FilterTrie::default();
+            trie.insert("apple", vec![json!({"type": "fruit"})]);
+            trie.insert("app", vec![json!({"type": "abbreviation"})]);
+
+            let result = trie.exact_search("application");
+            assert!(result.is_none());
+        }
+    }
+
     mod prefix_search {
         use super::*;
         use serde_json::json;
 
         #[test]
-        fn test() {
+        fn test_with_exact_prefix() {
             let mut trie = FilterTrie::default();
             trie.insert("apple", vec![json!({"type": "fruit"})]);
             trie.insert("app", vec![json!({"type": "abbreviation"})]);
@@ -36,18 +68,39 @@ mod tests {
             let values = result.unwrap();
             assert_eq!(values.len(), 1);
             assert_eq!(values[0], json!({"type": "abbreviation"}));
+        }
+
+        #[test]
+        fn test_with_longer_query_than_keys() {
+            let mut trie = FilterTrie::default();
+            trie.insert("apple", vec![json!({"type": "fruit"})]);
+            trie.insert("app", vec![json!({"type": "abbreviation"})]);
 
             let result = trie.prefix_search("application");
             assert!(result.is_some());
             let values = result.unwrap();
             assert_eq!(values.len(), 1);
             assert_eq!(values[0], json!({"type": "abbreviation"}));
+        }
+
+        #[test]
+        fn test_with_full_key_match() {
+            let mut trie = FilterTrie::default();
+            trie.insert("apple", vec![json!({"type": "fruit"})]);
+            trie.insert("app", vec![json!({"type": "abbreviation"})]);
 
             let result = trie.prefix_search("apple");
             assert!(result.is_some());
             let values = result.unwrap();
             assert_eq!(values.len(), 1);
             assert_eq!(values[0], json!({"type": "fruit"}));
+        }
+
+        #[test]
+        fn test_with_shorter_query_than_any_key() {
+            let mut trie = FilterTrie::default();
+            trie.insert("apple", vec![json!({"type": "fruit"})]);
+            trie.insert("app", vec![json!({"type": "abbreviation"})]);
 
             let result = trie.prefix_search("ap");
             assert!(result.is_none());