From 076f5564ee376656ce58be34df67d927771303ad Mon Sep 17 00:00:00 2001
From: daimond113 <72147841+daimond113@users.noreply.github.com>
Date: Fri, 10 Jan 2025 16:06:27 +0100
Subject: [PATCH] feat(registry): set content-length header for fs storage

---
 registry/CHANGELOG.md      | 1 +
 registry/src/storage/fs.rs | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/registry/CHANGELOG.md b/registry/CHANGELOG.md
index 92abb8c..291b502 100644
--- a/registry/CHANGELOG.md
+++ b/registry/CHANGELOG.md
@@ -11,6 +11,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Add yanking & deprecating to registry by @daimond113
 - Log more information about configured auth & storage by @daimond113
 - Add individual endpoints for package data over using `Accept` header conditional returns by @daimond113
+- Set `Content-Length` header for FS storage backend by @daimond113
 
 ### Performance
 - Switch to using a `RwLock` over a `Mutex` to store repository data by @daimond113
diff --git a/registry/src/storage/fs.rs b/registry/src/storage/fs.rs
index 25ce90a..e99767a 100644
--- a/registry/src/storage/fs.rs
+++ b/registry/src/storage/fs.rs
@@ -1,6 +1,6 @@
 use crate::{error::RegistryError, storage::StorageImpl};
 use actix_web::{
-	http::header::{CONTENT_ENCODING, CONTENT_TYPE},
+	http::header::{CONTENT_ENCODING, CONTENT_LENGTH, CONTENT_TYPE},
 	HttpResponse,
 };
 use fs_err::tokio as fs;
@@ -24,6 +24,7 @@ async fn read_file_to_response(
 		Ok(file) => HttpResponse::Ok()
 			.append_header((CONTENT_TYPE, content_type))
 			.append_header((CONTENT_ENCODING, "gzip"))
+			.append_header((CONTENT_LENGTH, file.metadata().await?.len()))
 			.streaming(ReaderStream::new(file)),
 		Err(e) if e.kind() == std::io::ErrorKind::NotFound => HttpResponse::NotFound().finish(),
 		Err(e) => return Err(e.into()),