Added support for mime types while uploading

1 files changed, 5 insertions, 2 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)