You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
1.7 KiB
116 lines
1.7 KiB
package Archive
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func UnGzip(source, target string) error {
|
|
var (
|
|
reader, writer *os.File
|
|
archive *gzip.Reader
|
|
e error
|
|
)
|
|
|
|
reader, e = os.Open(source)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
defer reader.Close()
|
|
|
|
archive, e = gzip.NewReader(reader)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
defer archive.Close()
|
|
|
|
target = filepath.Join(target, archive.Name)
|
|
|
|
writer, e = os.Create(target)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
defer writer.Close()
|
|
|
|
_, e = io.Copy(writer, archive)
|
|
return e
|
|
}
|
|
|
|
func Untar(tarball, target string) error {
|
|
var (
|
|
reader *os.File
|
|
tarReader *tar.Reader
|
|
header *tar.Header
|
|
info fs.FileInfo
|
|
file *os.File
|
|
path string
|
|
e error
|
|
)
|
|
|
|
reader, e = os.Open(tarball)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
defer reader.Close()
|
|
tarReader = tar.NewReader(reader)
|
|
|
|
for {
|
|
header, e = tarReader.Next()
|
|
if e == io.EOF {
|
|
break
|
|
}
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
path = filepath.Join(target, header.Name)
|
|
|
|
info = header.FileInfo()
|
|
if info.IsDir() {
|
|
e = os.MkdirAll(path, info.Mode())
|
|
if e != nil {
|
|
return e
|
|
}
|
|
continue
|
|
}
|
|
|
|
file, e = os.OpenFile(path, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, info.Mode())
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
_, e = io.Copy(file, tarReader)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func UntarGzip(source, target string) error {
|
|
var (
|
|
tarPath string = strings.ReplaceAll(source, ".gz", "")
|
|
e error
|
|
)
|
|
|
|
e = UnGzip(source, tarPath)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
e = Untar(tarPath, target)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
return os.Remove(tarPath)
|
|
}
|