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

This commit is contained in:
Felicitas Pojtinger
2021-12-17 15:30:09 +01:00
parent e11a3bffcc
commit 2a67c243fd
4 changed files with 59 additions and 8 deletions

View File

@@ -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),