bedrocktool/utils/zip.go

53 lines
1.0 KiB
Go
Raw Normal View History

2022-09-04 14:26:32 +00:00
package utils
import (
"archive/zip"
"fmt"
"io"
"io/fs"
"log"
"os"
"path"
"path/filepath"
2023-01-27 12:07:53 +00:00
"strings"
2022-09-04 14:26:32 +00:00
)
2023-01-29 21:20:13 +00:00
func UnpackZip(r io.ReaderAt, size int64, unpackFolder string) {
2022-09-04 14:26:32 +00:00
zr, _ := zip.NewReader(r, size)
2023-01-29 21:20:13 +00:00
for _, srcFile := range zr.File {
srcName := strings.ReplaceAll(srcFile.Name, "\\", "/")
outPath := path.Join(unpackFolder, srcName)
if srcFile.Mode().IsDir() {
os.Mkdir(outPath, 0o755)
2022-09-04 14:26:32 +00:00
} else {
2023-01-29 21:20:13 +00:00
os.MkdirAll(path.Dir(outPath), 0o755)
fr, _ := srcFile.Open()
f, _ := os.Create(path.Join(unpackFolder, srcName))
2022-09-04 14:26:32 +00:00
io.Copy(f, fr)
}
}
}
func ZipFolder(filename, folder string) error {
f, err := os.Create(filename)
if err != nil {
log.Fatal(err)
}
zw := zip.NewWriter(f)
err = filepath.WalkDir(folder, func(path string, d fs.DirEntry, err error) error {
if !d.Type().IsDir() {
rel := path[len(folder)+1:]
zwf, _ := zw.Create(rel)
data, err := os.ReadFile(path)
if err != nil {
fmt.Println(err)
}
zwf.Write(data)
}
return nil
})
zw.Close()
f.Close()
return err
}