mirror of
https://github.com/restic/restic.git
synced 2026-06-21 16:14:18 +00:00
Michael Eischer
2f31cde517
The restorer imported ui/restorer which leaks ui logic into the restorer core. Swap the direction by letting the restorer use a ProgressReporter interface + associated constants.
97 lines
3.0 KiB
Go
97 lines
3.0 KiB
Go
package restore
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
"github.com/restic/restic/internal/restorer"
|
|
"github.com/restic/restic/internal/ui"
|
|
"github.com/restic/restic/internal/ui/progress"
|
|
)
|
|
|
|
type textPrinter struct {
|
|
restic.Printer
|
|
|
|
terminal ui.Terminal
|
|
}
|
|
|
|
func NewTextProgress(terminal ui.Terminal, verbosity uint) ProgressPrinter {
|
|
return &textPrinter{
|
|
Printer: progress.NewTerminalPrinter(false, verbosity, terminal),
|
|
terminal: terminal,
|
|
}
|
|
}
|
|
|
|
func (t *textPrinter) Update(p State, duration time.Duration) {
|
|
timeLeft := ui.FormatDuration(duration)
|
|
formattedAllBytesWritten := ui.FormatBytes(p.AllBytesWritten)
|
|
formattedAllBytesTotal := ui.FormatBytes(p.AllBytesTotal)
|
|
allPercent := ui.FormatPercent(p.AllBytesWritten, p.AllBytesTotal)
|
|
progress := fmt.Sprintf("[%s] %s %v files/dirs %s, total %v files/dirs %v",
|
|
timeLeft, allPercent, p.FilesFinished, formattedAllBytesWritten, p.FilesTotal, formattedAllBytesTotal)
|
|
if p.FilesSkipped > 0 {
|
|
progress += fmt.Sprintf(", skipped %v files/dirs %v", p.FilesSkipped, ui.FormatBytes(p.AllBytesSkipped))
|
|
}
|
|
if p.FilesDeleted > 0 {
|
|
progress += fmt.Sprintf(", deleted %v files/dirs", p.FilesDeleted)
|
|
}
|
|
|
|
t.terminal.SetStatus([]string{progress})
|
|
}
|
|
|
|
func (t *textPrinter) Error(item string, err error) error {
|
|
t.E("ignoring error for %s: %s\n", item, err)
|
|
return nil
|
|
}
|
|
|
|
func (t *textPrinter) CompleteItem(messageType restorer.ItemAction, item string, size uint64) {
|
|
var action string
|
|
switch messageType {
|
|
case restorer.ActionDirRestored:
|
|
action = "restored"
|
|
case restorer.ActionFileRestored:
|
|
action = "restored"
|
|
case restorer.ActionOtherRestored:
|
|
action = "restored"
|
|
case restorer.ActionFileUpdated:
|
|
action = "updated"
|
|
case restorer.ActionFileUnchanged:
|
|
action = "unchanged"
|
|
case restorer.ActionDeleted:
|
|
action = "deleted"
|
|
default:
|
|
panic("unknown message type")
|
|
}
|
|
|
|
if messageType == restorer.ActionDirRestored || messageType == restorer.ActionOtherRestored || messageType == restorer.ActionDeleted {
|
|
t.VV("%-9v %v", action, item)
|
|
} else {
|
|
t.VV("%-9v %v with size %v", action, item, ui.FormatBytes(size))
|
|
}
|
|
}
|
|
|
|
func (t *textPrinter) Finish(p State, duration time.Duration) {
|
|
t.terminal.SetStatus(nil)
|
|
|
|
timeLeft := ui.FormatDuration(duration)
|
|
formattedAllBytesTotal := ui.FormatBytes(p.AllBytesTotal)
|
|
|
|
var summary string
|
|
if p.FilesFinished == p.FilesTotal && p.AllBytesWritten == p.AllBytesTotal {
|
|
summary = fmt.Sprintf("Summary: Restored %d files/dirs (%s) in %s", p.FilesTotal, formattedAllBytesTotal, timeLeft)
|
|
} else {
|
|
formattedAllBytesWritten := ui.FormatBytes(p.AllBytesWritten)
|
|
summary = fmt.Sprintf("Summary: Restored %d / %d files/dirs (%s / %s) in %s",
|
|
p.FilesFinished, p.FilesTotal, formattedAllBytesWritten, formattedAllBytesTotal, timeLeft)
|
|
}
|
|
if p.FilesSkipped > 0 {
|
|
summary += fmt.Sprintf(", skipped %v files/dirs %v", p.FilesSkipped, ui.FormatBytes(p.AllBytesSkipped))
|
|
}
|
|
if p.FilesDeleted > 0 {
|
|
summary += fmt.Sprintf(", deleted %v files/dirs", p.FilesDeleted)
|
|
}
|
|
|
|
t.terminal.Print(summary)
|
|
}
|