gogcli/internal/cmd/drive_listing.go

package cmd

import (
	"context"
	"fmt"
	"os"
	"strings"

	"google.golang.org/api/drive/v3"
	gapi "google.golang.org/api/googleapi"

	"github.com/steipete/gogcli/internal/outfmt"
	"github.com/steipete/gogcli/internal/ui"
)

type driveFileListOptions struct {
	query     string
	max       int64
	page      string
	allDrives bool
	driveID   string
	fields    string // optional field mask override
}

func (c *DriveLsCmd) Run(ctx context.Context, flags *RootFlags) error {
	if c.All && strings.TrimSpace(c.Parent) != "" {
		return usage("--all cannot be combined with --parent")
	}

	folderID := strings.TrimSpace(c.Parent)
	if folderID == "" {
		folderID = "root"
	}

	_, svc, err := requireDriveService(ctx, flags)
	if err != nil {
		return err
	}

	query := buildDriveListQuery(folderID, c.Query)
	if c.All {
		query = buildDriveAllListQuery(c.Query)
	}

	resp, err := listDriveFiles(ctx, svc, driveFileListOptions{
		query:     query,
		max:       c.Max,
		page:      c.Page,
		allDrives: c.AllDrives,
		fields:    c.Fields,
	})
	if err != nil {
		return err
	}

	return writeDriveFileList(ctx, resp, "No files")
}

func (c *DriveSearchCmd) Run(ctx context.Context, flags *RootFlags) error {
	query := strings.TrimSpace(strings.Join(c.Query, " "))
	if query == "" {
		return usage("missing query")
	}
	driveID := strings.TrimSpace(c.Drive)
	parentID := strings.TrimSpace(c.Parent)

	if driveID != "" && !c.AllDrives {
		return usage("--drive cannot be combined with --no-all-drives")
	}
	if parentID != "" && c.RawQuery {
		return usage("--parent cannot be combined with --raw-query; include the \"'<parentId>' in parents\" clause in your raw query instead")
	}

	_, svc, err := requireDriveService(ctx, flags)
	if err != nil {
		return err
	}

	finalQuery := buildDriveSearchQuery(query, c.RawQuery)
	if parentID != "" {
		finalQuery = fmt.Sprintf("'%s' in parents and %s", escapeDriveQueryString(parentID), finalQuery)
	}

	resp, err := listDriveFiles(ctx, svc, driveFileListOptions{
		query:     finalQuery,
		max:       c.Max,
		page:      c.Page,
		allDrives: c.AllDrives,
		driveID:   driveID,
	})
	if err != nil {
		return err
	}

	return writeDriveFileList(ctx, resp, "No results")
}

func listDriveFiles(ctx context.Context, svc *drive.Service, opts driveFileListOptions) (*drive.FileList, error) {
	call := svc.Files.List().
		Q(opts.query).
		PageSize(opts.max).
		PageToken(opts.page).
		OrderBy("modifiedTime desc")
	call = driveFilesListCallWithDriveSupport(call, opts.allDrives, opts.driveID)
	mask := driveFileListFields
	if strings.TrimSpace(opts.fields) != "" {
		mask = opts.fields
	}
	return call.Fields(gapi.Field(mask)).Context(ctx).Do()
}

func writeDriveFileList(ctx context.Context, resp *drive.FileList, emptyMessage string) error {
	u := ui.FromContext(ctx)
	if outfmt.IsJSON(ctx) {
		return outfmt.WriteJSON(ctx, os.Stdout, map[string]any{
			"files":         resp.Files,
			"nextPageToken": resp.NextPageToken,
		})
	}

	if len(resp.Files) == 0 {
		u.Err().Println(emptyMessage)
		return nil
	}

	w, flush := tableWriter(ctx)
	defer flush()
	fmt.Fprintln(w, "ID\tNAME\tTYPE\tSIZE\tMODIFIED\tOWNER")
	for _, f := range resp.Files {
		fmt.Fprintf(
			w,
			"%s\t%s\t%s\t%s\t%s\t%s\n",
			f.Id,
			f.Name,
			driveType(f.MimeType),
			formatDriveSize(f.Size),
			formatDateTime(f.ModifiedTime),
			driveOwnerEmail(f.Owners),
		)
	}
	printNextPageHint(u, resp.NextPageToken)
	return nil
}

func driveOwnerEmail(owners []*drive.User) string {
	if len(owners) == 0 || owners[0] == nil || strings.TrimSpace(owners[0].EmailAddress) == "" {
		return "-"
	}

	return owners[0].EmailAddress
}