moosicbox_image 0.2.0

//! Image resizing command-line tool.
//!
//! This binary provides a command-line interface for resizing images using
//! the `moosicbox_image` library. It supports various encodings and quality settings.

#![cfg_attr(feature = "fail-on-warnings", deny(warnings))]
#![warn(clippy::all, clippy::pedantic, clippy::nursery, clippy::cargo)]
#![allow(clippy::multiple_crate_versions)]

use std::{
    io::{Seek as _, Write as _},
    path::Path,
    str::FromStr,
};

use clap::Parser;
use moosicbox_image::Encoding;
use thiserror::Error;

mod image;

#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
    #[arg(index = 1)]
    file: String,

    #[arg(short, long)]
    output: String,

    #[arg(short, long)]
    encoding: Option<String>,

    #[arg(short, long)]
    width: Option<u32>,

    #[arg(short, long)]
    height: Option<u32>,

    #[arg(short, long, default_value_t = 80)]
    quality: u8,
}

#[switchy_async::main]
async fn main() {
    let args = Args::parse();

    let encoding = args
        .encoding
        .map(|x| Encoding::from_str(&x.to_uppercase()))
        .transpose()
        .expect("Invalid encoding {}");

    try_resize_local_file(
        args.width,
        args.height,
        &args.file,
        &args.output,
        encoding,
        args.quality,
    )
    .await
    .expect("Failed to compress image");
}

/// Error type for local file resizing operations.
#[derive(Error, Debug)]
pub enum ResizeLocalFileError {
    /// Image resize error from the image module.
    #[error(transparent)]
    ResizeImage(#[from] image::ResizeImageError),
    /// Image processing error from the `image` crate.
    #[error(transparent)]
    Image(#[from] ::image::error::ImageError),
    /// I/O error during file operations.
    #[error(transparent)]
    IO(#[from] std::io::Error),
    /// Generic failure during resize operation.
    #[error("Failed")]
    Failed,
}

/// Resizes a local image file with optional width and height constraints.
///
/// If only one dimension is provided, the other is calculated to maintain aspect ratio.
/// If neither dimension is provided, the original dimensions are used.
///
/// # Errors
///
/// * [`ResizeLocalFileError::Image`] - If the image file cannot be opened or decoded
/// * [`ResizeLocalFileError::ResizeImage`] - If the image resize operation fails
/// * [`ResizeLocalFileError::IO`] - If file I/O operations fail
/// * [`ResizeLocalFileError::Failed`] - If the resize operation returns no data
async fn try_resize_local_file(
    width: Option<u32>,
    height: Option<u32>,
    path: &str,
    output: &str,
    encoding: Option<Encoding>,
    quality: u8,
) -> Result<(), ResizeLocalFileError> {
    let (width, height) = if let (Some(width), Some(height)) = (width, height) {
        (width, height)
    } else {
        let file_path = Path::new(path);
        let reader = ::image::ImageReader::open(file_path)?;
        let dimensions = reader.into_dimensions()?;

        width.map_or_else(
            || {
                height.map_or_else(
                    || {
                        (
                            width.unwrap_or(dimensions.0),
                            height.unwrap_or(dimensions.1),
                        )
                    },
                    |height| {
                        #[allow(clippy::cast_sign_loss, clippy::cast_possible_truncation)]
                        (
                            (f64::from(dimensions.0)
                                * (f64::from(height) / f64::from(dimensions.1)))
                            .round() as u32,
                            height,
                        )
                    },
                )
            },
            |width| {
                #[allow(clippy::cast_sign_loss, clippy::cast_possible_truncation)]
                (
                    width,
                    (f64::from(dimensions.1) * (f64::from(width) / f64::from(dimensions.0))).round()
                        as u32,
                )
            },
        )
    };

    let output = Path::new(output);

    let encoding = encoding.unwrap_or_else(|| {
        output
            .extension()
            .and_then(|ext| ext.to_ascii_uppercase().to_str().map(str::to_string))
            .and_then(|ext| Encoding::from_str(&ext).ok())
            .unwrap_or_else(|| {
                log::debug!("Defaulting encoding to Jpeg");
                Encoding::Jpeg
            })
    });

    log::debug!(
        "Resizing local image file path={path} width={width} height={height} encoding={encoding} quality={quality}"
    );

    let bytes = image::try_resize_local_file_async(width, height, path, encoding, quality)
        .await?
        .ok_or(ResizeLocalFileError::Failed)?;

    save_bytes_to_file(&bytes, output, None)?;

    Ok(())
}

/// Saves bytes to a file, creating parent directories if needed.
///
/// # Errors
///
/// * [`std::io::Error`] - If parent directory creation fails
/// * [`std::io::Error`] - If file cannot be opened or written to
/// * [`std::io::Error`] - If seeking to the start position fails
///
/// # Panics
///
/// Panics if the path has no parent directory.
fn save_bytes_to_file(bytes: &[u8], path: &Path, start: Option<u64>) -> Result<(), std::io::Error> {
    std::fs::create_dir_all(path.parent().expect("No parent directory"))?;

    let file = switchy_fs::sync::OpenOptions::new()
        .create(true)
        .write(true)
        .truncate(start.is_none_or(|start| start == 0))
        .open(path)?;

    let mut writer = std::io::BufWriter::new(file);

    if let Some(start) = start {
        writer.seek(std::io::SeekFrom::Start(start))?;
    }

    writer.write_all(bytes)
}