forward more response parts

Author: ealmloff

packages/fullstack/src/server/mod.rs

@@ -274,13 +274,16 @@ where
 }
 
 fn apply_request_parts_to_response<B>(
-    headers: hyper::header::HeaderMap,
+    parts: &http::response::Parts,
     response: &mut axum::response::Response<B>,
 ) {
     let mut_headers = response.headers_mut();
-    for (key, value) in headers.iter() {
+    for (key, value) in parts.headers.iter() {
         mut_headers.insert(key, value.clone());
     }
+    *response.status_mut() = parts.status;
+    *response.version_mut() = parts.version;
+    *response.extensions_mut() = parts.extensions.clone();
 }
 
 fn add_server_context(server_context: &DioxusServerContext, context_providers: &ContextProviders) {
@@ -410,8 +413,8 @@ pub async fn render_handler(
         Ok((freshness, rx)) => {
             let mut response = axum::response::Html::from(Body::from_stream(rx)).into_response();
             freshness.write(response.headers_mut());
-            let headers = server_context.response_parts().headers.clone();
-            apply_request_parts_to_response(headers, &mut response);
+            let parts = server_context.response_parts();
+            apply_request_parts_to_response(&parts, &mut response);
             Result::<http::Response<axum::body::Body>, StatusCode>::Ok(response)
         }
         Err(SSRError::Incremental(e)) => {