From 3428a637ce420baef9aa9f9803e71bd587867005 Mon Sep 17 00:00:00 2001
From: FxQnLr <felixquinn03@gmail.com>
Date: Wed, 10 Apr 2024 00:16:55 +0200
Subject: Closes #24. Changed postgres to json directory storage

---
 src/storage.rs | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 src/storage.rs

(limited to 'src/storage.rs')

diff --git a/src/storage.rs b/src/storage.rs
new file mode 100644
index 0000000..6ba5ee1
--- /dev/null
+++ b/src/storage.rs
@@ -0,0 +1,65 @@
+use std::{
+    fs::{create_dir_all, File},
+    io::{Read, Write},
+    path::Path,
+};
+
+use ipnetwork::IpNetwork;
+use mac_address::MacAddress;
+use serde::{Deserialize, Serialize};
+use serde_json::json;
+use tracing::{debug, warn};
+use utoipa::ToSchema;
+
+use crate::error::Error;
+
+#[derive(Serialize, Deserialize, Clone, Debug)]
+pub struct Device {
+    pub id: String,
+    pub mac: MacAddress,
+    pub broadcast_addr: String,
+    pub ip: IpNetwork,
+    pub times: Option<Vec<i64>>,
+}
+
+impl Device {
+    const STORAGE_PATH: &'static str = "devices";
+
+    pub fn setup() -> Result<String, Error> {
+        let sp = Path::new(Self::STORAGE_PATH);
+        if !sp.exists() {
+            warn!("device storage path doesn't exist, creating it");
+            create_dir_all(Self::STORAGE_PATH)?;
+        };
+
+        debug!("device storage at '{}'", Self::STORAGE_PATH);
+
+        Ok(Self::STORAGE_PATH.to_string())
+    }
+
+    pub fn read(id: &str) -> Result<Self, Error> {
+        let mut file = File::open(format!("{}/{id}.json", Self::STORAGE_PATH))?;
+        let mut buf = String::new();
+        file.read_to_string(&mut buf)?;
+
+        let dev = serde_json::from_str(&buf)?;
+        Ok(dev)
+    }
+
+    pub fn write(&self) -> Result<(), Error> {
+        let mut file = File::create(format!("{}/{}.json", Self::STORAGE_PATH, self.id))?;
+        file.write_all(json!(self).to_string().as_bytes())?;
+
+        Ok(())
+    }
+}
+
+#[derive(ToSchema)]
+#[schema(as = Device)]
+pub struct DeviceSchema {
+    pub id: String,
+    pub mac: String,
+    pub broadcast_addr: String,
+    pub ip: String,
+    pub times: Option<Vec<i64>>,
+}
-- 
cgit v1.2.3