/
create_achive.go
106 lines (92 loc) · 2.2 KB
/
create_achive.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package ziputil
import (
"archive/zip"
"fmt"
"io"
"os"
"regexp"
"github.com/grokify/mogo/errors/errorsutil"
"github.com/grokify/mogo/os/osutil"
)
func ZipFilesRx(zipfile, dir string, rx *regexp.Regexp, removePaths bool) error {
entries, err := osutil.ReadDirMore(dir, rx, false, true, false)
if err != nil {
return err
}
filepaths := entries.Names(dir)
return ZipFiles(zipfile, removePaths, filepaths)
}
// ZipFiles compresses one or many files into a single zip archive file.
func ZipFiles(zipfile string, removePaths bool, srcfiles []string) error {
flags := os.O_WRONLY | os.O_CREATE | os.O_TRUNC
zfile, err := os.OpenFile(zipfile, flags, 0644)
if err != nil {
return fmt.Errorf("E_FAILED_TO_OPEN_FILE [%s]", err)
}
zipw := zip.NewWriter(zfile)
for _, filename := range srcfiles {
if err := AddFileToZip(zipw, filename, removePaths); err != nil {
return closeFileAndZipOnError(
zfile,
zipw,
errorsutil.Wrap(err, fmt.Sprintf("Failed to add file %s to zip", filename)))
}
}
err = zipw.Close()
if err != nil {
return closeFileOnError(zfile, err)
}
return zfile.Close()
}
func closeFileAndZipOnError(f *os.File, zipw *zip.Writer, err error) error {
if zipw != nil {
zipwErr := zipw.Close()
if zipwErr != nil {
err = errorsutil.Wrap(err, zipwErr.Error())
}
}
if f != nil {
fErr := f.Close()
if fErr != nil {
err = errorsutil.Wrap(err, fErr.Error())
}
}
return err
}
func closeFileOnError(f *os.File, err error) error {
if f != nil {
fErr := f.Close()
if fErr != nil {
err = errorsutil.Wrap(err, fErr.Error())
}
}
return err
}
func AddFileToZip(zipWriter *zip.Writer, filename string, removePaths bool) error {
fileToZip, err := os.Open(filename)
if err != nil {
return err
}
info, err := fileToZip.Stat()
if err != nil {
return err
}
header, err := zip.FileInfoHeader(info)
if err != nil {
return err
}
if !removePaths {
header.Name = filename
}
// See http://golang.org/pkg/archive/zip/#pkg-constants
header.Method = zip.Deflate
writer, err := zipWriter.CreateHeader(header)
if err != nil {
return err
}
_, err = io.Copy(writer, fileToZip)
if err != nil {
return closeFileOnError(fileToZip, err)
}
return fileToZip.Close()
}