mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 16:57:49 +03:00
ac31713c84
Summary: Don't use failure's bail!() and ensure!() macros. Instead, failure_ext provides: - bail_err!(err) - Converts its single parameter to the expected error and returns; ie `return Err(From::from(err));` - bail_msg!(fmt, ...) - takes format string parameters and returns a `failure::err_msg()` error - ensure_err!(), ensure_msg!() - corresponding changes Also: - remove all stray references to error-chain - remove direct references to failure_derive (it's reexported via failure and failure_ext) - replace uses of `Err(foo)?;` with `bail_err!()` (since `bail_err` unconditionally returns, but `Err(x)?` does not in principle, which can affect type inference) Reviewed By: kulshrax Differential Revision: D6507717 fbshipit-source-id: 635fb6f8c96d185b195dff171ea9c8db9e83af10
116 lines
3.7 KiB
Rust
116 lines
3.7 KiB
Rust
// Copyright (c) 2004-present, Facebook, Inc.
|
|
// All Rights Reserved.
|
|
//
|
|
// This software may be used and distributed according to the terms of the
|
|
// GNU General Public License version 2 or any later version.
|
|
|
|
extern crate clap;
|
|
#[macro_use]
|
|
extern crate failure_ext as failure;
|
|
extern crate mercurial;
|
|
|
|
// Import symbols from std:: (standard library)
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::str::FromStr;
|
|
|
|
// Just need `App` from clap
|
|
use clap::App;
|
|
|
|
// Get `Revlog` from the mercurial revlog module
|
|
use failure::{Error, Result, ResultExt};
|
|
use mercurial::revlog::Revlog;
|
|
|
|
fn run() -> Result<()> {
|
|
// Define command line args and parse command line
|
|
let matches = App::new("dumprev")
|
|
.version("0.0.0")
|
|
.about("extract a revision from a revlog")
|
|
.args_from_usage(concat!(
|
|
"-d, --data=[DATAFILE] 'Data file if not inline'\n",
|
|
"-w, --write=[DUMPFILE] 'Write data to file'\n",
|
|
"<IDXFILE> 'index file'\n",
|
|
"<REV> 'revision index'"
|
|
))
|
|
.get_matches();
|
|
// Get path of index file; `unwrap()` is safe because parameter is non-optional
|
|
let idxpath = matches.value_of("IDXFILE").unwrap();
|
|
|
|
// Get optional datapath
|
|
let datapath = matches.value_of("DATAFILE");
|
|
|
|
// Also optional dumpfile
|
|
let dumpfile = matches.value_of("write");
|
|
|
|
// Get non-optional revision
|
|
let revidx = FromStr::from_str(matches.value_of("REV").unwrap())
|
|
.map_err(Error::from)
|
|
.with_context(|_| "idx malformed")?;
|
|
|
|
// Construct a `Revlog`
|
|
let revlog =
|
|
Revlog::from_idx_data(idxpath, datapath).with_context(|_| "failed to load idx and data")?;
|
|
println!("made revlog {:?}", revlog.get_header());
|
|
|
|
let entry = revlog
|
|
.get_entry(revidx)
|
|
.with_context(|_| "failed to get entry")?;
|
|
|
|
println!("Revlog[{:?}] = {:?}", revidx, entry);
|
|
match revlog.get_rev(revidx) {
|
|
Ok(ref rev) if rev.nodeid().is_some() => {
|
|
if entry.nodeid() != &rev.nodeid().unwrap() {
|
|
println!(
|
|
"NOTE: hash mismatch: expected {}, got {}",
|
|
entry.nodeid(),
|
|
rev.nodeid().unwrap()
|
|
)
|
|
}
|
|
if let Some(revdata) = rev.as_blob().as_slice() {
|
|
if let Some(dumpfile) = dumpfile {
|
|
let mut file = match File::create(dumpfile) {
|
|
Ok(file) => file,
|
|
Err(err) => bail_msg!("Failed to create file {}: {:?}", dumpfile, err),
|
|
};
|
|
println!(
|
|
"Writing rev {:?} to {}",
|
|
rev.nodeid().expect("no id"),
|
|
dumpfile
|
|
);
|
|
if let Err(err) = file.write_all(revdata) {
|
|
bail_msg!("Failed to write {}: {:?}", dumpfile, err);
|
|
}
|
|
} else {
|
|
println!(
|
|
"rev {:?}:\n{}",
|
|
rev.nodeid().expect("no id"),
|
|
String::from_utf8_lossy(revdata)
|
|
);
|
|
}
|
|
} else {
|
|
println!("Dataless rev {:?}", rev.nodeid().expect("no id"));
|
|
}
|
|
}
|
|
Ok(rev) => bail_msg!(
|
|
"Nodeid missing: got {:?} expected {:?}",
|
|
rev.nodeid(),
|
|
entry.nodeid()
|
|
),
|
|
Err(err) => bail_msg!("failed to get chunk {:?}: {}", revidx, err),
|
|
};
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn main() {
|
|
if let Err(ref e) = run() {
|
|
println!("Failed: {}", e);
|
|
|
|
for e in e.causes() {
|
|
println!("caused by: {}", e);
|
|
}
|
|
|
|
std::process::exit(1);
|
|
}
|
|
}
|