get-request-parser-fix
All checks were successful
deploy / deploy (push) Successful in 1m1s

This commit is contained in:
Untone 2024-10-23 13:31:05 +03:00
parent 911284c8ae
commit d48141fee2
2 changed files with 156 additions and 152 deletions

View File

@ -20,11 +20,8 @@ pub async fn proxy_handler(
}; };
// парсим GET запрос // парсим GET запрос
if let Some((base_filename, requested_width, extension)) = parse_image_request(&normalized_path) let (base_filename, requested_width, extension) = parse_image_request(&normalized_path);
{
warn!("base_filename: {}", base_filename);
warn!("requested_width: {}", requested_width);
warn!("extension: {}", extension);
let filekey = format!("{}.{}", base_filename, extension); let filekey = format!("{}.{}", base_filename, extension);
let content_type = match extension.as_str() { let content_type = match extension.as_str() {
"jpg" | "jpeg" => "image/jpeg", "jpg" | "jpeg" => "image/jpeg",
@ -154,12 +151,7 @@ pub async fn proxy_handler(
} }
}, },
Err(e) => Err(ErrorInternalServerError(e)) Err(e) => Err(ErrorInternalServerError(e))
};
} else {
warn!("cant parse file request");
} }
Err(ErrorInternalServerError("invalid file key"))
} }
async fn thumbdata_save( async fn thumbdata_save(
@ -171,7 +163,7 @@ async fn thumbdata_save(
let state = state.clone(); let state = state.clone();
if content_type.starts_with("image") { if content_type.starts_with("image") {
warn!("original file name: {}", original_filename); warn!("original file name: {}", original_filename);
let (base_filename, _, extension) = parse_image_request(&original_filename).unwrap(); let (base_filename, _, extension) = parse_image_request(&original_filename);
let filename = format!("{}.{}", base_filename, extension); let filename = format!("{}.{}", base_filename, extension);
warn!("file extension: {}", extension); warn!("file extension: {}", extension);

View File

@ -5,33 +5,45 @@ use std::{cmp::max, collections::HashMap, io::Cursor};
pub const THUMB_WIDTHS: [u32; 7] = [10, 40, 110, 300, 600, 800, 1400]; pub const THUMB_WIDTHS: [u32; 7] = [10, 40, 110, 300, 600, 800, 1400];
/// Парсит запрос на миниатюру, извлекая оригинальное имя файла и требуемую ширину. /// Парсит запрос на миниатюру, извлекая оригинальное имя файла и требуемую ширину.
/// Пример: "filename_150.ext" -> ("filename", 150, "ext") /// Пример: "filename_150.ext" -> ("filename.ext", 150, "ext")
/// unsafe/1440x/production/image/439efaa0-816f-11ef-b201-439da98539bc.jpg -> ("439efaa0-816f-11ef-b201-439da98539bc", 1440, "jpg") /// unsafe/1440x/production/image/439efaa0-816f-11ef-b201-439da98539bc.jpg -> ("439efaa0-816f-11ef-b201-439da98539bc.jpg", 1440, "jpg")
pub fn parse_image_request(path: &str) -> Option<(String, u32, String)> { /// unsafe/production/image/5627e002-0c53-11ee-9565-0242ac110006.png -> ("5627e002-0c53-11ee-9565-0242ac110006.png", 0, "png")
pub fn parse_image_request(path: &str) -> (String, u32, String) {
let mut path_parts = path.rsplit('/').collect::<Vec<&str>>(); let mut path_parts = path.rsplit('/').collect::<Vec<&str>>();
if let Some(filename_part) = path_parts.pop() { let filename_part = path_parts.pop().unwrap_or("");
let mut oldwidth = 0; let mut width = 0;
if path.starts_with("unsafe") { if path.starts_with("unsafe") {
if let Some(old_width_str) = path_parts.get(1) { if let Some(old_width_str) = path_parts.get(1) {
let mut old_width_str = old_width_str.to_string(); let mut old_width_str = old_width_str.to_string();
if old_width_str.ends_with('x') { if old_width_str.ends_with('x') {
old_width_str.pop(); old_width_str.pop();
if let Ok(width) = old_width_str.parse::<u32>() { if let Ok(w) = old_width_str.parse::<u32>() {
oldwidth = width; width = w;
} }
} }
} }
} }
if let Some((name_part, ext_part)) = filename_part.rsplit_once('.') { if let Some((name_part, ext_part)) = filename_part.rsplit_once('.') {
if let Some((base_name, width_str)) = name_part.rsplit_once('_') { if let Some((base_name, width_str)) = name_part.rsplit_once('_') {
if let Ok(width) = width_str.parse::<u32>() { if let Ok(w) = width_str.parse::<u32>() {
return Some((base_name.to_string(), max(width, oldwidth), ext_part.to_string())); return (
format!("{}.{}", base_name, ext_part),
max(w, width),
ext_part.to_string(),
);
} }
} }
return Some((name_part.to_string(), 0, ext_part.to_string().to_lowercase())); (
format!("{}.{}", name_part, ext_part),
width,
ext_part.to_string().to_lowercase(),
)
} else {
// Если расширение отсутствует, возвращаем имя файла как есть, ширину 0 и пустую строку для расширения
(filename_part.to_string(), width, "".to_string())
} }
}
None
} }
/// Выбирает ближайший подходящий размер из предопределённых. /// Выбирает ближайший подходящий размер из предопределённых.