fix prasmussen#154

[harsh: Fix formatting] Signed-off-by: Harsh Shandilya <msfjarvis@gmail.com>
growtopiajaw · Nov 25, 2022 · 39e3417 · 39e3417
1 parent 4f9df5e
commit 39e3417
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/drive/upload.go b/drive/upload.go
@@ -175,6 +175,9 @@ func (self *Drive) uploadFile(args UploadArgs) (*drive.File, int64, error) {
 	// Chunk size option
 	chunkSize := googleapi.ChunkSize(int(args.ChunkSize))
 
+	// Content type option
+	contentType := googleapi.ContentType(args.Mime)
+
 	// Wrap file in progress reader
 	progressReader := getProgressReader(srcFile, args.Progress, srcFileInfo.Size())
 
@@ -184,7 +187,7 @@ func (self *Drive) uploadFile(args UploadArgs) (*drive.File, int64, error) {
 	fmt.Fprintf(args.Out, "Uploading %s\n", args.Path)
 	started := time.Now()
 
-	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "md5Checksum", "webContentLink").Context(ctx).Media(reader, chunkSize).Do()
+	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "md5Checksum", "webContentLink").Context(ctx).Media(reader, chunkSize, contentType).Do()
 	if err != nil {
 		if isTimeoutError(err) {
 			return nil, 0, fmt.Errorf("Failed to upload file: timeout, no data was transferred for %v", args.Timeout)
@@ -230,6 +233,9 @@ func (self *Drive) UploadStream(args UploadStreamArgs) error {
 	// Chunk size option
 	chunkSize := googleapi.ChunkSize(int(args.ChunkSize))
 
+	// Content type option
+	contentType := googleapi.ContentType(args.Mime)
+
 	// Wrap file in progress reader
 	progressReader := getProgressReader(args.In, args.Progress, 0)
 
@@ -239,7 +245,7 @@ func (self *Drive) UploadStream(args UploadStreamArgs) error {
 	fmt.Fprintf(args.Out, "Uploading %s\n", dstFile.Name)
 	started := time.Now()
 
-	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "webContentLink").Context(ctx).Media(reader, chunkSize).Do()
+	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "webContentLink").Context(ctx).Media(reader, chunkSize, contentType).Do()
 	if err != nil {
 		if isTimeoutError(err) {
 			return fmt.Errorf("Failed to upload file: timeout, no data was transferred for %v", args.Timeout)