files.rs
55 lines
| 1.4 KiB
| application/rls-services+xml
|
RustLexer
Simon Sapin
|
r47163 | use crate::error::CommandError; | ||
Antoine Cezar
|
r45923 | use crate::ui::Ui; | ||
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}; | ||
Antoine Cezar
|
r45923 | |||
pub const HELP_TEXT: &str = " | ||||
List tracked files. | ||||
Returns 0 on success. | ||||
"; | ||||
Simon Sapin
|
r47228 | pub fn run( | ||
ui: &Ui, | ||||
config: &Config, | ||||
args: &ArgMatches, | ||||
) -> Result<(), CommandError> { | ||||
let rev = args.value_of("rev"); | ||||
Antoine Cezar
|
r45923 | |||
Simon Sapin
|
r47228 | let repo = Repo::find(config)?; | ||
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> { | ||||
let cwd = hg::utils::current_dir()?; | ||||
let rooted_cwd = cwd | ||||
.strip_prefix(repo.working_directory_path()) | ||||
.expect("cwd was already checked within the repository"); | ||||
let rooted_cwd = HgPathBuf::from(get_bytes_from_path(rooted_cwd)); | ||||
let mut stdout = ui.stdout_buffer(); | ||||
for file in files { | ||||
stdout.write_all(relativize_path(file, &rooted_cwd).as_ref())?; | ||||
stdout.write_all(b"\n")?; | ||||
Antoine Cezar
|
r46108 | } | ||
Simon Sapin
|
r47228 | stdout.flush()?; | ||
Ok(()) | ||||
Antoine Cezar
|
r46108 | } | ||