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

merged_tree: make ConflictIterator a struct

Browse source
This commit is contained in:
Martin von Zweigbergk 2024-07-24 01:11:43 -07:00 committed by Martin von Zweigbergk
parent 109391f9c7
commit 7596935285
1 changed files with 18 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
lib/src/merged_tree.rs
View file

@ -556,18 +556,15 @@ impl From<&Merge<Tree>> for ConflictsDirItem {
}
}
// TODO: replace by a struct
enum ConflictIterator {
Merge {
store: Arc<Store>,
stack: Vec<ConflictsDirItem>,
},
struct ConflictIterator {
store: Arc<Store>,
stack: Vec<ConflictsDirItem>,
}
impl ConflictIterator {
fn new(tree: &MergedTree) -> Self {
match tree {
MergedTree::Merge(trees) => ConflictIterator::Merge {
MergedTree::Merge(trees) => ConflictIterator {
store: tree.store().clone(),
stack: vec![ConflictsDirItem::from(trees)],
},
@ -579,27 +576,23 @@ impl Iterator for ConflictIterator {
type Item = (RepoPathBuf, MergedTreeValue);
fn next(&mut self) -> Option<Self::Item> {
match self {
ConflictIterator::Merge { store, stack } => {
while let Some(top) = stack.last_mut() {
if let Some((path, tree_values)) = top.entries.pop() {
// TODO: propagate errors
if let Some(trees) = tree_values.to_tree_merge(store, &path).unwrap() {
// If all sides are trees or missing, descend into the merged tree
stack.push(ConflictsDirItem::from(&trees));
} else {
// Otherwise this is a conflict between files, trees, etc. If they could
// be automatically resolved, they should have been when the top-level
// tree conflict was written, so we assume that they can't be.
return Some((path, tree_values));
}
} else {
stack.pop();
}
while let Some(top) = self.stack.last_mut() {
if let Some((path, tree_values)) = top.entries.pop() {
// TODO: propagate errors
if let Some(trees) = tree_values.to_tree_merge(&self.store, &path).unwrap() {
// If all sides are trees or missing, descend into the merged tree
self.stack.push(ConflictsDirItem::from(&trees));
} else {
// Otherwise this is a conflict between files, trees, etc. If they could
// be automatically resolved, they should have been when the top-level
// tree conflict was written, so we assume that they can't be.
return Some((path, tree_values));
}
None
} else {
self.stack.pop();
}
}
None
}
}