Add opencsg check back to git http handler

OpenCSGs · Dec 25, 2024 · 378004a · 378004a
1 parent 45af473
commit 378004a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/handler/git_http.go b/api/handler/git_http.go
@@ -175,6 +175,11 @@ func (h *GitHTTPHandler) LfsBatch(ctx *gin.Context) {
 		return
 	}
 
+	s3Internal := ctx.GetHeader("X-OPENCSG-S3-Internal")
+	if s3Internal == "true" {
+		ctx.Set("X-OPENCSG-S3-Internal", true)
+	}
+
 	objectResponse, err := h.gitHttp.BuildObjectResponse(ctx, batchRequest, isUpload)
 	if err != nil {
 		if errors.Is(err, component.ErrUnauthorized) {
@@ -235,6 +240,11 @@ func (h *GitHTTPHandler) LfsDownload(ctx *gin.Context) {
 	downloadRequest.CurrentUser = httpbase.GetCurrentUser(ctx)
 	downloadRequest.SaveAs = ctx.Query("save_as")
 
+	s3Internal := ctx.GetHeader("X-OPENCSG-S3-Internal")
+	if s3Internal == "true" {
+		ctx.Set("X-OPENCSG-S3-Internal", true)
+	}
+
 	url, err := h.gitHttp.LfsDownload(ctx, downloadRequest)
 	if err != nil {
 		httpbase.ServerError(ctx, err)