mirror of https://github.com/Trivernis/nenv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.6 KiB
Rust
92 lines
2.6 KiB
Rust
use std::{
|
|
fs::File,
|
|
io::{self, BufReader},
|
|
path::Path,
|
|
};
|
|
|
|
use miette::Diagnostic;
|
|
use miette::Result;
|
|
use thiserror::Error;
|
|
|
|
use crate::utils::progress_spinner;
|
|
type ExtractResult<T> = Result<T, ExtractError>;
|
|
|
|
/// An error that can occur during extraction
|
|
#[derive(Error, Debug, Diagnostic)]
|
|
pub enum ExtractError {
|
|
#[diagnostic(code(nenv::extract::io))]
|
|
#[error("IO error when extracting: {0}")]
|
|
Io(#[from] io::Error),
|
|
|
|
#[diagnostic(code(nenv::extract::zip))]
|
|
#[error("Failed to extract zip: {0}")]
|
|
Zip(#[from] zip::result::ZipError),
|
|
}
|
|
|
|
pub fn extract_file(src: &Path, dst: &Path) -> ExtractResult<()> {
|
|
#[cfg(target_os = "windows")]
|
|
extract_zip(src, dst)?;
|
|
#[cfg(not(target_os = "windows"))]
|
|
extract_tar_gz(src, dst)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(target_os = "windows"))]
|
|
fn extract_tar_gz(src: &Path, dst: &Path) -> ExtractResult<()> {
|
|
use libflate::gzip::Decoder;
|
|
use tar::Archive;
|
|
|
|
let reader = BufReader::new(File::open(src)?);
|
|
let decoder = Decoder::new(reader)?;
|
|
let mut archive = Archive::new(decoder);
|
|
let pb = progress_spinner();
|
|
pb.set_message("Extracting tar.gz archive");
|
|
|
|
archive.unpack(dst)?;
|
|
pb.finish_with_message("Archive extracted.");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(target_os = "windows")]
|
|
fn extract_zip(src: &Path, dst: &Path) -> ExtractResult<()> {
|
|
use crate::utils::progress_bar;
|
|
use std::fs;
|
|
use zip::ZipArchive;
|
|
let mut archive = ZipArchive::new(File::open(src)?)?;
|
|
|
|
let pb = progress_bar(archive.len() as u64);
|
|
pb.set_message("Extracting zip archive");
|
|
|
|
for i in 0..archive.len() {
|
|
let mut file = archive.by_index(i)?;
|
|
let Some(path) = file.enclosed_name() else {
|
|
tracing::error!(
|
|
"Cannot extract {:?} because it has an invalid name",
|
|
file.name()
|
|
);
|
|
continue;
|
|
};
|
|
let output_path = dst.join(path);
|
|
if (*file.name()).ends_with('/') {
|
|
tracing::debug!("Creating directory {output_path:?}");
|
|
fs::create_dir_all(output_path)?;
|
|
} else {
|
|
if let Some(parent) = output_path.parent() {
|
|
if !parent.exists() {
|
|
tracing::debug!("Creating parent directory {parent:?}");
|
|
fs::create_dir_all(parent)?;
|
|
}
|
|
}
|
|
let mut file_output = File::create(&output_path)?;
|
|
tracing::debug!("Extracting to {output_path:?}");
|
|
io::copy(&mut file, &mut file_output)?;
|
|
}
|
|
pb.inc(1);
|
|
}
|
|
pb.finish_with_message("Archive extracted.");
|
|
|
|
Ok(())
|
|
}
|