/
zip_files.go
99 lines (75 loc) · 1.71 KB
/
zip_files.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
package anvil
import (
"archive/zip"
"io"
"sort"
"strings"
)
const uint32max = 4294967295
// Reads a Stream from the given zip file stream.
func FromZip(r *zip.Reader, source string) Stream {
return makeStream(func(result chan<- Blob) {
blobs := make([]Blob, 0)
for _, f := range r.File {
// FIXME: figure out what to do with weird metadata.
if strings.HasPrefix(f.Name, "__MACOSX") {
continue
}
r, err := f.Open()
// FIXME: deal with bugs in Zip format
// t := f.ModTime().UTC()
// utcd := time.Date(t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second(), t.Nanosecond(), time.Local).Local()
sz := int64(f.UncompressedSize64)
if strings.HasSuffix(f.Name, "/") {
sz = 0
}
blob := &memBlob{
name: f.Name,
modtime: f.ModTime(),
mode: f.Mode(),
size: sz,
contents: r,
err: err,
source: source,
}
if err != nil {
result <- blob
return
}
blobs = append(blobs, blob)
}
sort.Sort(sortBlobs(blobs))
var prev Blob
for _, blob := range blobs {
if prev == nil || prev.Name() != blob.Name() {
// OS X sometimes generates duplicate directories
result <- blob
}
prev = blob
}
}).withValidation()
}
// Writes this tree to the given zip file, returning an error on failure.
func (t Stream) ToZip(w *zip.Writer) error {
for blob := range t {
hdr, err := zip.FileInfoHeader(blob)
if err != nil {
return err
}
writer, err := w.CreateHeader(hdr)
if err != nil {
return err
}
if blob.Contents() != nil {
_, err := io.Copy(writer, blob.Contents())
if err != nil {
return err
}
}
if blob.Error() != nil {
return blob.Error()
}
}
w.Close()
return nil
}