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

pkg/operations/restore.go

@@ -4,6 +4,8 @@ import (
 	"archive/tar"
 	"context"
 	"database/sql"
+	"io"
+	"io/fs"
 	"path"
 	"path/filepath"
 	"strings"
@@ -13,7 +15,14 @@ import (
 	"github.com/pojntfx/stfs/pkg/recovery"
 )
 
-func (o *Operations) Restore(from string, to string, flatten bool) error {
+func (o *Operations) Restore(
+	getDst func(path string, mode fs.FileMode) (io.WriteCloser, error),
+	mkdirAll func(path string, mode fs.FileMode) error,
+
+	from string,
+	to string,
+	flatten bool,
+) error {
 	o.diskOperationLock.Lock()
 	defer o.diskOperationLock.Unlock()
 
@@ -84,6 +93,9 @@ func (o *Operations) Restore(from string, to string, flatten bool) error {
 			o.pipes,
 			o.crypto,
 
+			getDst,
+			mkdirAll,
+
 			o.pipes.RecordSize,
 			int(dbhdr.Record),
 			int(dbhdr.Block),

pkg/recovery/fetch.go

@@ -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 {