From d660882b8d9632a29243644c7f63930c1a8c1e72 Mon Sep 17 00:00:00 2001
From: Jean-Michel Rouet <36963549+jmrouet@users.noreply.github.com>
Date: Tue, 11 Mar 2025 15:54:30 +0100
Subject: [PATCH] publish with sized stream to comply with WSGI pypi server
 constraints. (#12111)

<!--
Thank you for contributing to uv! To help us out with reviewing, please
consider the following:

- Does this pull request include a summary of the change? (See below.)
- Does this pull request include a descriptive title?
- Does this pull request include references to any relevant issues?
-->

## Summary

This PR is meant to fix issue #11862

It allows to send sized bodies during `publish`

<!-- What's the purpose of the change? What does it do, and why? -->

## Test Plan

the PR was tested on the MRE from #11862

<!-- How was it tested? -->
---
 crates/uv-publish/src/lib.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/crates/uv-publish/src/lib.rs b/crates/uv-publish/src/lib.rs
index 6cd51cbd8..db0290953 100644
--- a/crates/uv-publish/src/lib.rs
+++ b/crates/uv-publish/src/lib.rs
@@ -754,7 +754,8 @@ async fn build_request(
     }
 
     let file = File::open(file).await?;
-    let idx = reporter.on_upload_start(&filename.to_string(), Some(file.metadata().await?.len()));
+    let file_size = file.metadata().await?.len();
+    let idx = reporter.on_upload_start(&filename.to_string(), Some(file_size));
     let reader = ProgressReader::new(file, move |read| {
         reporter.on_upload_progress(idx, read as u64);
     });
@@ -762,7 +763,7 @@ async fn build_request(
     // a lifetime) -> callback needs to be static -> reporter reference needs to be Arc'd.
     let file_reader = Body::wrap_stream(ReaderStream::new(reader));
     // See [`files_for_publishing`] on `raw_filename`
-    let part = Part::stream(file_reader).file_name(raw_filename.to_string());
+    let part = Part::stream_with_length(file_reader, file_size).file_name(raw_filename.to_string());
     form = form.part("content", part);
 
     let url = if let Some(username) = username {
@@ -988,6 +989,7 @@ mod tests {
                 },
                 headers: {
                     "content-type": "multipart/form-data; boundary=[...]",
+                    "content-length": "6803",
                     "accept": "application/json;q=0.9, text/plain;q=0.8, text/html;q=0.7",
                     "authorization": "Basic ZmVycmlzOkYzUlIhUw==",
                 },
@@ -1138,6 +1140,7 @@ mod tests {
                 },
                 headers: {
                     "content-type": "multipart/form-data; boundary=[...]",
+                    "content-length": "19330",
                     "accept": "application/json;q=0.9, text/plain;q=0.8, text/html;q=0.7",
                     "authorization": "Basic ZmVycmlzOkYzUlIhUw==",
                 },