repartee 0.9.1

//! Image preview support — renders inline image previews in the terminal.
//!
//! Orchestrates the image preview pipeline: URL detection, async fetching,
//! disk caching, image decoding, and protocol encoding for ratatui-image.

pub mod cache;
pub mod detect;
pub mod fetch;

use std::io::Cursor;

use image::ImageReader;
use ratatui_image::picker::Picker;
use ratatui_image::protocol::StatefulProtocol;
use tokio::sync::mpsc;
use tracing::{error, warn};

use crate::config::ImagePreviewConfig;

// ---------------------------------------------------------------------------
// Preview status (lives on App, driven by events from background tasks)
// ---------------------------------------------------------------------------

/// Current state of the image preview overlay.
#[derive(Default)]
pub enum PreviewStatus {
    /// No preview is active.
    #[default]
    Hidden,
    /// A preview is being fetched/decoded in the background.
    Loading { url: String },
    /// The image is ready to display.
    Ready {
        url: String,
        title: Option<String>,
        /// Pre-encoded image for the ratatui-image widget.
        image: Box<StatefulProtocol>,
        /// Raw PNG bytes for direct-write path (iTerm2+tmux).
        raw_png: Vec<u8>,
        /// Width in terminal cells (including border).
        width: u16,
        /// Height in terminal cells (including border).
        height: u16,
        /// Whether the direct-write (stdout) path has already sent the image.
        /// Prevents re-sending ~500KB of PNG on every render frame.
        direct_written: bool,
    },
    /// Fetching or decoding failed.
    Error { url: String, message: String },
}

// ---------------------------------------------------------------------------
// Events sent from background tasks back to the main loop
// ---------------------------------------------------------------------------

/// Result of an async image preview task, sent via channel.
pub enum ImagePreviewEvent {
    /// Image is ready to display.
    Ready {
        url: String,
        title: Option<String>,
        image: Box<StatefulProtocol>,
        raw_png: Vec<u8>,
        width: u16,
        height: u16,
    },
    /// Fetching or decoding failed.
    Error { url: String, message: String },
}

// ---------------------------------------------------------------------------
// Spawn a background task to fetch, cache, decode, and encode an image
// ---------------------------------------------------------------------------

/// Spawn an async task to fetch, cache, and encode an image for preview.
///
/// Results are sent back via the provided channel. The caller should set the
/// preview status to `Loading` before calling this.
///
/// The pipeline is split into two phases:
/// 1. **Async I/O** (fetch from network or read from disk cache) — runs in the
///    tokio async context with `.await`.
/// 2. **CPU-bound** (image decode + protocol encode) — runs in
///    `spawn_blocking` to avoid blocking the async runtime.
///
/// # Arguments
///
/// * `url` - The image URL to fetch.
/// * `config` - Image preview configuration (limits, timeouts).
/// * `picker` - The ratatui-image picker (cloned into the task).
/// * `http_client` - Shared reqwest client.
/// * `tx` - Channel sender for delivering results.
/// * `term_size` - Terminal dimensions `(cols, rows)` for sizing the popup.
pub fn spawn_preview(
    url: &str,
    config: &ImagePreviewConfig,
    picker: &Picker,
    http_client: &reqwest::Client,
    tx: mpsc::Sender<ImagePreviewEvent>,
    term_size: (u16, u16),
) {
    let config = config.clone();
    let picker = picker.clone();
    let client = http_client.clone();
    let url = url.to_owned();

    tokio::spawn(async move {
        // Phase 1: Async I/O — fetch image bytes (network or disk cache).
        let fetch_result = fetch_image_data(&url, &config, &client).await;

        let event = match fetch_result {
            Ok((data, title)) => {
                // Phase 2: CPU-bound — decode image + encode for terminal protocol.
                // Run in spawn_blocking to avoid blocking the async runtime.
                let decode_result = tokio::task::spawn_blocking(move || {
                    decode_and_encode(&data, &config, &picker, term_size)
                })
                .await;

                match decode_result {
                    Ok(Ok((protocol, png_buf, width, height))) => ImagePreviewEvent::Ready {
                        url,
                        title,
                        image: Box::new(protocol),
                        raw_png: png_buf,
                        width,
                        height,
                    },
                    Ok(Err(e)) => {
                        error!(url = %url, error = %e, "image preview decode failed");
                        ImagePreviewEvent::Error {
                            url,
                            message: e.to_string(),
                        }
                    }
                    Err(e) => {
                        error!(url = %url, error = %e, "image preview task panicked");
                        ImagePreviewEvent::Error {
                            url,
                            message: e.to_string(),
                        }
                    }
                }
            }
            Err(e) => {
                error!(url = %url, error = %e, "image preview fetch failed");
                ImagePreviewEvent::Error {
                    url,
                    message: e.to_string(),
                }
            }
        };

        if tx.send(event).await.is_err() {
            warn!("image preview channel closed before result could be sent");
        }
    });
}

/// Phase 1: Fetch image bytes from the network or disk cache (async I/O).
///
/// Returns the raw image data and an optional title extracted from the URL.
async fn fetch_image_data(
    url: &str,
    config: &ImagePreviewConfig,
    client: &reqwest::Client,
) -> color_eyre::eyre::Result<(Vec<u8>, Option<String>)> {
    // 1. Check the disk cache first.
    if let Some(cached_path) = cache::is_cached(url) {
        let data = tokio::fs::read(&cached_path).await?;
        let title = detect::classify_url(url).and_then(|c| c.title);
        return Ok((data, title));
    }

    // 2. Fetch from network (async).
    let fetch_config = fetch::FetchConfig {
        timeout_secs: config.fetch_timeout,
        max_file_size: config.max_file_size,
    };
    let result = fetch::fetch_image(url, &fetch_config, client).await?;

    // 3. Validate magic bytes.
    if !cache::validate_magic_bytes(&result.data) {
        return Err(color_eyre::eyre::eyre!(
            "downloaded data does not appear to be a valid image"
        ));
    }

    // 4. Store in cache.
    if let Err(e) = cache::store(url, &result.data, &result.content_type) {
        warn!(url, error = %e, "failed to cache image");
    }

    let title = detect::classify_url(url).and_then(|c| c.title);
    Ok((result.data, title))
}

/// Phase 2: Decode image and encode for terminal display (CPU-bound).
///
/// Called inside `spawn_blocking` because image decoding and protocol encoding
/// are CPU-intensive operations.
/// Returns (protocol, `raw_png`, width, height).
type DecodeResult = (StatefulProtocol, Vec<u8>, u16, u16);

fn decode_and_encode(
    data: &[u8],
    config: &ImagePreviewConfig,
    picker: &Picker,
    term_size: (u16, u16),
) -> color_eyre::eyre::Result<DecodeResult> {
    // 5. Decode the image.
    let dyn_img = ImageReader::new(Cursor::new(data))
        .with_guessed_format()?
        .decode()?;

    // 6. Calculate display dimensions using the picker's font_size so the
    //    popup matches what ratatui-image's StatefulImage will actually render.
    let font_size = picker.font_size();
    tracing::debug!(
        img_w = dyn_img.width(),
        img_h = dyn_img.height(),
        term_cols = term_size.0,
        term_rows = term_size.1,
        font_w = font_size.0,
        font_h = font_size.1,
        protocol = ?picker.protocol_type(),
        "image decode: input dimensions"
    );
    let (width, height) = calculate_display_size(config, term_size, &dyn_img, font_size);
    tracing::debug!(
        popup_w = width,
        popup_h = height,
        inner_w = width.saturating_sub(2),
        inner_h = height.saturating_sub(2),
        "image decode: popup dimensions"
    );

    // 7. Encode as PNG for the direct-write path (iTerm2+tmux).
    let mut png_buf: Vec<u8> = Vec::new();
    dyn_img.write_to(&mut Cursor::new(&mut png_buf), image::ImageFormat::Png)?;

    // 8. Create the protocol image via the picker.
    let protocol = picker.new_resize_protocol(dyn_img);

    Ok((protocol, png_buf, width, height))
}

/// Calculate the popup dimensions in terminal cells.
///
/// The popup includes a 1-cell border on each side, so the inner image area
/// is `(width - 2, height - 2)`. The image is scaled to fit while preserving
/// its aspect ratio.
///
/// Uses the picker's `font_size` to convert between pixels and cells — this
/// must match what `ratatui-image`'s `StatefulImage` uses internally, otherwise
/// the popup will be larger than the rendered image (empty space) or smaller
/// (image clipped).
fn calculate_display_size(
    config: &ImagePreviewConfig,
    term_size: (u16, u16),
    img: &image::DynamicImage,
    font_size: (u16, u16),
) -> (u16, u16) {
    let max_cols = if config.max_width > 0 {
        u16::try_from(config.max_width).unwrap_or(u16::MAX)
    } else {
        term_size.0 * 3 / 4
    };

    let max_rows = if config.max_height > 0 {
        u16::try_from(config.max_height).unwrap_or(u16::MAX)
    } else {
        term_size.1 * 3 / 4
    };

    // Reserve 2 cells on each axis for the border.
    let inner_cols = max_cols.saturating_sub(2).max(1);
    let inner_rows = max_rows.saturating_sub(2).max(1);

    let img_w = img.width();
    let img_h = img.height();

    if img_w == 0 || img_h == 0 {
        return (max_cols.min(10), max_rows.min(5));
    }

    let fw = f64::from(font_size.0.max(1));
    let fh = f64::from(font_size.1.max(1));

    // Convert image pixel dimensions to cell dimensions at 1:1 pixel mapping.
    // This matches ratatui-image's Resize::Fit, which doesn't upscale beyond
    // the image's native size in cells.
    let img_cells_w = (f64::from(img_w) / fw).ceil();
    let img_cells_h = (f64::from(img_h) / fh).ceil();

    // Fit the image into the available inner area, preserving aspect ratio.
    // Never upscale — cap at the image's natural cell size.
    let fitted_cols = img_cells_w.min(f64::from(inner_cols));
    let fitted_rows = img_cells_h.min(f64::from(inner_rows));

    // If the image doesn't fit in one dimension, scale down proportionally.
    let scale_x = fitted_cols / img_cells_w;
    let scale_y = fitted_rows / img_cells_h;
    let scale = scale_x.min(scale_y);

    #[expect(
        clippy::cast_possible_truncation,
        clippy::cast_sign_loss,
        reason = "dimensions are small positive values; truncation is intentional"
    )]
    let final_cols = (img_cells_w * scale).round().max(1.0) as u16;
    #[expect(
        clippy::cast_possible_truncation,
        clippy::cast_sign_loss,
        reason = "dimensions are small positive values; truncation is intentional"
    )]
    let final_rows = (img_cells_h * scale).round().max(1.0) as u16;

    // Add the border back.
    (final_cols + 2, final_rows + 2)
}