/
tar_utils.go
158 lines (124 loc) · 2.87 KB
/
tar_utils.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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
package tar_utils
import (
"archive/tar"
"io"
"os"
"path/filepath"
"strings"
"syscall"
)
func Untar(tarStream io.Reader, targetDir string) error {
tarBallReader := tar.NewReader(tarStream)
// Extracting tarred files
for {
header, err := tarBallReader.Next()
if err != nil {
if err == io.EOF {
break
}
return err
}
// get the individual filename and extract to the target directory
filename := header.Name
switch header.Typeflag {
case tar.TypeDir:
// handle directory
err = MkdirAll(targetDir, filename, os.FileMode(header.Mode)) // or use 0755 if you prefer
if err != nil {
return err
}
case tar.TypeReg:
fullFilename := filepath.Join(targetDir, filename)
dir := filepath.Dir(filename)
err = MkdirAll(targetDir, dir, os.FileMode(header.Mode)) // or use 0755 if you prefer
if err != nil {
return err
}
// handle normal file
writer, err := os.Create(fullFilename)
if err != nil {
return err
}
io.Copy(writer, tarBallReader)
err = os.Chmod(fullFilename, os.FileMode(header.Mode))
if err != nil {
return err
}
writer.Close()
}
}
return nil
}
// Source: http://blog.ralch.com/tutorial/golang-working-with-tar-and-gzip/
func Tarit(source string, target io.WriteCloser) error {
defer target.Close()
tarball := tar.NewWriter(target)
defer tarball.Close()
info, err := os.Stat(source)
if err != nil {
return nil
}
var baseDir string
if info.IsDir() {
baseDir = filepath.Base(source)
}
return filepath.Walk(source,
func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
header, err := tar.FileInfoHeader(info, info.Name())
if err != nil {
return err
}
if baseDir != "" {
header.Name = filepath.Join(baseDir, strings.TrimPrefix(path, source))
}
if err := tarball.WriteHeader(header); err != nil {
return err
}
if info.IsDir() {
return nil
}
file, err := os.Open(path)
if err != nil {
return err
}
defer file.Close()
_, err = io.Copy(tarball, file)
return err
})
}
func MkdirAll(basePath, targetDir string, fileMode os.FileMode) error {
parents := []string{filepath.Clean(targetDir)}
prevParent := ""
for {
parent := filepath.Dir(targetDir)
if parent == prevParent {
break
}
prevParent = parent
parents = append(parents, parent)
targetDir = parent
}
for i := len(parents) - 1; i >= 0; i-- {
fullDirPath := filepath.Join(basePath, parents[i])
dir, err := os.Stat(fullDirPath)
if err == nil {
if dir.IsDir() {
continue
}
return &os.PathError{"mkdir", fullDirPath, syscall.ENOTDIR}
}
err = os.Mkdir(fullDirPath, fileMode)
if err != nil {
// Handle arguments like "foo/." by
// double-checking that directory doesn't exist.
dir, err1 := os.Lstat(fullDirPath)
if err1 != nil || !dir.IsDir() {
return err
}
}
}
return nil
}