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, trace, 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: Option, pub times: Option>, } impl Device { const STORAGE_PATH: &'static str = "devices"; pub fn setup() -> Result { trace!("check for storage at {}", Self::STORAGE_PATH); 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 { trace!(?id, "attempt to read file"); let mut file = File::open(format!("{}/{id}.json", Self::STORAGE_PATH))?; let mut buf = String::new(); file.read_to_string(&mut buf)?; trace!(?id, ?buf, "read successfully from file"); let dev = serde_json::from_str(&buf)?; Ok(dev) } pub fn write(&self) -> Result<(), Error> { trace!(?self.id, ?self, "attempt to write to file"); let mut file = File::create(format!("{}/{}.json", Self::STORAGE_PATH, self.id))?; file.write_all(json!(self).to_string().as_bytes())?; trace!(?self.id, "wrote successfully to file"); 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>, }