Enable progress indicator

1 files changed, 13 insertions, 0 deletions

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