Sebastian Thielfinish refactoring git-odbec282ae5 years agoHistory
107 lines · 2.9 KB · Rust
RawDownload
use crate::OutputFormat;use git_features::progress::Progress;use git_odb::pack;use std::{fs, io, path::PathBuf, str::FromStr}; #[derive(PartialEq, Debug)]pub enum IterationMode {    AsIs,    Verify,    Restore,} impl IterationMode {    pub fn variants() -> &'static [&'static str] {        &["as-is", "verify", "restore"]    }} impl Default for IterationMode {    fn default() -> Self {        IterationMode::Verify    }} impl FromStr for IterationMode {    type Err = String;     fn from_str(s: &str) -> Result<Self, Self::Err> {        use IterationMode::*;        let slc = s.to_ascii_lowercase();        Ok(match slc.as_str() {            "as-is" => AsIs,            "verify" => Verify,            "restore" => Restore,            _ => return Err("invalid value".into()),        })    }} impl From<IterationMode> for pack::data::iter::Mode {    fn from(v: IterationMode) -> Self {        use pack::data::iter::Mode::*;        match v {            IterationMode::AsIs => AsIs,            IterationMode::Verify => Verify,            IterationMode::Restore => Restore,        }    }} pub struct Context<W: io::Write> {    pub thread_limit: Option<usize>,    pub iteration_mode: IterationMode,    pub format: OutputFormat,    pub out: W,} pub fn stream_len(mut s: impl io::Seek) -> io::Result<u64> {    use io::SeekFrom;    let old_pos = s.seek(SeekFrom::Current(0))?;    let len = s.seek(SeekFrom::End(0))?;    if old_pos != len {        s.seek(SeekFrom::Start(old_pos))?;    }    Ok(len)} pub const PROGRESS_RANGE: std::ops::RangeInclusive<u8> = 2..=3; pub fn from_pack(    pack: Option<PathBuf>,    directory: Option<PathBuf>,    progress: impl Progress,    ctx: Context<impl io::Write>,) -> anyhow::Result<()> {    use anyhow::Context;    let options = pack::bundle::write::Options {        thread_limit: ctx.thread_limit,        iteration_mode: ctx.iteration_mode.into(),        index_kind: pack::index::Version::default(),    };    let out = ctx.out;    let format = ctx.format;    let res = match pack {        Some(pack) => {            let pack_len = pack.metadata()?.len();            let pack_file = fs::File::open(pack)?;            pack::Bundle::write_to_directory_eagerly(pack_file, Some(pack_len), directory, progress, options)        }        None => {            let stdin = io::stdin();            pack::Bundle::write_to_directory_eagerly(stdin, None, directory, progress, options)        }    }    .with_context(|| "Failed to write pack and index")?;    match format {        OutputFormat::Human => drop(human_output(out, res)),        #[cfg(feature = "serde1")]        OutputFormat::Json => serde_json::to_writer_pretty(out, &res)?,    };    Ok(())} fn human_output(mut out: impl io::Write, res: pack::bundle::write::Outcome) -> io::Result<()> {    writeln!(&mut out, "index: {}", res.index.index_hash)?;    writeln!(&mut out, "pack: {}", res.index.data_hash)}