files.rs
70 lines
| 1.8 KiB
| application/rls-services+xml
|
RustLexer
Simon Sapin
|
r47163 | use crate::error::CommandError; | ||
Antoine Cezar
|
r45923 | use crate::ui::Ui; | ||
Simon Sapin
|
r47229 | use clap::Arg; | ||
Simon Sapin
|
r47228 | use clap::ArgMatches; | ||
Simon Sapin
|
r47213 | use hg::config::Config; | ||
Simon Sapin
|
r47165 | use hg::operations::list_rev_tracked_files; | ||
use hg::operations::Dirstate; | ||||
Simon Sapin
|
r46782 | use hg::repo::Repo; | ||
Raphaël Gomès
|
r46007 | use hg::utils::files::{get_bytes_from_path, relativize_path}; | ||
Antoine Cezar
|
r46108 | use hg::utils::hg_path::{HgPath, HgPathBuf}; | ||
Simon Sapin
|
r47231 | use std::path::Path; | ||
Antoine Cezar
|
r45923 | |||
pub const HELP_TEXT: &str = " | ||||
List tracked files. | ||||
Returns 0 on success. | ||||
"; | ||||
Simon Sapin
|
r47229 | pub fn args() -> clap::App<'static, 'static> { | ||
clap::SubCommand::with_name("files") | ||||
.arg( | ||||
Arg::with_name("rev") | ||||
.help("search the repository as it is in REV") | ||||
.short("-r") | ||||
.long("--revision") | ||||
.value_name("REV") | ||||
.takes_value(true), | ||||
) | ||||
.about(HELP_TEXT) | ||||
} | ||||
Simon Sapin
|
r47228 | pub fn run( | ||
ui: &Ui, | ||||
config: &Config, | ||||
Simon Sapin
|
r47231 | repo_path: Option<&Path>, | ||
Simon Sapin
|
r47228 | args: &ArgMatches, | ||
) -> Result<(), CommandError> { | ||||
let rev = args.value_of("rev"); | ||||
Antoine Cezar
|
r45923 | |||
Simon Sapin
|
r47231 | let repo = Repo::find(config, repo_path)?; | ||
Simon Sapin
|
r47228 | if let Some(rev) = rev { | ||
let files = | ||||
list_rev_tracked_files(&repo, rev).map_err(|e| (e, rev))?; | ||||
display_files(ui, &repo, files.iter()) | ||||
} else { | ||||
let distate = Dirstate::new(&repo)?; | ||||
let files = distate.tracked_files()?; | ||||
display_files(ui, &repo, files) | ||||
Antoine Cezar
|
r45923 | } | ||
} | ||||
Antoine Cezar
|
r46106 | |||
Simon Sapin
|
r47228 | fn display_files<'a>( | ||
ui: &Ui, | ||||
repo: &Repo, | ||||
files: impl IntoIterator<Item = &'a HgPath>, | ||||
) -> Result<(), CommandError> { | ||||
Simon Sapin
|
r47231 | let cwd = HgPathBuf::from(get_bytes_from_path(hg::utils::current_dir()?)); | ||
let working_directory = | ||||
HgPathBuf::from(get_bytes_from_path(repo.working_directory_path())); | ||||
Simon Sapin
|
r47228 | |||
let mut stdout = ui.stdout_buffer(); | ||||
for file in files { | ||||
Simon Sapin
|
r47231 | let file = working_directory.join(file); | ||
stdout.write_all(relativize_path(&file, &cwd).as_ref())?; | ||||
Simon Sapin
|
r47228 | stdout.write_all(b"\n")?; | ||
Antoine Cezar
|
r46108 | } | ||
Simon Sapin
|
r47228 | stdout.flush()?; | ||
Ok(()) | ||||
Antoine Cezar
|
r46108 | } | ||