mirror of
https://github.com/cargo-bins/cargo-binstall.git
synced 2025-04-22 13:38:43 +00:00
Refactor: Extract create_tar_decoder
from `extract_compressed_from_readable`. Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
This commit is contained in:
parent
8ef1e56fcc
commit
9584c8d35e
1 changed files with 30 additions and 40 deletions
|
@ -32,9 +32,10 @@ pub(super) enum Op<'a, V: TarEntriesVisitor> {
|
||||||
/// * `f` - If Some, then this function will pass
|
/// * `f` - If Some, then this function will pass
|
||||||
/// the entries of the `dat` to it and let it decides
|
/// the entries of the `dat` to it and let it decides
|
||||||
/// what to do with the tar.
|
/// what to do with the tar.
|
||||||
fn untar<R: Read, V: TarEntriesVisitor>(dat: R, op: Op<'_, V>) -> Result<(), BinstallError> {
|
fn untar<R: Read, V: TarEntriesVisitor>(
|
||||||
let mut tar = Archive::new(dat);
|
mut tar: Archive<R>,
|
||||||
|
op: Op<'_, V>,
|
||||||
|
) -> Result<(), BinstallError> {
|
||||||
match op {
|
match op {
|
||||||
Op::Visit(mut visitor) => {
|
Op::Visit(mut visitor) => {
|
||||||
debug!("Untaring with callback");
|
debug!("Untaring with callback");
|
||||||
|
@ -52,6 +53,28 @@ fn untar<R: Read, V: TarEntriesVisitor>(dat: R, op: Op<'_, V>) -> Result<(), Bin
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(super) fn create_tar_decoder(
|
||||||
|
dat: impl BufRead + 'static,
|
||||||
|
fmt: TarBasedFmt,
|
||||||
|
) -> Result<Archive<Box<dyn Read>>, BinstallError> {
|
||||||
|
use TarBasedFmt::*;
|
||||||
|
|
||||||
|
let r: Box<dyn Read> = match fmt {
|
||||||
|
Tar => Box::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))
|
||||||
|
}
|
||||||
|
|
||||||
/// Extract files from the specified source onto the specified path.
|
/// Extract files from the specified source onto the specified path.
|
||||||
///
|
///
|
||||||
/// * `fmt` - must not be `PkgFmt::Bin` or `PkgFmt::Zip`.
|
/// * `fmt` - must not be `PkgFmt::Bin` or `PkgFmt::Zip`.
|
||||||
|
@ -59,54 +82,21 @@ fn untar<R: Read, V: TarEntriesVisitor>(dat: R, op: Op<'_, V>) -> Result<(), Bin
|
||||||
/// and only extract ones which filter returns `true`.
|
/// and only extract ones which filter returns `true`.
|
||||||
/// Note that this is a best-effort and it only works when `fmt`
|
/// Note that this is a best-effort and it only works when `fmt`
|
||||||
/// is not `PkgFmt::Bin` or `PkgFmt::Zip`.
|
/// is not `PkgFmt::Bin` or `PkgFmt::Zip`.
|
||||||
pub(super) fn extract_compressed_from_readable<V: TarEntriesVisitor, R: BufRead>(
|
pub(super) fn extract_compressed_from_readable<V: TarEntriesVisitor, R: BufRead + 'static>(
|
||||||
dat: R,
|
dat: R,
|
||||||
fmt: TarBasedFmt,
|
fmt: TarBasedFmt,
|
||||||
op: Op<'_, V>,
|
op: Op<'_, V>,
|
||||||
) -> Result<(), BinstallError> {
|
) -> Result<(), BinstallError> {
|
||||||
use TarBasedFmt::*;
|
|
||||||
|
|
||||||
let msg = if let Op::UnpackToPath(path) = op {
|
let msg = if let Op::UnpackToPath(path) = op {
|
||||||
format!("destination: {path:#?}")
|
format!("destination: {path:#?}")
|
||||||
} else {
|
} else {
|
||||||
"process in-memory".to_string()
|
"process in-memory".to_string()
|
||||||
};
|
};
|
||||||
|
|
||||||
match fmt {
|
debug!("Extracting from {fmt} archive: {msg}");
|
||||||
Tar => {
|
|
||||||
// Extract to install dir
|
|
||||||
debug!("Extracting from tar archive: {msg}");
|
|
||||||
|
|
||||||
untar(dat, op)?
|
let tar = create_tar_decoder(dat, fmt)?;
|
||||||
}
|
untar(tar, op)
|
||||||
Tgz => {
|
|
||||||
// Extract to install dir
|
|
||||||
debug!("Decompressing from tgz archive: {msg}");
|
|
||||||
|
|
||||||
let tar = GzDecoder::new(dat);
|
|
||||||
untar(tar, op)?;
|
|
||||||
}
|
|
||||||
Txz => {
|
|
||||||
// Extract to install dir
|
|
||||||
debug!("Decompressing from txz archive: {msg}");
|
|
||||||
|
|
||||||
let tar = XzDecoder::new(dat);
|
|
||||||
untar(tar, op)?;
|
|
||||||
}
|
|
||||||
Tzstd => {
|
|
||||||
// Extract to install dir
|
|
||||||
debug!("Decompressing from tzstd archive: {msg}");
|
|
||||||
|
|
||||||
// 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.
|
|
||||||
let tar = ZstdDecoder::with_buffer(dat)?;
|
|
||||||
untar(tar, op)?;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(super) fn unzip(dat: File, dst: &Path) -> Result<(), BinstallError> {
|
pub(super) fn unzip(dat: File, dst: &Path) -> Result<(), BinstallError> {
|
||||||
|
|
Loading…
Add table
Reference in a new issue