upload: Show upload size properly in log

Implement size calculation for streaming upload.

Signed-off-by: Denys Fedoryshchenko <denys.f@collabora.com>

Author: nuclearcat

src/azure.rs

@@ -83,7 +83,7 @@ async fn write_file_to_blob_streaming(
     data: &mut (dyn tokio::io::AsyncRead + Unpin + Send),
     cont_type: String,
     owner_email: Option<String>,
-) -> &'static str {
+) -> (&'static str, usize) {
     let azure_cfg = Arc::new(get_azure_credentials("azure"));
 
     let storage_account = azure_cfg.account.as_str();
@@ -177,7 +177,7 @@ async fn write_file_to_blob_streaming(
             eprintln!("Error uploading block list: {:?}", e);
         }
     }
-    "OK"
+    ("OK", total_bytes_uploaded)
 }
 
 /// Write file to Azure blob storage (legacy version using Vec<u8>)
@@ -557,9 +557,9 @@ impl super::Driver for AzureDriver {
         data: &mut (dyn tokio::io::AsyncRead + Unpin + Send),
         cont_type: String,
         owner_email: Option<String>,
-    ) -> String {
-        write_file_to_blob_streaming(filename.clone(), data, cont_type, owner_email).await;
-        filename
+    ) -> (String, usize) {
+        let (_status, size) = write_file_to_blob_streaming(filename.clone(), data, cont_type, owner_email).await;
+        (filename, size)
     }
     fn tag_file(
         &self,

src/local.rs

@@ -99,7 +99,7 @@ async fn write_file_to_local_streaming(
     data: &mut (dyn tokio::io::AsyncRead + Unpin + Send),
     cont_type: String,
     owner_email: Option<String>,
-) -> Result<String, String> {
+) -> Result<(String, usize), String> {
     let file_path = get_storage_file_path(&filename);
 
     // Ensure directory structure exists
@@ -140,7 +140,7 @@ async fn write_file_to_local_streaming(
     }
 
     debug_log!("File written to local storage: {}", file_path.display());
-    Ok(filename)
+    Ok((filename, total_bytes))
 }
 
 /// Write file to local storage (legacy version using Vec<u8>)
@@ -348,12 +348,12 @@ impl super::Driver for LocalDriver {
         data: &mut (dyn tokio::io::AsyncRead + Unpin + Send),
         cont_type: String,
         owner_email: Option<String>,
-    ) -> String {
+    ) -> (String, usize) {
         match write_file_to_local_streaming(filename.clone(), data, cont_type, owner_email).await {
-            Ok(_) => filename,
+            Ok((fname, size)) => (fname, size),
             Err(e) => {
                 eprintln!("Local storage streaming write error: {}", e);
-                String::new()
+                (String::new(), 0)
             }
         }
     }

src/main.rs

@@ -194,7 +194,7 @@ trait Driver: Send + Sync {
         data: &mut (dyn tokio::io::AsyncRead + Unpin + Send),
         cont_type: String,
         owner_email: Option<String>,
-    ) -> String;
+    ) -> (String, usize);
     fn get_file(&self, filename: String) -> ReceivedFile;
     fn tag_file(
         &self,
@@ -630,7 +630,7 @@ async fn ax_post_file(
                     let driver_name = get_driver_type();
                     let driver = init_driver(&driver_name);
 
-                    let result = driver.write_file_streaming(
+                    let (result, file_size) = driver.write_file_streaming(
                         full_path.clone(),
                         &mut field_stream,
                         content_type.to_string(),
@@ -651,7 +651,7 @@ async fn ax_post_file(
                         "{} {} {} {} {} {} {} {}",
                         client_ip,
                         status.as_u16(),
-                        "streaming",
+                        file_size,
                         human_time,
                         Method::POST,
                         request_target,