|
|
- 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)
- }
|