aboutsummaryrefslogtreecommitdiffstats
path: root/drive/changes.go
blob: ffd782428da6eeea44ae3ab7c9c515ffbd4efe78 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package drive

import (
	"fmt"
	"google.golang.org/api/drive/v3"
	"io"
	"text/tabwriter"
)

type ListChangesArgs struct {
	Out        io.Writer
	PageToken  string
	MaxChanges int64
	Now        bool
	NameWidth  int64
	SkipHeader bool
}

func (self *Drive) ListChanges(args ListChangesArgs) error {
	if args.Now {
		pageToken, err := self.GetChangesStartPageToken()
		if err != nil {
			return err
		}

		fmt.Fprintf(args.Out, "Page token: %s\n", pageToken)
		return nil
	}

	changeList, err := self.service.Changes.List(args.PageToken).PageSize(args.MaxChanges).RestrictToMyDrive(true).Fields("newStartPageToken", "nextPageToken", "changes(fileId,removed,time,file(id,name,md5Checksum,mimeType,createdTime,modifiedTime))").Do()
	if err != nil {
		return fmt.Errorf("Failed listing changes: %s", err)
	}

	PrintChanges(PrintChangesArgs{
		Out:        args.Out,
		ChangeList: changeList,
		NameWidth:  int(args.NameWidth),
		SkipHeader: args.SkipHeader,
	})

	return nil
}

func (self *Drive) GetChangesStartPageToken() (string, error) {
	res, err := self.service.Changes.GetStartPageToken().Do()
	if err != nil {
		return "", fmt.Errorf("Failed getting start page token: %s", err)
	}

	return res.StartPageToken, nil
}

type PrintChangesArgs struct {
	Out        io.Writer
	ChangeList *drive.ChangeList
	NameWidth  int
	SkipHeader bool
}

func PrintChanges(args PrintChangesArgs) {
	w := new(tabwriter.Writer)
	w.Init(args.Out, 0, 0, 3, ' ', 0)

	if !args.SkipHeader {
		fmt.Fprintln(w, "Id\tName\tAction\tTime")
	}

	for _, c := range args.ChangeList.Changes {
		var name string
		var action string

		if c.Removed {
			action = "remove"
		} else {
			name = c.File.Name
			action = "update"
		}

		fmt.Fprintf(w, "%s\t%s\t%s\t%s\n",
			c.FileId,
			truncateString(name, args.NameWidth),
			action,
			formatDatetime(c.Time),
		)
	}

	if len(args.ChangeList.Changes) > 0 {
		w.Flush()
		pageToken, hasMore := nextChangesPageToken(args.ChangeList)
		fmt.Fprintf(args.Out, "\nToken: %s, more: %t\n", pageToken, hasMore)
	} else {
		fmt.Fprintln(args.Out, "No changes")
	}
}

func nextChangesPageToken(cl *drive.ChangeList) (string, bool) {
	if cl.NextPageToken != "" {
		return cl.NextPageToken, true
	}

	return cl.NewStartPageToken, false
}