Mercurial > hg
view rust/rhg/src/commands/cat.rs @ 46446:1dcd9c9975ed
rust: Fold find_root and check_requirements into Repo::find
Differential Revision: https://phab.mercurial-scm.org/D9906
author | Simon Sapin <simon.sapin@octobus.net> |
---|---|
date | Thu, 28 Jan 2021 20:31:42 +0100 |
parents | ca3f73cc3cf4 |
children | a6e4e4650bac |
line wrap: on
line source
use crate::commands::Command; use crate::error::CommandError; use crate::ui::Ui; use hg::operations::cat; use hg::repo::Repo; 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 "; pub struct CatCommand<'a> { rev: Option<&'a str>, files: Vec<&'a str>, } impl<'a> CatCommand<'a> { pub fn new(rev: Option<&'a str>, files: Vec<&'a str>) -> Self { Self { rev, files } } fn display(&self, ui: &Ui, data: &[u8]) -> Result<(), CommandError> { ui.write_stdout(data)?; Ok(()) } } impl<'a> Command for CatCommand<'a> { #[timed] fn run(&self, ui: &Ui) -> Result<(), CommandError> { let repo = Repo::find()?; let cwd = hg::utils::current_dir()?; let mut files = vec![]; for file in self.files.iter() { // TODO: actually normalize `..` path segments etc? let normalized = cwd.join(&file); let stripped = normalized .strip_prefix(&repo.working_directory_path()) // 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); } match self.rev { Some(rev) => { let data = cat(&repo, rev, &files).map_err(|e| (e, rev))?; self.display(ui, &data) } None => Err(CommandError::Unimplemented.into()), } } }