ax/cmd/list.go

package cmd

import (
	"axolotl/models"
	"axolotl/output"
	"axolotl/service"
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

var lTags, lRels []string
var lStatus, lPrio, lType, lNamespace, lAssignee, lMention string

var listCmd = &cobra.Command{
	Use: "list", Short: "List nodes",
	Run: func(cmd *cobra.Command, args []string) {
		svc, err := service.GetNodeService(cfg)
		if err != nil {
			fmt.Fprintln(os.Stderr, err)
			return
		}

		var filter service.ListFilter

		// --tag is an alias for a label filter with no target.
		for _, tag := range lTags {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelType(tag), Target: ""})
		}

		// Shorthand flags expand to property filters or edge filters.
		if lStatus != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelType("_status::" + lStatus), Target: ""})
		}
		if lPrio != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelType("_prio::" + lPrio), Target: ""})
		}
		if lType != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelType("_type::" + lType), Target: ""})
		}
		if lNamespace != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelInNamespace, Target: lNamespace})
		}
		if lAssignee != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelAssignee, Target: lAssignee})
		}
		if lMention != "" {
			filter.Rels = append(filter.Rels, service.RelInput{Type: models.RelMentions, Target: lMention})
		}

		for _, r := range lRels {
			ri, err := parseRelInput(r)
			if err != nil {
				fmt.Fprintf(os.Stderr, "failed to parse relation flag: %v", err)
				return
			}
			filter.Rels = append(filter.Rels, ri)
		}

		if nodes, err := svc.List(filter); err == nil {
			output.PrintNodes(cmd.OutOrStdout(), svc, nodes, jsonFlag)
		} else {
			fmt.Fprintf(os.Stderr, "err: %v\n", err)
		}
	},
}

func init() {
	rootCmd.AddCommand(listCmd)
	f := listCmd.Flags()
	f.StringArrayVar(&lTags, "tag", nil, "filter by label tag")
	f.StringArrayVar(&lRels, "rel", nil, "filter by relation (prefix::value or relname:target)")
	f.StringVar(&lStatus, "status", "", "filter by status")
	f.StringVar(&lPrio, "prio", "", "filter by priority")
	f.StringVar(&lType, "type", "", "filter by type")
	f.StringVar(&lNamespace, "namespace", "", "filter by namespace")
	f.StringVar(&lAssignee, "assignee", "", "filter by assignee")
	f.StringVar(&lMention, "mention", "", "filter by mention")
}