From 06c6857d7d8837e28cdc45a33f4f69ec882c6fe2 Mon Sep 17 00:00:00 2001
From: Petter Rasmussen
Date: Sun, 24 Jan 2016 12:39:16 +0100
Subject: Enable progress indicator

---
 drive/progress.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'drive/progress.go')

diff --git a/drive/progress.go b/drive/progress.go
index ee63b6c..6187058 100644
--- a/drive/progress.go
+++ b/drive/progress.go
@@ -2,10 +2,23 @@ package drive
 
 import (
     "io"
+    "io/ioutil"
     "fmt"
     "time"
 )
 
+func getProgressReader(r io.Reader, w io.Writer, size int64) io.Reader {
+    if w == ioutil.Discard || size < 1024 * 1024 {
+        return r
+    }
+
+    return &Progress{
+        Reader: r,
+        Writer: w,
+        Size: size,
+    }
+}
+
 type Progress struct {
     Writer io.Writer
     Reader io.Reader
-- 
cgit v1.2.3