From 4a9dcd35dea2cfa226a5a504cddc713bd16b373d Mon Sep 17 00:00:00 2001
From: Trivernis <trivernis@gmail.com>
Date: Thu, 12 Mar 2020 11:08:56 +0100
Subject: [PATCH] Add serialization of meta chunks

---
 src/lib/rainbowutils.rs | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/src/lib/rainbowutils.rs b/src/lib/rainbowutils.rs
index 368be82..d3df94e 100644
--- a/src/lib/rainbowutils.rs
+++ b/src/lib/rainbowutils.rs
@@ -12,7 +12,6 @@ pub const HTBL_CHUNK_NAME: &str = "HTBL";
 pub const DTBL_CHUNK_NAME: &str = "DTBL";
 
 pub struct BinaryDictionaryFile {
-    name: String,
     reader: BufReader<File>,
     metadata: Option<MetaChunk>,
     lookup_table: Option<HashLookupTable>,
@@ -55,7 +54,6 @@ pub struct DataEntry {
 impl BinaryDictionaryFile {
     pub fn new(reader: BufReader<File>) -> Self {
         Self {
-            name: "".to_string(),
             metadata: None,
             lookup_table: None,
             reader,
@@ -167,6 +165,33 @@ impl GenericChunk {
     }
 }
 
+impl From<MetaChunk> for GenericChunk {
+    fn from(chunk: MetaChunk) -> GenericChunk {
+        let mut serialized_data: Vec<u8> = Vec::new();
+        let mut chunk_count_raw = [0u8; 4];
+        BigEndian::write_u32(&mut chunk_count_raw, chunk.chunk_count);
+        serialized_data.append(&mut chunk_count_raw.to_vec());
+        let mut entries_pc_raw = [0u8; 4];
+        BigEndian::write_u32(&mut entries_pc_raw, chunk.entries_per_chunk);
+        serialized_data.append(&mut entries_pc_raw.to_vec());
+        let mut total_entries_raw = [0u8; 4];
+        BigEndian::write_u32(&mut total_entries_raw, chunk.entry_count);
+        serialized_data.append(&mut total_entries_raw.to_vec());
+        if let Some(method) = chunk.compression_method {
+            serialized_data.append(&mut method.into_bytes());
+        } else {
+            serialized_data.append(&mut vec![0, 0, 0, 0]);
+        }
+
+        GenericChunk {
+            length: serialized_data.len() as u32,
+            name: META_CHUNK_NAME.to_string(),
+            data: serialized_data,
+            crc: 0,
+        }
+    }
+}
+
 impl TryFrom<GenericChunk> for MetaChunk {
     type Error = Error;