From 3215eab199a5a55a3254217da0aa10e9bd92c696 Mon Sep 17 00:00:00 2001
From: HouXiaoxuan <Hou_Xiaoxuan@163.com>
Date: Mon, 18 Dec 2023 01:07:03 +0800
Subject: [PATCH] simple store

---
 src/lib.rs   |  3 ++-
 src/store.rs | 57 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 src/store.rs

diff --git a/src/lib.rs b/src/lib.rs
index 88f4974..82e690a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,4 +1,5 @@
 // 不使用lib.rs的话，就无法在tests里引用到src中的模块
 pub mod objects;
 pub mod utils;
-pub mod commands;
\ No newline at end of file
+pub mod commands;
+mod store;
\ No newline at end of file
diff --git a/src/store.rs b/src/store.rs
new file mode 100644
index 0000000..7812526
--- /dev/null
+++ b/src/store.rs
@@ -0,0 +1,57 @@
+use super::utils::util;
+
+pub struct Store {
+    store_path: String,
+}
+
+impl Store {
+    pub fn new() -> Store {
+        if !util::storage_exist() {
+            panic!("不是合法的mit仓库");
+        }
+        let store_path = util::get_storage_path().unwrap();
+        Store {
+            store_path: store_path,
+        }
+    }
+    pub fn load(&self, hash: &String) -> String {
+        /* 读取文件内容 */
+        let mut path = self.store_path.clone();
+        path.push_str("/objects/");
+        path.push_str(hash);
+        match std::fs::read_to_string(path) {
+            Ok(content) => content,
+            Err(_) => panic!("储存库疑似损坏，无法读取文件"),
+        }
+    }
+    pub fn save(&self, content: &String) -> String {
+        /* 保存文件内容 */
+        let hash = util::calc_hash(content);
+        let mut path = self.store_path.clone();
+
+        path.push_str("/objects/");
+        path.push_str(&hash);
+        match std::fs::write(path, content) {
+            Ok(_) => hash,
+            Err(_) => panic!("储存库疑似损坏，无法写入文件"),
+        }
+    }
+}
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_new() {
+        let _ = Store::new();
+    }
+
+    #[test]
+    fn test_save_and_load() {
+        let store = Store::new();
+        let content = "hello world".to_string();
+        let hash = store.save(&content);
+        let content2 = store.load(&hash);
+        assert_eq!(content, content2, "内容不一致");
+    }
+}