ouch/src/archive/rar.rs

//! Contains RAR-specific building and unpacking functions

use std::path::Path;

use unrar::{self, Archive};

use crate::{error::Error, info, list::FileInArchive};

/// Unpacks the archive given by `archive_path` into the folder given by `output_folder`.
/// Assumes that output_folder is empty
pub fn unpack_archive(archive_path: &Path, output_folder: &Path, quiet: bool) -> crate::Result<usize> {
    assert!(output_folder.read_dir().expect("dir exists").count() == 0);

    let mut archive = Archive::new(archive_path).open_for_processing()?;
    let mut unpacked = 0;

    while let Some(header) = archive.read_header()? {
        let entry = header.entry();
        archive = if entry.is_file() {
            if !quiet {
                info!(
                    inaccessible,
                    "{} extracted. ({})",
                    entry.filename.display(),
                    entry.unpacked_size
                );
            }
            unpacked += 1;
            header.extract_with_base(output_folder)?
        } else {
            header.skip()?
        };
    }

    Ok(unpacked)
}

/// List contents of `archive_path`, returning a vector of archive entries
pub fn list_archive(archive_path: &Path) -> impl Iterator<Item = crate::Result<FileInArchive>> {
    Archive::new(archive_path)
        .open_for_listing()
        .expect("cannot open archive")
        .map(|item| {
            let item = item?;
            let is_dir = item.is_directory();
            let path = item.filename;

            Ok(FileInArchive { path, is_dir })
        })
}

pub fn no_compression() -> Error {
    Error::UnsupportedFormat {
        reason: "Creating RAR archives is not allowed due to licensing restrictions.".into(),
    }
}