refactor: Decompose restore func

pkg/operations/operations.go

@@ -4,19 +4,6 @@ import (
 	"github.com/pojntfx/stfs/pkg/config"
 )
 
-func Restore(
-	state config.StateConfig,
-	pipes config.PipeConfig,
-	crypto config.CryptoConfig,
-
-	recordSize int,
-	from string,
-	to string,
-	flatten bool,
-) error {
-	return nil
-}
-
 func Update(
 	state config.StateConfig,
 	pipes config.PipeConfig,

pkg/operations/restore.go

@@ -0,0 +1,108 @@
+package operations
+
+import (
+	"archive/tar"
+	"context"
+	"database/sql"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/pojntfx/stfs/internal/converters"
+	models "github.com/pojntfx/stfs/internal/db/sqlite/models/metadata"
+	"github.com/pojntfx/stfs/internal/formatting"
+	"github.com/pojntfx/stfs/internal/persisters"
+	"github.com/pojntfx/stfs/pkg/config"
+	"github.com/pojntfx/stfs/pkg/recovery"
+)
+
+func Restore(
+	state config.StateConfig,
+	pipes config.PipeConfig,
+	crypto config.CryptoConfig,
+
+	recordSize int,
+	from string,
+	to string,
+	flatten bool,
+) error {
+	metadataPersister := persisters.NewMetadataPersister(state.Metadata)
+	if err := metadataPersister.Open(); err != nil {
+		return err
+	}
+
+	headersToRestore := []*models.Header{}
+	src := strings.TrimSuffix(from, "/")
+	dbhdr, err := metadataPersister.GetHeader(context.Background(), src)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			src = src + "/"
+
+			dbhdr, err = metadataPersister.GetHeader(context.Background(), src)
+			if err != nil {
+				return err
+			}
+		} else {
+			return err
+		}
+	}
+	headersToRestore = append(headersToRestore, dbhdr)
+
+	// If the header refers to a directory, get it's children
+	if dbhdr.Typeflag == tar.TypeDir {
+		dbhdrs, err := metadataPersister.GetHeaderChildren(context.Background(), src)
+		if err != nil {
+			return err
+		}
+
+		headersToRestore = append(headersToRestore, dbhdrs...)
+	}
+
+	for i, dbhdr := range headersToRestore {
+		if i == 0 {
+			if err := formatting.PrintCSV(formatting.TARHeaderCSV); err != nil {
+				return err
+			}
+		}
+
+		hdr, err := converters.DBHeaderToTarHeader(dbhdr)
+		if err != nil {
+			return err
+		}
+
+		if err := formatting.PrintCSV(formatting.GetTARHeaderAsCSV(dbhdr.Record, dbhdr.Lastknownrecord, dbhdr.Block, dbhdr.Lastknownblock, hdr)); err != nil {
+			return err
+		}
+
+		dst := dbhdr.Name
+		if to != "" {
+			if flatten {
+				dst = to
+			} else {
+				dst = filepath.Join(to, strings.TrimPrefix(dst, from))
+
+				if strings.TrimSuffix(dst, "/") == strings.TrimSuffix(to, "/") {
+					dst = filepath.Join(dst, path.Base(dbhdr.Name)) // Append the name so we don't overwrite
+				}
+			}
+		}
+
+		if err := recovery.Fetch(
+			state,
+			pipes,
+			crypto,
+
+			recordSize,
+			int(dbhdr.Record),
+			int(dbhdr.Block),
+			dst,
+			false,
+
+			false,
+		); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}