quoter/src/thumbnail.rs

use actix_web::error::ErrorInternalServerError;
use image::{imageops::FilterType, DynamicImage};
use std::{collections::HashMap, io::Cursor};

pub const THUMB_WIDTHS: [u32; 7] = [10, 40, 110, 300, 600, 800, 1400];

/// Парсит запрос на миниатюру, извлекая оригинальное имя файла, требуемую ширину и формат.
/// Примеры:
/// - "filename_150.ext" -> ("filename", 150, "ext")
/// - "unsafe/1440x/production/image/439efaa0-816f-11ef-b201-439da98539bc.jpg" -> ("439efaa0-816f-11ef-b201-439da98539bc", 1440, "jpg")
/// - "unsafe/production/image/5627e002-0c53-11ee-9565-0242ac110006.png" -> ("5627e002-0c53-11ee-9565-0242ac110006", 0, "png")
/// - "unsafe/development/image/439efaa0-816f-11ef-b201-439da98539bc.jpg/webp" -> ("439efaa0-816f-11ef-b201-439da98539bc", 0, "webp")
pub fn parse_image_request(path: &str) -> (String, u32, String) {
    let mut path_parts: Vec<&str> = path.split('/').collect();
    let mut extension = String::new();
    let mut width = 0;
    let mut base_filename = String::new();

    if path_parts.is_empty() {
        return (path.to_string(), width, extension);
    }

    // Проверка, содержит ли последний сегмент известный формат
    let known_formats = vec!["jpg", "jpeg", "png", "webp", "gif", "mp3", "wav", "ogg", "aac", "m4a", "flac"];
    let last_part = path_parts.last().unwrap().to_lowercase();
    if known_formats.contains(&last_part.as_str()) {
        extension = last_part.clone();
        path_parts.pop(); // Удаляем формат из пути
    } else {
        // Если формат не указан отдельно, пытаемся извлечь из имени файла
        if let Some(filename_part) = path_parts.pop() {
            if let Some((_, ext_part)) = filename_part.rsplit_once('.') {
                extension = ext_part.to_lowercase();
                path_parts.push(filename_part); // Возвращаем имя файла обратно
            }
        }
    }

    // Извлечение имени файла и ширины
    if let Some(filename_part) = path_parts.pop() {
        if let Some((name_part, width_str)) = filename_part.rsplit_once('_') {
            if let Ok(w) = width_str.parse::<u32>() {
                width = w;
                base_filename = name_part.to_string();
            } else {
                base_filename = filename_part.to_string();
            }
        } else {
            base_filename = filename_part.to_string();
        }
    }

    // Проверка на старую ширину в путях, начинающихся с "unsafe"
    if path.starts_with("unsafe") && width == 0 {
        if path_parts.len() >= 2 {
            if let Some(old_width_str) = path_parts.get(path_parts.len() - 1) {
                let old_width_str = old_width_str.trim_end_matches('x');
                if let Ok(w) = old_width_str.parse::<u32>() {
                    width = w;
                }
            }
        }
    }

    (base_filename, width, extension)
}

/// Выбирает ближайший подходящий размер из предопределённых.
pub fn find_closest_width(requested_width: u32) -> u32 {
    *THUMB_WIDTHS
        .iter()
        .min_by_key(|&&width| (width as i32 - requested_width as i32).abs())
        .unwrap_or(&THUMB_WIDTHS[0]) // Возвращаем самый маленький размер, если ничего не подошло
}

/// Генерирует миниатюры изображения.
pub async fn generate_thumbnails(image: &DynamicImage) -> Result<HashMap<u32, Vec<u8>>, actix_web::Error> {
    let mut thumbnails = HashMap::new();

    for &width in THUMB_WIDTHS.iter().filter(|&&w| w < image.width()) {
        let thumbnail = image.resize(width, u32::MAX, FilterType::Lanczos3); // Ресайз изображения по ширине
        let mut buffer = Vec::new();
        thumbnail
            .write_to(&mut Cursor::new(&mut buffer), image::ImageFormat::Jpeg)
            .map_err(|_| ErrorInternalServerError("Failed to generate thumbnail"))?; // Сохранение изображения в формате JPEG
        thumbnails.insert(width, buffer);
    }

    Ok(thumbnails)
}