2019-02-07 12:12:20 -02:00
|
|
|
// Package gzip implements the Archive interface providing gz archiving
|
2018-08-05 10:23:39 -03:00
|
|
|
// and compression.
|
2019-02-07 12:12:20 -02:00
|
|
|
package gzip
|
2018-08-05 10:23:39 -03:00
|
|
|
|
|
|
|
import (
|
|
|
|
"compress/gzip"
|
2019-02-07 12:12:20 -02:00
|
|
|
"fmt"
|
2018-08-05 10:23:39 -03:00
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
2020-05-26 00:48:10 -03:00
|
|
|
// Archive as gz.
|
2018-08-05 10:23:39 -03:00
|
|
|
type Archive struct {
|
|
|
|
gw *gzip.Writer
|
|
|
|
}
|
|
|
|
|
2020-05-26 00:48:10 -03:00
|
|
|
// Close all closeables.
|
2018-08-05 10:23:39 -03:00
|
|
|
func (a Archive) Close() error {
|
|
|
|
return a.gw.Close()
|
|
|
|
}
|
|
|
|
|
2020-05-26 00:48:10 -03:00
|
|
|
// New gz archive.
|
2018-08-05 10:23:39 -03:00
|
|
|
func New(target io.Writer) Archive {
|
2020-04-01 15:05:20 -04:00
|
|
|
// the error will be nil since the compression level is valid
|
|
|
|
gw, _ := gzip.NewWriterLevel(target, gzip.BestCompression)
|
2018-08-05 10:23:39 -03:00
|
|
|
return Archive{
|
|
|
|
gw: gw,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-26 00:48:10 -03:00
|
|
|
// Add file to the archive.
|
2018-08-05 10:23:39 -03:00
|
|
|
func (a Archive) Add(name, path string) error {
|
2019-02-07 12:12:20 -02:00
|
|
|
if a.gw.Header.Name != "" {
|
|
|
|
return fmt.Errorf("gzip: failed to add %s, only one file can be archived in gz format", name)
|
|
|
|
}
|
2018-09-04 09:48:22 -03:00
|
|
|
file, err := os.Open(path) // #nosec
|
2018-08-05 10:23:39 -03:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-05-26 00:48:10 -03:00
|
|
|
defer file.Close()
|
2018-08-05 10:23:39 -03:00
|
|
|
info, err := file.Stat()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if info.IsDir() {
|
|
|
|
return nil
|
|
|
|
}
|
2019-02-07 12:12:20 -02:00
|
|
|
a.gw.Header.Name = name
|
|
|
|
a.gw.Header.ModTime = info.ModTime()
|
|
|
|
_, err = io.Copy(a.gw, file)
|
2018-08-05 10:23:39 -03:00
|
|
|
return err
|
|
|
|
}
|