use std::{
borrow::Cow,
collections::{HashMap, VecDeque},
convert::TryFrom,
fmt::Display,
path::{Path, PathBuf},
};
use crate::{
file::{
error::{GitConfigError, GitConfigFromEnvError},
section::{MutableSection, SectionBody},
value::{EntryData, MutableMultiValue, MutableValue},
Index, Size,
},
parser::{
parse_from_bytes, parse_from_path, parse_from_str, Error, Event, Key, ParsedSectionHeader, Parser,
ParserOrIoError, SectionHeaderName,
},
};
#[derive(PartialEq, Eq, Hash, Copy, Clone, PartialOrd, Ord, Debug)]
pub(super) struct SectionId(usize);
#[derive(PartialEq, Eq, Clone, Debug)]
pub(super) enum LookupTreeNode<'a> {
Terminal(Vec<SectionId>),
NonTerminal(HashMap<Cow<'a, str>, Vec<SectionId>>),
}
#[derive(PartialEq, Eq, Clone, Debug, Default)]
pub struct GitConfig<'event> {
frontmatter_events: SectionBody<'event>,
pub(super) section_lookup_tree: HashMap<SectionHeaderName<'event>, Vec<LookupTreeNode<'event>>>,
pub(super) sections: HashMap<SectionId, SectionBody<'event>>,
section_headers: HashMap<SectionId, ParsedSectionHeader<'event>>,
section_id_counter: usize,
section_order: VecDeque<SectionId>,
}
impl<'event> GitConfig<'event> {
#[inline]
#[must_use]
pub fn new() -> Self {
Self::default()
}
#[inline]
pub fn open<P: AsRef<Path>>(path: P) -> Result<Self, ParserOrIoError<'static>> {
parse_from_path(path).map(Self::from)
}
#[inline]
pub fn from_paths(paths: &[&Path]) -> Result<Self, ParserOrIoError<'static>> {
let mut config = Self::new();
for path in paths {
let other = Self::open(path)?;
for (section_id, section_header) in other.section_headers {
config.push_section(
section_header.name.0.to_owned(),
section_header.subsection_name.to_owned(),
other.sections[§ion_id].clone(),
);
}
}
Ok(config)
}
pub fn from_env_paths() -> Result<Self, ParserOrIoError<'static>> {
use std::env;
let mut paths = vec![];
if env::var("GIT_CONFIG_NO_SYSTEM").is_err() {
if let Ok(git_config_system) = env::var("GIT_CONFIG_SYSTEM") {
paths.push(PathBuf::from(git_config_system))
} else {
paths.push(PathBuf::from("/etc/gitconfig"));
}
}
if let Ok(git_config_global) = env::var("GIT_CONFIG_GLOBAL") {
paths.push(PathBuf::from(git_config_global));
} else {
if let Ok(xdg_config_home) = env::var("XDG_CONFIG_HOME") {
paths.push(PathBuf::from(xdg_config_home).join("git/config"));
} else if let Ok(home) = env::var("HOME") {
paths.push(PathBuf::from(home).join(".config/git/config"));
}
if let Ok(home) = env::var("HOME") {
paths.push(PathBuf::from(home).join(".gitconfig"));
}
}
if let Ok(git_dir) = env::var("GIT_DIR") {
paths.push(PathBuf::from(git_dir).join("config"));
}
let paths = paths.iter().map(PathBuf::as_path).collect::<Vec<&Path>>();
Self::from_paths(&paths)
}
pub fn from_env() -> Result<Option<Self>, GitConfigFromEnvError> {
use std::env;
let count: usize = match env::var("GIT_CONFIG_COUNT") {
Ok(v) => v.parse().map_err(|_| GitConfigFromEnvError::ParseError(v))?,
Err(_) => return Ok(None),
};
let mut config = Self::new();
for i in 0..count {
let key = env::var(format!("GIT_CONFIG_KEY_{}", i)).map_err(|_| GitConfigFromEnvError::InvalidKeyId(i))?;
let value =
env::var(format!("GIT_CONFIG_VALUE_{}", i)).map_err(|_| GitConfigFromEnvError::InvalidValueId(i))?;
if let Some((section, maybe_subsection)) = key.split_once('.') {
let (subsection, key) = if let Some((subsection, key)) = maybe_subsection.rsplit_once('.') {
(Some(subsection), key)
} else {
(None, maybe_subsection)
};
let mut section = if let Ok(section) = config.section_mut(section, subsection) {
section
} else {
config.new_section(
section.to_string(),
subsection.map(|subsection| Cow::Owned(subsection.to_string())),
)
};
section.push(
Cow::<str>::Owned(key.to_string()).into(),
Cow::Owned(value.into_bytes()),
);
} else {
return Err(GitConfigFromEnvError::InvalidKeyValue(i, key.to_string()));
}
}
if config.is_empty() {
Ok(None)
} else {
Ok(Some(config))
}
}
#[inline]
pub fn value<'lookup, T: TryFrom<Cow<'event, [u8]>>>(
&'event self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<T, GitConfigError<'lookup>> {
T::try_from(self.get_raw_value(section_name, subsection_name, key)?)
.map_err(|_| GitConfigError::FailedConversion)
}
#[inline]
pub fn multi_value<'lookup, T: TryFrom<Cow<'event, [u8]>>>(
&'event self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<Vec<T>, GitConfigError<'lookup>> {
self.get_raw_multi_value(section_name, subsection_name, key)?
.into_iter()
.map(T::try_from)
.collect::<Result<Vec<_>, _>>()
.map_err(|_| GitConfigError::FailedConversion)
}
pub fn section<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
) -> Result<&SectionBody<'event>, GitConfigError<'lookup>> {
let section_ids = self.get_section_ids_by_name_and_subname(section_name, subsection_name)?;
let id = section_ids
.last()
.expect("Section lookup vec was empty, internal invariant violated");
Ok(self
.sections
.get(id)
.expect("Section did not have id from lookup, internal invariant violated"))
}
pub fn section_mut<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
) -> Result<MutableSection<'_, 'event>, GitConfigError<'lookup>> {
let section_ids = self.get_section_ids_by_name_and_subname(section_name, subsection_name)?;
let id = section_ids
.last()
.expect("Section lookup vec was empty, internal invariant violated");
Ok(MutableSection::new(self.sections.get_mut(id).expect(
"Section did not have id from lookup, internal invariant violated",
)))
}
#[must_use]
pub fn sections_by_name<'lookup>(&self, section_name: &'lookup str) -> Vec<&SectionBody<'event>> {
self.get_section_ids_by_name(section_name)
.unwrap_or_default()
.into_iter()
.map(|id| {
self.sections
.get(&id)
.expect("section doesn't have id from from lookup")
})
.collect()
}
pub fn new_section(
&mut self,
section_name: impl Into<Cow<'event, str>>,
subsection_name: impl Into<Option<Cow<'event, str>>>,
) -> MutableSection<'_, 'event> {
let mut section = self.push_section(section_name, subsection_name, SectionBody::new());
section.push_newline();
section
}
pub fn remove_section<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: impl Into<Option<&'lookup str>>,
) -> Option<SectionBody> {
let id = self
.get_section_ids_by_name_and_subname(section_name, subsection_name.into())
.ok()?
.pop()?;
self.section_order.remove(
self.section_order
.iter()
.position(|v| *v == id)
.expect("Section order does not contain section that we were trying to remove"),
);
self.sections.remove(&id)
}
pub fn push_section(
&mut self,
section_name: impl Into<Cow<'event, str>>,
subsection_name: impl Into<Option<Cow<'event, str>>>,
section: SectionBody<'event>,
) -> MutableSection<'_, 'event> {
let subsection_name = subsection_name.into();
if subsection_name.is_some() {
self.push_section_internal(
ParsedSectionHeader {
name: SectionHeaderName(section_name.into()),
separator: Some(" ".into()),
subsection_name,
},
section,
)
} else {
self.push_section_internal(
ParsedSectionHeader {
name: SectionHeaderName(section_name.into()),
separator: None,
subsection_name: None,
},
section,
)
}
}
pub fn rename_section<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: impl Into<Option<&'lookup str>>,
new_section_name: impl Into<SectionHeaderName<'event>>,
new_subsection_name: impl Into<Option<Cow<'event, str>>>,
) -> Result<(), GitConfigError<'lookup>> {
let id = self.get_section_ids_by_name_and_subname(section_name, subsection_name.into())?;
let id = id
.last()
.expect("list of sections were empty, which violates invariant");
let header = self
.section_headers
.get_mut(id)
.expect("sections does not have section id from section ids");
header.name = new_section_name.into();
header.subsection_name = new_subsection_name.into();
Ok(())
}
#[must_use]
pub fn len(&self) -> usize {
self.sections.values().fold(0, |acc, section| acc + section.len())
}
#[must_use]
pub fn is_empty(&self) -> bool {
self.sections.values().all(SectionBody::is_empty)
}
}
impl<'event> GitConfig<'event> {
pub fn get_raw_value<'lookup>(
&self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<Cow<'_, [u8]>, GitConfigError<'lookup>> {
let key = Key(key.into());
for section_id in self
.get_section_ids_by_name_and_subname(section_name, subsection_name)?
.iter()
.rev()
{
if let Some(v) = self
.sections
.get(section_id)
.expect("sections does not have section id from section ids")
.value(&key)
{
return Ok(v.to_vec().into());
}
}
Err(GitConfigError::KeyDoesNotExist)
}
pub fn get_raw_value_mut<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<MutableValue<'_, 'lookup, 'event>, GitConfigError<'lookup>> {
let section_ids = self.get_section_ids_by_name_and_subname(section_name, subsection_name)?;
let key = Key(key.into());
for section_id in section_ids.iter().rev() {
let mut size = Size(0);
let mut index = Index(0);
let mut found_key = false;
for (i, event) in self
.sections
.get(section_id)
.expect("sections does not have section id from section ids")
.as_ref()
.iter()
.enumerate()
{
match event {
Event::Key(event_key) if *event_key == key => {
found_key = true;
size = Size(1);
index = Index(i);
}
Event::Newline(_) | Event::Whitespace(_) | Event::ValueNotDone(_) if found_key => {
size += 1;
}
Event::ValueDone(_) | Event::Value(_) if found_key => {
found_key = false;
size += 1;
}
_ => (),
}
}
if size.0 == 0 {
continue;
}
return Ok(MutableValue::new(
MutableSection::new(
self.sections
.get_mut(section_id)
.expect("sections does not have section id from section ids"),
),
key,
index,
size,
));
}
Err(GitConfigError::KeyDoesNotExist)
}
pub fn get_raw_multi_value<'lookup>(
&self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<Vec<Cow<'_, [u8]>>, GitConfigError<'lookup>> {
let mut values = vec![];
for section_id in self.get_section_ids_by_name_and_subname(section_name, subsection_name)? {
values.extend(
self.sections
.get(§ion_id)
.expect("sections does not have section id from section ids")
.values(&Key(key.into()))
.iter()
.map(|v| Cow::Owned(v.to_vec())),
);
}
if values.is_empty() {
Err(GitConfigError::KeyDoesNotExist)
} else {
Ok(values)
}
}
pub fn get_raw_multi_value_mut<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
) -> Result<MutableMultiValue<'_, 'lookup, 'event>, GitConfigError<'lookup>> {
let section_ids = self.get_section_ids_by_name_and_subname(section_name, subsection_name)?;
let key = Key(key.into());
let mut offsets = HashMap::new();
let mut entries = vec![];
for section_id in section_ids.iter().rev() {
let mut last_boundary = 0;
let mut found_key = false;
let mut offset_list = vec![];
let mut offset_index = 0;
for (i, event) in self
.sections
.get(section_id)
.expect("sections does not have section id from section ids")
.as_ref()
.iter()
.enumerate()
{
match event {
Event::Key(event_key) if *event_key == key => {
found_key = true;
offset_list.push(i - last_boundary);
offset_index += 1;
last_boundary = i;
}
Event::Value(_) | Event::ValueDone(_) if found_key => {
found_key = false;
entries.push(EntryData::new(*section_id, offset_index));
offset_list.push(i - last_boundary + 1);
offset_index += 1;
last_boundary = i + 1;
}
_ => (),
}
}
offsets.insert(*section_id, offset_list);
}
entries.sort();
if entries.is_empty() {
Err(GitConfigError::KeyDoesNotExist)
} else {
Ok(MutableMultiValue::new(&mut self.sections, key, entries, offsets))
}
}
pub fn set_raw_value<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
new_value: Vec<u8>,
) -> Result<(), GitConfigError<'lookup>> {
self.get_raw_value_mut(section_name, subsection_name, key)
.map(|mut entry| entry.set_bytes(new_value))
}
pub fn set_raw_multi_value<'lookup>(
&mut self,
section_name: &'lookup str,
subsection_name: Option<&'lookup str>,
key: &'lookup str,
new_values: impl Iterator<Item = Cow<'event, [u8]>>,
) -> Result<(), GitConfigError<'lookup>> {
self.get_raw_multi_value_mut(section_name, subsection_name, key)
.map(|mut v| v.set_values(new_values))
}
}
impl<'event> GitConfig<'event> {
fn push_section_internal(
&mut self,
header: ParsedSectionHeader<'event>,
section: SectionBody<'event>,
) -> MutableSection<'_, 'event> {
let new_section_id = SectionId(self.section_id_counter);
self.section_headers.insert(new_section_id, header.clone());
self.sections.insert(new_section_id, section);
let lookup = self.section_lookup_tree.entry(header.name).or_default();
let mut found_node = false;
if let Some(subsection_name) = header.subsection_name {
for node in lookup.iter_mut() {
if let LookupTreeNode::NonTerminal(subsection) = node {
found_node = true;
subsection
.entry(subsection_name.clone())
.or_default()
.push(new_section_id);
break;
}
}
if !found_node {
let mut map = HashMap::new();
map.insert(subsection_name, vec![new_section_id]);
lookup.push(LookupTreeNode::NonTerminal(map));
}
} else {
for node in lookup.iter_mut() {
if let LookupTreeNode::Terminal(vec) = node {
found_node = true;
vec.push(new_section_id);
break;
}
}
if !found_node {
lookup.push(LookupTreeNode::Terminal(vec![new_section_id]));
}
}
self.section_order.push_back(new_section_id);
self.section_id_counter += 1;
self.sections.get_mut(&new_section_id).map(MutableSection::new).unwrap()
}
fn get_section_ids_by_name_and_subname<'lookup>(
&self,
section_name: impl Into<SectionHeaderName<'lookup>>,
subsection_name: Option<&'lookup str>,
) -> Result<Vec<SectionId>, GitConfigError<'lookup>> {
let section_name = section_name.into();
let section_ids = self
.section_lookup_tree
.get(§ion_name)
.ok_or(GitConfigError::SectionDoesNotExist(section_name))?;
let mut maybe_ids = None;
if let Some(subsection_name) = subsection_name {
for node in section_ids {
if let LookupTreeNode::NonTerminal(subsection_lookup) = node {
maybe_ids = subsection_lookup.get(subsection_name);
break;
}
}
} else {
for node in section_ids {
if let LookupTreeNode::Terminal(subsection_lookup) = node {
maybe_ids = Some(subsection_lookup);
break;
}
}
}
maybe_ids
.map(Vec::to_owned)
.ok_or(GitConfigError::SubSectionDoesNotExist(subsection_name))
}
fn get_section_ids_by_name<'lookup>(
&self,
section_name: impl Into<SectionHeaderName<'lookup>>,
) -> Result<Vec<SectionId>, GitConfigError<'lookup>> {
let section_name = section_name.into();
self.section_lookup_tree
.get(§ion_name)
.map(|lookup| {
lookup
.iter()
.flat_map(|node| match node {
LookupTreeNode::Terminal(v) => v.clone(),
LookupTreeNode::NonTerminal(v) => v.values().flatten().copied().collect(),
})
.collect()
})
.ok_or(GitConfigError::SectionDoesNotExist(section_name))
}
}
impl<'a> TryFrom<&'a str> for GitConfig<'a> {
type Error = Error<'a>;
#[inline]
fn try_from(s: &'a str) -> Result<GitConfig<'a>, Self::Error> {
parse_from_str(s).map(Self::from)
}
}
impl<'a> TryFrom<&'a [u8]> for GitConfig<'a> {
type Error = Error<'a>;
#[inline]
fn try_from(value: &'a [u8]) -> Result<GitConfig<'a>, Self::Error> {
parse_from_bytes(value).map(GitConfig::from)
}
}
impl<'a> TryFrom<&'a Vec<u8>> for GitConfig<'a> {
type Error = Error<'a>;
#[inline]
fn try_from(value: &'a Vec<u8>) -> Result<GitConfig<'a>, Self::Error> {
parse_from_bytes(value).map(GitConfig::from)
}
}
impl<'a> From<Parser<'a>> for GitConfig<'a> {
fn from(parser: Parser<'a>) -> Self {
let mut new_self = Self::default();
let mut prev_section_header = None;
let mut section_events = SectionBody::new();
#[allow(clippy::explicit_into_iter_loop)] for event in parser.into_iter() {
match event {
Event::SectionHeader(header) => {
if let Some(prev_header) = prev_section_header.take() {
new_self.push_section_internal(prev_header, section_events);
} else {
new_self.frontmatter_events = section_events;
}
prev_section_header = Some(header);
section_events = SectionBody::new();
}
e @ Event::Key(_)
| e @ Event::Value(_)
| e @ Event::ValueNotDone(_)
| e @ Event::ValueDone(_)
| e @ Event::KeyValueSeparator => section_events.as_mut().push(e),
e @ Event::Comment(_) | e @ Event::Newline(_) | e @ Event::Whitespace(_) => {
section_events.as_mut().push(e);
}
}
}
if let Some(header) = prev_section_header {
new_self.push_section_internal(header, section_events);
} else {
new_self.frontmatter_events = section_events;
}
new_self
}
}
impl From<GitConfig<'_>> for Vec<u8> {
#[inline]
fn from(c: GitConfig) -> Self {
c.into()
}
}
impl From<&GitConfig<'_>> for Vec<u8> {
fn from(config: &GitConfig) -> Self {
let mut value = Self::new();
for events in config.frontmatter_events.as_ref() {
value.extend(events.to_vec());
}
for section_id in &config.section_order {
value.extend(
config
.section_headers
.get(section_id)
.expect("section_header does not contain section id from section_order")
.to_vec(),
);
for event in config
.sections
.get(section_id)
.expect("sections does not contain section id from section_order")
.as_ref()
{
value.extend(event.to_vec());
}
}
value
}
}
impl Display for GitConfig<'_> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
for front_matter in self.frontmatter_events.as_ref() {
front_matter.fmt(f)?;
}
for section_id in &self.section_order {
self.section_headers.get(section_id).unwrap().fmt(f)?;
for event in self.sections.get(section_id).unwrap().as_ref() {
event.fmt(f)?;
}
}
Ok(())
}
}
#[cfg(test)]
mod mutable_value {
use std::convert::TryFrom;
use super::GitConfig;
fn init_config() -> GitConfig<'static> {
GitConfig::try_from(
r#"[core]
a=b"100"
[core]
c=d
e=f"#,
)
.unwrap()
}
#[test]
fn value_is_correct() {
let mut git_config = init_config();
let value = git_config.get_raw_value_mut("core", None, "a").unwrap();
assert_eq!(&*value.get().unwrap(), b"b100");
}
#[test]
fn set_string_cleanly_updates() {
let mut git_config = init_config();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
value.set_string("hello world".to_string());
assert_eq!(
git_config.to_string(),
r#"[core]
a=hello world
[core]
c=d
e=f"#,
);
let mut value = git_config.get_raw_value_mut("core", None, "e").unwrap();
value.set_string(String::new());
assert_eq!(
git_config.to_string(),
r#"[core]
a=hello world
[core]
c=d
e="#,
);
}
#[test]
fn delete_value() {
let mut git_config = init_config();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
value.delete();
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]
c=d
e=f",
);
let mut value = git_config.get_raw_value_mut("core", None, "c").unwrap();
value.delete();
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]\n \n e=f",
);
}
#[test]
fn get_value_after_deleted() {
let mut git_config = init_config();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
value.delete();
assert!(value.get().is_err());
}
#[test]
fn set_string_after_deleted() {
let mut git_config = init_config();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
value.delete();
value.set_string("hello world".to_string());
assert_eq!(
git_config.to_string(),
r#"[core]
a=hello world
[core]
c=d
e=f"#,
);
}
#[test]
fn subsequent_delete_calls_are_noop() {
let mut git_config = init_config();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
for _ in 0..10 {
value.delete();
}
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]
c=d
e=f",
);
}
#[test]
fn partial_values_are_supported() {
let mut git_config = GitConfig::try_from(
r#"[core]
a=b"100"\
b
[core]
c=d
e=f"#,
)
.unwrap();
let mut value = git_config.get_raw_value_mut("core", None, "a").unwrap();
assert_eq!(&*value.get().unwrap(), b"b100b");
value.delete();
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]
c=d
e=f",
);
}
}
#[cfg(test)]
mod mutable_multi_value {
use std::{borrow::Cow, convert::TryFrom};
use super::GitConfig;
fn init_config() -> GitConfig<'static> {
GitConfig::try_from(
r#"[core]
a=b"100"
[core]
a=d
a=f"#,
)
.unwrap()
}
#[test]
fn value_is_correct() {
let mut git_config = init_config();
let value = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
assert_eq!(
&*value.get().unwrap(),
vec![
Cow::<[u8]>::Owned(b"b100".to_vec()),
Cow::<[u8]>::Borrowed(b"d"),
Cow::<[u8]>::Borrowed(b"f"),
]
);
}
#[test]
fn non_empty_sizes_are_correct() {
let mut git_config = init_config();
assert_eq!(git_config.get_raw_multi_value_mut("core", None, "a").unwrap().len(), 3);
assert!(!git_config
.get_raw_multi_value_mut("core", None, "a")
.unwrap()
.is_empty());
}
#[test]
fn set_value_at_start() {
let mut git_config = init_config();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
values.set_string(0, "Hello".to_string());
assert_eq!(
git_config.to_string(),
r#"[core]
a=Hello
[core]
a=d
a=f"#,
);
}
#[test]
fn set_value_at_end() {
let mut git_config = init_config();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
values.set_string(2, "Hello".to_string());
assert_eq!(
git_config.to_string(),
r#"[core]
a=b"100"
[core]
a=d
a=Hello"#,
);
}
#[test]
fn set_values_all() {
let mut git_config = init_config();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
values.set_owned_values_all(b"Hello");
assert_eq!(
git_config.to_string(),
r#"[core]
a=Hello
[core]
a=Hello
a=Hello"#,
);
}
#[test]
fn delete() {
let mut git_config = init_config();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
values.delete(0);
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]
a=d
a=f",
);
}
#[test]
fn delete_all() {
let mut git_config = init_config();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
values.delete_all();
assert!(values.get().is_err());
assert_eq!(
git_config.to_string(),
"[core]\n \n [core]\n \n ",
);
}
#[test]
fn partial_values_are_supported() {
let mut git_config = GitConfig::try_from(
r#"[core]
a=b\
"100"
[core]
a=d\
b
a=f\
a"#,
)
.unwrap();
let mut values = git_config.get_raw_multi_value_mut("core", None, "a").unwrap();
assert_eq!(
&*values.get().unwrap(),
vec![
Cow::<[u8]>::Owned(b"b100".to_vec()),
Cow::<[u8]>::Borrowed(b"db"),
Cow::<[u8]>::Borrowed(b"fa"),
]
);
values.delete_all();
assert!(values.get().is_err());
}
}
#[cfg(test)]
mod from_paths {
use std::{fs, io};
use tempfile::tempdir;
use super::{Cow, GitConfig, ParserOrIoError};
#[test]
fn file_not_found() {
let dir = tempdir().unwrap();
let config_path = dir.path().join("config");
let paths = vec![config_path.as_path()];
let error = GitConfig::from_paths(&paths).unwrap_err();
assert!(matches!(error, ParserOrIoError::Io(io_error) if io_error.kind() == io::ErrorKind::NotFound));
}
#[test]
fn single_path() {
let dir = tempdir().unwrap();
let config_path = dir.path().join("config");
fs::write(config_path.as_path(), b"[core]\nboolean = true").expect("Unable to write config file");
let paths = vec![config_path.as_path()];
let config = GitConfig::from_paths(&paths).unwrap();
assert_eq!(
config.get_raw_value("core", None, "boolean"),
Ok(Cow::<[u8]>::Borrowed(b"true"))
);
assert_eq!(config.len(), 1);
}
#[test]
fn multiple_paths_single_value() {
let dir = tempdir().unwrap();
let a_path = dir.path().join("a");
fs::write(a_path.as_path(), b"[core]\na = true").expect("Unable to write config file");
let b_path = dir.path().join("b");
fs::write(b_path.as_path(), b"[core]\nb = true").expect("Unable to write config file");
let c_path = dir.path().join("c");
fs::write(c_path.as_path(), b"[core]\nc = true").expect("Unable to write config file");
let paths = vec![a_path.as_path(), b_path.as_path(), c_path.as_path()];
let config = GitConfig::from_paths(&paths).unwrap();
assert_eq!(
config.get_raw_value("core", None, "a"),
Ok(Cow::<[u8]>::Borrowed(b"true"))
);
assert_eq!(
config.get_raw_value("core", None, "b"),
Ok(Cow::<[u8]>::Borrowed(b"true"))
);
assert_eq!(
config.get_raw_value("core", None, "c"),
Ok(Cow::<[u8]>::Borrowed(b"true"))
);
assert_eq!(config.len(), 3);
}
#[test]
fn multiple_paths_multi_value() {
let dir = tempdir().unwrap();
let a_path = dir.path().join("a");
fs::write(a_path.as_path(), b"[core]\nkey = a").expect("Unable to write config file");
let b_path = dir.path().join("b");
fs::write(b_path.as_path(), b"[core]\nkey = b").expect("Unable to write config file");
let c_path = dir.path().join("c");
fs::write(c_path.as_path(), b"[core]\nkey = c").expect("Unable to write config file");
let paths = vec![a_path.as_path(), b_path.as_path(), c_path.as_path()];
let config = GitConfig::from_paths(&paths).unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "key").unwrap(),
vec![Cow::Borrowed(b"a"), Cow::Borrowed(b"b"), Cow::Borrowed(b"c")]
);
assert_eq!(config.len(), 3);
}
}
#[cfg(test)]
pub mod from_env_paths {}
#[cfg(test)]
mod from_env {
use std::env;
use serial_test::serial;
use super::{Cow, GitConfig, GitConfigFromEnvError};
struct Env {
altered_vars: Vec<&'static str>,
}
impl Env {
fn new() -> Self {
Env {
altered_vars: Vec::new(),
}
}
fn set(mut self, var: &'static str, value: &'static str) -> Self {
env::set_var(var, value);
self.altered_vars.push(var);
self
}
}
impl Drop for Env {
fn drop(&mut self) {
for var in &self.altered_vars {
env::remove_var(var);
}
}
}
#[test]
#[serial]
fn empty_without_relevant_environment() {
let config = GitConfig::from_env().unwrap();
assert!(config.is_none());
}
#[test]
#[serial]
fn empty_with_zero_count() {
let _env = Env::new().set("GIT_CONFIG_COUNT", "0");
let config = GitConfig::from_env().unwrap();
assert!(config.is_none());
}
#[test]
#[serial]
fn parse_error_with_invalid_count() {
let _env = Env::new().set("GIT_CONFIG_COUNT", "invalid");
let err = GitConfig::from_env().unwrap_err();
assert!(matches!(err, GitConfigFromEnvError::ParseError(_)));
}
#[test]
#[serial]
fn single_key_value_pair() {
let _env = Env::new()
.set("GIT_CONFIG_COUNT", "1")
.set("GIT_CONFIG_KEY_0", "core.key")
.set("GIT_CONFIG_VALUE_0", "value");
let config = GitConfig::from_env().unwrap().unwrap();
assert_eq!(
config.get_raw_value("core", None, "key"),
Ok(Cow::<[u8]>::Borrowed(b"value"))
);
assert_eq!(config.len(), 1);
}
#[test]
#[serial]
fn multiple_key_value_pairs() {
let _env = Env::new()
.set("GIT_CONFIG_COUNT", "3")
.set("GIT_CONFIG_KEY_0", "core.a")
.set("GIT_CONFIG_VALUE_0", "a")
.set("GIT_CONFIG_KEY_1", "core.b")
.set("GIT_CONFIG_VALUE_1", "b")
.set("GIT_CONFIG_KEY_2", "core.c")
.set("GIT_CONFIG_VALUE_2", "c");
let config = GitConfig::from_env().unwrap().unwrap();
assert_eq!(config.get_raw_value("core", None, "a"), Ok(Cow::<[u8]>::Borrowed(b"a")));
assert_eq!(config.get_raw_value("core", None, "b"), Ok(Cow::<[u8]>::Borrowed(b"b")));
assert_eq!(config.get_raw_value("core", None, "c"), Ok(Cow::<[u8]>::Borrowed(b"c")));
assert_eq!(config.len(), 3);
}
}
#[cfg(test)]
mod from_parser {
use super::{Cow, Event, GitConfig, HashMap, LookupTreeNode, SectionBody, SectionId, TryFrom};
use crate::{
parser::SectionHeaderName,
test_util::{name_event, newline_event, section_header, value_event},
};
#[test]
fn parse_empty() {
let config = GitConfig::try_from("").unwrap();
assert!(config.section_headers.is_empty());
assert_eq!(config.section_id_counter, 0);
assert!(config.section_lookup_tree.is_empty());
assert!(config.sections.is_empty());
assert!(config.section_order.is_empty());
}
#[test]
fn parse_single_section() {
let mut config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
let expected_separators = {
let mut map = HashMap::new();
map.insert(SectionId(0), section_header("core", None));
map
};
assert_eq!(config.section_headers, expected_separators);
assert_eq!(config.section_id_counter, 1);
let expected_lookup_tree = {
let mut tree = HashMap::new();
tree.insert(
SectionHeaderName(Cow::Borrowed("core")),
vec![LookupTreeNode::Terminal(vec![SectionId(0)])],
);
tree
};
assert_eq!(config.section_lookup_tree, expected_lookup_tree);
let expected_sections = {
let mut sections = HashMap::new();
sections.insert(
SectionId(0),
SectionBody::from(vec![
newline_event(),
name_event("a"),
Event::KeyValueSeparator,
value_event("b"),
newline_event(),
name_event("c"),
Event::KeyValueSeparator,
value_event("d"),
]),
);
sections
};
assert_eq!(config.sections, expected_sections);
assert_eq!(config.section_order.make_contiguous(), &[SectionId(0)]);
}
#[test]
fn parse_single_subsection() {
let mut config = GitConfig::try_from("[core.sub]\na=b\nc=d").unwrap();
let expected_separators = {
let mut map = HashMap::new();
map.insert(SectionId(0), section_header("core", (".", "sub")));
map
};
assert_eq!(config.section_headers, expected_separators);
assert_eq!(config.section_id_counter, 1);
let expected_lookup_tree = {
let mut tree = HashMap::new();
let mut inner_tree = HashMap::new();
inner_tree.insert(Cow::Borrowed("sub"), vec![SectionId(0)]);
tree.insert(
SectionHeaderName(Cow::Borrowed("core")),
vec![LookupTreeNode::NonTerminal(inner_tree)],
);
tree
};
assert_eq!(config.section_lookup_tree, expected_lookup_tree);
let expected_sections = {
let mut sections = HashMap::new();
sections.insert(
SectionId(0),
SectionBody::from(vec![
newline_event(),
name_event("a"),
Event::KeyValueSeparator,
value_event("b"),
newline_event(),
name_event("c"),
Event::KeyValueSeparator,
value_event("d"),
]),
);
sections
};
assert_eq!(config.sections, expected_sections);
assert_eq!(config.section_order.make_contiguous(), &[SectionId(0)]);
}
#[test]
fn parse_multiple_sections() {
let mut config = GitConfig::try_from("[core]\na=b\nc=d\n[other]e=f").unwrap();
let expected_separators = {
let mut map = HashMap::new();
map.insert(SectionId(0), section_header("core", None));
map.insert(SectionId(1), section_header("other", None));
map
};
assert_eq!(config.section_headers, expected_separators);
assert_eq!(config.section_id_counter, 2);
let expected_lookup_tree = {
let mut tree = HashMap::new();
tree.insert(
SectionHeaderName(Cow::Borrowed("core")),
vec![LookupTreeNode::Terminal(vec![SectionId(0)])],
);
tree.insert(
SectionHeaderName(Cow::Borrowed("other")),
vec![LookupTreeNode::Terminal(vec![SectionId(1)])],
);
tree
};
assert_eq!(config.section_lookup_tree, expected_lookup_tree);
let expected_sections = {
let mut sections = HashMap::new();
sections.insert(
SectionId(0),
SectionBody::from(vec![
newline_event(),
name_event("a"),
Event::KeyValueSeparator,
value_event("b"),
newline_event(),
name_event("c"),
Event::KeyValueSeparator,
value_event("d"),
newline_event(),
]),
);
sections.insert(
SectionId(1),
SectionBody::from(vec![name_event("e"), Event::KeyValueSeparator, value_event("f")]),
);
sections
};
assert_eq!(config.sections, expected_sections);
assert_eq!(config.section_order.make_contiguous(), &[SectionId(0), SectionId(1)]);
}
#[test]
fn parse_multiple_duplicate_sections() {
let mut config = GitConfig::try_from("[core]\na=b\nc=d\n[core]e=f").unwrap();
let expected_separators = {
let mut map = HashMap::new();
map.insert(SectionId(0), section_header("core", None));
map.insert(SectionId(1), section_header("core", None));
map
};
assert_eq!(config.section_headers, expected_separators);
assert_eq!(config.section_id_counter, 2);
let expected_lookup_tree = {
let mut tree = HashMap::new();
tree.insert(
SectionHeaderName(Cow::Borrowed("core")),
vec![LookupTreeNode::Terminal(vec![SectionId(0), SectionId(1)])],
);
tree
};
assert_eq!(config.section_lookup_tree, expected_lookup_tree);
let expected_sections = {
let mut sections = HashMap::new();
sections.insert(
SectionId(0),
SectionBody::from(vec![
newline_event(),
name_event("a"),
Event::KeyValueSeparator,
value_event("b"),
newline_event(),
name_event("c"),
Event::KeyValueSeparator,
value_event("d"),
newline_event(),
]),
);
sections.insert(
SectionId(1),
SectionBody::from(vec![name_event("e"), Event::KeyValueSeparator, value_event("f")]),
);
sections
};
assert_eq!(config.sections, expected_sections);
assert_eq!(config.section_order.make_contiguous(), &[SectionId(0), SectionId(1)]);
}
}
#[cfg(test)]
mod get_raw_value {
use super::{Cow, GitConfig, GitConfigError, TryFrom};
use crate::parser::SectionHeaderName;
#[test]
fn single_section() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(config.get_raw_value("core", None, "a"), Ok(Cow::<[u8]>::Borrowed(b"b")));
assert_eq!(config.get_raw_value("core", None, "c"), Ok(Cow::<[u8]>::Borrowed(b"d")));
}
#[test]
fn last_one_wins_respected_in_section() {
let config = GitConfig::try_from("[core]\na=b\na=d").unwrap();
assert_eq!(config.get_raw_value("core", None, "a"), Ok(Cow::<[u8]>::Borrowed(b"d")));
}
#[test]
fn last_one_wins_respected_across_section() {
let config = GitConfig::try_from("[core]\na=b\n[core]\na=d").unwrap();
assert_eq!(config.get_raw_value("core", None, "a"), Ok(Cow::<[u8]>::Borrowed(b"d")));
}
#[test]
fn section_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_value("foo", None, "a"),
Err(GitConfigError::SectionDoesNotExist(SectionHeaderName("foo".into())))
);
}
#[test]
fn subsection_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_value("core", Some("a"), "a"),
Err(GitConfigError::SubSectionDoesNotExist(Some("a")))
);
}
#[test]
fn key_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_value("core", None, "aaaaaa"),
Err(GitConfigError::KeyDoesNotExist)
);
}
#[test]
fn subsection_must_be_respected() {
let config = GitConfig::try_from("[core]a=b\n[core.a]a=c").unwrap();
assert_eq!(config.get_raw_value("core", None, "a"), Ok(Cow::<[u8]>::Borrowed(b"b")));
assert_eq!(
config.get_raw_value("core", Some("a"), "a"),
Ok(Cow::<[u8]>::Borrowed(b"c"))
);
}
}
#[cfg(test)]
mod get_value {
use std::error::Error;
use super::{Cow, GitConfig, TryFrom};
use crate::values::{Boolean, TrueVariant, Value};
#[test]
fn single_section() -> Result<(), Box<dyn Error>> {
let config = GitConfig::try_from("[core]\na=b\nc").unwrap();
let first_value: Value = config.value("core", None, "a")?;
let second_value: Boolean = config.value("core", None, "c")?;
assert_eq!(first_value, Value::Other(Cow::Borrowed(b"b")));
assert_eq!(second_value, Boolean::True(TrueVariant::Implicit));
Ok(())
}
#[test]
fn sections_by_name() {
let config = r#"
[core]
repositoryformatversion = 0
filemode = true
bare = false
logallrefupdates = true
[remote "origin"]
url = git@github.com:Byron/gitoxide.git
fetch = +refs/heads/*:refs/remotes/origin/*
"#;
let config = GitConfig::try_from(config).unwrap();
let value = config.value::<Value>("remote", Some("origin"), "url").unwrap();
assert_eq!(value, Value::Other(Cow::Borrowed(b"git@github.com:Byron/gitoxide.git")));
}
}
#[cfg(test)]
mod get_raw_multi_value {
use super::{Cow, GitConfig, GitConfigError, TryFrom};
use crate::parser::SectionHeaderName;
#[test]
fn single_value_is_identical_to_single_value_query() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
vec![config.get_raw_value("core", None, "a").unwrap()],
config.get_raw_multi_value("core", None, "a").unwrap()
);
}
#[test]
fn multi_value_in_section() {
let config = GitConfig::try_from("[core]\na=b\na=c").unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "a").unwrap(),
vec![Cow::Borrowed(b"b"), Cow::Borrowed(b"c")]
);
}
#[test]
fn multi_value_across_sections() {
let config = GitConfig::try_from("[core]\na=b\na=c\n[core]a=d").unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "a").unwrap(),
vec![Cow::Borrowed(b"b"), Cow::Borrowed(b"c"), Cow::Borrowed(b"d")]
);
}
#[test]
fn section_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_multi_value("foo", None, "a"),
Err(GitConfigError::SectionDoesNotExist(SectionHeaderName("foo".into())))
);
}
#[test]
fn subsection_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_multi_value("core", Some("a"), "a"),
Err(GitConfigError::SubSectionDoesNotExist(Some("a")))
);
}
#[test]
fn key_not_found() {
let config = GitConfig::try_from("[core]\na=b\nc=d").unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "aaaaaa"),
Err(GitConfigError::KeyDoesNotExist)
);
}
#[test]
fn subsection_must_be_respected() {
let config = GitConfig::try_from("[core]a=b\n[core.a]a=c").unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "a").unwrap(),
vec![Cow::Borrowed(b"b")]
);
assert_eq!(
config.get_raw_multi_value("core", Some("a"), "a").unwrap(),
vec![Cow::Borrowed(b"c")]
);
}
#[test]
fn non_relevant_subsection_is_ignored() {
let config = GitConfig::try_from("[core]\na=b\na=c\n[core]a=d\n[core]g=g").unwrap();
assert_eq!(
config.get_raw_multi_value("core", None, "a").unwrap(),
vec![Cow::Borrowed(b"b"), Cow::Borrowed(b"c"), Cow::Borrowed(b"d")]
);
}
}
#[cfg(test)]
mod display {
use super::{GitConfig, TryFrom};
#[test]
fn can_reconstruct_empty_config() {
let config = r#"
"#;
assert_eq!(GitConfig::try_from(config).unwrap().to_string(), config);
}
#[test]
fn can_reconstruct_non_empty_config() {
let config = r#"
[user]
email = code@eddie.sh
[core]
autocrlf = input
[push]
default = simple
[commit]
gpgsign = true
[gpg]
program = gpg
[url "ssh://git@github.com/"]
insteadOf = "github://"
[url "ssh://git@git.eddie.sh/edward/"]
insteadOf = "gitea://"
[pull]
ff = only
[init]
defaultBranch = master
"#;
assert_eq!(GitConfig::try_from(config).unwrap().to_string(), config);
}
#[test]
fn can_reconstruct_configs_with_implicits() {
let config = r#"
[user]
email
name
[core]
autocrlf
[push]
default
[commit]
gpgsign
"#;
assert_eq!(GitConfig::try_from(config).unwrap().to_string(), config);
}
#[test]
fn can_reconstruct_configs_without_whitespace_in_middle() {
let config = r#"
[core]
autocrlf=input
[push]
default=simple
[commit]
gpgsign=true
[pull]
ff = only
[init]
defaultBranch = master
"#;
assert_eq!(GitConfig::try_from(config).unwrap().to_string(), config);
}
}