Implement update

3 files changed, 130 insertions, 0 deletions

diff --git a/drive/update.go b/drive/update.go
new file mode 100644
index 0000000..4fa33ed
--- /dev/null
+++ b/drive/update.go
@@ -0,0 +1,71 @@
+package drive
+
+import (
+    "fmt"
+    "mime"
+    "os"
+    "io"
+    "path/filepath"
+    "google.golang.org/api/drive/v3"
+    "golang.org/x/net/context"
+)
+
+type UpdateArgs struct {
+    Out io.Writer
+    Id string
+    Path string
+    Name string
+    Parents []string
+    Mime string
+    Recursive bool
+    Stdin bool
+    Share bool
+    NoProgress bool
+}
+
+func (self *Drive) Update(args UpdateArgs) (err error) {
+    //if args.Stdin {
+    //    self.uploadStdin()
+    //}
+
+    srcFile, err := os.Open(args.Path)
+    if err != nil {
+        return fmt.Errorf("Failed to open file: %s", err)
+    }
+
+    srcFileInfo, err := srcFile.Stat()
+    if err != nil {
+        return fmt.Errorf("Failed to read file metadata: %s", err)
+    }
+
+    // Instantiate empty drive file
+    dstFile := &drive.File{}
+
+    // Use provided file name or use filename
+    if args.Name == "" {
+        dstFile.Name = filepath.Base(srcFileInfo.Name())
+    } else {
+        dstFile.Name = args.Name
+    }
+
+    // Set provided mime type or get type based on file extension
+    if args.Mime == "" {
+        dstFile.MimeType = mime.TypeByExtension(filepath.Ext(dstFile.Name))
+    } else {
+        dstFile.MimeType = args.Mime
+    }
+
+    // Set parent folders
+    dstFile.Parents = args.Parents
+
+    f, err := self.service.Files.Update(args.Id, dstFile).ResumableMedia(context.Background(), srcFile, srcFileInfo.Size(), dstFile.MimeType).Do()
+    if err != nil {
+        return fmt.Errorf("Failed to upload file: %s", err)
+    }
+
+    fmt.Fprintf(args.Out, "Uploaded '%s' at %s, total %d\n", f.Name, "x/s", f.Size)
+    //if args.Share {
+    //    self.Share(TODO)
+    //}
+    return
+}
diff --git a/gdrive.go b/gdrive.go
index 967f6ec..354f272 100644
--- a/gdrive.go
+++ b/gdrive.go
@@ -202,6 +202,49 @@ func main() {
             },
         },
         &cli.Handler{
+            Pattern: "[global options] update [options] <id> <path>",
+            Description: "Update file, this creates a new revision of the file",
+            Callback: updateHandler,
+            Flags: cli.Flags{
+                "global options": globalFlags,
+                "options": []cli.Flag{
+                    cli.StringSliceFlag{
+                        Name: "parent",
+                        Patterns: []string{"-p", "--parent"},
+                        Description: "Parent id, used to upload file to a specific directory, can be specified multiple times to give many parents",
+                    },
+                    cli.StringFlag{
+                        Name: "name",
+                        Patterns: []string{"--name"},
+                        Description: "Filename",
+                    },
+                    cli.BoolFlag{
+                        Name: "noProgress",
+                        Patterns: []string{"--no-progress"},
+                        Description: "Hide progress",
+                        OmitValue: true,
+                    },
+                    cli.BoolFlag{
+                        Name: "stdin",
+                        Patterns: []string{"--stdin"},
+                        Description: "Use stdin as file content",
+                        OmitValue: true,
+                    },
+                    cli.StringFlag{
+                        Name: "mime",
+                        Patterns: []string{"--mime"},
+                        Description: "Force mime type",
+                    },
+                    cli.BoolFlag{
+                        Name: "share",
+                        Patterns: []string{"--share"},
+                        Description: "Share file",
+                        OmitValue: true,
+                    },
+                },
+            },
+        },
+        &cli.Handler{
             Pattern: "[global options] info [options] <id>",
             Description: "Show file info",
             Callback: infoHandler,
diff --git a/handlers_drive.go b/handlers_drive.go
index 1b01996..7491996 100644
--- a/handlers_drive.go
+++ b/handlers_drive.go
@@ -67,6 +67,22 @@ func uploadHandler(ctx cli.Context) {
     checkErr(err)
 }
 
+func updateHandler(ctx cli.Context) {
+    args := ctx.Args()
+    err := newDrive(args).Update(drive.UpdateArgs{
+        Out: os.Stdout,
+        Id: args.String("id"),
+        Path: args.String("path"),
+        Name: args.String("name"),
+        Parents: args.StringSlice("parent"),
+        Mime: args.String("mime"),
+        Stdin: args.Bool("stdin"),
+        Share: args.Bool("share"),
+        NoProgress: args.Bool("noProgress"),
+    })
+    checkErr(err)
+}
+
 func infoHandler(ctx cli.Context) {
     args := ctx.Args()
     err := newDrive(args).Info(drive.FileInfoArgs{