discours.io/quoter
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

thumbget-fix
All checks were successful
deploy / deploy (push) Successful in 1m1s
Details

Browse Source
This commit is contained in:
Untone 2024-10-23 15:11:08 +03:00
parent 4ee178686a
commit b8c5a15bec
1 changed files with 50 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
src/thumbnail.rs
View File

@ -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::<Vec<&str>>();
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::<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;
}
@ -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::<u32>() {
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)
}
/// Выбирает ближайший подходящий размер из предопределённых.