need to do more checks

lxaw · Nov 9, 2023 · 7048bca · 7048bca
1 parent a6c5cb4
commit 7048bca
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 52 deletions.
diff --git a/decoded.txt b/decoded.txt
diff --git a/out.bin b/out.bin
diff --git a/out.binc b/out.binc
diff --git a/out.txt b/out.txt
diff --git a/src/huffman.rs b/src/huffman.rs
@@ -112,7 +112,12 @@ pub fn decode_encoded_str(encoded_msg: &String, map: &BTreeMap<char,String>) ->
         for (key,value) in map.iter(){
             if value.len() <= msg_copy.len() && msg_copy.starts_with(value) {
                 // check if this key matches the current substring
-                ret.push(*key);
+                if *key == NEW_LINE{
+                    ret.push('\n');
+                }
+                else{
+                    ret.push(*key);
+                }
                 // remove len chars
                 msg_copy.drain(..value.len());
             }