unarc-rs 0.6.1

//! TBZ (tar.bz2) archive format support
//!
//! TBZ is a combination of TAR archive format with bzip2 compression.
//! Files with extensions .tar.bz2 or .tbz/.tbz2 are bzip2-compressed TAR archives.
//!
//! This module provides read-only access to TBZ archives by first decompressing
//! the bzip2 layer and then parsing the TAR content.

use std::io::{Cursor, Read};

use bzip2::read::BzDecoder;

use crate::error::Result;
use crate::tar::{TarArchive, TarFileHeader};

/// TBZ archive reader
///
/// This wraps a TAR archive after bzip2 decompression.
/// Note: The entire archive is decompressed into memory on construction.
pub struct TbzArchive {
    inner: TarArchive<Cursor<Vec<u8>>>,
}

impl TbzArchive {
    /// Create a new TBZ archive reader
    ///
    /// This will decompress the entire bzip2 stream into memory,
    /// then create a TAR archive reader from the decompressed data.
    pub fn new<T: Read>(reader: T) -> Result<Self> {
        // Decompress the bzip2 data
        let mut decoder = BzDecoder::new(reader);
        let mut decompressed = Vec::new();
        decoder
            .read_to_end(&mut decompressed)
            .map_err(|e| crate::error::ArchiveError::io_error(format!("Failed to decompress bzip2: {}", e)))?;

        // Create a cursor for the decompressed data
        let cursor = Cursor::new(decompressed);

        // Create TAR archive from decompressed data
        let inner = TarArchive::new(cursor)?;

        Ok(Self { inner })
    }

    /// Get the next entry in the archive
    pub fn get_next_entry(&mut self) -> Result<Option<TarFileHeader>> {
        self.inner.get_next_entry()
    }

    /// Skip the current entry
    pub fn skip(&mut self, header: &TarFileHeader) -> Result<()> {
        self.inner.skip(header)
    }

    /// Read the contents of the current entry
    pub fn read(&mut self, header: &TarFileHeader) -> Result<Vec<u8>> {
        self.inner.read(header)
    }

    /// Get the total number of entries
    pub fn entry_count(&self) -> usize {
        self.inner.entry_count()
    }
}