From ac306333209459f0f695555d31be10e1e879f3d4 Mon Sep 17 00:00:00 2001
From: arnaucube <arnaucube@gmail.com>
Date: Sat, 25 May 2019 10:28:05 +0200
Subject: [PATCH] add db insert&get, add merkletree struct

---
 src/lib.rs | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index ab93cd1..ed7926c 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -10,6 +10,10 @@ extern crate rustc_hex;
 mod utils;
 mod node;
 
+const TYPENODEEMPTY: u8 = 0;
+const TYPENODENORMAL: u8 = 1;
+const TYPENODEFINAL: u8 = 2;
+const TYPENODEVALUE: u8 = 3;
 // const TYPENODEROOT: u8 = 4;
 const EMPTYNODEVALUE: [u8;32] = [0;32];
 
@@ -30,3 +34,58 @@ impl Value for TestValue {
    }
 }
 
+#[allow(dead_code)]
+pub struct Db {
+   storage: HashMap<[u8;32], Vec<u8>>,
+}
+impl Db {
+   pub fn insert(&mut self, k: [u8; 32], t: u8, il: u32, b: &mut Vec<u8>) {
+      let mut v: Vec<u8>;
+      v = [t].to_vec();
+      let il_bytes = il.to_le_bytes();
+      v.append(&mut il_bytes.to_vec()); // il_bytes are [u8;4] (4 bytes)
+      v.append(b);
+      self.storage.insert(k, v);
+   }
+   pub fn get(&self, k: &[u8;32]) -> (u8, u32, Vec<u8>) {
+      if k.to_vec() == EMPTYNODEVALUE.to_vec() {
+         return (0, 0, EMPTYNODEVALUE.to_vec());
+      }
+      match self.storage.get(k) {
+         Some(x) => {
+            let t = x[0];
+            let il_bytes: [u8; 4] = [x[1], x[2], x[3], x[4]];
+            let il = u32::from_le_bytes(il_bytes);
+            let b = &x[5..];
+            return (t, il, b.to_vec());
+         },
+         None => return (TYPENODEEMPTY, 0, EMPTYNODEVALUE.to_vec()),
+      }
+   }
+} 
+
+pub fn new_db()-> Db {
+   let db = Db {
+      storage: HashMap::new(),
+   };
+   db
+}
+
+pub struct MerkleTree {
+   #[allow(dead_code)]
+   root: [u8; 32],
+   #[allow(dead_code)]
+   num_levels: u32,
+   #[allow(dead_code)]
+   sto: Db,
+}
+
+
+pub fn new(num_levels: u32) -> MerkleTree {
+   let mt = MerkleTree {
+      root: EMPTYNODEVALUE,
+      num_levels: num_levels,
+      sto: new_db(),
+   };
+   mt
+}