mit_commit_message_lints/external/
in_memory.rs

1use std::{collections::BTreeMap, convert::TryFrom};
2
3use glob::Pattern;
4use miette::{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 const 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 compiled_glob = glob.map(Pattern::new).transpose().into_diagnostic()?;
30
31        Ok(self
32            .store
33            .keys()
34            .filter(|key| {
35                compiled_glob
36                    .as_ref()
37                    .is_none_or(|pattern| pattern.matches(key))
38            })
39            .cloned()
40            .collect())
41    }
42
43    fn get_bool(&self, name: &str) -> Result<Option<bool>> {
44        match self.store.get(name) {
45            None => Ok(None),
46            Some(raw_value) => Ok(Some(raw_value.parse().into_diagnostic()?)),
47        }
48    }
49
50    fn get_str(&self, name: &str) -> Result<Option<&str>> {
51        Ok(self.store.get(name).map(String::as_str))
52    }
53
54    fn get_i64(&self, name: &str) -> Result<Option<i64>> {
55        match self.store.get(name) {
56            None => Ok(None),
57            Some(raw_value) => Ok(Some(raw_value.parse().into_diagnostic()?)),
58        }
59    }
60
61    fn set_str(&mut self, name: &str, value: &str) -> Result<()> {
62        self.store.insert(name.into(), value.into());
63        Ok(())
64    }
65
66    fn set_i64(&mut self, name: &str, value: i64) -> Result<()> {
67        self.store.insert(name.into(), format!("{value}"));
68        Ok(())
69    }
70
71    fn remove(&mut self, name: &str) -> Result<()> {
72        self.store.remove(name);
73        Ok(())
74    }
75
76    fn state(&self) -> Option<RepoState> {
77        None
78    }
79}
80
81impl TryFrom<&'_ InMemory<'_>> for Authors<'_> {
82    type Error = Report;
83
84    fn try_from(vcs: &'_ InMemory<'_>) -> Result<Self, Self::Error> {
85        let raw_entries: BTreeMap<String, BTreeMap<String, String>> = vcs
86            .entries(Some("mit.author.config.*"))?
87            .iter()
88            .map(|key| (key, key.trim_start_matches("mit.author.config.")))
89            .map(|(key, parts)| (key, parts.split_terminator('.').collect::<Vec<_>>()))
90            .try_fold::<_, _, Result<_, Self::Error>>(
91                BTreeMap::new(),
92                |mut acc, (key, fragments)| {
93                    let mut fragment_iterator = fragments.iter();
94                    let initial = String::from(
95                        *fragment_iterator
96                            .next()
97                            .ok_or_else(|| miette!("Malformed config key: {key}"))?,
98                    );
99                    let part = String::from(
100                        *fragment_iterator
101                            .next()
102                            .ok_or_else(|| miette!("Malformed config key: {key}"))?,
103                    );
104
105                    let mut existing: BTreeMap<String, String> =
106                        acc.get(&initial).cloned().unwrap_or_default();
107                    existing.insert(part, String::from(vcs.get_str(key)?.unwrap()));
108
109                    acc.insert(initial, existing);
110                    Ok(acc)
111                },
112            )?;
113
114        Ok(Self::new(
115            raw_entries
116                .iter()
117                .filter_map(|(key, cfg)| {
118                    let name = cfg.get("name").cloned();
119                    let email = cfg.get("email").cloned();
120                    let signingkey: Option<String> = cfg.get("signingkey").cloned();
121
122                    match (name, email, signingkey) {
123                        (Some(name), Some(email), None) => {
124                            Some((key, Author::new(name.into(), email.into(), None)))
125                        }
126                        (Some(name), Some(email), Some(signingkey)) => Some((
127                            key,
128                            Author::new(name.into(), email.into(), Some(signingkey.into())),
129                        )),
130                        _ => None,
131                    }
132                })
133                .map(|(key, value): (&String, Author<'_>)| (key.clone(), value))
134                .collect(),
135        ))
136    }
137}
mit_commit_message_lints/external/in_memory.rs

mit_commit_message_lints/external/
in_memory.rs
