use figment::{providers::Serialized, value, Figment, Metadata, Profile, Provider}; use serde::{Deserialize, Serialize}; use std::{convert::Infallible, path::PathBuf, str::FromStr}; #[serde_with::serde_as] #[derive(Debug, PartialEq, Eq, Clone, Default, Serialize, Deserialize)] pub struct Config { #[serde_as(as = "Vec>")] pub(crate) paths: Vec, } impl Config { // Extract the configuration from any `Provider` pub fn extract(provider: T) -> figment::error::Result { Figment::from(provider).extract() } // Provide a default provider, a `Figment`. pub fn figment() -> Figment { Figment::from(Config::default()) } } impl Provider for Config { fn metadata(&self) -> Metadata { Metadata::named("Projectr config") } fn data(&self) -> figment::error::Result> { Serialized::defaults(self).data() } } #[derive(Debug, PartialEq, Eq, Clone, Default, Serialize, Deserialize)] #[serde(default)] pub struct Entry { pub path_buf: PathBuf, pub hidden: bool, pub max_depth: Option, pub pattern: Option, #[cfg(feature = "git")] pub git: bool, } impl Entry { pub fn new(path_buf: PathBuf) -> Self { Self { path_buf, ..Default::default() } } } impl From for Entry { fn from(path_buf: PathBuf) -> Self { Self::new(path_buf) } } impl FromStr for Entry { type Err = Infallible; fn from_str(s: &str) -> Result { s.parse().map(|path_buf| Self { path_buf, ..Default::default() }) } } #[cfg(test)] mod tests { use super::*; use figment::providers::{Format, Serialized, Toml}; use pretty_assertions::assert_eq; #[test] fn test_extract_config() { figment::Jail::expect_with(|jail| { jail.create_file( "file.toml", r#" paths = [ "/path/to/projects", { path_buf = "/path/to/other_projects", hidden = true, max_depth = 1 }, { path_buf = "/path/to/another_project", max_depth = 0 } ] "#, )?; let config: Config = Figment::from(Serialized::defaults(Config::default())) .merge(Toml::file("file.toml")) .extract()?; assert_eq!( config, Config { paths: Vec::from([ Entry { path_buf: "/path/to/projects".into(), hidden: false, max_depth: None, ..Default::default() }, Entry { path_buf: "/path/to/other_projects".into(), hidden: true, max_depth: Some(1), ..Default::default() }, Entry { path_buf: "/path/to/another_project".into(), hidden: false, max_depth: Some(0), ..Default::default() }, ]), } ); Ok(()) }); } }