-
Notifications
You must be signed in to change notification settings - Fork 469
/
zipwritersystem.go
92 lines (82 loc) · 2.17 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
81
82
83
84
85
86
87
88
89
90
91
92
package chezmoi
import (
"io"
"io/fs"
"os/exec"
"time"
"github.com/klauspost/compress/zip"
)
// A ZIPWriterSystem is a System that writes to a ZIP archive.
type ZIPWriterSystem struct {
emptySystemMixin
noUpdateSystemMixin
zipWriter *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{
zipWriter: zip.NewWriter(w),
modified: modified,
}
}
// Close closes m.
func (s *ZIPWriterSystem) Close() error {
return s.zipWriter.Close()
}
// Mkdir implements System.Mkdir.
func (s *ZIPWriterSystem) Mkdir(name AbsPath, perm fs.FileMode) error {
fileHeader := zip.FileHeader{
Name: name.String(),
Modified: s.modified,
}
fileHeader.SetMode(fs.ModeDir | perm)
_, err := s.zipWriter.CreateHeader(&fileHeader)
return err
}
// RunCmd implements System.RunCmd.
func (s *ZIPWriterSystem) RunCmd(cmd *exec.Cmd) error {
return nil
}
// RunScript implements System.RunScript.
func (s *ZIPWriterSystem) RunScript(
scriptname RelPath,
dir AbsPath,
data []byte,
options RunScriptOptions,
) error {
return s.WriteFile(NewAbsPath(scriptname.String()), data, 0o700)
}
// WriteFile implements System.WriteFile.
func (s *ZIPWriterSystem) WriteFile(filename AbsPath, data []byte, perm fs.FileMode) error {
fileHeader := zip.FileHeader{
Name: filename.String(),
Method: zip.Deflate,
Modified: s.modified,
UncompressedSize64: uint64(len(data)),
}
fileHeader.SetMode(perm)
fileWriter, err := s.zipWriter.CreateHeader(&fileHeader)
if err != nil {
return err
}
_, err = fileWriter.Write(data)
return err
}
// WriteSymlink implements System.WriteSymlink.
func (s *ZIPWriterSystem) WriteSymlink(oldname string, newname AbsPath) error {
data := []byte(oldname)
fileHeader := zip.FileHeader{
Name: newname.String(),
Modified: s.modified,
UncompressedSize64: uint64(len(data)),
}
fileHeader.SetMode(fs.ModeSymlink)
fileWriter, err := s.zipWriter.CreateHeader(&fileHeader)
if err != nil {
return err
}
_, err = fileWriter.Write(data)
return err
}