feat: add zip download for folders in file manager

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

internal/controllers/file_controller.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"time"
 
 	"github.com/engigu/baihu-panel/internal/utils"
 
@@ -492,3 +493,50 @@ func (fc *FileController) DownloadFile(c *gin.Context) {
 	c.Header("Content-Type", "application/octet-stream")
 	c.File(fullPath)
 }
+
+func (fc *FileController) DownloadZip(c *gin.Context) {
+	paths := c.QueryArray("path")
+	if len(paths) == 0 || c.ContentType() == "application/json" {
+		var req struct {
+			Paths []string `json:"paths"`
+		}
+		if err := c.ShouldBindJSON(&req); err == nil && len(paths) == 0 {
+			paths = req.Paths
+		}
+	}
+
+	if len(paths) == 0 {
+		utils.BadRequest(c, "path参数必填")
+		return
+	}
+
+	validatedAbsPaths := make([]string, 0, len(paths))
+	for _, path := range paths {
+		fullPath, safe := fc.checkPath(path, false)
+		if !safe {
+			utils.Forbidden(c, "访问被拒绝")
+			return
+		}
+
+		if _, err := os.Stat(fullPath); err != nil {
+			utils.NotFound(c, "文件不存在")
+			return
+		}
+
+		validatedAbsPaths = append(validatedAbsPaths, fullPath)
+	}
+
+	fileName := "baihu-export-" + time.Now().Format("20060102-150405") + ".zip"
+	if len(validatedAbsPaths) == 1 {
+		fileName = filepath.Base(validatedAbsPaths[0]) + ".zip"
+	}
+
+	c.Header("Content-Description", "File Transfer")
+	c.Header("Content-Transfer-Encoding", "binary")
+	c.Header("Content-Disposition", "attachment; filename="+fileName)
+	c.Header("Content-Type", "application/zip")
+
+	if err := utils.CreateZip(c.Writer, validatedAbsPaths); err != nil {
+		return
+	}
+}

internal/router/api_routes.go

@@ -129,6 +129,7 @@ func registerFileRoutes(g *gin.RouterGroup, c *Controllers) {
 		files.GET("/tree", c.File.GetFileTree)
 		files.GET("/content", c.File.GetFileContent)
 		files.GET("/download", c.File.DownloadFile)
+		files.GET("/download-zip", c.File.DownloadZip)
 		files.POST("/content", c.File.SaveFileContent)
 		files.POST("/create", c.File.CreateFile)
 		files.POST("/delete", c.File.DeleteFile)

internal/utils/archive.go

@@ -5,11 +5,109 @@ import (
 	"archive/zip"
 	"compress/gzip"
 	"io"
+	"io/fs"
 	"os"
 	"path/filepath"
 	"strings"
 )
 
+func CreateZip(dst io.Writer, basePaths []string) (err error) {
+	w := zip.NewWriter(dst)
+	defer func() {
+		if closeErr := w.Close(); err == nil {
+			err = closeErr
+		}
+	}()
+
+	for _, basePath := range basePaths {
+		info, err := os.Lstat(basePath)
+		if err != nil {
+			return err
+		}
+
+		if info.Mode()&os.ModeSymlink != 0 {
+			continue
+		}
+
+		if info.Mode().IsRegular() {
+			if err := addZipFile(w, basePath, filepath.Base(basePath), info); err != nil {
+				return err
+			}
+			continue
+		}
+
+		if info.IsDir() {
+			baseName := filepath.Base(basePath)
+			if err := filepath.WalkDir(basePath, func(path string, d fs.DirEntry, err error) error {
+				if err != nil {
+					return err
+				}
+				if d.Type()&fs.ModeSymlink != 0 {
+					return nil
+				}
+
+				rel, err := filepath.Rel(basePath, path)
+				if err != nil || strings.HasPrefix(rel, ".."+string(filepath.Separator)) || rel == ".." {
+					return nil
+				}
+
+				name := baseName
+				if rel != "." {
+					name = filepath.Join(baseName, rel)
+				}
+				name = filepath.ToSlash(name)
+
+				info, err := d.Info()
+				if err != nil {
+					return err
+				}
+
+				if d.IsDir() {
+					header, err := zip.FileInfoHeader(info)
+					if err != nil {
+						return err
+					}
+					header.Name = name + "/"
+					_, err = w.CreateHeader(header)
+					return err
+				}
+
+				if info.Mode().IsRegular() {
+					return addZipFile(w, path, name, info)
+				}
+				return nil
+			}); err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func addZipFile(w *zip.Writer, path, name string, info os.FileInfo) error {
+	header, err := zip.FileInfoHeader(info)
+	if err != nil {
+		return err
+	}
+	header.Name = filepath.ToSlash(name)
+	header.Method = zip.Deflate
+
+	writer, err := w.CreateHeader(header)
+	if err != nil {
+		return err
+	}
+
+	file, err := os.Open(path)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	_, err = io.Copy(writer, file)
+	return err
+}
+
 func ExtractZip(src, dest string) error {
 	r, err := zip.OpenReader(src)
 	if err != nil {