Where appropriate retrieves multiple pages when listing files

jcruzcruz · Jul 14, 2015 · a4bfad1 · a4bfad1
1 parent 385b8b9
commit a4bfad1
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/cli/cli.go b/cli/cli.go
@@ -34,9 +34,20 @@ func List(d *gdrive.Drive, query, titleFilter string, maxResults int, sharedStat
 		return err
 	}
 
+	files := list.Items
+
+	for(list.NextPageToken != "") {
+		caller.PageToken(list.NextPageToken)
+		list, err = caller.Do()
+		if err != nil {
+			return err
+		}
+		files = append(files,list.Items...)
+	}
+
 	items := make([]map[string]string, 0, 0)
 
-	for _, f := range list.Items {
+	for _, f := range files {
 		if f.DownloadUrl == "" && !includeDocs {
 			if f.MimeType != "application/vnd.google-apps.folder" {
 				continue