From 1b779459f1e40fcae389619855ed48fb5c3ee3a5 Mon Sep 17 00:00:00 2001
From: Coin de Gamma <cg@corpglory.com>
Date: Mon, 16 Sep 2024 14:29:28 +0000
Subject: [PATCH] basic bae64 storage

---
 Cargo.toml  |  1 +
 src/db.rs   | 23 +++++++++++++++++++++--
 src/main.rs |  3 ---
 3 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 87cd199..b3d73f6 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -6,6 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+base64 = "0.22.1"
 clap = { version = "4.5.16", features = ["derive"] }
 once_cell = "1.19.0"
 tempfile = "3.12.0"
diff --git a/src/db.rs b/src/db.rs
index 65f2b14..93b044c 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -1,3 +1,5 @@
+use base64::prelude::*;
+
 use std::collections::HashSet;
 use std::hash::{Hash, Hasher};
 
@@ -55,6 +57,14 @@ impl Ord for Item {
     }
 }
 
+fn decode(line: String) -> Result<String, io::Error> {
+    let content = BASE64_STANDARD.decode(line).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;
+    match String::from_utf8(content) {
+        Ok(s) => Ok(s),
+        Err(e) => Err(io::Error::new(io::ErrorKind::InvalidData, e))
+    }
+}
+
 pub struct DB {
     pub items: HashSet::<Item>,
     path: String
@@ -65,11 +75,18 @@ impl DB {
         let file = fs::File::open(path)?;
         let reader = io::BufReader::new(file);
         let mut items = HashSet::<Item>::new();
+        let mut id: Option<String> = None;
         for line in reader.lines() {
             match line {
-                Ok(id) => {
+                Ok(line) => {
                     // TODO: check content
-                    items.insert(Item::from_d(id));
+                    if id.is_none() {
+                        id = Some(line);
+                    } else {
+                        let content = decode(line)?;
+                        items.insert(Item::from(id.unwrap(), content));
+                        id = None;
+                    }
                 },
                 Err(e) => {
                     eprintln!("Error reading line, {}", e);
@@ -97,6 +114,8 @@ impl DB {
 
         for item in self.items.iter() {
             writeln!(file, "{}", item.id)?;
+            let content = BASE64_STANDARD.encode(item.content.clone());
+            writeln!(file, "{}", content)?;
         }
 
         Ok(())
diff --git a/src/main.rs b/src/main.rs
index 9bb3fb8..62f61ec 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -77,9 +77,6 @@ fn add(id: &String) -> io::Result<()> {
     
     // TODO: implement encryption
     let content = editor::open_to_edit()?;
-    println!("Content");
-    println!("--------");
-    println!("{}", content);
     db.items.insert(Item::from(id.clone(), content));
     db.dump()?;