gogcli/internal/cmd/forms_modify.go

package cmd

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

	formsapi "google.golang.org/api/forms/v1"

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

// FormsAddQuestionCmd adds a question to an existing form via batchUpdate.
type FormsAddQuestionCmd struct {
	FormID   string   `arg:"" name:"formId" help:"Form ID"`
	Title    string   `name:"title" help:"Question title/text" required:""`
	Type     string   `name:"type" help:"Question type: text|paragraph|radio|checkbox|dropdown|scale|date|time" default:"text"`
	Required bool     `name:"required" help:"Whether an answer is required"`
	Options  []string `name:"option" help:"Choice options (for radio/checkbox/dropdown, repeat for each)" short:"o"`
	Index    int      `name:"index" help:"Position to insert (0-based, default append)" default:"-1"`

	// Scale-specific
	ScaleLow       int    `name:"scale-low" help:"Scale minimum value" default:"1"`
	ScaleHigh      int    `name:"scale-high" help:"Scale maximum value" default:"5"`
	ScaleLowLabel  string `name:"scale-low-label" help:"Label for low end of scale"`
	ScaleHighLabel string `name:"scale-high-label" help:"Label for high end of scale"`

	// Date/time specific
	IncludeTime bool `name:"include-time" help:"Include time picker (for date type)"`
	IncludeYear bool `name:"include-year" help:"Include year field (for date type)"`
	Duration    bool `name:"duration" help:"Ask for duration instead of time (for time type)"`

	Description string `name:"description" help:"Question description/help text"`
}

func (c *FormsAddQuestionCmd) Run(ctx context.Context, flags *RootFlags) error {
	account, err := requireAccount(flags)
	if err != nil {
		return err
	}
	formID := strings.TrimSpace(normalizeGoogleID(c.FormID))
	if formID == "" {
		return usage("empty formId")
	}
	title := strings.TrimSpace(c.Title)
	if title == "" {
		return usage("empty --title")
	}
	qType := strings.ToLower(strings.TrimSpace(c.Type))

	question, err := buildQuestion(qType, c)
	if err != nil {
		return err
	}

	if dryRunErr := dryRunExit(ctx, flags, "forms.addQuestion", map[string]any{
		"form_id":     formID,
		"title":       title,
		"type":        qType,
		"required":    c.Required,
		"options":     c.Options,
		"index":       c.Index,
		"description": c.Description,
	}); dryRunErr != nil {
		return dryRunErr
	}

	svc, err := newFormsService(ctx, account)
	if err != nil {
		return err
	}

	item := &formsapi.Item{
		Title:       title,
		Description: strings.TrimSpace(c.Description),
		QuestionItem: &formsapi.QuestionItem{
			Question: question,
		},
	}

	createReq := &formsapi.CreateItemRequest{
		Item: item,
	}

	// Determine the insertion index. The API requires a Location.
	// For index 0 we must use ForceSendFields since 0 is Go's zero-value.
	var insertAt int64
	if c.Index >= 0 {
		insertAt = int64(c.Index)
	} else {
		// Append: get the form to find current item count.
		currentForm, getErr := svc.Forms.Get(formID).Context(ctx).Do()
		if getErr != nil {
			return getErr
		}
		insertAt = int64(len(currentForm.Items))
	}
	createReq.Location = &formsapi.Location{
		Index:           insertAt,
		ForceSendFields: []string{"Index"},
	}

	batchReq := &formsapi.BatchUpdateFormRequest{
		Requests: []*formsapi.Request{
			{CreateItem: createReq},
		},
		IncludeFormInResponse: true,
	}

	resp, err := svc.Forms.BatchUpdate(formID, batchReq).Context(ctx).Do()
	if err != nil {
		return err
	}

	// Determine the actual index used for output.
	var insertIndex int64 = -1
	if createReq.Location != nil {
		insertIndex = createReq.Location.Index
	} else if resp.Form != nil {
		// Appended — index is last item position.
		insertIndex = int64(len(resp.Form.Items) - 1)
	}

	if outfmt.IsJSON(ctx) {
		return outfmt.WriteJSON(ctx, os.Stdout, map[string]any{
			"created":  true,
			"form_id":  formID,
			"title":    title,
			"type":     qType,
			"index":    insertIndex,
			"form":     resp.Form,
			"edit_url": formEditURL(formID),
		})
	}

	u := ui.FromContext(ctx)
	u.Out().Printf("created\ttrue")
	u.Out().Printf("form_id\t%s", formID)
	u.Out().Printf("question\t%s", title)
	u.Out().Printf("type\t%s", qType)
	u.Out().Printf("index\t%d", insertIndex)
	u.Out().Printf("edit_url\t%s", formEditURL(formID))
	return nil
}

func buildQuestion(qType string, c *FormsAddQuestionCmd) (*formsapi.Question, error) {
	q := &formsapi.Question{
		Required: c.Required,
	}

	switch qType {
	case "text":
		q.TextQuestion = &formsapi.TextQuestion{Paragraph: false}
	case "paragraph":
		q.TextQuestion = &formsapi.TextQuestion{Paragraph: true}
	case "radio", "checkbox", "dropdown":
		if len(c.Options) == 0 {
			return nil, usage("--option is required for " + qType + " questions (repeat for each choice)")
		}
		apiType := map[string]string{
			"radio":    "RADIO",
			"checkbox": "CHECKBOX",
			"dropdown": "DROP_DOWN",
		}[qType]
		opts := make([]*formsapi.Option, len(c.Options))
		for i, v := range c.Options {
			opts[i] = &formsapi.Option{Value: v}
		}
		q.ChoiceQuestion = &formsapi.ChoiceQuestion{
			Type:    apiType,
			Options: opts,
		}
	case "scale":
		q.ScaleQuestion = &formsapi.ScaleQuestion{
			Low:       int64(c.ScaleLow),
			High:      int64(c.ScaleHigh),
			LowLabel:  c.ScaleLowLabel,
			HighLabel: c.ScaleHighLabel,
		}
	case "date":
		q.DateQuestion = &formsapi.DateQuestion{
			IncludeTime: c.IncludeTime,
			IncludeYear: c.IncludeYear,
		}
	case "time":
		q.TimeQuestion = &formsapi.TimeQuestion{
			Duration: c.Duration,
		}
	default:
		return nil, usage("unknown question type: " + qType + " (use text|paragraph|radio|checkbox|dropdown|scale|date|time)")
	}

	return q, nil
}

// FormsDeleteQuestionCmd removes a question from a form by index.
type FormsDeleteQuestionCmd struct {
	FormID string `arg:"" name:"formId" help:"Form ID"`
	Index  int    `arg:"" name:"index" help:"Question index (0-based)"`
}

func (c *FormsDeleteQuestionCmd) Run(ctx context.Context, flags *RootFlags) error {
	account, err := requireAccount(flags)
	if err != nil {
		return err
	}
	formID := strings.TrimSpace(normalizeGoogleID(c.FormID))
	if formID == "" {
		return usage("empty formId")
	}
	if c.Index < 0 {
		return usage("index must be >= 0")
	}

	svc, err := newFormsService(ctx, account)
	if err != nil {
		return err
	}

	form, err := svc.Forms.Get(formID).Context(ctx).Do()
	if err != nil {
		return err
	}
	if c.Index >= len(form.Items) {
		return usagef("question index %d out of range (form has %d items)", c.Index, len(form.Items))
	}

	if dryRunErr := dryRunExit(ctx, flags, "forms.deleteQuestion", map[string]any{
		"form_id":    formID,
		"index":      c.Index,
		"item_count": len(form.Items),
	}); dryRunErr != nil {
		return dryRunErr
	}

	if confirmErr := confirmDestructiveChecked(ctx, flagsWithoutDryRun(flags), fmt.Sprintf("delete question %d from form %s", c.Index, formID)); confirmErr != nil {
		return confirmErr
	}

	batchReq := &formsapi.BatchUpdateFormRequest{
		Requests: []*formsapi.Request{
			{
				DeleteItem: &formsapi.DeleteItemRequest{
					Location: &formsapi.Location{Index: int64(c.Index)},
				},
			},
		},
	}

	_, err = svc.Forms.BatchUpdate(formID, batchReq).Context(ctx).Do()
	if err != nil {
		return err
	}

	if outfmt.IsJSON(ctx) {
		return outfmt.WriteJSON(ctx, os.Stdout, map[string]any{
			"deleted": true,
			"form_id": formID,
			"index":   c.Index,
		})
	}

	u := ui.FromContext(ctx)
	u.Out().Printf("deleted\ttrue")
	u.Out().Printf("form_id\t%s", formID)
	u.Out().Printf("index\t%d", c.Index)
	return nil
}

// FormsMoveQuestionCmd moves a question to a new position.
type FormsMoveQuestionCmd struct {
	FormID   string `arg:"" name:"formId" help:"Form ID"`
	OldIndex int    `arg:"" name:"oldIndex" help:"Current question index (0-based)"`
	NewIndex int    `arg:"" name:"newIndex" help:"Target question index (0-based)"`
}

func (c *FormsMoveQuestionCmd) Run(ctx context.Context, flags *RootFlags) error {
	account, err := requireAccount(flags)
	if err != nil {
		return err
	}
	formID := strings.TrimSpace(normalizeGoogleID(c.FormID))
	if formID == "" {
		return usage("empty formId")
	}
	if c.OldIndex < 0 || c.NewIndex < 0 {
		return usage("indices must be >= 0")
	}

	if dryRunErr := dryRunExit(ctx, flags, "forms.moveQuestion", map[string]any{
		"form_id":   formID,
		"old_index": c.OldIndex,
		"new_index": c.NewIndex,
	}); dryRunErr != nil {
		return dryRunErr
	}

	svc, err := newFormsService(ctx, account)
	if err != nil {
		return err
	}

	batchReq := &formsapi.BatchUpdateFormRequest{
		Requests: []*formsapi.Request{
			{
				MoveItem: &formsapi.MoveItemRequest{
					OriginalLocation: &formsapi.Location{Index: int64(c.OldIndex)},
					NewLocation:      &formsapi.Location{Index: int64(c.NewIndex)},
				},
			},
		},
	}

	_, err = svc.Forms.BatchUpdate(formID, batchReq).Context(ctx).Do()
	if err != nil {
		return err
	}

	if outfmt.IsJSON(ctx) {
		return outfmt.WriteJSON(ctx, os.Stdout, map[string]any{
			"moved":     true,
			"form_id":   formID,
			"old_index": c.OldIndex,
			"new_index": c.NewIndex,
		})
	}

	u := ui.FromContext(ctx)
	u.Out().Printf("moved\ttrue")
	u.Out().Printf("form_id\t%s", formID)
	u.Out().Printf("old_index\t%d", c.OldIndex)
	u.Out().Printf("new_index\t%d", c.NewIndex)
	return nil
}

// FormsUpdateCmd modifies form title, description, or settings.
type FormsUpdateCmd struct {
	FormID      string `arg:"" name:"formId" help:"Form ID"`
	Title       string `name:"title" help:"New form title"`
	Description string `name:"description" help:"New form description"`
	IsQuiz      string `name:"quiz" help:"Enable quiz mode (true/false)"`
}

func (c *FormsUpdateCmd) Run(ctx context.Context, flags *RootFlags) error {
	account, err := requireAccount(flags)
	if err != nil {
		return err
	}
	formID := strings.TrimSpace(normalizeGoogleID(c.FormID))
	if formID == "" {
		return usage("empty formId")
	}

	title := strings.TrimSpace(c.Title)
	description := strings.TrimSpace(c.Description)
	quiz := strings.TrimSpace(strings.ToLower(c.IsQuiz))

	if title == "" && description == "" && quiz == "" {
		return usage("at least one of --title, --description, or --quiz is required")
	}

	if dryRunErr := dryRunExit(ctx, flags, "forms.update", map[string]any{
		"form_id":     formID,
		"title":       title,
		"description": description,
		"quiz":        quiz,
	}); dryRunErr != nil {
		return dryRunErr
	}

	svc, err := newFormsService(ctx, account)
	if err != nil {
		return err
	}

	var requests []*formsapi.Request

	if title != "" || description != "" {
		info := &formsapi.Info{}
		var masks []string
		if title != "" {
			info.Title = title
			masks = append(masks, "title")
		}
		if description != "" {
			info.Description = description
			masks = append(masks, "description")
		}
		requests = append(requests, &formsapi.Request{
			UpdateFormInfo: &formsapi.UpdateFormInfoRequest{
				Info:       info,
				UpdateMask: strings.Join(masks, ","),
			},
		})
	}

	if quiz != "" {
		isQuiz, parseErr := strconv.ParseBool(quiz)
		if parseErr != nil {
			return usage("--quiz must be true or false")
		}
		requests = append(requests, &formsapi.Request{
			UpdateSettings: &formsapi.UpdateSettingsRequest{
				Settings: &formsapi.FormSettings{
					QuizSettings: &formsapi.QuizSettings{
						IsQuiz: isQuiz,
					},
				},
				UpdateMask: "quizSettings.isQuiz",
			},
		})
	}

	batchReq := &formsapi.BatchUpdateFormRequest{
		Requests:              requests,
		IncludeFormInResponse: true,
	}

	resp, err := svc.Forms.BatchUpdate(formID, batchReq).Context(ctx).Do()
	if err != nil {
		return err
	}

	if outfmt.IsJSON(ctx) {
		return outfmt.WriteJSON(ctx, os.Stdout, map[string]any{
			"updated":  true,
			"form_id":  formID,
			"form":     resp.Form,
			"edit_url": formEditURL(formID),
		})
	}

	u := ui.FromContext(ctx)
	u.Out().Printf("updated\ttrue")
	printFormSummary(u, resp.Form, formID)
	return nil
}