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/upload.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'drive/upload.go')

diff --git a/drive/upload.go b/drive/upload.go
index 03cb5ad..0c1df0c 100644
--- a/drive/upload.go
+++ b/drive/upload.go
@@ -12,13 +12,13 @@ import (
 
 type UploadFileArgs struct {
     Out io.Writer
+    Progress io.Writer
     Path string
     Name string
     Parents []string
     Mime string
     Recursive bool
     Share bool
-    NoProgress bool
     ChunkSize int64
 }
 
@@ -60,7 +60,10 @@ func (self *Drive) Upload(args UploadFileArgs) (err error) {
     // Chunk size option
     chunkSize := googleapi.ChunkSize(int(args.ChunkSize))
 
-    f, err := self.service.Files.Create(dstFile).Media(srcFile, chunkSize).Do()
+    // Wrap file in progress reader
+    srcReader := getProgressReader(srcFile, args.Progress, srcFileInfo.Size())
+
+    f, err := self.service.Files.Create(dstFile).Media(srcReader, chunkSize).Do()
     if err != nil {
         return fmt.Errorf("Failed to upload file: %s", err)
     }
-- 
cgit v1.2.3