files.rs
61 lines
| 1.7 KiB
| application/rls-services+xml
|
RustLexer
Antoine Cezar
|
r45923 | use crate::commands::Command; | ||
Simon Sapin
|
r47163 | use crate::error::CommandError; | ||
Antoine Cezar
|
r45923 | use crate::ui::Ui; | ||
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. | ||||
"; | ||||
Antoine Cezar
|
r46108 | pub struct FilesCommand<'a> { | ||
rev: Option<&'a str>, | ||||
Antoine Cezar
|
r45923 | } | ||
Antoine Cezar
|
r46108 | impl<'a> FilesCommand<'a> { | ||
pub fn new(rev: Option<&'a str>) -> Self { | ||||
FilesCommand { rev } | ||||
} | ||||
Antoine Cezar
|
r45923 | |||
Antoine Cezar
|
r46108 | fn display_files( | ||
&self, | ||||
ui: &Ui, | ||||
Simon Sapin
|
r46782 | repo: &Repo, | ||
Antoine Cezar
|
r46108 | files: impl IntoIterator<Item = &'a HgPath>, | ||
) -> Result<(), CommandError> { | ||||
Simon Sapin
|
r47174 | let cwd = hg::utils::current_dir()?; | ||
Raphaël Gomès
|
r46007 | let rooted_cwd = cwd | ||
Simon Sapin
|
r46782 | .strip_prefix(repo.working_directory_path()) | ||
Raphaël Gomès
|
r46007 | .expect("cwd was already checked within the repository"); | ||
let rooted_cwd = HgPathBuf::from(get_bytes_from_path(rooted_cwd)); | ||||
Antoine Cezar
|
r46009 | let mut stdout = ui.stdout_buffer(); | ||
Raphaël Gomès
|
r46007 | |||
Antoine Cezar
|
r45923 | for file in files { | ||
Raphaël Gomès
|
r46007 | stdout.write_all(relativize_path(file, &rooted_cwd).as_ref())?; | ||
Antoine Cezar
|
r45923 | stdout.write_all(b"\n")?; | ||
} | ||||
stdout.flush()?; | ||||
Ok(()) | ||||
} | ||||
} | ||||
Antoine Cezar
|
r46106 | |||
Antoine Cezar
|
r46108 | impl<'a> Command for FilesCommand<'a> { | ||
fn run(&self, ui: &Ui) -> Result<(), CommandError> { | ||||
Simon Sapin
|
r46782 | let repo = Repo::find()?; | ||
Antoine Cezar
|
r46108 | if let Some(rev) = self.rev { | ||
Simon Sapin
|
r47165 | let files = | ||
list_rev_tracked_files(&repo, rev).map_err(|e| (e, rev))?; | ||||
Simon Sapin
|
r46782 | self.display_files(ui, &repo, files.iter()) | ||
Antoine Cezar
|
r46108 | } else { | ||
Simon Sapin
|
r47165 | let distate = Dirstate::new(&repo)?; | ||
let files = distate.tracked_files()?; | ||||
Simon Sapin
|
r46782 | self.display_files(ui, &repo, files) | ||
Antoine Cezar
|
r46108 | } | ||
} | ||||
} | ||||