From b3a04c4a143c34ba92008cf018eed159f87a0c6e Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Fri, 10 Apr 2020 00:12:40 -0400
Subject: move some basic stuff into config

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

(limited to 'src/config.rs')

diff --git a/src/config.rs b/src/config.rs
new file mode 100644
index 0000000..d02bc11
--- /dev/null
+++ b/src/config.rs
@@ -0,0 +1,76 @@
+use crate::prelude::*;
+
+use std::io::{Read as _, Write as _};
+use tokio::io::AsyncReadExt as _;
+
+#[derive(serde::Serialize, serde::Deserialize, Default, Debug)]
+pub struct Config {
+    pub email: Option<String>,
+    pub base_url: Option<String>,
+    pub identity_url: Option<String>,
+}
+
+impl Config {
+    pub fn new() -> Self {
+        Self::default()
+    }
+
+    pub fn load() -> Result<Self> {
+        let mut fh = std::fs::File::open(Self::filename())
+            .context(crate::error::LoadConfig)?;
+        let mut json = String::new();
+        fh.read_to_string(&mut json)
+            .context(crate::error::LoadConfig)?;
+        let slf: Self = serde_json::from_str(&json)
+            .context(crate::error::LoadConfigJson)?;
+        Ok(slf)
+    }
+
+    pub async fn load_async() -> Result<Self> {
+        let mut fh = tokio::fs::File::open(Self::filename())
+            .await
+            .context(crate::error::LoadConfigAsync)?;
+        let mut json = String::new();
+        fh.read_to_string(&mut json)
+            .await
+            .context(crate::error::LoadConfigAsync)?;
+        let slf: Self = serde_json::from_str(&json)
+            .context(crate::error::LoadConfigJson)?;
+        Ok(slf)
+    }
+
+    pub fn save(&self) -> Result<()> {
+        let filename = Self::filename();
+        std::fs::create_dir_all(filename.parent().unwrap())
+            .context(crate::error::SaveConfig)?;
+        let mut fh = std::fs::File::create(filename)
+            .context(crate::error::SaveConfig)?;
+        fh.write_all(
+            serde_json::to_string(self)
+                .context(crate::error::SaveConfigJson)?
+                .as_bytes(),
+        )
+        .context(crate::error::SaveConfig)?;
+        Ok(())
+    }
+
+    pub fn base_url(&self) -> String {
+        self.base_url.clone().map_or_else(
+            || "https://api.bitwarden.com".to_string(),
+            |url| format!("{}/api", url),
+        )
+    }
+
+    pub fn identity_url(&self) -> String {
+        self.identity_url.clone().unwrap_or_else(|| {
+            self.base_url.clone().map_or_else(
+                || "https://identity.bitwarden.com".to_string(),
+                |url| format!("{}/identity", url),
+            )
+        })
+    }
+
+    fn filename() -> std::path::PathBuf {
+        crate::dirs::config_dir().join("config.json")
+    }
+}
-- 
cgit v1.2.3-54-g00ecf