From 8ff994ab8419a243b051a07feabefdad3a4e1e3b Mon Sep 17 00:00:00 2001
From: Nathan Reiner <nathan@nathanreiner.xyz>
Date: Thu, 6 Jul 2023 12:48:30 +0200
Subject: add cli

---
 src/dictionary.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/dictionary.rs')

diff --git a/src/dictionary.rs b/src/dictionary.rs
index a8d9b28..b522c9e 100644
--- a/src/dictionary.rs
+++ b/src/dictionary.rs
@@ -31,8 +31,8 @@ impl Dictionary {
         }
     }
 
-    pub fn get(&self, name : String) -> u64 {
-        *self.data.get(&name).unwrap()
+    pub fn get(&self, name : String) -> Option<&u64> {
+        self.data.get(&name)
     }
 
     pub fn iter(&self) -> &HashMap<String, u64> {
@@ -55,7 +55,7 @@ impl Dictionary {
         let mut fv = FileVector::new();
 
         for word in words {
-            let i = self.get(word);
+            let i = *self.get(word).unwrap();
             if !fv.contains_key(&i) {
                 fv.insert(i, 1);
             } else {
-- 
cgit v1.2.3-70-g09d2