1package commands
  2
  3import (
  4	"fmt"
  5	"strings"
  6
  7	"github.com/MichaelMure/git-bug/cache"
  8	"github.com/MichaelMure/git-bug/util/colors"
  9	"github.com/MichaelMure/git-bug/util/interrupt"
 10	"github.com/spf13/cobra"
 11)
 12
 13var (
 14	lsStatusQuery   []string
 15	lsAuthorQuery   []string
 16	lsTitleQuery    []string
 17	lsLabelQuery    []string
 18	lsNoQuery       []string
 19	lsSortBy        string
 20	lsSortDirection string
 21)
 22
 23func runLsBug(cmd *cobra.Command, args []string) error {
 24	backend, err := cache.NewRepoCache(repo)
 25	if err != nil {
 26		return err
 27	}
 28	defer backend.Close()
 29	interrupt.RegisterCleaner(backend.Close)
 30
 31	var query *cache.Query
 32	if len(args) >= 1 {
 33		query, err = cache.ParseQuery(strings.Join(args, " "))
 34
 35		if err != nil {
 36			return err
 37		}
 38	} else {
 39		query, err = lsQueryFromFlags()
 40		if err != nil {
 41			return err
 42		}
 43	}
 44
 45	allIds := backend.QueryBugs(query)
 46
 47	for _, id := range allIds {
 48		b, err := backend.ResolveBugExcerpt(id)
 49		if err != nil {
 50			return err
 51		}
 52
 53		// truncate + pad if needed
 54		titleFmt := fmt.Sprintf("%-50.50s", b.Title)
 55		authorFmt := fmt.Sprintf("%-15.15s", b.LegacyAuthor.Name)
 56
 57		fmt.Printf("%s %s\t%s\t%s\tC:%d L:%d\n",
 58			colors.Cyan(b.HumanId()),
 59			colors.Yellow(b.Status),
 60			titleFmt,
 61			colors.Magenta(authorFmt),
 62			b.LenComments,
 63			len(b.Labels),
 64		)
 65	}
 66
 67	return nil
 68}
 69
 70// Transform the command flags into a query
 71func lsQueryFromFlags() (*cache.Query, error) {
 72	query := cache.NewQuery()
 73
 74	for _, status := range lsStatusQuery {
 75		f, err := cache.StatusFilter(status)
 76		if err != nil {
 77			return nil, err
 78		}
 79		query.Status = append(query.Status, f)
 80	}
 81
 82	for _, title := range lsTitleQuery {
 83		f := cache.TitleFilter(title)
 84		query.Title = append(query.Title, f)
 85	}
 86
 87	for _, author := range lsAuthorQuery {
 88		f := cache.AuthorFilter(author)
 89		query.Author = append(query.Author, f)
 90	}
 91
 92	for _, label := range lsLabelQuery {
 93		f := cache.LabelFilter(label)
 94		query.Label = append(query.Label, f)
 95	}
 96
 97	for _, no := range lsNoQuery {
 98		switch no {
 99		case "label":
100			query.NoFilters = append(query.NoFilters, cache.NoLabelFilter())
101		default:
102			return nil, fmt.Errorf("unknown \"no\" filter %s", no)
103		}
104	}
105
106	switch lsSortBy {
107	case "id":
108		query.OrderBy = cache.OrderById
109	case "creation":
110		query.OrderBy = cache.OrderByCreation
111	case "edit":
112		query.OrderBy = cache.OrderByEdit
113	default:
114		return nil, fmt.Errorf("unknown sort flag %s", lsSortBy)
115	}
116
117	switch lsSortDirection {
118	case "asc":
119		query.OrderDirection = cache.OrderAscending
120	case "desc":
121		query.OrderDirection = cache.OrderDescending
122	default:
123		return nil, fmt.Errorf("unknown sort direction %s", lsSortDirection)
124	}
125
126	return query, nil
127}
128
129var lsCmd = &cobra.Command{
130	Use:   "ls [<query>]",
131	Short: "List bugs.",
132	Long: `Display a summary of each bugs.
133
134You can pass an additional query to filter and order the list. This query can be expressed either with a simple query language or with flags.`,
135	Example: `List open bugs sorted by last edition with a query:
136git bug ls status:open sort:edit-desc
137
138List closed bugs sorted by creation with flags:
139git bug ls --status closed --by creation
140`,
141	PreRunE: loadRepo,
142	RunE:    runLsBug,
143}
144
145func init() {
146	RootCmd.AddCommand(lsCmd)
147
148	lsCmd.Flags().SortFlags = false
149
150	lsCmd.Flags().StringSliceVarP(&lsStatusQuery, "status", "s", nil,
151		"Filter by status. Valid values are [open,closed]")
152	lsCmd.Flags().StringSliceVarP(&lsAuthorQuery, "author", "a", nil,
153		"Filter by author")
154	lsCmd.Flags().StringSliceVarP(&lsLabelQuery, "label", "l", nil,
155		"Filter by label")
156	lsCmd.Flags().StringSliceVarP(&lsTitleQuery, "title", "t", nil,
157		"Filter by title")
158	lsCmd.Flags().StringSliceVarP(&lsNoQuery, "no", "n", nil,
159		"Filter by absence of something. Valid values are [label]")
160	lsCmd.Flags().StringVarP(&lsSortBy, "by", "b", "creation",
161		"Sort the results by a characteristic. Valid values are [id,creation,edit]")
162	lsCmd.Flags().StringVarP(&lsSortDirection, "direction", "d", "asc",
163		"Select the sorting direction. Valid values are [asc,desc]")
164}