gix_diff/tree/
recorder.rsuse gix_hash::ObjectId;
use gix_object::{
bstr::{BStr, BString, ByteSlice, ByteVec},
tree,
};
use crate::tree::visit::Relation;
use crate::tree::{visit, Recorder, Visit};
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum Location {
Path,
FileName,
}
#[derive(Clone, Debug, PartialEq, Eq)]
#[allow(missing_docs)]
pub enum Change {
Addition {
entry_mode: tree::EntryMode,
oid: ObjectId,
path: BString,
relation: Option<Relation>,
},
Deletion {
entry_mode: tree::EntryMode,
oid: ObjectId,
path: BString,
relation: Option<Relation>,
},
Modification {
previous_entry_mode: tree::EntryMode,
previous_oid: ObjectId,
entry_mode: tree::EntryMode,
oid: ObjectId,
path: BString,
},
}
impl Default for Recorder {
fn default() -> Self {
Recorder {
path_deque: Default::default(),
path: Default::default(),
location: Some(Location::Path),
records: vec![],
}
}
}
impl Recorder {
pub fn track_location(mut self, location: Option<Location>) -> Self {
self.location = location;
self
}
}
impl Recorder {
pub fn path_clone(&self) -> BString {
self.path.clone()
}
pub fn path(&self) -> &BStr {
self.path.as_ref()
}
}
impl Recorder {
fn pop_element(&mut self) {
if let Some(pos) = self.path.rfind_byte(b'/') {
self.path.resize(pos, 0);
} else {
self.path.clear();
}
}
fn push_element(&mut self, name: &BStr) {
if !self.path.is_empty() {
self.path.push(b'/');
}
self.path.push_str(name);
}
}
impl Visit for Recorder {
fn pop_front_tracked_path_and_set_current(&mut self) {
if let Some(Location::Path) = self.location {
self.path = self.path_deque.pop_front().expect("every parent is set only once");
}
}
fn push_back_tracked_path_component(&mut self, component: &BStr) {
match self.location {
None => {}
Some(Location::Path) => {
self.push_element(component);
self.path_deque.push_back(self.path.clone());
}
Some(Location::FileName) => {
self.path.clear();
self.path.extend_from_slice(component);
}
}
}
fn push_path_component(&mut self, component: &BStr) {
match self.location {
None => {}
Some(Location::Path) => {
self.push_element(component);
}
Some(Location::FileName) => {
self.path.clear();
self.path.extend_from_slice(component);
}
}
}
fn pop_path_component(&mut self) {
if let Some(Location::Path) = self.location {
self.pop_element();
}
}
fn visit(&mut self, change: visit::Change) -> visit::Action {
use visit::Change::*;
self.records.push(match change {
Deletion {
entry_mode,
oid,
relation,
} => Change::Deletion {
entry_mode,
oid,
path: self.path_clone(),
relation,
},
Addition {
entry_mode,
oid,
relation,
} => Change::Addition {
entry_mode,
oid,
path: self.path_clone(),
relation,
},
Modification {
previous_entry_mode,
previous_oid,
entry_mode,
oid,
} => Change::Modification {
previous_entry_mode,
previous_oid,
entry_mode,
oid,
path: self.path_clone(),
},
});
visit::Action::Continue
}
}