diff options
Diffstat (limited to 'drive/download.go')
| -rw-r--r-- | drive/download.go | 12 | 
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  } | 
