kopia/cli/progress.go

package cli

import (
	"log"
	"path/filepath"
	"time"

	"github.com/cheggaaa/pb"
	"github.com/kopia/kopia/internal/units"
	"github.com/kopia/kopia/snapshot"
)

type uploadProgress struct {
	currentDir string

	bar *pb.ProgressBar
}

func (p *uploadProgress) Cached(path string, length int64) {
	log.Printf("  Cached: %v %v", path, units.BytesString(length))
}

func (p *uploadProgress) StartedDir(path string) {
}

func (p *uploadProgress) FinishedDir(path string) {
}

func (p *uploadProgress) Started(path string, length int64) {
	dir := filepath.Dir(path)

	if p.currentDir != dir {
		p.currentDir = dir
		log.Printf("Processing directory: %v", dir)
	}

	p.bar = pb.New64(length).Prefix("  " + filepath.Base(path))
	p.bar.SetRefreshRate(time.Second)
	p.bar.ShowSpeed = true
	p.bar.ShowTimeLeft = true
	p.bar.SetUnits(pb.U_BYTES)
	p.bar.Start()
}

func (p *uploadProgress) Finished(path string, length int64, err error) {
	if p.bar != nil {
		p.bar.Finish()
		p.bar = nil
	}
	//log.Printf("FINISHED %v %v", path, units.BytesString(length))
}

func (p *uploadProgress) Progress(path string, completed, total int64) {
	//log.Printf("PROGRESS %v %v/%v", path, units.BytesString(completed), units.BytesString(total))
	p.bar.Set64(completed)
}

var up snapshot.UploadProgress = &uploadProgress{}