use std::borrow::Cow;
use std::iter::FromIterator;
use std::iter::IntoIterator;
use regex::Regex;
#[derive(Debug, Default, Clone)]
#[cfg_attr(docsrs, doc(cfg(feature = "filters")))]
pub struct Filters {
rules: Vec<(Regex, String)>,
}
impl<'a, I> From<I> for Filters
where
I: IntoIterator<Item = (&'a str, &'a str)>,
{
fn from(value: I) -> Self {
Self::from_iter(value)
}
}
impl<'a> FromIterator<(&'a str, &'a str)> for Filters {
fn from_iter<I: IntoIterator<Item = (&'a str, &'a str)>>(iter: I) -> Self {
let mut rv = Filters::default();
for (regex, replacement) in iter {
rv.add(regex, replacement);
}
rv
}
}
impl Filters {
pub(crate) fn add<S: Into<String>>(&mut self, regex: &str, replacement: S) {
self.rules.push((
Regex::new(regex).expect("invalid regex for snapshot filter rule"),
replacement.into(),
));
}
pub(crate) fn clear(&mut self) {
self.rules.clear();
}
pub(crate) fn apply_to<'s>(&self, s: &'s str) -> Cow<'s, str> {
let mut rv = Cow::Borrowed(s);
for (regex, replacement) in &self.rules {
match regex.replace_all(&rv, replacement) {
Cow::Borrowed(_) => continue,
Cow::Owned(value) => rv = Cow::Owned(value),
};
}
rv
}
}
#[test]
fn test_filters() {
let mut filters = Filters::default();
filters.add("\\bhello\\b", "[NAME]");
filters.add("(a)", "[$1]");
assert_eq!(
filters.apply_to("hellohello hello abc"),
"hellohello [NAME] [a]bc"
);
}
#[test]
fn test_static_str_array_conversion() {
let arr: [(&'static str, &'static str); 2] = [("a1", "b1"), ("a2", "b2")];
let _ = Filters::from_iter(arr);
}
#[test]
fn test_vec_str_conversion() {
let vec: Vec<(&str, &str)> = Vec::from([("a1", "b1"), ("a2", "b2")]);
let _ = Filters::from(vec);
}