mirror of
https://github.com/cargo-bins/cargo-binstall.git
synced 2025-05-05 19:50:02 +00:00
46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
use std::{
|
|
fs::File,
|
|
io::{self, BufRead, Read},
|
|
path::Path,
|
|
};
|
|
|
|
use bzip2::bufread::BzDecoder;
|
|
use flate2::bufread::GzDecoder;
|
|
use log::debug;
|
|
use tar::Archive;
|
|
use xz2::bufread::XzDecoder;
|
|
use zip::read::ZipArchive;
|
|
use zstd::stream::Decoder as ZstdDecoder;
|
|
|
|
use crate::{errors::BinstallError, manifests::cargo_toml_binstall::TarBasedFmt};
|
|
|
|
pub(super) fn create_tar_decoder(
|
|
dat: impl BufRead + 'static,
|
|
fmt: TarBasedFmt,
|
|
) -> io::Result<Archive<Box<dyn Read>>> {
|
|
use TarBasedFmt::*;
|
|
|
|
let r: Box<dyn Read> = match fmt {
|
|
Tar => Box::new(dat),
|
|
Tbz2 => Box::new(BzDecoder::new(dat)),
|
|
Tgz => Box::new(GzDecoder::new(dat)),
|
|
Txz => Box::new(XzDecoder::new(dat)),
|
|
Tzstd => {
|
|
// The error can only come from raw::Decoder::with_dictionary as of zstd 0.10.2 and
|
|
// 0.11.2, which is specified as `&[]` by `ZstdDecoder::new`, thus `ZstdDecoder::new`
|
|
// should not return any error.
|
|
Box::new(ZstdDecoder::with_buffer(dat)?)
|
|
}
|
|
};
|
|
|
|
Ok(Archive::new(r))
|
|
}
|
|
|
|
pub(super) fn unzip(dat: File, dst: &Path) -> Result<(), BinstallError> {
|
|
debug!("Decompressing from zip archive to `{dst:?}`");
|
|
|
|
let mut zip = ZipArchive::new(dat)?;
|
|
zip.extract(dst)?;
|
|
|
|
Ok(())
|
|
}
|