use crate::prelude::*; use std::io::{Read as _, Write as _}; use tokio::io::AsyncReadExt as _; #[derive(serde::Serialize, serde::Deserialize, Debug)] pub struct Config { pub email: Option, pub base_url: Option, pub identity_url: Option, #[serde(default = "default_lock_timeout")] pub lock_timeout: u64, } impl Default for Config { fn default() -> Self { Self { email: Default::default(), base_url: Default::default(), identity_url: Default::default(), lock_timeout: default_lock_timeout(), } } } pub fn default_lock_timeout() -> u64 { 3600 } impl Config { pub fn new() -> Self { Self::default() } pub fn load() -> Result { let file = crate::dirs::config_file(); let mut fh = std::fs::File::open(&file).with_context(|| { crate::error::LoadConfig { file: file.clone() } })?; let mut json = String::new(); fh.read_to_string(&mut json).with_context(|| { crate::error::LoadConfig { file: file.clone() } })?; let mut slf: Self = serde_json::from_str(&json) .context(crate::error::LoadConfigJson { file })?; if slf.lock_timeout == 0 { log::warn!("lock_timeout must be greater than 0"); slf.lock_timeout = default_lock_timeout(); } Ok(slf) } pub async fn load_async() -> Result { let file = crate::dirs::config_file(); let mut fh = tokio::fs::File::open(&file).await.with_context(|| { crate::error::LoadConfigAsync { file: file.clone() } })?; let mut json = String::new(); fh.read_to_string(&mut json).await.with_context(|| { crate::error::LoadConfigAsync { file: file.clone() } })?; let mut slf: Self = serde_json::from_str(&json) .context(crate::error::LoadConfigJson { file })?; if slf.lock_timeout == 0 { log::warn!("lock_timeout must be greater than 0"); slf.lock_timeout = default_lock_timeout(); } Ok(slf) } pub fn save(&self) -> Result<()> { let file = crate::dirs::config_file(); // unwrap is safe here because Self::filename is explicitly // constructed as a filename in a directory std::fs::create_dir_all(file.parent().unwrap()).with_context( || crate::error::SaveConfig { file: file.clone() }, )?; let mut fh = std::fs::File::create(&file).with_context(|| { crate::error::SaveConfig { file: file.clone() } })?; fh.write_all( serde_json::to_string(self) .with_context(|| crate::error::SaveConfigJson { file: file.clone(), })? .as_bytes(), ) .context(crate::error::SaveConfig { file })?; Ok(()) } pub fn validate() -> Result<()> { let config = Self::load()?; if config.email.is_none() { return Err(Error::ConfigMissingEmail); } 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), ) }) } pub fn server_name(&self) -> String { self.base_url .clone() .unwrap_or_else(|| "default".to_string()) } }