docs: add rustdoc to compression_engine module

Co-authored-by: aider (openai/andrew/openrouter/deepseek/deepseek-chat-v3.1) <aider@aider.chat>
This commit is contained in:
Andrew Phillips
2025-09-10 11:14:03 -03:00
parent 67a61093d5
commit 93d99a644a

View File

@@ -22,6 +22,7 @@ use crate::compression_engine::lz4::CompressionEngineLZ4;
use crate::compression_engine::none::CompressionEngineNone; use crate::compression_engine::none::CompressionEngineNone;
use crate::compression_engine::program::CompressionEngineProgram; use crate::compression_engine::program::CompressionEngineProgram;
/// Enum representing different compression types supported by the system
#[derive(Debug, Eq, PartialEq, Clone, strum::EnumIter, strum::Display, strum::EnumString, Enum)] #[derive(Debug, Eq, PartialEq, Clone, strum::EnumIter, strum::Display, strum::EnumString, Enum)]
#[strum(ascii_case_insensitive)] #[strum(ascii_case_insensitive)]
pub enum CompressionType { pub enum CompressionType {
@@ -33,14 +34,49 @@ pub enum CompressionType {
None, None,
} }
/// Trait defining the interface for compression engines
pub trait CompressionEngine { pub trait CompressionEngine {
/// Opens a compressed file for reading
///
/// # Arguments
///
/// * `file_path` - Path to the compressed file
///
/// # Returns
///
/// * `Result<Box<dyn Read>>` - A boxed reader that decompresses the file on read
fn open(&self, file_path: PathBuf) -> Result<Box<dyn Read>>; fn open(&self, file_path: PathBuf) -> Result<Box<dyn Read>>;
/// Creates a new compressed file for writing
///
/// # Arguments
///
/// * `file_path` - Path where the compressed file will be created
///
/// # Returns
///
/// * `Result<Box<dyn Write>>` - A boxed writer that compresses data on write
fn create(&self, file_path: PathBuf) -> Result<Box<dyn Write>>; fn create(&self, file_path: PathBuf) -> Result<Box<dyn Write>>;
/// Checks if this compression engine is supported on the current system
///
/// # Returns
///
/// * `bool` - True if supported, false otherwise
fn is_supported(&self) -> bool { fn is_supported(&self) -> bool {
true true
} }
/// Copies decompressed content from a file to a writer
///
/// # Arguments
///
/// * `file_path` - Path to the compressed file
/// * `writer` - Writer to receive decompressed content
///
/// # Returns
///
/// * `Result<()>` - Success or error
fn copy(&self, file_path: PathBuf, writer: &mut dyn Write) -> Result<()> { fn copy(&self, file_path: PathBuf, writer: &mut dyn Write) -> Result<()> {
let mut reader = self.open(file_path)?; let mut reader = self.open(file_path)?;
io::copy(&mut reader, writer)?; io::copy(&mut reader, writer)?;
@@ -48,11 +84,29 @@ pub trait CompressionEngine {
Ok(()) Ok(())
} }
/// Decompresses and outputs file content to stdout
///
/// # Arguments
///
/// * `file_path` - Path to the compressed file
///
/// # Returns
///
/// * `Result<()>` - Success or error
fn cat(&self, file_path: PathBuf) -> Result<()> { fn cat(&self, file_path: PathBuf) -> Result<()> {
let mut stdout = io::stdout().lock(); let mut stdout = io::stdout().lock();
self.copy(file_path, &mut stdout) self.copy(file_path, &mut stdout)
} }
/// Calculates the decompressed size of a file
///
/// # Arguments
///
/// * `file_path` - Path to the compressed file
///
/// # Returns
///
/// * `Result<usize>` - The decompressed size in bytes
fn size(&self, file_path: PathBuf) -> Result<usize> { fn size(&self, file_path: PathBuf) -> Result<usize> {
let mut reader = self.open(file_path)?; let mut reader = self.open(file_path)?;
let mut buffer = [0; libc::BUFSIZ as usize]; let mut buffer = [0; libc::BUFSIZ as usize];
@@ -73,6 +127,7 @@ pub trait CompressionEngine {
} }
lazy_static! { lazy_static! {
/// Mapping of compression types to their external program implementations
pub static ref COMPRESSION_PROGRAMS: EnumMap<CompressionType, Option<CompressionEngineProgram>> = enum_map! { pub static ref COMPRESSION_PROGRAMS: EnumMap<CompressionType, Option<CompressionEngineProgram>> = enum_map! {
CompressionType::LZ4 => { CompressionType::LZ4 => {
let program = CompressionEngineProgram::new("lz4", vec!["-c"], vec!["-d", "-c"]); let program = CompressionEngineProgram::new("lz4", vec!["-c"], vec!["-d", "-c"]);
@@ -104,6 +159,15 @@ lazy_static! {
}; };
} }
/// Gets a compression engine for the specified compression type
///
/// # Arguments
///
/// * `compression_type` - The type of compression to use
///
/// # Returns
///
/// * `Result<Box<dyn CompressionEngine>>` - A boxed compression engine instance
pub fn get_compression_engine( pub fn get_compression_engine(
compression_type: CompressionType, compression_type: CompressionType,
) -> Result<Box<dyn CompressionEngine>> { ) -> Result<Box<dyn CompressionEngine>> {
@@ -148,6 +212,11 @@ pub fn get_compression_engine(
} }
} }
/// Gets the default compression type based on available support
///
/// # Returns
///
/// * `CompressionType` - The first supported compression type found
pub fn default_compression_type() -> CompressionType { pub fn default_compression_type() -> CompressionType {
let mut default = CompressionType::None; let mut default = CompressionType::None;
for compression_type in CompressionType::iter() { for compression_type in CompressionType::iter() {