refactor everything

1 files changed, 49 insertions, 0 deletions

diff --git a/src/primitive/variantlist.rs b/src/primitive/variantlist.rs
new file mode 100644
index 0000000..452b927
--- /dev/null
+++ b/src/primitive/variantlist.rs
@@ -0,0 +1,49 @@
+use std::convert::TryInto;
+use std::vec::Vec;
+
+use failure::Error;
+
+use log::trace;
+
+use crate::{Deserialize, Serialize};
+
+extern crate bytes;
+
+use crate::primitive::Variant;
+
+/// VariantLists are represented as a Vec of Variants.
+///
+/// They are serialized as the amount of entries as a i32 and then a Variant for each entry
+pub type VariantList = Vec<Variant>;
+
+impl Serialize for VariantList {
+    fn serialize(&self) -> Result<Vec<u8>, Error> {
+        let len: i32 = self.len().try_into()?;
+        let mut res: Vec<u8> = Vec::new();
+
+        res.extend(len.to_be_bytes().iter());
+        for v in self {
+            res.extend(v.serialize()?.iter());
+        }
+
+        return Ok(res);
+    }
+}
+
+impl Deserialize for VariantList {
+    fn parse(b: &[u8]) -> Result<(usize, Self), Error> {
+        let (_, len) = i32::parse(&b[0..4])?;
+        trace!(target: "primitive::VariantList", "Parsing VariantList with {:?} elements", len);
+
+        let mut res: VariantList = VariantList::new();
+        let mut pos: usize = 4;
+        for i in 0..len {
+            trace!(target: "primitive::VariantList", "Parsing VariantList element: {:?}", i);
+            let (vlen, val) = Variant::parse(&b[pos..])?;
+            res.push(val);
+            pos += vlen;
+        }
+
+        return Ok((pos, res));
+    }
+}