Write tar.gz when requesting to download a directory

backend/browser/handlers.go

@@ -1,6 +1,7 @@
 package browser
 
 import (
+	"fmt"
 	"io"
 	"log"
 	"net/http"
@@ -13,7 +14,8 @@ import (
 // GET /stream
 // StreamFile streams the content of the requested file
 func StreamFile(c *gin.Context) {
-	pathInDataDir := getAbsolutePath(c.Param("path"))
+	path := c.Param("path")
+	pathInDataDir := getAbsolutePath(path)
 
 	dir, err := isDir(pathInDataDir)
 	if err != nil {
@@ -24,6 +26,11 @@ func StreamFile(c *gin.Context) {
 	}
 
 	if dir {
+		c.Header(
+			"Content-Disposition",
+			fmt.Sprintf("attachment; filename=%s.tar.gz", path[1:len(path)-1]),
+		)
+		writeTarGz(pathInDataDir, c.Writer)
 	} else {
 		c.File(pathInDataDir)
 	}

backend/browser/utils.go

@@ -1,6 +1,9 @@
 package browser
 
 import (
+	"archive/tar"
+	"compress/gzip"
+	"io"
 	"io/fs"
 	"os"
 	"path/filepath"
@@ -76,6 +79,73 @@ func isDir(filepath string) (bool, error) {
 	return fileInfo.IsDir(), nil
 }
 
-func sendTarGz() error {
+func writeTarGz(filepath string, w io.Writer) error {
+	gzipWriter := gzip.NewWriter(w)
+	defer gzipWriter.Close()
+	tarWriter := tar.NewWriter(gzipWriter)
+	defer tarWriter.Close()
+
+	 writeDirectoryToTarGz(filepath, tarWriter, filepath)
+
+	return nil
+}
+
+func writeDirectoryToTarGz(directory string, tarWriter *tar.Writer, subPath string) error {
+	files, err := os.ReadDir(directory)
+	if err != nil {
+		return err
+	}
+
+	for _, file := range files {
+		currentPath := filepath.Join(directory, file.Name())
+		if file.IsDir() {
+			err := writeDirectoryToTarGz(currentPath, tarWriter, subPath)
+			if err != nil {
+				return err
+			}
+		} else {
+			fileInfo, err := file.Info()
+			if err != nil {
+				return err
+			}
+			err = writeFileToTarGz(currentPath, tarWriter, fileInfo, subPath)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
 	return nil
 }
+
+func writeFileToTarGz(path string, tarWriter *tar.Writer, fileInfo os.FileInfo, subPath string) error {
+	file, err := os.Open(path)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	path, err = filepath.EvalSymlinks(path)
+	if err != nil {
+		return err
+	}
+
+	subPath, err = filepath.EvalSymlinks(subPath)
+	if err != nil {
+		return err
+	}
+
+	header, err := tar.FileInfoHeader(fileInfo, path)
+	if err != nil {
+		return err
+	}
+	header.Name = path[len(subPath):]
+
+	err = tarWriter.WriteHeader(header)
+	if err != nil {
+		return err
+	}
+
+	_, err = io.Copy(tarWriter, file)
+	return err
+}