/
zipwritersystem.go
80 lines (72 loc) · 1.84 KB
/
zipwritersystem.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
package chezmoi
import (
"archive/zip"
"io"
"os"
"time"
)
// A ZIPWriterSystem is a System that writes to a ZIP archive.
type ZIPWriterSystem struct {
emptySystemMixin
noUpdateSystemMixin
w *zip.Writer
modified time.Time
}
// NewZIPWriterSystem returns a new ZIPWriterSystem that writes a ZIP archive to
// w.
func NewZIPWriterSystem(w io.Writer, modified time.Time) *ZIPWriterSystem {
return &ZIPWriterSystem{
w: zip.NewWriter(w),
modified: modified,
}
}
// Close closes m.
func (s *ZIPWriterSystem) Close() error {
return s.w.Close()
}
// Mkdir implements System.Mkdir.
func (s *ZIPWriterSystem) Mkdir(name AbsPath, perm os.FileMode) error {
fh := zip.FileHeader{
Name: string(name),
Modified: s.modified,
}
fh.SetMode(os.ModeDir | perm)
_, err := s.w.CreateHeader(&fh)
return err
}
// RunScript implements System.RunScript.
func (s *ZIPWriterSystem) RunScript(scriptname RelPath, dir AbsPath, data []byte) error {
return s.WriteFile(AbsPath(scriptname), data, 0o700)
}
// WriteFile implements System.WriteFile.
func (s *ZIPWriterSystem) WriteFile(filename AbsPath, data []byte, perm os.FileMode) error {
fh := zip.FileHeader{
Name: string(filename),
Method: zip.Deflate,
Modified: s.modified,
UncompressedSize64: uint64(len(data)),
}
fh.SetMode(perm)
fw, err := s.w.CreateHeader(&fh)
if err != nil {
return err
}
_, err = fw.Write(data)
return err
}
// WriteSymlink implements System.WriteSymlink.
func (s *ZIPWriterSystem) WriteSymlink(oldname string, newname AbsPath) error {
data := []byte(oldname)
fh := zip.FileHeader{
Name: string(newname),
Modified: s.modified,
UncompressedSize64: uint64(len(data)),
}
fh.SetMode(os.ModeSymlink)
fw, err := s.w.CreateHeader(&fh)
if err != nil {
return err
}
_, err = fw.Write(data)
return err
}