gitea-tea/internal/app/milestone_issues.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package app

import (
	"errors"
	"fmt"
	"strconv"
	"strings"

	"gitea.dev/tea/internal/services"
	"github.com/spf13/cobra"
)

func (a *application) milestoneIssueCommand() *cobra.Command {
	group := &cobra.Command{Use: "issue", Short: "Manage the issues and pull requests of a milestone", Long: `List the issues and pull requests in a milestone, or add them to it and remove them from
it. Milestones are addressed by title or ID; issues and pull requests by number or URL.`}
	opts := services.ListOptions{Page: 1}
	var byName bool
	list := &cobra.Command{Use: "list <milestone>", Short: "List issues and pull requests in a milestone", Example: `$ tea milestone issue list v1.0
$ tea milestone issue list v1.0 --state all --kind issues`, Annotations: map[string]string{"tea.arguments": milestoneArguments}, Args: argsExact(1), RunE: func(cmd *cobra.Command, args []string) error {
		target, err := a.resolve(cmd.Context(), true)
		if err != nil {
			return err
		}
		milestone, err := findMilestone(cmd, services.Milestones{Client: target.client, Repository: target.Repo}, args[0], byName)
		if err != nil {
			return err
		}
		opts.Milestones = strconv.FormatInt(milestone.ID, 10)
		page, err := (services.Issues{Client: target.client, Repository: target.Repo}).ListPage(cmd.Context(), opts)
		if err != nil {
			return err
		}
		where := fmt.Sprintf("in milestone %q of %s", milestone.Title, target.Repo)
		return writeTable(a, cmd, page.Items, issueListSpec(where, opts.State, opts.Kind, false), servicePage(page, opts.Page))
	}}
	pageFlags(list, &opts.Limit, &opts.Page)
	enumFlag(list, &opts.State, "state", "s", "open", "Filter by state", "open", "closed", "all")
	enumFlag(list, &opts.Kind, "kind", "", "all", "List issues, pull requests or both", "issues", "pulls", "all")
	list.Flags().BoolVar(&byName, "by-name", false, "Match a numeric argument as a milestone title rather than an ID")
	list.ValidArgsFunction = a.completeMilestoneArgs(false)
	group.AddCommand(list)
	for _, verb := range []string{"add", "remove"} {
		var byName bool
		short, preposition, past := "Add issues or pull requests to a milestone", "to", "Added"
		if verb == "remove" {
			short, preposition, past = "Remove issues or pull requests from a milestone", "from", "Removed"
		}
		command := &cobra.Command{Use: verb + " <milestone> {<number> | <url>}...", Short: short, Example: fmt.Sprintf(`$ tea milestone issue %[1]s v1.0 12
$ tea milestone issue %[1]s v1.0 12 13 '#14'`, verb), Annotations: map[string]string{"tea.arguments": milestoneArguments + "\n\n" + issueArguments}, Args: usageArgs(cobra.MinimumNArgs(2)), RunE: func(cmd *cobra.Command, args []string) error {
			target, numbers, err := a.resolveRefs(cmd, args[1:], issueRef)
			if err != nil {
				return err
			}
			milestone, err := findMilestone(cmd, services.Milestones{Client: target.client, Repository: target.Repo}, args[0], byName)
			if err != nil {
				return err
			}
			service := services.Issues{Client: target.client, Repository: target.Repo}
			results, failures := applyBatch(cmd, numbers, issueLabel, func(n int64) (services.Issue, error) {
				if verb == "remove" {
					return service.RemoveMilestone(cmd.Context(), n, milestone)
				}
				return service.Edit(cmd.Context(), n, services.IssueEdit{Milestone: &milestone.ID})
			}, func(services.Issue) {})
			if len(results) > 0 {
				changed := make([]string, len(results))
				for i, item := range results {
					changed[i] = "#" + strconv.FormatInt(item.Number, 10)
				}
				a.done(cmd, "%s %s %s %s", past, strings.Join(changed, ", "), preposition, milestoneName(target, milestone))
			}
			if a.options.json != "" {
				failures = append(failures, writeBatchJSON(a, cmd, len(numbers), results))
			}
			return errors.Join(failures...)
		}}
		command.Flags().BoolVar(&byName, "by-name", false, "Match a numeric milestone argument as a title rather than an ID")
		milestones, issues := a.completeMilestones(), a.completeIssues("open")
		command.ValidArgsFunction = func(cmd *cobra.Command, args []string, toComplete string) ([]cobra.Completion, cobra.ShellCompDirective) {
			if len(args) == 0 {
				return milestones(cmd, args, toComplete)
			}
			return issues(cmd, args, toComplete)
		}
		group.AddCommand(command)
	}
	return group
}