feat: Add io.Writer-based API to restore and fetch cmds

This commit is contained in:
Felix Pojtinger
2021-12-17 15:30:09 +01:00
parent 99359e09fd
commit 66a5cd4b9c
4 changed files with 59 additions and 8 deletions
+6 -7
View File
@@ -4,7 +4,7 @@ import (
"archive/tar"
"bufio"
"io"
"os"
"io/fs"
"path/filepath"
"github.com/pojntfx/stfs/internal/compression"
@@ -23,6 +23,9 @@ func Fetch(
pipes config.PipeConfig,
crypto config.CryptoConfig,
getDst func(path string, mode fs.FileMode) (io.WriteCloser, error),
mkdirAll func(path string, mode fs.FileMode) error,
recordSize int,
record int,
block int,
@@ -82,18 +85,14 @@ func Fetch(
}
if hdr.Typeflag == tar.TypeDir {
return os.MkdirAll(to, hdr.FileInfo().Mode())
return mkdirAll(to, hdr.FileInfo().Mode())
}
dstFile, err := os.OpenFile(to, os.O_WRONLY|os.O_CREATE, hdr.FileInfo().Mode())
dstFile, err := getDst(to, hdr.FileInfo().Mode())
if err != nil {
return err
}
if err := dstFile.Truncate(0); err != nil {
return err
}
// Don't decompress non-regular files
if !hdr.FileInfo().Mode().IsRegular() {
if _, err := io.Copy(dstFile, tr); err != nil {