cat.rs
71 lines
| 2.1 KiB
| application/rls-services+xml
|
RustLexer
|
r46112 | // list_tracked_files.rs | ||
// | ||||
// Copyright 2020 Antoine Cezar <antoine.cezar@octobus.net> | ||||
// | ||||
// This software may be used and distributed according to the terms of the | ||||
// GNU General Public License version 2 or any later version. | ||||
|
r46782 | use crate::repo::Repo; | ||
|
r46112 | use crate::revlog::revlog::RevlogError; | ||
|
r46647 | use crate::revlog::Node; | ||
|
r48775 | |||
use crate::utils::hg_path::HgPathBuf; | ||||
|
r46112 | |||
|
r47478 | pub struct CatOutput { | ||
/// Whether any file in the manifest matched the paths given as CLI | ||||
/// arguments | ||||
pub found_any: bool, | ||||
/// The contents of matching files, in manifest order | ||||
pub concatenated: Vec<u8>, | ||||
/// Which of the CLI arguments did not match any manifest file | ||||
pub missing: Vec<HgPathBuf>, | ||||
/// The node ID that the given revset was resolved to | ||||
pub node: Node, | ||||
} | ||||
/// Output the given revision of files | ||||
|
r46751 | /// | ||
/// * `root`: Repository root | ||||
/// * `rev`: The revision to cat the files from. | ||||
/// * `files`: The files to output. | ||||
|
r47478 | pub fn cat<'a>( | ||
|
r46782 | repo: &Repo, | ||
|
r47162 | revset: &str, | ||
|
r47478 | files: &'a [HgPathBuf], | ||
) -> Result<CatOutput, RevlogError> { | ||||
|
r47162 | let rev = crate::revset::resolve_single(revset, repo)?; | ||
|
r48778 | let manifest = repo.manifest_for_rev(rev)?; | ||
|
r48774 | let node = *repo | ||
.changelog()? | ||||
|
r47478 | .node_from_rev(rev) | ||
|
r48774 | .expect("should succeed when repo.manifest did"); | ||
|
r46751 | let mut bytes = vec![]; | ||
|
r47478 | let mut matched = vec![false; files.len()]; | ||
let mut found_any = false; | ||||
|
r46112 | |||
|
r48774 | for (manifest_file, node_bytes) in manifest.files_with_nodes() { | ||
|
r47478 | for (cat_file, is_matched) in files.iter().zip(&mut matched) { | ||
|
r46751 | if cat_file.as_bytes() == manifest_file.as_bytes() { | ||
|
r47478 | *is_matched = true; | ||
found_any = true; | ||||
|
r48775 | let file_log = repo.filelog(manifest_file)?; | ||
|
r47172 | let file_node = Node::from_hex_for_repo(node_bytes)?; | ||
|
r48783 | let entry = file_log.data_for_node(file_node)?; | ||
|
r48775 | bytes.extend(entry.data()?) | ||
|
r46751 | } | ||
} | ||||
|
r46112 | } | ||
|
r47478 | let missing: Vec<_> = files | ||
.iter() | ||||
.zip(&matched) | ||||
.filter(|pair| !*pair.1) | ||||
.map(|pair| pair.0.clone()) | ||||
.collect(); | ||||
Ok(CatOutput { | ||||
found_any, | ||||
concatenated: bytes, | ||||
missing, | ||||
node, | ||||
}) | ||||
|
r46112 | } | ||