files.rs
136 lines
| 4.4 KiB
| application/rls-services+xml
|
RustLexer
|
r45923 | use crate::commands::Command; | ||
use crate::error::{CommandError, CommandErrorKind}; | ||||
|
r46106 | use crate::ui::utf8_to_local; | ||
|
r45923 | use crate::ui::Ui; | ||
|
r46106 | use hg::operations::FindRoot; | ||
use hg::operations::{ | ||||
ListDirstateTrackedFiles, ListDirstateTrackedFilesError, | ||||
ListDirstateTrackedFilesErrorKind, | ||||
}; | ||||
|
r46108 | use hg::operations::{ | ||
ListRevTrackedFiles, ListRevTrackedFilesError, | ||||
ListRevTrackedFilesErrorKind, | ||||
}; | ||||
|
r46007 | use hg::utils::files::{get_bytes_from_path, relativize_path}; | ||
|
r46108 | use hg::utils::hg_path::{HgPath, HgPathBuf}; | ||
use std::path::PathBuf; | ||||
|
r45923 | |||
pub const HELP_TEXT: &str = " | ||||
List tracked files. | ||||
Returns 0 on success. | ||||
"; | ||||
|
r46108 | pub struct FilesCommand<'a> { | ||
rev: Option<&'a str>, | ||||
|
r45923 | } | ||
|
r46108 | impl<'a> FilesCommand<'a> { | ||
pub fn new(rev: Option<&'a str>) -> Self { | ||||
FilesCommand { rev } | ||||
} | ||||
|
r45923 | |||
|
r46108 | fn display_files( | ||
&self, | ||||
ui: &Ui, | ||||
root: &PathBuf, | ||||
files: impl IntoIterator<Item = &'a HgPath>, | ||||
) -> Result<(), CommandError> { | ||||
|
r46007 | let cwd = std::env::current_dir() | ||
.or_else(|e| Err(CommandErrorKind::CurrentDirNotFound(e)))?; | ||||
let rooted_cwd = cwd | ||||
|
r46106 | .strip_prefix(&root) | ||
|
r46007 | .expect("cwd was already checked within the repository"); | ||
let rooted_cwd = HgPathBuf::from(get_bytes_from_path(rooted_cwd)); | ||||
|
r46009 | let mut stdout = ui.stdout_buffer(); | ||
|
r46007 | |||
|
r45923 | for file in files { | ||
|
r46007 | stdout.write_all(relativize_path(file, &rooted_cwd).as_ref())?; | ||
|
r45923 | stdout.write_all(b"\n")?; | ||
} | ||||
stdout.flush()?; | ||||
Ok(()) | ||||
} | ||||
} | ||||
|
r46106 | |||
|
r46108 | impl<'a> Command for FilesCommand<'a> { | ||
fn run(&self, ui: &Ui) -> Result<(), CommandError> { | ||||
let root = FindRoot::new().run()?; | ||||
if let Some(rev) = self.rev { | ||||
let mut operation = ListRevTrackedFiles::new(&root, rev) | ||||
.map_err(|e| map_rev_error(rev, e))?; | ||||
let files = operation.run().map_err(|e| map_rev_error(rev, e))?; | ||||
self.display_files(ui, &root, files) | ||||
} else { | ||||
let mut operation = ListDirstateTrackedFiles::new(&root) | ||||
.map_err(map_dirstate_error)?; | ||||
let files = operation.run().map_err(map_dirstate_error)?; | ||||
self.display_files(ui, &root, files) | ||||
} | ||||
} | ||||
} | ||||
/// Convert `ListRevTrackedFilesErrorKind` to `CommandError` | ||||
fn map_rev_error(rev: &str, err: ListRevTrackedFilesError) -> CommandError { | ||||
CommandError { | ||||
kind: match err.kind { | ||||
ListRevTrackedFilesErrorKind::IoError(err) => { | ||||
CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!("abort: {}\n", err)).into(), | ||||
)) | ||||
} | ||||
ListRevTrackedFilesErrorKind::InvalidRevision => { | ||||
CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: invalid revision identifier{}\n", | ||||
rev | ||||
)) | ||||
.into(), | ||||
)) | ||||
} | ||||
ListRevTrackedFilesErrorKind::UnsuportedRevlogVersion(version) => { | ||||
CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: unsupported revlog version {}\n", | ||||
version | ||||
)) | ||||
.into(), | ||||
)) | ||||
} | ||||
ListRevTrackedFilesErrorKind::CorruptedRevlog => { | ||||
CommandErrorKind::Abort(Some( | ||||
"abort: corrupted revlog\n".into(), | ||||
)) | ||||
} | ||||
ListRevTrackedFilesErrorKind::UnknowRevlogDataFormat(format) => { | ||||
CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: unknow revlog dataformat {:?}\n", | ||||
format | ||||
)) | ||||
.into(), | ||||
)) | ||||
} | ||||
}, | ||||
} | ||||
} | ||||
/// Convert `ListDirstateTrackedFilesError` to `CommandError` | ||||
|
r46106 | fn map_dirstate_error(err: ListDirstateTrackedFilesError) -> CommandError { | ||
CommandError { | ||||
kind: match err.kind { | ||||
ListDirstateTrackedFilesErrorKind::IoError(err) => { | ||||
CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!("abort: {}\n", err)).into(), | ||||
)) | ||||
} | ||||
ListDirstateTrackedFilesErrorKind::ParseError(_) => { | ||||
CommandErrorKind::Abort(Some( | ||||
// TODO find a better error message | ||||
b"abort: parse error\n".to_vec(), | ||||
)) | ||||
} | ||||
}, | ||||
} | ||||
} | ||||