manifest.rs
119 lines
| 3.6 KiB
| application/rls-services+xml
|
RustLexer
|
r48777 | use crate::errors::HgError; | ||
|
r46782 | use crate::repo::Repo; | ||
|
r46104 | use crate::revlog::revlog::{Revlog, RevlogError}; | ||
use crate::revlog::Revision; | ||||
|
r48778 | use crate::revlog::{Node, NodePrefix}; | ||
|
r46104 | use crate::utils::hg_path::HgPath; | ||
|
r49166 | use crate::utils::SliceExt; | ||
|
r46104 | |||
/// A specialized `Revlog` to work with `manifest` data format. | ||||
|
r48771 | pub struct Manifestlog { | ||
|
r46104 | /// The generic `revlog` format. | ||
revlog: Revlog, | ||||
} | ||||
|
r48771 | impl Manifestlog { | ||
|
r46104 | /// Open the `manifest` of a repository given by its root. | ||
|
r48777 | pub fn open(repo: &Repo) -> Result<Self, HgError> { | ||
|
r46782 | let revlog = Revlog::open(repo, "00manifest.i", None)?; | ||
|
r46104 | Ok(Self { revlog }) | ||
} | ||||
|
r48783 | /// Return the `Manifest` for the given node ID. | ||
/// | ||||
/// Note: this is a node ID in the manifestlog, typically found through | ||||
/// `ChangelogEntry::manifest_node`. It is *not* the node ID of any | ||||
/// changeset. | ||||
/// | ||||
/// See also `Repo::manifest_for_node` | ||||
pub fn data_for_node( | ||||
&self, | ||||
node: NodePrefix, | ||||
) -> Result<Manifest, RevlogError> { | ||||
|
r48782 | let rev = self.revlog.rev_from_node(node)?; | ||
|
r48783 | self.data_for_rev(rev) | ||
|
r46104 | } | ||
|
r48783 | /// Return the `Manifest` of a given revision number. | ||
/// | ||||
/// Note: this is a revision number in the manifestlog, *not* of any | ||||
/// changeset. | ||||
/// | ||||
/// See also `Repo::manifest_for_rev` | ||||
pub fn data_for_rev( | ||||
&self, | ||||
rev: Revision, | ||||
) -> Result<Manifest, RevlogError> { | ||||
|
r46104 | let bytes = self.revlog.get_rev_data(rev)?; | ||
|
r48771 | Ok(Manifest { bytes }) | ||
|
r46104 | } | ||
} | ||||
|
r48771 | /// `Manifestlog` entry which knows how to interpret the `manifest` data bytes. | ||
|
r46104 | #[derive(Debug)] | ||
|
r48771 | pub struct Manifest { | ||
|
r46104 | bytes: Vec<u8>, | ||
} | ||||
|
r48771 | impl Manifest { | ||
|
r49166 | pub fn iter( | ||
&self, | ||||
) -> impl Iterator<Item = Result<ManifestEntry, HgError>> { | ||||
|
r46104 | self.bytes | ||
.split(|b| b == &b'\n') | ||||
.filter(|line| !line.is_empty()) | ||||
|
r49166 | .map(|line| { | ||
let (path, rest) = line.split_2(b'\0').ok_or_else(|| { | ||||
|
r49165 | HgError::corrupted("manifest line should contain \\0") | ||
})?; | ||||
|
r49166 | let path = HgPath::new(path); | ||
let (hex_node_id, flags) = match rest.split_last() { | ||||
Some((&b'x', rest)) => (rest, Some(b'x')), | ||||
Some((&b'l', rest)) => (rest, Some(b'l')), | ||||
Some((&b't', rest)) => (rest, Some(b't')), | ||||
_ => (rest, None), | ||||
}; | ||||
Ok(ManifestEntry { | ||||
path, | ||||
hex_node_id, | ||||
flags, | ||||
}) | ||||
}) | ||||
|
r46111 | } | ||
|
r48778 | |||
/// If the given path is in this manifest, return its filelog node ID | ||||
|
r49166 | pub fn find_file( | ||
&self, | ||||
path: &HgPath, | ||||
) -> Result<Option<ManifestEntry>, HgError> { | ||||
|
r48778 | // TODO: use binary search instead of linear scan. This may involve | ||
// building (and caching) an index of the byte indicex of each manifest | ||||
// line. | ||||
|
r49166 | |||
// TODO: use try_find when available (if still using linear scan) | ||||
// https://github.com/rust-lang/rust/issues/63178 | ||||
for entry in self.iter() { | ||||
let entry = entry?; | ||||
if entry.path == path { | ||||
return Ok(Some(entry)); | ||||
|
r48778 | } | ||
} | ||||
Ok(None) | ||||
} | ||||
|
r46104 | } | ||
|
r49166 | |||
/// `Manifestlog` entry which knows how to interpret the `manifest` data bytes. | ||||
#[derive(Debug)] | ||||
pub struct ManifestEntry<'manifest> { | ||||
pub path: &'manifest HgPath, | ||||
pub hex_node_id: &'manifest [u8], | ||||
/// `Some` values are b'x', b'l', or 't' | ||||
pub flags: Option<u8>, | ||||
} | ||||
impl ManifestEntry<'_> { | ||||
pub fn node_id(&self) -> Result<Node, HgError> { | ||||
Node::from_hex_for_repo(self.hex_node_id) | ||||
} | ||||
} | ||||