Fix missed method DownloadArchive

Author: ChristopherHX

routers/api/v1/repo/download.go

@@ -8,6 +8,7 @@ import (
 
 	"code.gitea.io/gitea/modules/git"
 	"code.gitea.io/gitea/modules/gitrepo"
+	"code.gitea.io/gitea/modules/setting"
 	"code.gitea.io/gitea/services/context"
 	archiver_service "code.gitea.io/gitea/services/repository/archiver"
 )
@@ -41,11 +42,18 @@ func DownloadArchive(ctx *context.APIContext) {
 		return
 	}
 
+	downloadName := prepareDownload(ctx, r)
+
+	if setting.Repository.StreamArchives {
+		streamDownload(ctx, downloadName, r)
+		return
+	}
+
 	archive, err := r.Await(ctx)
 	if err != nil {
 		ctx.APIErrorInternal(err)
 		return
 	}
 
-	download(ctx, r.GetArchiveName(), archive)
+	download(ctx, downloadName, archive)
 }

routers/api/v1/repo/file.go

@@ -307,6 +307,23 @@ func archiveDownload(ctx *context.APIContext) {
 		return
 	}
 
+	downloadName := prepareDownload(ctx, aReq)
+
+	if setting.Repository.StreamArchives {
+		streamDownload(ctx, downloadName, aReq)
+		return
+	}
+
+	archiver, err := aReq.Await(ctx)
+	if err != nil {
+		ctx.APIErrorInternal(err)
+		return
+	}
+
+	download(ctx, downloadName, archiver)
+}
+
+func prepareDownload(ctx *context.APIContext, aReq *archiver_service.ArchiveRequest) (downloadName string) {
 	// Add nix format link header so tarballs lock correctly:
 	// https://github.com/nixos/nix/blob/56763ff918eb308db23080e560ed2ea3e00c80a7/doc/manual/src/protocols/tarball-fetcher.md
 	ctx.Resp.Header().Add("Link", fmt.Sprintf(`<%s/archive/%s.%s?rev=%s>; rel="immutable"`,
@@ -316,26 +333,17 @@ func archiveDownload(ctx *context.APIContext) {
 		aReq.CommitID,
 	))
 
-	downloadName := ctx.Repo.Repository.Name + "-" + aReq.GetArchiveName()
-
-	if setting.Repository.StreamArchives {
-		ctx.SetServeHeaders(&context.ServeHeaderOptions{
-			Filename: downloadName,
-		})
+	return ctx.Repo.Repository.Name + "-" + aReq.GetArchiveName()
+}
 
-		if err := aReq.Stream(ctx, ctx.Repo.GitRepo, ctx.Resp); err != nil && !ctx.Written() {
-			ctx.APIErrorInternal(err)
-		}
-		return
-	}
+func streamDownload(ctx *context.APIContext, downloadName string, r *archiver_service.ArchiveRequest) {
+	ctx.SetServeHeaders(&context.ServeHeaderOptions{
+		Filename: downloadName,
+	})
 
-	archiver, err := aReq.Await(ctx)
-	if err != nil {
+	if err := r.Stream(ctx, ctx.Repo.GitRepo, ctx.Resp); err != nil && !ctx.Written() {
 		ctx.APIErrorInternal(err)
-		return
 	}
-
-	download(ctx, downloadName, archiver)
 }
 
 func download(ctx *context.APIContext, downloadName string, archiver *repo_model.RepoArchiver) {