diff --git a/go.sum b/go.sum index 8f23b48..07b72b6 100644 --- a/go.sum +++ b/go.sum @@ -83,7 +83,6 @@ github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrU github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8= -github.com/golang/protobuf v1.4.2 h1:+Z5KGCizgyZCbGh1KZqA0fcLLkwbsjIzS4aV2v7wJX0= github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= github.com/golang/protobuf v1.4.3 h1:JjCZWpVbqXDqFVmTfYWEVTMIYrL/NPdPSCHPJ0T/raM= github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= @@ -371,7 +370,6 @@ google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7 google.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= google.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0= google.golang.org/appengine v1.6.5/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc= -google.golang.org/appengine v1.6.6 h1:lMO5rYAqUxkmaj76jAkRUvt5JZgFymx/+Q5Mzfivuhc= google.golang.org/appengine v1.6.6/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc= google.golang.org/appengine v1.6.7 h1:FZR1q0exgwxzPzp/aF+VccGrSfxfPpkBqjIIEq3ru6c= google.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc= diff --git a/main.go b/main.go index 5d26e03..df8642d 100644 --- a/main.go +++ b/main.go @@ -10,6 +10,7 @@ import ( "golang.org/x/oauth2" "golang.org/x/oauth2/google" "google.golang.org/api/drive/v3" + "google.golang.org/api/googleapi" "google.golang.org/api/option" "log" "os" @@ -169,6 +170,10 @@ func main() { "check-token", false, "True to exit after getting the token.") + contentType := flag.String( + "content-type", + "", + "The content type to upload as. If omitted, no explicit content type is set.") flag.Parse() var dataFile *os.File var dataFilename string @@ -219,14 +224,19 @@ func main() { Name: dataFilename, } + var mediaOptions []googleapi.MediaOption + if *contentType != "" { + mediaOptions = append(mediaOptions, googleapi.ContentType(*contentType)) + } + if *fileId != "" { - fileOut, err := srv.Files.Update(*fileId, &metadata).Media(dataFile).Do() + fileOut, err := srv.Files.Update(*fileId, &metadata).Media(dataFile, mediaOptions...).Do() if err != nil { log.Fatalf("Failed updating file %s: %v", *fileId, err) } fmt.Println(fileOut.Id) } else { - fileOut, err := srv.Files.Create(&metadata).Media(dataFile).Do() + fileOut, err := srv.Files.Create(&metadata).Media(dataFile, mediaOptions...).Do() if err != nil { log.Fatalf("Failed uploading file: %v", err) }