smallsolar/postmarketos-mkinitfs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

archive: allow selecting different formats when writing compressed file (MR 25)

Browse Source
This commit is contained in:
Clayton Craft
2022-11-17 16:36:16 -08:00
parent 4e97990804
commit 31b7eb34ee
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/archive/archive.go
View File

@@ -223,19 +223,26 @@ func (archive *Archive) addFile(source string, dest string) error {
} }
func (archive *Archive) writeCompressed(path string, mode os.FileMode) error { func (archive *Archive) writeCompressed(path string, mode os.FileMode) error {
// TODO: support other compression formats, based on deviceinfo
fd, err := os.Create(path) fd, err := os.Create(path)
if err != nil { if err != nil {
return err return err
} }
gz := gzip.NewWriter(fd) var compressor io.WriteCloser
if _, err = io.Copy(gz, archive.buf); err != nil { switch archive.compress_format {
case FormatGzip:
compressor = gzip.NewWriter(fd)
default:
log.Print("Unknown or no compression format set, using gzip")
compressor = gzip.NewWriter(fd)
}
if _, err = io.Copy(compressor, archive.buf); err != nil {
return err return err
} }
if err := gz.Close(); err != nil { if err := compressor.Close(); err != nil {
return err return err
} }