debugdata.rs
91 lines
| 2.7 KiB
| application/rls-services+xml
|
RustLexer
|
r46099 | use crate::commands::Command; | ||
use crate::error::{CommandError, CommandErrorKind}; | ||||
use crate::ui::utf8_to_local; | ||||
use crate::ui::Ui; | ||||
|
r46751 | use hg::operations::find_root; | ||
|
r46099 | use hg::operations::{ | ||
|
r46751 | debug_data, DebugDataError, DebugDataErrorKind, DebugDataKind, | ||
|
r46099 | }; | ||
|
r46101 | use micro_timer::timed; | ||
|
r46099 | |||
pub const HELP_TEXT: &str = " | ||||
Dump the contents of a data file revision | ||||
"; | ||||
pub struct DebugDataCommand<'a> { | ||||
rev: &'a str, | ||||
kind: DebugDataKind, | ||||
} | ||||
impl<'a> DebugDataCommand<'a> { | ||||
pub fn new(rev: &'a str, kind: DebugDataKind) -> Self { | ||||
DebugDataCommand { rev, kind } | ||||
} | ||||
} | ||||
impl<'a> Command for DebugDataCommand<'a> { | ||||
|
r46101 | #[timed] | ||
|
r46099 | fn run(&self, ui: &Ui) -> Result<(), CommandError> { | ||
|
r46751 | let root = find_root()?; | ||
let data = debug_data(&root, self.rev, self.kind) | ||||
.map_err(|e| to_command_error(self.rev, e))?; | ||||
|
r46099 | |||
let mut stdout = ui.stdout_buffer(); | ||||
stdout.write_all(&data)?; | ||||
stdout.flush()?; | ||||
Ok(()) | ||||
} | ||||
} | ||||
/// Convert operation errors to command errors | ||||
fn to_command_error(rev: &str, err: DebugDataError) -> CommandError { | ||||
match err.kind { | ||||
DebugDataErrorKind::IoError(err) => CommandError { | ||||
kind: CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!("abort: {}\n", err)).into(), | ||||
)), | ||||
}, | ||||
DebugDataErrorKind::InvalidRevision => CommandError { | ||||
kind: CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: invalid revision identifier{}\n", | ||||
rev | ||||
)) | ||||
.into(), | ||||
)), | ||||
}, | ||||
|
r46646 | DebugDataErrorKind::AmbiguousPrefix => CommandError { | ||
kind: CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: ambiguous revision identifier{}\n", | ||||
rev | ||||
)) | ||||
.into(), | ||||
)), | ||||
}, | ||||
|
r46099 | DebugDataErrorKind::UnsuportedRevlogVersion(version) => CommandError { | ||
kind: CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: unsupported revlog version {}\n", | ||||
version | ||||
)) | ||||
.into(), | ||||
)), | ||||
}, | ||||
DebugDataErrorKind::CorruptedRevlog => CommandError { | ||||
kind: CommandErrorKind::Abort(Some( | ||||
"abort: corrupted revlog\n".into(), | ||||
)), | ||||
}, | ||||
DebugDataErrorKind::UnknowRevlogDataFormat(format) => CommandError { | ||||
kind: CommandErrorKind::Abort(Some( | ||||
utf8_to_local(&format!( | ||||
"abort: unknow revlog dataformat {:?}\n", | ||||
format | ||||
)) | ||||
.into(), | ||||
)), | ||||
}, | ||||
} | ||||
} | ||||