entry.rs
400 lines
| 11.6 KiB
| application/rls-services+xml
|
RustLexer
Simon Sapin
|
r48830 | use crate::errors::HgError; | ||
Simon Sapin
|
r48856 | use bitflags::bitflags; | ||
Simon Sapin
|
r48830 | use std::convert::TryFrom; | ||
#[derive(Copy, Clone, Debug, Eq, PartialEq)] | ||||
pub enum EntryState { | ||||
Normal, | ||||
Added, | ||||
Removed, | ||||
Merged, | ||||
} | ||||
/// The C implementation uses all signed types. This will be an issue | ||||
/// either when 4GB+ source files are commonplace or in 2038, whichever | ||||
/// comes first. | ||||
#[derive(Debug, PartialEq, Copy, Clone)] | ||||
pub struct DirstateEntry { | ||||
Simon Sapin
|
r48856 | flags: Flags, | ||
Simon Sapin
|
r48834 | mode: i32, | ||
size: i32, | ||||
mtime: i32, | ||||
Simon Sapin
|
r48830 | } | ||
Simon Sapin
|
r48856 | bitflags! { | ||
Simon Sapin
|
r48857 | pub struct Flags: u8 { | ||
Simon Sapin
|
r48856 | const WDIR_TRACKED = 1 << 0; | ||
const P1_TRACKED = 1 << 1; | ||||
const P2_TRACKED = 1 << 2; | ||||
const POSSIBLY_DIRTY = 1 << 3; | ||||
const MERGED = 1 << 4; | ||||
const CLEAN_P1 = 1 << 5; | ||||
const CLEAN_P2 = 1 << 6; | ||||
const ENTRYLESS_TREE_NODE = 1 << 7; | ||||
} | ||||
} | ||||
Simon Sapin
|
r48830 | pub const V1_RANGEMASK: i32 = 0x7FFFFFFF; | ||
pub const MTIME_UNSET: i32 = -1; | ||||
/// A `DirstateEntry` with a size of `-2` means that it was merged from the | ||||
/// other parent. This allows revert to pick the right status back during a | ||||
/// merge. | ||||
pub const SIZE_FROM_OTHER_PARENT: i32 = -2; | ||||
/// A special value used for internal representation of special case in | ||||
/// dirstate v1 format. | ||||
pub const SIZE_NON_NORMAL: i32 = -1; | ||||
impl DirstateEntry { | ||||
Simon Sapin
|
r48857 | pub fn new( | ||
flags: Flags, | ||||
mode_size_mtime: Option<(i32, i32, i32)>, | ||||
) -> Self { | ||||
let (mode, size, mtime) = | ||||
mode_size_mtime.unwrap_or((0, SIZE_NON_NORMAL, MTIME_UNSET)); | ||||
Self { | ||||
flags, | ||||
mode, | ||||
size, | ||||
mtime, | ||||
} | ||||
} | ||||
Simon Sapin
|
r48834 | pub fn from_v1_data( | ||
state: EntryState, | ||||
mode: i32, | ||||
size: i32, | ||||
mtime: i32, | ||||
) -> Self { | ||||
Simon Sapin
|
r48856 | match state { | ||
EntryState::Normal => { | ||||
if size == SIZE_FROM_OTHER_PARENT { | ||||
Self::new_from_p2() | ||||
} else if size == SIZE_NON_NORMAL { | ||||
Self::new_possibly_dirty() | ||||
} else if mtime == MTIME_UNSET { | ||||
Self { | ||||
flags: Flags::WDIR_TRACKED | ||||
| Flags::P1_TRACKED | ||||
| Flags::POSSIBLY_DIRTY, | ||||
mode, | ||||
size, | ||||
mtime: 0, | ||||
} | ||||
} else { | ||||
Simon Sapin
|
r48865 | Self::new_normal(mode, size, mtime) | ||
Simon Sapin
|
r48856 | } | ||
} | ||||
EntryState::Added => Self::new_added(), | ||||
EntryState::Removed => Self { | ||||
flags: if size == SIZE_NON_NORMAL { | ||||
Flags::P1_TRACKED // might not be true because of rename ? | ||||
| Flags::P2_TRACKED // might not be true because of rename ? | ||||
| Flags::MERGED | ||||
} else if size == SIZE_FROM_OTHER_PARENT { | ||||
// We don’t know if P1_TRACKED should be set (file history) | ||||
Flags::P2_TRACKED | Flags::CLEAN_P2 | ||||
} else { | ||||
Flags::P1_TRACKED | ||||
}, | ||||
mode: 0, | ||||
size: 0, | ||||
mtime: 0, | ||||
}, | ||||
EntryState::Merged => Self::new_merged(), | ||||
} | ||||
} | ||||
Simon Sapin
|
r48865 | pub fn new_from_p2() -> Self { | ||
Simon Sapin
|
r48834 | Self { | ||
Simon Sapin
|
r48856 | // might be missing P1_TRACKED | ||
flags: Flags::WDIR_TRACKED | Flags::P2_TRACKED | Flags::CLEAN_P2, | ||||
mode: 0, | ||||
size: SIZE_FROM_OTHER_PARENT, | ||||
mtime: MTIME_UNSET, | ||||
} | ||||
} | ||||
Simon Sapin
|
r48865 | pub fn new_possibly_dirty() -> Self { | ||
Simon Sapin
|
r48856 | Self { | ||
flags: Flags::WDIR_TRACKED | ||||
| Flags::P1_TRACKED | ||||
| Flags::POSSIBLY_DIRTY, | ||||
mode: 0, | ||||
size: SIZE_NON_NORMAL, | ||||
mtime: MTIME_UNSET, | ||||
} | ||||
} | ||||
Simon Sapin
|
r48865 | pub fn new_added() -> Self { | ||
Simon Sapin
|
r48856 | Self { | ||
flags: Flags::WDIR_TRACKED, | ||||
mode: 0, | ||||
size: SIZE_NON_NORMAL, | ||||
mtime: MTIME_UNSET, | ||||
} | ||||
} | ||||
Simon Sapin
|
r48865 | pub fn new_merged() -> Self { | ||
Simon Sapin
|
r48856 | Self { | ||
flags: Flags::WDIR_TRACKED | ||||
| Flags::P1_TRACKED // might not be true because of rename ? | ||||
| Flags::P2_TRACKED // might not be true because of rename ? | ||||
| Flags::MERGED, | ||||
mode: 0, | ||||
size: SIZE_NON_NORMAL, | ||||
mtime: MTIME_UNSET, | ||||
Simon Sapin
|
r48834 | } | ||
} | ||||
Simon Sapin
|
r48865 | pub fn new_normal(mode: i32, size: i32, mtime: i32) -> Self { | ||
Self { | ||||
flags: Flags::WDIR_TRACKED | Flags::P1_TRACKED, | ||||
mode, | ||||
size, | ||||
mtime, | ||||
} | ||||
} | ||||
Simon Sapin
|
r48834 | /// Creates a new entry in "removed" state. | ||
/// | ||||
/// `size` is expected to be zero, `SIZE_NON_NORMAL`, or | ||||
/// `SIZE_FROM_OTHER_PARENT` | ||||
pub fn new_removed(size: i32) -> Self { | ||||
Simon Sapin
|
r48856 | Self::from_v1_data(EntryState::Removed, 0, size, 0) | ||
Simon Sapin
|
r48834 | } | ||
Simon Sapin
|
r48857 | pub fn tracked(&self) -> bool { | ||
self.flags.contains(Flags::WDIR_TRACKED) | ||||
} | ||||
Simon Sapin
|
r48856 | fn tracked_in_any_parent(&self) -> bool { | ||
self.flags.intersects(Flags::P1_TRACKED | Flags::P2_TRACKED) | ||||
} | ||||
Simon Sapin
|
r48857 | pub fn removed(&self) -> bool { | ||
Simon Sapin
|
r48856 | self.tracked_in_any_parent() | ||
&& !self.flags.contains(Flags::WDIR_TRACKED) | ||||
} | ||||
Simon Sapin
|
r48857 | pub fn merged(&self) -> bool { | ||
Simon Sapin
|
r48856 | self.flags.contains(Flags::WDIR_TRACKED | Flags::MERGED) | ||
} | ||||
Simon Sapin
|
r48857 | pub fn added(&self) -> bool { | ||
Simon Sapin
|
r48856 | self.flags.contains(Flags::WDIR_TRACKED) | ||
&& !self.tracked_in_any_parent() | ||||
} | ||||
Simon Sapin
|
r48857 | pub fn from_p2(&self) -> bool { | ||
Simon Sapin
|
r48856 | self.flags.contains(Flags::WDIR_TRACKED | Flags::CLEAN_P2) | ||
Simon Sapin
|
r48834 | } | ||
r48898 | pub fn maybe_clean(&self) -> bool { | |||
if !self.flags.contains(Flags::WDIR_TRACKED) { | ||||
false | ||||
} else if self.added() { | ||||
false | ||||
} else if self.flags.contains(Flags::MERGED) { | ||||
false | ||||
} else if self.flags.contains(Flags::CLEAN_P2) { | ||||
false | ||||
} else { | ||||
true | ||||
} | ||||
} | ||||
r48899 | pub fn any_tracked(&self) -> bool { | |||
self.flags.intersects( | ||||
Flags::WDIR_TRACKED | Flags::P1_TRACKED | Flags::P2_TRACKED, | ||||
) | ||||
} | ||||
Simon Sapin
|
r48834 | pub fn state(&self) -> EntryState { | ||
Simon Sapin
|
r48856 | if self.removed() { | ||
EntryState::Removed | ||||
} else if self.merged() { | ||||
EntryState::Merged | ||||
} else if self.added() { | ||||
EntryState::Added | ||||
} else { | ||||
EntryState::Normal | ||||
} | ||||
Simon Sapin
|
r48834 | } | ||
pub fn mode(&self) -> i32 { | ||||
self.mode | ||||
} | ||||
pub fn size(&self) -> i32 { | ||||
r48878 | if self.removed() && self.flags.contains(Flags::MERGED) { | |||
Simon Sapin
|
r48856 | SIZE_NON_NORMAL | ||
r48880 | } else if self.removed() && self.flags.contains(Flags::CLEAN_P2) { | |||
Simon Sapin
|
r48856 | SIZE_FROM_OTHER_PARENT | ||
} else if self.removed() { | ||||
0 | ||||
} else if self.merged() { | ||||
SIZE_FROM_OTHER_PARENT | ||||
} else if self.added() { | ||||
SIZE_NON_NORMAL | ||||
} else if self.from_p2() { | ||||
SIZE_FROM_OTHER_PARENT | ||||
} else if self.flags.contains(Flags::POSSIBLY_DIRTY) { | ||||
self.size // TODO: SIZE_NON_NORMAL ? | ||||
} else { | ||||
self.size | ||||
} | ||||
Simon Sapin
|
r48834 | } | ||
pub fn mtime(&self) -> i32 { | ||||
Simon Sapin
|
r48856 | if self.removed() { | ||
0 | ||||
} else if self.flags.contains(Flags::POSSIBLY_DIRTY) { | ||||
MTIME_UNSET | ||||
} else if self.merged() { | ||||
MTIME_UNSET | ||||
} else if self.added() { | ||||
MTIME_UNSET | ||||
} else if self.from_p2() { | ||||
MTIME_UNSET | ||||
} else { | ||||
self.mtime | ||||
} | ||||
Simon Sapin
|
r48834 | } | ||
Simon Sapin
|
r48857 | pub fn set_possibly_dirty(&mut self) { | ||
self.flags.insert(Flags::POSSIBLY_DIRTY) | ||||
} | ||||
pub fn set_clean(&mut self, mode: i32, size: i32, mtime: i32) { | ||||
self.flags.insert(Flags::WDIR_TRACKED | Flags::P1_TRACKED); | ||||
self.flags.remove( | ||||
Flags::P2_TRACKED // This might be wrong | ||||
| Flags::MERGED | ||||
| Flags::CLEAN_P2 | ||||
| Flags::POSSIBLY_DIRTY, | ||||
); | ||||
self.mode = mode; | ||||
self.size = size; | ||||
self.mtime = mtime; | ||||
} | ||||
pub fn set_tracked(&mut self) { | ||||
self.flags | ||||
.insert(Flags::WDIR_TRACKED | Flags::POSSIBLY_DIRTY); | ||||
// size = None on the python size turn into size = NON_NORMAL when | ||||
// accessed. So the next line is currently required, but a some future | ||||
// clean up would be welcome. | ||||
self.size = SIZE_NON_NORMAL; | ||||
} | ||||
pub fn set_untracked(&mut self) { | ||||
self.flags.remove(Flags::WDIR_TRACKED); | ||||
self.mode = 0; | ||||
self.size = 0; | ||||
self.mtime = 0; | ||||
} | ||||
Simon Sapin
|
r48834 | /// Returns `(state, mode, size, mtime)` for the puprose of serialization | ||
/// in the dirstate-v1 format. | ||||
/// | ||||
/// This includes marker values such as `mtime == -1`. In the future we may | ||||
/// want to not represent these cases that way in memory, but serialization | ||||
/// will need to keep the same format. | ||||
pub fn v1_data(&self) -> (u8, i32, i32, i32) { | ||||
Simon Sapin
|
r48856 | (self.state().into(), self.mode(), self.size(), self.mtime()) | ||
Simon Sapin
|
r48834 | } | ||
r48875 | pub(crate) fn is_from_other_parent(&self) -> bool { | |||
Simon Sapin
|
r48856 | self.state() == EntryState::Normal | ||
&& self.size() == SIZE_FROM_OTHER_PARENT | ||||
Simon Sapin
|
r48830 | } | ||
// TODO: other platforms | ||||
#[cfg(unix)] | ||||
pub fn mode_changed( | ||||
&self, | ||||
filesystem_metadata: &std::fs::Metadata, | ||||
) -> bool { | ||||
use std::os::unix::fs::MetadataExt; | ||||
const EXEC_BIT_MASK: u32 = 0o100; | ||||
Simon Sapin
|
r48856 | let dirstate_exec_bit = (self.mode() as u32) & EXEC_BIT_MASK; | ||
Simon Sapin
|
r48830 | let fs_exec_bit = filesystem_metadata.mode() & EXEC_BIT_MASK; | ||
dirstate_exec_bit != fs_exec_bit | ||||
} | ||||
/// Returns a `(state, mode, size, mtime)` tuple as for | ||||
/// `DirstateMapMethods::debug_iter`. | ||||
pub fn debug_tuple(&self) -> (u8, i32, i32, i32) { | ||||
Simon Sapin
|
r48856 | let state = if self.flags.contains(Flags::ENTRYLESS_TREE_NODE) { | ||
b' ' | ||||
} else { | ||||
self.state().into() | ||||
}; | ||||
(state, self.mode(), self.size(), self.mtime()) | ||||
Simon Sapin
|
r48830 | } | ||
pub fn mtime_is_ambiguous(&self, now: i32) -> bool { | ||||
Simon Sapin
|
r48856 | self.state() == EntryState::Normal && self.mtime() == now | ||
Simon Sapin
|
r48830 | } | ||
pub fn clear_ambiguous_mtime(&mut self, now: i32) -> bool { | ||||
let ambiguous = self.mtime_is_ambiguous(now); | ||||
if ambiguous { | ||||
// The file was last modified "simultaneously" with the current | ||||
// write to dirstate (i.e. within the same second for file- | ||||
// systems with a granularity of 1 sec). This commonly happens | ||||
// for at least a couple of files on 'update'. | ||||
// The user could change the file without changing its size | ||||
// within the same second. Invalidate the file's mtime in | ||||
// dirstate, forcing future 'status' calls to compare the | ||||
// contents of the file if the size is the same. This prevents | ||||
// mistakenly treating such files as clean. | ||||
self.clear_mtime() | ||||
} | ||||
ambiguous | ||||
} | ||||
pub fn clear_mtime(&mut self) { | ||||
self.mtime = -1; | ||||
} | ||||
} | ||||
impl EntryState { | ||||
pub fn is_tracked(self) -> bool { | ||||
use EntryState::*; | ||||
match self { | ||||
Normal | Added | Merged => true, | ||||
Simon Sapin
|
r48838 | Removed => false, | ||
Simon Sapin
|
r48830 | } | ||
} | ||||
} | ||||
impl TryFrom<u8> for EntryState { | ||||
type Error = HgError; | ||||
fn try_from(value: u8) -> Result<Self, Self::Error> { | ||||
match value { | ||||
b'n' => Ok(EntryState::Normal), | ||||
b'a' => Ok(EntryState::Added), | ||||
b'r' => Ok(EntryState::Removed), | ||||
b'm' => Ok(EntryState::Merged), | ||||
_ => Err(HgError::CorruptedRepository(format!( | ||||
"Incorrect dirstate entry state {}", | ||||
value | ||||
))), | ||||
} | ||||
} | ||||
} | ||||
impl Into<u8> for EntryState { | ||||
fn into(self) -> u8 { | ||||
match self { | ||||
EntryState::Normal => b'n', | ||||
EntryState::Added => b'a', | ||||
EntryState::Removed => b'r', | ||||
EntryState::Merged => b'm', | ||||
} | ||||
} | ||||
} | ||||