quoter/src/thumbnail.rs

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

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_file_path(requested_path: &str) -> (String, u32, String) {
    let mut path = requested_path.to_string();
    if requested_path.ends_with("/webp") {
        path = path.replace("/webp", "");
    }
    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);
    }

    // пытаемся извлечь формат из имени файла
    if let Some(filename_part) = path_parts.pop() {
        if let Some((base, ext_part)) = filename_part.rsplit_once('.') {
            extension = ext_part.to_lowercase();
            base_filename = base.to_string(); // Устанавливаем base_filename без расширения
        } else {
            base_filename = filename_part.to_string();
        }
    }

    // Если base_filename ещё не установлено, извлекаем его
    if base_filename.is_empty() {
        if let Some(filename_part) = path_parts.pop() {
            if let Some((base, ext_part)) = filename_part.rsplit_once('.') {
                extension = ext_part.to_lowercase();
                base_filename = base.to_string();
            } else {
                base_filename = filename_part.to_string();
            }
        }
    }

    // Извлечение ширины из base_filename, если она есть
    if let Some((name_part, width_str)) = base_filename.rsplit_once('_') {
        if let Ok(w) = width_str.parse::<u32>() {
            width = w;
            base_filename = name_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(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 async fn generate_thumbnails(image: &DynamicImage) -> Result<HashMap<u32, Vec<u8>>, actix_web::Error> {
    let mut thumbnails = HashMap::new();
    let format = guess_format(&image.as_bytes()).unwrap();
    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), format)
            .map_err(|_| ErrorInternalServerError("Не удалось сгенерировать миниатюру"))?; // Сохранение изображения в указанном формате
        thumbnails.insert(width, buffer);
    }

    Ok(thumbnails)
}