diff --git a/rust/hg-core/src/matchers.rs b/rust/hg-core/src/matchers.rs --- a/rust/hg-core/src/matchers.rs +++ b/rust/hg-core/src/matchers.rs @@ -7,8 +7,10 @@ //! Structs and types for matching files and directories. -use crate::utils::hg_path::{HgPath, HgPathBuf}; +use crate::utils::hg_path::HgPath; +use crate::DirsMultiset; use std::collections::HashSet; +use std::iter::FromIterator; pub enum VisitChildrenSet<'a> { /// Don't visit anything @@ -114,3 +116,62 @@ false } } + +/// Matches the input files exactly. They are interpreted as paths, not +/// patterns. +/// +///``` +/// use hg::{ matchers::{Matcher, FileMatcher}, utils::hg_path::HgPath }; +/// +/// let files = [HgPath::new(b"a.txt"), HgPath::new(br"re:.*\.c$")]; +/// let matcher = FileMatcher::new(&files); +/// +/// assert_eq!(true, matcher.matches(HgPath::new(b"a.txt"))); +/// assert_eq!(false, matcher.matches(HgPath::new(b"b.txt"))); +/// assert_eq!(false, matcher.matches(HgPath::new(b"main.c"))); +/// assert_eq!(true, matcher.matches(HgPath::new(br"re:.*\.c$"))); +/// ``` +#[derive(Debug)] +pub struct FileMatcher<'a> { + files: HashSet<&'a HgPath>, + dirs: DirsMultiset, +} + +impl<'a> FileMatcher<'a> { + pub fn new(files: &'a [impl AsRef]) -> Self { + Self { + files: HashSet::from_iter(files.iter().map(|f| f.as_ref())), + dirs: DirsMultiset::from_manifest(files), + } + } + fn inner_matches(&self, filename: impl AsRef) -> bool { + self.files.contains(filename.as_ref()) + } +} + +impl<'a> Matcher for FileMatcher<'a> { + fn file_set(&self) -> Option<&HashSet<&HgPath>> { + Some(&self.files) + } + fn exact_match(&self, filename: impl AsRef) -> bool { + self.inner_matches(filename) + } + fn matches(&self, filename: impl AsRef) -> bool { + self.inner_matches(filename) + } + fn visit_children_set( + &self, + _directory: impl AsRef, + ) -> VisitChildrenSet { + // TODO implement once we have `status.traverse` + // This is useless until unknown files are taken into account + // Which will not need to happen before the `PatternMatcher`. + unimplemented!() + } + fn matches_everything(&self) -> bool { + false + } + fn is_exact(&self) -> bool { + true + } +}