Upload-Limit in HEAD responses

tus · Nov 2, 2024 · a8bfbac · a8bfbac
1 parent a4287f3
commit a8bfbac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/handler/head_test.go b/pkg/handler/head_test.go
@@ -177,6 +177,7 @@ func TestHead(t *testing.T) {
 							"Upload-Draft-Interop-Version": interopVersion,
 							"Upload-Offset":                "5",
 							"Upload-Length":                "10",
+							"Upload-Limit":                 "min-size=0,max-size=10",
 						}, false, interopVersion),
 					}).Run(handler, t)
 				})
@@ -211,6 +212,7 @@ func TestHead(t *testing.T) {
 							"Upload-Draft-Interop-Version": interopVersion,
 							"Upload-Offset":                "10",
 							"Upload-Length":                "10",
+							"Upload-Limit":                 "min-size=0,max-size=10",
 						}, true, interopVersion),
 					}).Run(handler, t)
 				})
@@ -231,6 +233,7 @@ func TestHead(t *testing.T) {
 					handler, _ := NewHandler(Config{
 						StoreComposer:              composer,
 						EnableExperimentalProtocol: true,
+						MaxSize:                    400,
 					})
 
 					(&httpTest{
@@ -244,6 +247,7 @@ func TestHead(t *testing.T) {
 							"Upload-Draft-Interop-Version": interopVersion,
 							"Upload-Offset":                "5",
 							"Upload-Length":                "",
+							"Upload-Limit":                 "min-size=0,max-size=400",
 						}, false, interopVersion),
 					}).Run(handler, t)
 				})

diff --git a/pkg/handler/unrouted_handler.go b/pkg/handler/unrouted_handler.go
@@ -706,6 +706,8 @@ func (handler *UnroutedHandler) HeadFile(w http.ResponseWriter, r *http.Request)
 			resp.Header["Upload-Length"] = strconv.FormatInt(info.Size, 10)
 		}
 
+		resp.Header["Upload-Limit"] = handler.getIETFDraftUploadLimits(info)
+
 		// Draft -01 and -02 require a 204 No Content response. Version -03 allows 200 OK as well,
 		// but we stick to 204 to not make the logic less complex.
 		resp.StatusCode = http.StatusNoContent