/
zipper.go
62 lines (49 loc) · 1.19 KB
/
zipper.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
package archivist
import (
"archive/zip"
"io"
"os"
"path/filepath"
"strings"
)
type zipper struct {
src string
dst string
writer *zip.Writer
buffer []byte
}
// zipper.do initialises output file, archive and directory tree walk function
func (z *zipper) do() error {
out, err := os.Create(z.dst)
if err != nil {
return err
}
z.buffer = make([]byte, bufSize)
z.writer = zip.NewWriter(out)
if err := filepath.Walk(z.src, z.walk); err != nil {
return err
}
if err := z.writer.Close(); err != nil {
return err
}
return out.Close()
}
// zipper.walk gets called for each file in given directory tree
func (z *zipper) walk(path string, info os.FileInfo, err error) error {
if err != nil { return err }
if !info.Mode().IsRegular() || info.Size() == 0 {
return nil
}
file, err := os.Open(path)
if err != nil {
return err
}
defer file.Close()
fileName := strings.TrimPrefix(path, z.src + string(filepath.Separator))
w, err := z.writer.Create(fileName)
if err != nil {
return err
}
_, err = io.CopyBuffer(w, file, z.buffer)
return err
}