Bind progress into Downloader.

utils/download.go

@@ -19,6 +19,7 @@ type Downloader interface {
 	Pause()
 	Resume()
 	Shutdown()
+	GetProgress() *Progress
 }
 
 // Check interface
@@ -33,6 +34,7 @@ type downloaderImpl struct {
 	stopped  chan bool
 	pause    chan bool
 	unpause  chan bool
+	progress *Progress
 	threads  int
 }
 
@@ -55,8 +57,11 @@ func NewDownloader(threads int) Downloader {
 		pause:    make(chan bool),
 		unpause:  make(chan bool),
 		threads:  threads,
+		progress: NewProgress(),
 	}
 
+	downloader.progress.Start()
+
 	for i := 0; i < downloader.threads; i++ {
 		go downloader.process()
 	}
@@ -74,6 +79,8 @@ func (downloader *downloaderImpl) Shutdown() {
 	for i := 0; i < downloader.threads; i++ {
 		<-downloader.stopped
 	}
+
+	downloader.progress.Shutdown()
 }
 
 // Pause pauses task processing
@@ -90,6 +97,11 @@ func (downloader *downloaderImpl) Resume() {
 	}
 }
 
+// Resume resumes task processing
+func (downloader *downloaderImpl) GetProgress() *Progress {
+	return downloader.progress
+}
+
 // Download starts new download task
 func (downloader *downloaderImpl) Download(url string, destination string, result chan<- error) {
 	downloader.DownloadWithChecksum(url, destination, result, ChecksumInfo{Size: -1}, false)
@@ -103,7 +115,7 @@ func (downloader *downloaderImpl) DownloadWithChecksum(url string, destination s
 
 // handleTask processes single download task
 func (downloader *downloaderImpl) handleTask(task *downloadTask) {
-	fmt.Printf("Downloading %s...\n", task.url)
+	downloader.progress.Printf("Downloading %s...\n", task.url)
 
 	resp, err := http.Get(task.url)
 	if err != nil {
@@ -132,16 +144,19 @@ func (downloader *downloaderImpl) handleTask(task *downloadTask) {
 	}
 	defer outfile.Close()
 
-	var w io.Writer
-
 	checksummer := NewChecksumWriter()
+	writers := []io.Writer{outfile}
 
 	if task.expected.Size != -1 {
-		w = io.MultiWriter(outfile, checksummer)
+		writers = append(writers, checksummer)
-	} else {
-		w = outfile
 	}
 
+	if downloader.progress != nil {
+		writers = append(writers, downloader.progress)
+	}
+
+	w := io.MultiWriter(writers...)
+
 	_, err = io.Copy(w, resp.Body)
 	if err != nil {
 		os.Remove(temppath)
@@ -164,7 +179,7 @@ func (downloader *downloaderImpl) handleTask(task *downloadTask) {
 
 		if err != nil {
 			if task.ignoreMismatch {
-				fmt.Printf("WARNING: %s\n", err.Error())
+				downloader.progress.Printf("WARNING: %s\n", err.Error())
 			} else {
 				os.Remove(temppath)
 				task.result <- err

utils/fake.go

@@ -117,3 +117,8 @@ func (f *FakeDownloader) Pause() {
 // Resume does nothing
 func (f *FakeDownloader) Resume() {
 }
+
+// GetProgress does nothing
+func (f *FakeDownloader) GetProgress() *Progress {
+	return nil
+}