gsheets.go

package main

import (
	"context"
	"fmt"
	"os"

	"os/exec"
	"runtime"
	"time"

	"go.bbkane.com/warg/command"
	"golang.org/x/oauth2/google"
	"google.golang.org/api/option"
	"google.golang.org/api/sheets/v4"
)

func gSheetsOpen(ctx command.Context) error {
	spreadsheetId := ctx.Flags["--spreadsheet-id"].(string)
	sheetID := ctx.Flags["--sheet-id"].(int)

	link := fmt.Sprintf(
		"https://docs.google.com/spreadsheets/d/%s/edit#gid=%d",
		spreadsheetId,
		sheetID,
	)
	fmt.Printf("Opening: %s\n", link)

	// https://stackoverflow.com/a/39324149/2958070
	var cmd string
	var args []string

	switch runtime.GOOS {
	case "windows":
		cmd = "cmd"
		args = []string{"/c", "start"}
	case "darwin":
		cmd = "open"
	default: // "linux", "freebsd", "openbsd", "netbsd"
		cmd = "xdg-open"
	}
	args = append(args, link)
	return exec.Command(cmd, args...).Start()

}

func gSheetsUpload(ctx command.Context) error {
	csvPath := ctx.Flags["--csv-path"].(string)
	spreadsheetId := ctx.Flags["--spreadsheet-id"].(string)
	sheetID := ctx.Flags["--sheet-id"].(int)
	timeout := ctx.Flags["--timeout"].(time.Duration)
	timeCtx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()

	csvBytes, err := os.ReadFile(csvPath)
	if err != nil {
		return fmt.Errorf("csv read error: %s: %w", csvPath, err)
	}
	csvStr := string(csvBytes)

	creds, err := google.FindDefaultCredentials(timeCtx, sheets.SpreadsheetsScope)
	if err != nil {
		return fmt.Errorf("can't find default credentials: %w", err)
	}

	srv, err := sheets.NewService(timeCtx, option.WithCredentials(creds))
	if err != nil {
		return fmt.Errorf("unable to retrieve Sheets client: %w", err)
	}

	requests := []*sheets.Request{
		// Erase current cells
		// https://stackoverflow.com/q/37928947/2958070
		{
			UpdateCells: &sheets.UpdateCellsRequest{
				Fields: "*",
				Range: &sheets.GridRange{
					SheetId:          0, // TODO: test this...
					EndColumnIndex:   0,
					EndRowIndex:      0,
					StartColumnIndex: 0,
					StartRowIndex:    0,
					ForceSendFields:  nil,
					NullFields:       nil,
				},
				Rows:            nil,
				Start:           nil,
				ForceSendFields: nil,
				NullFields:      nil,
			},
		},
		// https://stackoverflow.com/q/42362702/2958070
		{
			PasteData: &sheets.PasteDataRequest{
				Coordinate: &sheets.GridCoordinate{
					ColumnIndex: 0,
					RowIndex:    0,
					// https://developers.google.com/sheets/api/guides/concepts
					SheetId:         int64(sheetID), // TODO: prefer reading an int64 flag, not casting :)
					ForceSendFields: nil,
					NullFields:      nil,
				},
				Data:            csvStr,
				Delimiter:       ",",
				Type:            "PASTE_NORMAL",
				Html:            false,
				ForceSendFields: nil,
				NullFields:      nil,
			},
		},
	}

	rb := &sheets.BatchUpdateSpreadsheetRequest{
		Requests:                     requests,
		IncludeSpreadsheetInResponse: false,
		ResponseIncludeGridData:      false,
		ResponseRanges:               nil,
		ForceSendFields:              nil,
		NullFields:                   nil,
	}

	resp, err := srv.Spreadsheets.BatchUpdate(
		spreadsheetId,
		rb,
	).Context(timeCtx).Do()
	if err != nil {
		return fmt.Errorf("batch error failure: %w", err)
	}

	fmt.Printf("Status Code: %d\n", resp.HTTPStatusCode)
	return nil
}