Add delete option for upload and downloads

1 files changed, 12 insertions, 0 deletions

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
 }