feat: add zip download for folders in file manager

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
Matt Van Horn
2026-06-03 01:27:35 -07:00
parent 6c81c2a32f
commit 9a2ca526a7
8 changed files with 184 additions and 4 deletions
+98
View File
@@ -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 {