mirrors/jj
1
0
Fork 1
mirror of https://github.com/martinvonz/jj.git synced 2025-01-18 18:27:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

trees: pass a matcher to the entry iterator

Browse source
This commit is contained in:
Martin von Zweigbergk 2021-11-14 20:29:18 -08:00
parent f2d5aa188d
commit 41860692e9
1 changed files with 22 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
lib/src/tree.rs
View file

@ -100,8 +100,15 @@ impl Tree {
self.data.entries() self.data.entries()
} }
pub fn entries(&self) -> TreeEntriesIterator { pub fn entries(&self) -> TreeEntriesIterator<'static> {
TreeEntriesIterator::new(self.clone()) TreeEntriesIterator::new(self.clone(), &EverythingMatcher)
}
pub fn entries_matching<'matcher>(
&self,
matcher: &'matcher dyn Matcher,
) -> TreeEntriesIterator<'matcher> {
TreeEntriesIterator::new(self.clone(), matcher)
} }
pub fn entry(&self, basename: &RepoPathComponent) -> Option<TreeEntry> { pub fn entry(&self, basename: &RepoPathComponent) -> Option<TreeEntry> {
@ -206,15 +213,17 @@ impl Tree {
} }
} }
pub struct TreeEntriesIterator { pub struct TreeEntriesIterator<'matcher> {
tree: Pin<Box<Tree>>, tree: Pin<Box<Tree>>,
entry_iterator: TreeEntriesNonRecursiveIterator<'static>, entry_iterator: TreeEntriesNonRecursiveIterator<'static>,
subdir_iterator: Option<Box<TreeEntriesIterator>>, subdir_iterator: Option<Box<TreeEntriesIterator<'matcher>>>,
matcher: &'matcher dyn Matcher,
} }
impl TreeEntriesIterator { impl<'matcher> TreeEntriesIterator<'matcher> {
fn new(tree: Tree) -> Self { fn new(tree: Tree, matcher: &'matcher dyn Matcher) -> Self {
let tree = Box::pin(tree); let tree = Box::pin(tree);
// TODO: Restrict walk according to Matcher::visit()
let entry_iterator = tree.entries_non_recursive(); let entry_iterator = tree.entries_non_recursive();
let entry_iterator: TreeEntriesNonRecursiveIterator<'static> = let entry_iterator: TreeEntriesNonRecursiveIterator<'static> =
unsafe { std::mem::transmute(entry_iterator) }; unsafe { std::mem::transmute(entry_iterator) };
@ -222,11 +231,12 @@ impl TreeEntriesIterator {
tree, tree,
entry_iterator, entry_iterator,
subdir_iterator: None, subdir_iterator: None,
matcher,
} }
} }
} }
impl Iterator for TreeEntriesIterator { impl Iterator for TreeEntriesIterator<'_> {
type Item = (RepoPath, TreeValue); type Item = (RepoPath, TreeValue);
fn next(&mut self) -> Option<Self::Item> { fn next(&mut self) -> Option<Self::Item> {
@ -242,10 +252,14 @@ impl Iterator for TreeEntriesIterator {
match entry.value() { match entry.value() {
TreeValue::Tree(id) => { TreeValue::Tree(id) => {
let subtree = self.tree.known_sub_tree(entry.name(), id); let subtree = self.tree.known_sub_tree(entry.name(), id);
self.subdir_iterator = Some(Box::new(TreeEntriesIterator::new(subtree))); self.subdir_iterator =
Some(Box::new(TreeEntriesIterator::new(subtree, self.matcher)));
} }
other => { other => {
let path = self.tree.dir().join(entry.name()); let path = self.tree.dir().join(entry.name());
if !self.matcher.matches(&path) {
continue;
}
return Some((path, other.clone())); return Some((path, other.clone()));
} }
}; };