TheArchive/restic
1
0
Fork 0
mirror of https://github.com/restic/restic.git synced 2025-08-22 13:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

recover: refactor to use termstatus

Browse Source
This commit is contained in:
Michael Eischer
2025-03-23 17:46:49 +01:00
parent f7f48b3026
commit 0b6c355678
1 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
cmd/restic/cmd_recover.go
View File

@@ -6,7 +6,10 @@ import (
"time"
"github.com/restic/restic/internal/errors"
"github.com/restic/restic/internal/repository"
"github.com/restic/restic/internal/restic"
"github.com/restic/restic/internal/ui/progress"
"github.com/restic/restic/internal/ui/termstatus"
"github.com/spf13/cobra"
"golang.org/x/sync/errgroup"
)
@@ -32,13 +35,15 @@ Exit status is 12 if the password is incorrect.
GroupID: cmdGroupDefault,
DisableAutoGenTag: true,
RunE: func(cmd *cobra.Command, _ []string) error {
return runRecover(cmd.Context(), globalOptions)
term, cancel := setupTermstatus()
defer cancel()
return runRecover(cmd.Context(), globalOptions, term)
},
}
return cmd
}
func runRecover(ctx context.Context, gopts GlobalOptions) error {
func runRecover(ctx context.Context, gopts GlobalOptions, term *termstatus.Terminal) error {
hostname, err := os.Hostname()
if err != nil {
return err
@@ -50,13 +55,15 @@ func runRecover(ctx context.Context, gopts GlobalOptions) error {
}
defer unlock()
printer := newTerminalProgressPrinter(gopts.verbosity, term)
snapshotLister, err := restic.MemorizeList(ctx, repo, restic.SnapshotFile)
if err != nil {
return err
}
Verbosef("load index files\n")
bar := newIndexProgress(gopts.Quiet, gopts.JSON)
printer.P("load index files\n")
bar := newIndexTerminalProgress(gopts.Quiet, gopts.JSON, term)
if err = repo.LoadIndex(ctx, bar); err != nil {
return err
}
@@ -74,15 +81,15 @@ func runRecover(ctx context.Context, gopts GlobalOptions) error {
return err
}
Verbosef("load %d trees\n", len(trees))
bar = newProgressMax(!gopts.Quiet, uint64(len(trees)), "trees loaded")
printer.P("load %d trees\n", len(trees))
bar = newTerminalProgressMax(!gopts.Quiet, uint64(len(trees)), "trees loaded", term)
for id := range trees {
tree, err := restic.LoadTree(ctx, repo, id)
if ctx.Err() != nil {
return ctx.Err()
}
if err != nil {
Warnf("unable to load tree %v: %v\n", id.Str(), err)
printer.E("unable to load tree %v: %v\n", id.Str(), err)
continue
}
@@ -95,7 +102,7 @@ func runRecover(ctx context.Context, gopts GlobalOptions) error {
}
bar.Done()
Verbosef("load snapshots\n")
printer.P("load snapshots\n")
err = restic.ForAllSnapshots(ctx, snapshotLister, repo, nil, func(_ restic.ID, sn *restic.Snapshot, _ error) error {
trees[*sn.Tree] = true
return nil
@@ -103,19 +110,19 @@ func runRecover(ctx context.Context, gopts GlobalOptions) error {
if err != nil {
return err
}
Verbosef("done\n")
printer.P("done\n")
roots := restic.NewIDSet()
for id, seen := range trees {
if !seen {
Verboseff("found root tree %v\n", id.Str())
printer.V("found root tree %v\n", id.Str())
roots.Insert(id)
}
}
Printf("\nfound %d unreferenced roots\n", len(roots))
printer.S("\nfound %d unreferenced roots\n", len(roots))
if len(roots) == 0 {
Verbosef("no snapshot to write.\n")
printer.P("no snapshot to write.\n")
return nil
}
@@ -163,11 +170,11 @@ func runRecover(ctx context.Context, gopts GlobalOptions) error {
return err
}
return createSnapshot(ctx, "/recover", hostname, []string{"recovered"}, repo, &treeID)
return createSnapshot(ctx, printer, "/recover", hostname, []string{"recovered"}, repo, &treeID)
}
func createSnapshot(ctx context.Context, name, hostname string, tags []string, repo restic.SaverUnpacked[restic.WriteableFileType], tree *restic.ID) error {
func createSnapshot(ctx context.Context, printer progress.Printer, name, hostname string, tags []string, repo restic.SaverUnpacked[restic.WriteableFileType], tree *restic.ID) error {
sn, err := restic.NewSnapshot([]string{name}, tags, hostname, time.Now())
if err != nil {
return errors.Fatalf("unable to save snapshot: %v", err)
@@ -180,6 +187,6 @@ func createSnapshot(ctx context.Context, name, hostname string, tags []string, r
return errors.Fatalf("unable to save snapshot: %v", err)
}
Printf("saved new snapshot %v\n", id.Str())
printer.S("saved new snapshot %v\n", id.Str())
return nil
}