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 запрос
if let Some((base_filename, requested_width, extension)) = parse_image_request(&normalized_path)
{
warn!("base_filename: {}", base_filename);
warn!("requested_width: {}", requested_width);
warn!("extension: {}", extension);
let (base_filename, requested_width, extension) = parse_image_request(&normalized_path);
let filekey = format!("{}.{}", base_filename, extension);
let content_type = match extension.as_str() {
"jpg" | "jpeg" => "image/jpeg",
@ -154,12 +151,7 @@ pub async fn proxy_handler(
}
},
Err(e) => Err(ErrorInternalServerError(e))
};
} else {
warn!("cant parse file request");
}
Err(ErrorInternalServerError("invalid file key"))
}
async fn thumbdata_save(
@ -171,7 +163,7 @@ async fn thumbdata_save(
let state = state.clone();
if content_type.starts_with("image") {
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);
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];
/// Парсит запрос на миниатюру, извлекая оригинальное имя файла и требуемую ширину.
/// Пример: "filename_150.ext" -> ("filename", 150, "ext")
/// unsafe/1440x/production/image/439efaa0-816f-11ef-b201-439da98539bc.jpg -> ("439efaa0-816f-11ef-b201-439da98539bc", 1440, "jpg")
pub fn parse_image_request(path: &str) -> Option<(String, u32, String)> {
/// Пример: "filename_150.ext" -> ("filename.ext", 150, "ext")
/// unsafe/1440x/production/image/439efaa0-816f-11ef-b201-439da98539bc.jpg -> ("439efaa0-816f-11ef-b201-439da98539bc.jpg", 1440, "jpg")
/// 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>>();
if let Some(filename_part) = path_parts.pop() {
let mut oldwidth = 0;
let filename_part = path_parts.pop().unwrap_or("");
let mut width = 0;
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 let Ok(width) = old_width_str.parse::<u32>() {
oldwidth = width;
if let Ok(w) = old_width_str.parse::<u32>() {
width = w;
}
}
}
}
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(width) = width_str.parse::<u32>() {
return Some((base_name.to_string(), max(width, oldwidth), ext_part.to_string()));
if let Ok(w) = width_str.parse::<u32>() {
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
}
/// Выбирает ближайший подходящий размер из предопределённых.