Add flag --include-docs to include google docs in listing

2 files changed, 4 insertions, 4 deletions

diff --git a/cli/cli.go b/cli/cli.go
index 38de1ba..93bca81 100644
--- a/cli/cli.go
+++ b/cli/cli.go
@@ -13,7 +13,7 @@ import (
 	"strings"
 )
 
-func List(d *gdrive.Drive, query, titleFilter string, maxResults int, sharedStatus bool, noHeader bool) error {
+func List(d *gdrive.Drive, query, titleFilter string, maxResults int, sharedStatus bool, noHeader bool, includeDocs bool) error {
 	caller := d.Files.List()
 
 	if maxResults > 0 {
@@ -37,8 +37,7 @@ func List(d *gdrive.Drive, query, titleFilter string, maxResults int, sharedStat
 	items := make([]map[string]string, 0, 0)
 
 	for _, f := range list.Items {
-		// Skip files that dont have a download url (they are not stored on google drive)
-		if f.DownloadUrl == "" {
+		if f.DownloadUrl == "" && !includeDocs {
 			if f.MimeType != "application/vnd.google-apps.folder" {
 				continue
 			}
diff --git a/drive.go b/drive.go
index e11f7b2..3c3e78c 100644
--- a/drive.go
+++ b/drive.go
@@ -24,6 +24,7 @@ type Options struct {
 
 	List struct {
 		MaxResults   int    `goptions:"-m, --max, description='Max results'"`
+		IncludeDocs  bool   `goptions:"--include-docs, mutexgroup='query', description='Include google docs in listing'"`
 		TitleFilter  string `goptions:"-t, --title, mutexgroup='query', description='Title filter'"`
 		Query        string `goptions:"-q, --query, mutexgroup='query', description='Query (see https://developers.google.com/drive/search-parameters)'"`
 		SharedStatus bool   `goptions:"-s, --shared, description='Show shared status (Note: this will generate 1 http req per file)'"`
@@ -97,7 +98,7 @@ func main() {
 	switch opts.Verbs {
 	case "list":
 		args := opts.List
-		err = cli.List(drive, args.Query, args.TitleFilter, args.MaxResults, args.SharedStatus, args.NoHeader)
+		err = cli.List(drive, args.Query, args.TitleFilter, args.MaxResults, args.SharedStatus, args.NoHeader, args.IncludeDocs)
 
 	case "info":
 		err = cli.Info(drive, opts.Info.FileId)