Cargo.lock

@@ -224,7 +224,7 @@ dependencies = [
 
 [[package]]
 name = "breeze"
-version = "0.2.8"
+version = "0.2.7"
 dependencies = [
  "anyhow",
  "argh",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "breeze"
-version = "0.2.8"
+version = "0.2.7"
 edition = "2021"
 
 [dependencies]

src/engine.rs

@@ -240,13 +240,13 @@ impl Engine {
     /// Generate a new saved name for an upload.
     ///
     /// If it picks a name that already exists, it will try again.
-    pub async fn gen_saved_name(&self, ext: Option<String>) -> String {
+    pub async fn gen_saved_name(&self, ext: &str) -> String {
         loop {
             // generate a 6-character alphanumeric string
             let mut saved_name: String = Alphanumeric.sample_string(&mut rand::thread_rng(), 6);
 
             // if we have an extension, add it now
-            if let Some(ref ext) = ext {
+            if !ext.is_empty() {
                 saved_name.push('.');
                 saved_name.push_str(ext);
             }
@@ -389,7 +389,7 @@ impl Engine {
 
     pub async fn process(
         &self,
-        ext: Option<String>,
+        ext: &str,
         provided_len: u64,
         stream: BodyDataStream,
         lifetime: Option<Duration>,

src/new.rs

@@ -1,9 +1,4 @@
-use std::{
+use std::{ffi::OsStr, path::PathBuf, sync::Arc, time::Duration};
-    ffi::OsStr,
-    path::{Path, PathBuf},
-    sync::Arc,
-    time::Duration,
-};
 
 use axum::{
     body::Body,
@@ -37,6 +32,7 @@ pub struct NewRequest {
 
 /// The request handler for the /new path.
 /// This handles all new uploads.
+#[axum::debug_handler]
 pub async fn new(
     State(engine): State<Arc<crate::engine::Engine>>,
     Query(req): Query<NewRequest>,
@@ -53,37 +49,11 @@ pub async fn new(
         return Err(StatusCode::BAD_REQUEST);
     }
 
-    // -- try to figure out a file extension..
+    let extension = PathBuf::from(req.name)
-
+        .extension()
-    fn extension(pb: &Path) -> Option<String> {
+        .and_then(OsStr::to_str)
-        pb.extension().and_then(OsStr::to_str).map(str::to_string)
+        .unwrap_or_default()
-    }
+        .to_string();
-
-    let pb = PathBuf::from(req.name);
-    let mut ext = extension(&pb);
-
-    // common extensions that usually have a second extension before themselves
-    const ADDITIVE: &[&str] = &["gz", "xz", "bz2", "lz4", "zst"];
-
-    // if the extension is one of those, try to find that second extension
-    if ext
-        .as_ref()
-        .is_some_and(|ext| ADDITIVE.contains(&ext.as_str()))
-    {
-        // try to parse out another extension
-        let stem = pb.file_stem().unwrap(); // SAFETY: if extension is Some(), this will also be
-
-        if let Some(second_ext) = extension(&PathBuf::from(stem)) {
-            // there is another extension,
-            // try to make sure it's one we want
-            // 4 is enough for most common file extensions
-            // and not many false positives, hopefully
-            if second_ext.len() <= 4 {
-                // seems ok so combine them
-                ext = ext.as_ref().map(|first_ext| second_ext + "." + first_ext);
-            }
-        }
-    }
 
     // turn body into stream
     let stream = Body::into_data_stream(body);
@@ -94,7 +64,13 @@ pub async fn new(
     // they don't expect the connection to close before they're done uploading, i think
     // so it will just present the user with a "connection closed" error
     match engine
-        .process(ext, content_length, stream, req.last_for, req.keep_exif)
+        .process(
+            &extension,
+            content_length,
+            stream,
+            req.last_for,
+            req.keep_exif,
+        )
         .await
     {
         Ok(outcome) => match outcome {

src/view.rs

@@ -86,6 +86,7 @@ impl IntoResponse for UploadResponse {
 
 /// GET request handler for /p/* path.
 /// All file views are handled here.
+#[axum::debug_handler]
 pub async fn view(
     State(engine): State<Arc<crate::engine::Engine>>,
     Path(original_path): Path<PathBuf>,