From 5d561380f73f2b1bcb663191e363a752cc097dcc Mon Sep 17 00:00:00 2001
From: Petter Rasmussen
Date: Tue, 16 Feb 2016 22:17:54 +0100
Subject: Add delete option for upload and downloads

---
 drive/download.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'drive/download.go')

diff --git a/drive/download.go b/drive/download.go
index 3ed73df..7d3dc8b 100644
--- a/drive/download.go
+++ b/drive/download.go
@@ -17,6 +17,7 @@ type DownloadArgs struct {
     Path string
     Force bool
     Recursive bool
+    Delete bool
     Stdout bool
 }
 
@@ -43,6 +44,17 @@ func (self *Drive) Download(args DownloadArgs) error {
     if !args.Stdout {
         fmt.Fprintf(args.Out, "Downloaded %s at %s/s, total %s\n", f.Id, formatSize(rate, false), formatSize(bytes, false))
     }
+
+    if args.Delete {
+        err = self.deleteFile(args.Id)
+        if err != nil {
+            return fmt.Errorf("Failed to delete file: %s", err)
+        }
+
+        if !args.Stdout {
+            fmt.Fprintf(args.Out, "Removed %s\n", args.Id)
+        }
+    }
     return err
 }
 
-- 
cgit v1.2.3