44 lines
734 B
Go
44 lines
734 B
Go
package utils
|
|
|
|
import (
|
|
"archive/tar"
|
|
"errors"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
type TarballFileReader struct {
|
|
tarReader *tar.Reader
|
|
file *os.File
|
|
}
|
|
|
|
func ReadTarballContent(tarballPath, fileToExtract string) (*TarballFileReader, error) {
|
|
file, err := os.Open(tarballPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tr := tar.NewReader(file)
|
|
for {
|
|
header, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if header.Name == fileToExtract {
|
|
if header.Typeflag != tar.TypeReg {
|
|
return nil, errors.New("file to extract must be a regular file")
|
|
}
|
|
|
|
return &TarballFileReader{
|
|
tarReader: tr,
|
|
file: file,
|
|
}, nil
|
|
}
|
|
}
|
|
|
|
return nil, errors.New("could not file in tarball")
|
|
}
|