Added support for mime types while uploading

2 files changed, 8 insertions, 4 deletions

diff --git a/cli/cli.go b/cli/cli.go
index 58dfd28..6e95f59 100644
--- a/cli/cli.go
+++ b/cli/cli.go
@@ -158,7 +158,7 @@ func Folder(d *gdrive.Drive, title string, parentId string, share bool) {
 }
 
 // Upload file to drive
-func Upload(d *gdrive.Drive, input io.ReadCloser, title string, parentId string, share bool) {
+func Upload(d *gdrive.Drive, input io.ReadCloser, title string, parentId string, share bool, mimeType string) {
     // Use filename or 'untitled' as title if no title is specified
     if title == "" {
         if f, ok := input.(*os.File); ok && input != os.Stdin {
@@ -168,7 +168,10 @@ func Upload(d *gdrive.Drive, input io.ReadCloser, title string, parentId string,
         }
     }
 
-    var mimeType = mime.TypeByExtension(filepath.Ext(title))
+    if mimeType == "" {
+        mimeType = mime.TypeByExtension(filepath.Ext(title))
+    }
+
 	// File instance
 	f := &drive.File{Title: title, MimeType: mimeType}
 	// Set parent (if provided)
diff --git a/drive.go b/drive.go
index 389bbcd..4da72d1 100644
--- a/drive.go
+++ b/drive.go
@@ -45,6 +45,7 @@ type Options struct {
         Title string `goptions:"-t, --title, description='Title to give uploaded file. Defaults to filename'"`
         ParentId string `goptions:"-p, --parent, description='Parent Id of the file'"`
         Share bool `goptions:"--share, description='Share uploaded file'"`
+        MimeType string `goptions:"--mimetype, description='The mime type (default will try to figure it out)'"`
     } `goptions:"upload"`
 
     Download struct {
@@ -104,9 +105,9 @@ func main() {
         case "upload":
             args := opts.Upload
             if args.Stdin {
-                cli.Upload(drive, os.Stdin, args.Title, args.ParentId, args.Share)
+                cli.Upload(drive, os.Stdin, args.Title, args.ParentId, args.Share, args.MimeType)
             } else {
-                cli.Upload(drive, args.File, args.Title, args.ParentId, args.Share)
+                cli.Upload(drive, args.File, args.Title, args.ParentId, args.Share, args.MimeType)
             }
 
         case "download":