mit_commit_message_lints/external/
in_memory.rs

1use std::{collections::BTreeMap, convert::TryFrom};
2
3use glob::Pattern;
4use miette::{IntoDiagnostic, Report, Result};
5
6use crate::{
7    external::{vcs::RepoState, Vcs},
8    mit::{Author, Authors},
9};
10
11/// An in memory vcs implementation
12///
13/// Mostly used for testing
14#[derive(Debug)]
15pub struct InMemory<'a> {
16    store: &'a mut BTreeMap<String, String>,
17}
18
19impl InMemory<'_> {
20    /// Create a new in memory vcs
21    #[must_use]
22    pub fn new(store: &mut BTreeMap<String, String>) -> InMemory<'_> {
23        InMemory { store }
24    }
25}
26
27impl Vcs for InMemory<'_> {
28    fn entries(&self, glob: Option<&str>) -> Result<Vec<String>> {
29        let mut keys: Vec<String> = self.store.keys().map(String::from).collect();
30
31        if let Some(pattern) = glob {
32            let compiled_glob = glob::Pattern::new(pattern).into_diagnostic()?;
33
34            keys.retain(|value| Pattern::matches(&compiled_glob, value));
35        }
36
37        Ok(keys)
38    }
39
40    fn get_bool(&self, name: &str) -> Result<Option<bool>> {
41        match self.store.get(name) {
42            None => Ok(None),
43            Some(raw_value) => Ok(Some(raw_value.parse().into_diagnostic()?)),
44        }
45    }
46
47    fn get_str(&self, name: &str) -> Result<Option<&str>> {
48        Ok(self.store.get(name).map(String::as_str))
49    }
50
51    fn get_i64(&self, name: &str) -> Result<Option<i64>> {
52        match self.store.get(name) {
53            None => Ok(None),
54            Some(raw_value) => Ok(Some(raw_value.parse().into_diagnostic()?)),
55        }
56    }
57
58    fn set_str(&mut self, name: &str, value: &str) -> Result<()> {
59        self.store.insert(name.into(), value.into());
60        Ok(())
61    }
62
63    fn set_i64(&mut self, name: &str, value: i64) -> Result<()> {
64        self.store.insert(name.into(), format!("{value}"));
65        Ok(())
66    }
67
68    fn remove(&mut self, name: &str) -> Result<()> {
69        self.store.remove(name);
70        Ok(())
71    }
72
73    fn state(&self) -> Option<RepoState> {
74        None
75    }
76}
77
78impl TryFrom<&'_ InMemory<'_>> for Authors<'_> {
79    type Error = Report;
80
81    fn try_from(vcs: &'_ InMemory<'_>) -> Result<Self, Self::Error> {
82        let raw_entries: BTreeMap<String, BTreeMap<String, String>> = vcs
83            .entries(Some("mit.author.config.*"))?
84            .iter()
85            .map(|key| (key, key.trim_start_matches("mit.author.config.")))
86            .map(|(key, parts)| (key, parts.split_terminator('.').collect::<Vec<_>>()))
87            .try_fold::<_, _, Result<_, Self::Error>>(
88                BTreeMap::new(),
89                |mut acc, (key, fragments)| {
90                    let mut fragment_iterator = fragments.iter();
91                    let initial = String::from(*fragment_iterator.next().unwrap());
92                    let part = String::from(*fragment_iterator.next().unwrap());
93
94                    let mut existing: BTreeMap<String, String> =
95                        acc.get(&initial).cloned().unwrap_or_default();
96                    existing.insert(part, String::from(vcs.get_str(key)?.unwrap()));
97
98                    acc.insert(initial, existing);
99                    Ok(acc)
100                },
101            )?;
102
103        Ok(Self::new(
104            raw_entries
105                .iter()
106                .filter_map(|(key, cfg)| {
107                    let name = cfg.get("name").cloned();
108                    let email = cfg.get("email").cloned();
109                    let signingkey: Option<String> = cfg.get("signingkey").cloned();
110
111                    match (name, email, signingkey) {
112                        (Some(name), Some(email), None) => {
113                            Some((key, Author::new(name.into(), email.into(), None)))
114                        }
115                        (Some(name), Some(email), Some(signingkey)) => Some((
116                            key,
117                            Author::new(name.into(), email.into(), Some(signingkey.into())),
118                        )),
119                        _ => None,
120                    }
121                })
122                .fold(
123                    BTreeMap::new(),
124                    |mut acc: BTreeMap<String, Author<'_>>, (key, value): (&String, Author<'_>)| {
125                        acc.insert(key.clone(), value);
126                        acc
127                    },
128                ),
129        ))
130    }
131}
mit_commit_message_lints/external/in_memory.rs

mit_commit_message_lints/external/
in_memory.rs
