cat.rs
93 lines
| 3.1 KiB
| application/rls-services+xml
|
RustLexer
|
r47163 | use crate::error::CommandError; | ||
|
r47251 | use clap::Arg; | ||
|
r47478 | use format_bytes::format_bytes; | ||
|
r47165 | use hg::operations::cat; | ||
|
r46113 | use hg::utils::hg_path::HgPathBuf; | ||
use micro_timer::timed; | ||||
use std::convert::TryFrom; | ||||
pub const HELP_TEXT: &str = " | ||||
Output the current or given revision of files | ||||
"; | ||||
|
r47251 | pub fn args() -> clap::App<'static, 'static> { | ||
clap::SubCommand::with_name("cat") | ||||
.arg( | ||||
Arg::with_name("rev") | ||||
.help("search the repository as it is in REV") | ||||
.short("-r") | ||||
.long("--revision") | ||||
.value_name("REV") | ||||
.takes_value(true), | ||||
) | ||||
.arg( | ||||
clap::Arg::with_name("files") | ||||
.required(true) | ||||
.multiple(true) | ||||
.empty_values(false) | ||||
.value_name("FILE") | ||||
.help("Activity to start: activity@category"), | ||||
) | ||||
.about(HELP_TEXT) | ||||
} | ||||
|
r47250 | #[timed] | ||
|
r47334 | pub fn run(invocation: &crate::CliInvocation) -> Result<(), CommandError> { | ||
let rev = invocation.subcommand_args.value_of("rev"); | ||||
let file_args = match invocation.subcommand_args.values_of("files") { | ||||
|
r47250 | Some(files) => files.collect(), | ||
None => vec![], | ||||
}; | ||||
|
r46113 | |||
|
r47335 | let repo = invocation.repo?; | ||
|
r47250 | let cwd = hg::utils::current_dir()?; | ||
|
r47474 | let working_directory = repo.working_directory_path(); | ||
let working_directory = cwd.join(working_directory); // Make it absolute | ||||
|
r47250 | |||
let mut files = vec![]; | ||||
for file in file_args.iter() { | ||||
|
r48888 | if file.starts_with("set:") { | ||
let message = "fileset"; | ||||
return Err(CommandError::unsupported(message)); | ||||
} | ||||
|
r48887 | let normalized = cwd.join(&file); | ||
|
r47250 | // TODO: actually normalize `..` path segments etc? | ||
|
r48887 | let dotted = normalized.components().any(|c| c.as_os_str() == ".."); | ||
if file == &"." || dotted { | ||||
let message = "`..` or `.` path segment"; | ||||
return Err(CommandError::unsupported(message)); | ||||
} | ||||
|
r47250 | let stripped = normalized | ||
|
r47474 | .strip_prefix(&working_directory) | ||
|
r47250 | // TODO: error message for path arguments outside of the repo | ||
.map_err(|_| CommandError::abort(""))?; | ||||
let hg_file = HgPathBuf::try_from(stripped.to_path_buf()) | ||||
.map_err(|e| CommandError::abort(e.to_string()))?; | ||||
files.push(hg_file); | ||||
|
r46113 | } | ||
|
r48886 | // TODO probably move this to a util function like `repo.default_rev` or | ||
// something when it's used somewhere else | ||||
let rev = match rev { | ||||
Some(r) => r.to_string(), | ||||
None => format!("{:x}", repo.dirstate_parents()?.p1), | ||||
}; | ||||
|
r46113 | |||
|
r48886 | let output = cat(&repo, &rev, &files).map_err(|e| (e, rev.as_str()))?; | ||
invocation.ui.write_stdout(&output.concatenated)?; | ||||
if !output.missing.is_empty() { | ||||
let short = format!("{:x}", output.node.short()).into_bytes(); | ||||
for path in &output.missing { | ||||
invocation.ui.write_stderr(&format_bytes!( | ||||
b"{}: no such file in rev {}\n", | ||||
path.as_bytes(), | ||||
short | ||||
))?; | ||||
|
r47250 | } | ||
|
r48886 | } | ||
if output.found_any { | ||||
Ok(()) | ||||
} else { | ||||
Err(CommandError::Unsuccessful) | ||||
|
r46113 | } | ||
} | ||||