diff --git a/src/thumbnail.rs b/src/thumbnail.rs index b3aa337..a7f6cad 100644 --- a/src/thumbnail.rs +++ b/src/thumbnail.rs @@ -1,23 +1,60 @@ use actix_web::error::ErrorInternalServerError; use image::{imageops::FilterType, DynamicImage}; -use std::{cmp::max, collections::HashMap, io::Cursor}; +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") +/// Парсит запрос на миниатюру, извлекая оригинальное имя файла, требуемую ширину и формат. +/// Примеры: +/// - "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 = path.rsplit('/').collect::>(); - let filename_part = path_parts.pop().unwrap_or(""); + 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.starts_with("unsafe") { - if let Some(old_width_str) = path_parts.get(1) { - let mut old_width_str = old_width_str.to_string(); - if old_width_str.ends_with('x') { - old_width_str.pop(); + 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::() { + 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::() { width = w; } @@ -25,25 +62,7 @@ pub fn parse_image_request(path: &str) -> (String, u32, String) { } } - if let Some((name_part, ext_part)) = filename_part.rsplit_once('.') { - if let Some((base_name, width_str)) = name_part.rsplit_once('_') { - if let Ok(w) = width_str.parse::() { - return ( - base_name.to_string(), - max(w, width), - ext_part.to_string(), - ); - } - } - ( - name_part.to_string(), - width, - ext_part.to_string().to_lowercase(), - ) - } else { - // Если расширение отсутствует, возвращаем имя файла как есть, ширину 0 и пустую строку для расширения - (filename_part.to_string(), 0, "".to_string()) - } + (base_filename, width, extension) } /// Выбирает ближайший подходящий размер из предопределённых.