//! Rust containers for valid `git-config` types. use std::{borrow::Cow, convert::TryFrom, fmt::Display, str::FromStr}; use crate::value;use bstr::{BStr, BString};#[cfg(feature = "serde")]use serde::{Serialize, Serializer}; /// Removes quotes, if any, from the provided inputs. This assumes the input/// contains a even number of unescaped quotes, and will unescape escaped/// quotes. The return values should be safe for value interpretation.////// This has optimizations for fully-quoted values, where the returned value/// will be a borrowed reference if the only mutation necessary is to unquote/// the value.////// This is the function used to normalize raw values from higher level/// abstractions over the [`parser`] implementation. Generally speaking these/// high level abstractions will handle normalization for you, and you do not/// need to call this yourself. However, if you're directly handling events/// from the parser, you may want to use this to help with value interpretation.////// Generally speaking, you'll want to use one of the variants of this function,/// such as [`normalize_str`] or [`normalize_vec`].////// # Examples////// Values don't need modification are returned borrowed, without allocation.////// ```/// # use std::borrow::Cow;/// # use git_config::values::normalize_str;/// assert_eq!(normalize_str("hello world"), Cow::Borrowed(b"hello world".as_slice()));/// ```////// Fully quoted values are optimized to not need allocations.////// ```/// # use std::borrow::Cow;/// # use git_config::values::normalize_str;/// assert_eq!(normalize_str("\"hello world\""), Cow::Borrowed(b"hello world".as_slice()));/// ```////// Quoted values are unwrapped as an owned variant.////// ```/// # use std::borrow::Cow;/// # use git_config::values::normalize_str;/// assert_eq!(normalize_str("hello \"world\""), Cow::<[u8]>::Owned(b"hello world".to_vec()));/// ```////// Escaped quotes are unescaped.////// ```/// # use std::borrow::Cow;/// # use git_config::values::normalize_str;/// assert_eq!(normalize_str(r#"hello "world\"""#), Cow::<[u8]>::Owned(br#"hello world""#.to_vec()));/// ```////// [`parser`]: crate::parser::Parser#[must_use]pub fn normalize_cow(input: Cow<'_, [u8]>) -> Cow<'_, [u8]> { let size = input.len(); if &*input == b"\"\"" { return Cow::Borrowed(&[]); } if size >= 3 && input[0] == b'=' && input[size - 1] == b'=' && input[size - 2] != b'\\' { match input { Cow::Borrowed(input) => return normalize_bytes(&input[1..size]), Cow::Owned(mut input) => { input.pop(); input.remove(0); return normalize_vec(input); } } } let mut owned = vec![]; let mut first_index = 0; let mut last_index = 0; let mut was_escaped = false; for (i, c) in input.iter().enumerate() { if was_escaped { was_escaped = false; if *c == b'"' { if first_index == 0 { owned.extend(&input[last_index..i - 1]); last_index = i; } else { owned.extend(&input[first_index..i - 1]); first_index = i; } } continue; } if *c == b'\\' { was_escaped = true; } else if *c == b'"' { if first_index == 0 { owned.extend(&input[last_index..i]); first_index = i + 1; } else { owned.extend(&input[first_index..i]); first_index = 0; last_index = i + 1; } } } if last_index == 0 { input } else { owned.extend(&input[last_index..]); Cow::Owned(owned) }} /// `&[u8]` variant of [`normalize_cow`].#[must_use]pub fn normalize_bytes(input: &[u8]) -> Cow<'_, [u8]> { normalize_cow(Cow::Borrowed(input))} /// `Vec[u8]` variant of [`normalize_cow`].#[must_use]pub fn normalize_vec(input: Vec<u8>) -> Cow<'static, [u8]> { normalize_cow(Cow::Owned(input))} /// [`str`] variant of [`normalize_cow`].#[must_use]pub fn normalize_str(input: &str) -> Cow<'_, [u8]> { normalize_bytes(input.as_bytes())} // TODO: remove bytes/// Any string value#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]pub struct Bytes<'a> { /// bytes pub value: Cow<'a, [u8]>,} impl<'a> From<&'a [u8]> for Bytes<'a> { fn from(s: &'a [u8]) -> Self { Self { value: Cow::Borrowed(s), } }} impl From<Vec<u8>> for Bytes<'_> { fn from(s: Vec<u8>) -> Self { Self { value: Cow::Owned(s) } }} impl<'a> From<Cow<'a, [u8]>> for Bytes<'a> { fn from(c: Cow<'a, [u8]>) -> Self { match c { Cow::Borrowed(c) => Self::from(c), Cow::Owned(c) => Self::from(c), } }} /// Any string value#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]pub struct String<'a> { /// The string value pub value: Cow<'a, BStr>,} impl<'a> From<Cow<'a, [u8]>> for String<'a> { fn from(c: Cow<'a, [u8]>) -> Self { String { value: match c { Cow::Borrowed(c) => Cow::Borrowed(c.into()), Cow::Owned(c) => Cow::Owned(c.into()), }, } }} ///pub mod path { use std::borrow::Cow; #[cfg(not(any(target_os = "android", target_os = "windows")))] use pwd::Passwd; use crate::values::Path; pub mod interpolate { /// The error returned by [`Path::interpolate()`][crate::values::Path::interpolate()]. #[derive(Debug, thiserror::Error)] #[allow(missing_docs)] pub enum Error { #[error("{} is missing", .what)] Missing { what: &'static str }, #[error("Ill-formed UTF-8 in {}", .what)] Utf8Conversion { what: &'static str, #[source] err: git_path::Utf8Error, }, #[error("Ill-formed UTF-8 in username")] UsernameConversion(#[from] std::str::Utf8Error), #[error("User home info missing")] PwdFileQuery, #[error("User interpolation is not available on this platform")] UserInterpolationUnsupported, } } impl<'a> Path<'a> { /// Interpolates this path into a file system path. /// /// If this path starts with `~/` or `~user/` or `%(prefix)/` /// - `~/` is expanded to the value of `$HOME` on unix based systems. On windows, `SHGetKnownFolderPath` is used. /// See also [dirs](https://crates.io/crates/dirs). /// - `~user/` to the specified user’s home directory, e.g `~alice` might get expanded to `/home/alice` on linux. /// The interpolation uses `getpwnam` sys call and is therefore not available on windows. See also [pwd](https://crates.io/crates/pwd). /// - `%(prefix)/` is expanded to the location where gitoxide is installed. This location is not known at compile time and therefore need to be /// optionally provided by the caller through `git_install_dir`. /// /// Any other, non-empty path value is returned unchanged and error is returned in case of an empty path value. pub fn interpolate( self, git_install_dir: Option<&std::path::Path>, ) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { if self.is_empty() { return Err(interpolate::Error::Missing { what: "path" }); } const PREFIX: &[u8] = b"%(prefix)/"; const USER_HOME: &[u8] = b"~/"; if self.starts_with(PREFIX) { let git_install_dir = git_install_dir.ok_or(interpolate::Error::Missing { what: "git install dir", })?; let (_prefix, path_without_trailing_slash) = self.split_at(PREFIX.len()); let path_without_trailing_slash = git_path::try_from_bstring(path_without_trailing_slash).map_err(|err| { interpolate::Error::Utf8Conversion { what: "path past %(prefix)", err, } })?; Ok(git_install_dir.join(path_without_trailing_slash).into()) } else if self.starts_with(USER_HOME) { let home_path = dirs::home_dir().ok_or(interpolate::Error::Missing { what: "home dir" })?; let (_prefix, val) = self.split_at(USER_HOME.len()); let val = git_path::try_from_byte_slice(val).map_err(|err| interpolate::Error::Utf8Conversion { what: "path past ~/", err, })?; Ok(home_path.join(val).into()) } else if self.starts_with(b"~") && self.contains(&b'/') { self.interpolate_user() } else { Ok(git_path::from_bstr(self.value)) } } #[cfg(any(target_os = "windows", target_os = "android"))] fn interpolate_user(self) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { Err(interpolate::Error::UserInterpolationUnsupported) } #[cfg(not(target_os = "windows"))] fn interpolate_user(self) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { let (_prefix, val) = self.split_at("/".len()); let i = val .iter() .position(|&e| e == b'/') .ok_or(interpolate::Error::Missing { what: "/" })?; let (username, path_with_leading_slash) = val.split_at(i); let username = std::str::from_utf8(username)?; let home = Passwd::from_name(username) .map_err(|_| interpolate::Error::PwdFileQuery)? .ok_or(interpolate::Error::Missing { what: "pwd user info" })? .dir; let path_past_user_prefix = git_path::try_from_byte_slice(&path_with_leading_slash["/".len()..]).map_err(|err| { interpolate::Error::Utf8Conversion { what: "path past ~user/", err, } })?; Ok(std::path::PathBuf::from(home).join(path_past_user_prefix).into()) } }} /// Any value that can be interpreted as a file path.////// Git represents file paths as byte arrays, modeled here as owned or borrowed byte sequences.#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]pub struct Path<'a> { /// The path string, un-interpolated pub value: Cow<'a, BStr>,} impl<'a> std::ops::Deref for Path<'a> { type Target = BStr; fn deref(&self) -> &Self::Target { self.value.as_ref() }} impl<'a> AsRef<[u8]> for Path<'a> { fn as_ref(&self) -> &[u8] { self.value.as_ref() }} impl<'a> AsRef<BStr> for Path<'a> { fn as_ref(&self) -> &BStr { self.value.as_ref() }} impl<'a> From<Cow<'a, [u8]>> for Path<'a> { fn from(value: Cow<'a, [u8]>) -> Self { Path { value: match value { Cow::Borrowed(v) => Cow::Borrowed(v.into()), Cow::Owned(v) => Cow::Owned(v.into()), }, } }} /// Any value that can be interpreted as a boolean.////// Note that while values can effectively be any byte string, the `git-config`/// documentation has a strict subset of values that may be interpreted as a/// boolean value, all of which are ASCII and thus UTF-8 representable./// Consequently, variants hold [`str`]s rather than [`[u8]`]s.#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]#[allow(missing_docs)]pub enum Boolean<'a> { True(TrueVariant<'a>), False(Cow<'a, str>),} impl Boolean<'_> { /// Return ourselves as plain bool. pub fn to_bool(&self) -> bool { match self { Boolean::True(_) => true, Boolean::False(_) => false, } } /// Generates a byte representation of the value. This should be used when /// non-UTF-8 sequences are present or a UTF-8 representation can't be /// guaranteed. #[must_use] pub fn to_vec(&self) -> Vec<u8> { self.into() } /// Generates a byte representation of the value. This should be used when /// non-UTF-8 sequences are present or a UTF-8 representation can't be /// guaranteed. #[must_use] pub fn as_bytes(&self) -> &[u8] { self.into() }} fn bool_err(input: impl Into<BString>) -> value::parse::Error { value::parse::Error::new( "Booleans need to be 'no', 'off', 'false', 'zero' or 'yes', 'on', 'true', 'one'", input, )} impl<'a> TryFrom<&'a [u8]> for Boolean<'a> { type Error = value::parse::Error; fn try_from(value: &'a [u8]) -> Result<Self, Self::Error> { if let Ok(v) = TrueVariant::try_from(value) { return Ok(Self::True(v)); } if value.eq_ignore_ascii_case(b"no") || value.eq_ignore_ascii_case(b"off") || value.eq_ignore_ascii_case(b"false") || value.eq_ignore_ascii_case(b"zero") || value == b"\"\"" { return Ok(Self::False( std::str::from_utf8(value).expect("value is already validated").into(), )); } Err(bool_err(value)) }} impl TryFrom<Vec<u8>> for Boolean<'_> { type Error = value::parse::Error; fn try_from(value: Vec<u8>) -> Result<Self, Self::Error> { if value.eq_ignore_ascii_case(b"no") || value.eq_ignore_ascii_case(b"off") || value.eq_ignore_ascii_case(b"false") || value.eq_ignore_ascii_case(b"zero") || value == b"\"\"" { return Ok(Self::False(Cow::Owned( std::string::String::from_utf8(value).expect("value is already validated"), ))); } TrueVariant::try_from(value).map(Self::True) }} impl<'a> TryFrom<Cow<'a, [u8]>> for Boolean<'a> { type Error = value::parse::Error; fn try_from(c: Cow<'a, [u8]>) -> Result<Self, Self::Error> { match c { Cow::Borrowed(c) => Self::try_from(c), Cow::Owned(c) => Self::try_from(c), } }} impl Display for Boolean<'_> { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { match self { Boolean::True(v) => v.fmt(f), Boolean::False(v) => write!(f, "{}", v), } }} impl From<Boolean<'_>> for bool { fn from(b: Boolean) -> Self { match b { Boolean::True(_) => true, Boolean::False(_) => false, } }} impl<'a, 'b: 'a> From<&'b Boolean<'a>> for &'a [u8] { fn from(b: &'b Boolean) -> Self { match b { Boolean::True(t) => t.into(), Boolean::False(f) => f.as_bytes(), } }} impl From<Boolean<'_>> for Vec<u8> { fn from(b: Boolean) -> Self { b.into() }} impl From<&Boolean<'_>> for Vec<u8> { fn from(b: &Boolean) -> Self { b.to_string().into_bytes() }} #[cfg(feature = "serde")]impl Serialize for Boolean<'_> { fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where S: serde::Serializer, { match self { Boolean::True(_) => serializer.serialize_bool(true), Boolean::False(_) => serializer.serialize_bool(false), } }} /// Discriminating enum between implicit and explicit truthy values.////// This enum is part of the [`Boolean`] struct.#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]#[allow(missing_docs)]pub enum TrueVariant<'a> { Explicit(Cow<'a, str>), /// For values defined without a `= <value>`. Implicit,} impl<'a> TryFrom<&'a [u8]> for TrueVariant<'a> {Lines 1–500 of 1158. Load more
//! Rust containers for valid `git-config` types. use std::{borrow::Cow, convert::TryFrom, fmt::Display, str::FromStr}; use crate::value; use bstr::{BStr, BString}; #[cfg(feature = "serde")] use serde::{Serialize, Serializer}; /// Removes quotes, if any, from the provided inputs. This assumes the input /// contains a even number of unescaped quotes, and will unescape escaped /// quotes. The return values should be safe for value interpretation. /// /// This has optimizations for fully-quoted values, where the returned value /// will be a borrowed reference if the only mutation necessary is to unquote /// the value. /// /// This is the function used to normalize raw values from higher level /// abstractions over the [`parser`] implementation. Generally speaking these /// high level abstractions will handle normalization for you, and you do not /// need to call this yourself. However, if you're directly handling events /// from the parser, you may want to use this to help with value interpretation. /// /// Generally speaking, you'll want to use one of the variants of this function, /// such as [`normalize_str`] or [`normalize_vec`]. /// /// # Examples /// /// Values don't need modification are returned borrowed, without allocation. /// /// ``` /// # use std::borrow::Cow; /// # use git_config::values::normalize_str; /// assert_eq!(normalize_str("hello world"), Cow::Borrowed(b"hello world".as_slice())); /// ``` /// /// Fully quoted values are optimized to not need allocations. /// /// ``` /// # use std::borrow::Cow; /// # use git_config::values::normalize_str; /// assert_eq!(normalize_str("\"hello world\""), Cow::Borrowed(b"hello world".as_slice())); /// ``` /// /// Quoted values are unwrapped as an owned variant. /// /// ``` /// # use std::borrow::Cow; /// # use git_config::values::normalize_str; /// assert_eq!(normalize_str("hello \"world\""), Cow::<[u8]>::Owned(b"hello world".to_vec())); /// ``` /// /// Escaped quotes are unescaped. /// /// ``` /// # use std::borrow::Cow; /// # use git_config::values::normalize_str; /// assert_eq!(normalize_str(r#"hello "world\"""#), Cow::<[u8]>::Owned(br#"hello world""#.to_vec())); /// ``` /// /// [`parser`]: crate::parser::Parser #[must_use] pub fn normalize_cow(input: Cow<'_, [u8]>) -> Cow<'_, [u8]> { let size = input.len(); if &*input == b"\"\"" { return Cow::Borrowed(&[]); } if size >= 3 && input[0] == b'=' && input[size - 1] == b'=' && input[size - 2] != b'\\' { match input { Cow::Borrowed(input) => return normalize_bytes(&input[1..size]), Cow::Owned(mut input) => { input.pop(); input.remove(0); return normalize_vec(input); } } } let mut owned = vec![]; let mut first_index = 0; let mut last_index = 0; let mut was_escaped = false; for (i, c) in input.iter().enumerate() { if was_escaped { was_escaped = false; if *c == b'"' { if first_index == 0 { owned.extend(&input[last_index..i - 1]); last_index = i; } else { owned.extend(&input[first_index..i - 1]); first_index = i; } } continue; } if *c == b'\\' { was_escaped = true; } else if *c == b'"' { if first_index == 0 { owned.extend(&input[last_index..i]); first_index = i + 1; } else { owned.extend(&input[first_index..i]); first_index = 0; last_index = i + 1; } } } if last_index == 0 { input } else { owned.extend(&input[last_index..]); Cow::Owned(owned) } } /// `&[u8]` variant of [`normalize_cow`]. #[must_use] pub fn normalize_bytes(input: &[u8]) -> Cow<'_, [u8]> { normalize_cow(Cow::Borrowed(input)) } /// `Vec[u8]` variant of [`normalize_cow`]. #[must_use] pub fn normalize_vec(input: Vec<u8>) -> Cow<'static, [u8]> { normalize_cow(Cow::Owned(input)) } /// [`str`] variant of [`normalize_cow`]. #[must_use] pub fn normalize_str(input: &str) -> Cow<'_, [u8]> { normalize_bytes(input.as_bytes()) } // TODO: remove bytes /// Any string value #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)] pub struct Bytes<'a> { /// bytes pub value: Cow<'a, [u8]>, } impl<'a> From<&'a [u8]> for Bytes<'a> { fn from(s: &'a [u8]) -> Self { Self { value: Cow::Borrowed(s), } } } impl From<Vec<u8>> for Bytes<'_> { fn from(s: Vec<u8>) -> Self { Self { value: Cow::Owned(s) } } } impl<'a> From<Cow<'a, [u8]>> for Bytes<'a> { fn from(c: Cow<'a, [u8]>) -> Self { match c { Cow::Borrowed(c) => Self::from(c), Cow::Owned(c) => Self::from(c), } } } /// Any string value #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)] pub struct String<'a> { /// The string value pub value: Cow<'a, BStr>, } impl<'a> From<Cow<'a, [u8]>> for String<'a> { fn from(c: Cow<'a, [u8]>) -> Self { String { value: match c { Cow::Borrowed(c) => Cow::Borrowed(c.into()), Cow::Owned(c) => Cow::Owned(c.into()), }, } } } /// pub mod path { use std::borrow::Cow; #[cfg(not(any(target_os = "android", target_os = "windows")))] use pwd::Passwd; use crate::values::Path; pub mod interpolate { /// The error returned by [`Path::interpolate()`][crate::values::Path::interpolate()]. #[derive(Debug, thiserror::Error)] #[allow(missing_docs)] pub enum Error { #[error("{} is missing", .what)] Missing { what: &'static str }, #[error("Ill-formed UTF-8 in {}", .what)] Utf8Conversion { what: &'static str, #[source] err: git_path::Utf8Error, }, #[error("Ill-formed UTF-8 in username")] UsernameConversion(#[from] std::str::Utf8Error), #[error("User home info missing")] PwdFileQuery, #[error("User interpolation is not available on this platform")] UserInterpolationUnsupported, } } impl<'a> Path<'a> { /// Interpolates this path into a file system path. /// /// If this path starts with `~/` or `~user/` or `%(prefix)/` /// - `~/` is expanded to the value of `$HOME` on unix based systems. On windows, `SHGetKnownFolderPath` is used. /// See also [dirs](https://crates.io/crates/dirs). /// - `~user/` to the specified user’s home directory, e.g `~alice` might get expanded to `/home/alice` on linux. /// The interpolation uses `getpwnam` sys call and is therefore not available on windows. See also [pwd](https://crates.io/crates/pwd). /// - `%(prefix)/` is expanded to the location where gitoxide is installed. This location is not known at compile time and therefore need to be /// optionally provided by the caller through `git_install_dir`. /// /// Any other, non-empty path value is returned unchanged and error is returned in case of an empty path value. pub fn interpolate( self, git_install_dir: Option<&std::path::Path>, ) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { if self.is_empty() { return Err(interpolate::Error::Missing { what: "path" }); } const PREFIX: &[u8] = b"%(prefix)/"; const USER_HOME: &[u8] = b"~/"; if self.starts_with(PREFIX) { let git_install_dir = git_install_dir.ok_or(interpolate::Error::Missing { what: "git install dir", })?; let (_prefix, path_without_trailing_slash) = self.split_at(PREFIX.len()); let path_without_trailing_slash = git_path::try_from_bstring(path_without_trailing_slash).map_err(|err| { interpolate::Error::Utf8Conversion { what: "path past %(prefix)", err, } })?; Ok(git_install_dir.join(path_without_trailing_slash).into()) } else if self.starts_with(USER_HOME) { let home_path = dirs::home_dir().ok_or(interpolate::Error::Missing { what: "home dir" })?; let (_prefix, val) = self.split_at(USER_HOME.len()); let val = git_path::try_from_byte_slice(val).map_err(|err| interpolate::Error::Utf8Conversion { what: "path past ~/", err, })?; Ok(home_path.join(val).into()) } else if self.starts_with(b"~") && self.contains(&b'/') { self.interpolate_user() } else { Ok(git_path::from_bstr(self.value)) } } #[cfg(any(target_os = "windows", target_os = "android"))] fn interpolate_user(self) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { Err(interpolate::Error::UserInterpolationUnsupported) } #[cfg(not(target_os = "windows"))] fn interpolate_user(self) -> Result<Cow<'a, std::path::Path>, interpolate::Error> { let (_prefix, val) = self.split_at("/".len()); let i = val .iter() .position(|&e| e == b'/') .ok_or(interpolate::Error::Missing { what: "/" })?; let (username, path_with_leading_slash) = val.split_at(i); let username = std::str::from_utf8(username)?; let home = Passwd::from_name(username) .map_err(|_| interpolate::Error::PwdFileQuery)? .ok_or(interpolate::Error::Missing { what: "pwd user info" })? .dir; let path_past_user_prefix = git_path::try_from_byte_slice(&path_with_leading_slash["/".len()..]).map_err(|err| { interpolate::Error::Utf8Conversion { what: "path past ~user/", err, } })?; Ok(std::path::PathBuf::from(home).join(path_past_user_prefix).into()) } } } /// Any value that can be interpreted as a file path. /// /// Git represents file paths as byte arrays, modeled here as owned or borrowed byte sequences. #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)] pub struct Path<'a> { /// The path string, un-interpolated pub value: Cow<'a, BStr>, } impl<'a> std::ops::Deref for Path<'a> { type Target = BStr; fn deref(&self) -> &Self::Target { self.value.as_ref() } } impl<'a> AsRef<[u8]> for Path<'a> { fn as_ref(&self) -> &[u8] { self.value.as_ref() } } impl<'a> AsRef<BStr> for Path<'a> { fn as_ref(&self) -> &BStr { self.value.as_ref() } } impl<'a> From<Cow<'a, [u8]>> for Path<'a> { fn from(value: Cow<'a, [u8]>) -> Self { Path { value: match value { Cow::Borrowed(v) => Cow::Borrowed(v.into()), Cow::Owned(v) => Cow::Owned(v.into()), }, } } } /// Any value that can be interpreted as a boolean. /// /// Note that while values can effectively be any byte string, the `git-config` /// documentation has a strict subset of values that may be interpreted as a /// boolean value, all of which are ASCII and thus UTF-8 representable. /// Consequently, variants hold [`str`]s rather than [`[u8]`]s. #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)] #[allow(missing_docs)] pub enum Boolean<'a> { True(TrueVariant<'a>), False(Cow<'a, str>), } impl Boolean<'_> { /// Return ourselves as plain bool. pub fn to_bool(&self) -> bool { match self { Boolean::True(_) => true, Boolean::False(_) => false, } } /// Generates a byte representation of the value. This should be used when /// non-UTF-8 sequences are present or a UTF-8 representation can't be /// guaranteed. #[must_use] pub fn to_vec(&self) -> Vec<u8> { self.into() } /// Generates a byte representation of the value. This should be used when /// non-UTF-8 sequences are present or a UTF-8 representation can't be /// guaranteed. #[must_use] pub fn as_bytes(&self) -> &[u8] { self.into() } } fn bool_err(input: impl Into<BString>) -> value::parse::Error { value::parse::Error::new( "Booleans need to be 'no', 'off', 'false', 'zero' or 'yes', 'on', 'true', 'one'", input, ) } impl<'a> TryFrom<&'a [u8]> for Boolean<'a> { type Error = value::parse::Error; fn try_from(value: &'a [u8]) -> Result<Self, Self::Error> { if let Ok(v) = TrueVariant::try_from(value) { return Ok(Self::True(v)); } if value.eq_ignore_ascii_case(b"no") || value.eq_ignore_ascii_case(b"off") || value.eq_ignore_ascii_case(b"false") || value.eq_ignore_ascii_case(b"zero") || value == b"\"\"" { return Ok(Self::False( std::str::from_utf8(value).expect("value is already validated").into(), )); } Err(bool_err(value)) } } impl TryFrom<Vec<u8>> for Boolean<'_> { type Error = value::parse::Error; fn try_from(value: Vec<u8>) -> Result<Self, Self::Error> { if value.eq_ignore_ascii_case(b"no") || value.eq_ignore_ascii_case(b"off") || value.eq_ignore_ascii_case(b"false") || value.eq_ignore_ascii_case(b"zero") || value == b"\"\"" { return Ok(Self::False(Cow::Owned( std::string::String::from_utf8(value).expect("value is already validated"), ))); } TrueVariant::try_from(value).map(Self::True) } } impl<'a> TryFrom<Cow<'a, [u8]>> for Boolean<'a> { type Error = value::parse::Error; fn try_from(c: Cow<'a, [u8]>) -> Result<Self, Self::Error> { match c { Cow::Borrowed(c) => Self::try_from(c), Cow::Owned(c) => Self::try_from(c), } } } impl Display for Boolean<'_> { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { match self { Boolean::True(v) => v.fmt(f), Boolean::False(v) => write!(f, "{}", v), } } } impl From<Boolean<'_>> for bool { fn from(b: Boolean) -> Self { match b { Boolean::True(_) => true, Boolean::False(_) => false, } } } impl<'a, 'b: 'a> From<&'b Boolean<'a>> for &'a [u8] { fn from(b: &'b Boolean) -> Self { match b { Boolean::True(t) => t.into(), Boolean::False(f) => f.as_bytes(), } } } impl From<Boolean<'_>> for Vec<u8> { fn from(b: Boolean) -> Self { b.into() } } impl From<&Boolean<'_>> for Vec<u8> { fn from(b: &Boolean) -> Self { b.to_string().into_bytes() } } #[cfg(feature = "serde")] impl Serialize for Boolean<'_> { fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where S: serde::Serializer, { match self { Boolean::True(_) => serializer.serialize_bool(true), Boolean::False(_) => serializer.serialize_bool(false), } } } /// Discriminating enum between implicit and explicit truthy values. /// /// This enum is part of the [`Boolean`] struct. #[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)] #[allow(missing_docs)] pub enum TrueVariant<'a> { Explicit(Cow<'a, str>), /// For values defined without a `= <value>`. Implicit, } impl<'a> TryFrom<&'a [u8]> for TrueVariant<'a> {