filter.go

package cmd

import (
	"encoding/json"
	"fmt"

	"github.com/opslevel/opslevel-go/v2025"

	"github.com/opslevel/cli/common"
	"github.com/spf13/cobra"
)

var exampleFilterCmd = &cobra.Command{
	Use:   "filter",
	Short: "Example filter",
	Long:  `Example filter`,
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println(getExample[opslevel.FilterCreateInput]())
	},
}

var createFilterCmd = &cobra.Command{
	Use:   "filter",
	Short: "Create a filter",
	Long:  `Create a filter`,
	Example: `
cat << EOF | opslevel create filter -f -
name: "Tier 1 apps using RDS"
connective: "and"
predicates:
  - key: "tier_index"
    type: "equals"
    value: "1"
  - key: "tags"
    keyData: "db"
    type: "equals"
    value: "rds"
EOF`,
	Run: func(cmd *cobra.Command, args []string) {
		input, err := readResourceInput[opslevel.FilterCreateInput]()
		cobra.CheckErr(err)
		result, err := getClientGQL().CreateFilter(*input)
		cobra.CheckErr(err)
		fmt.Println(result.Id)
	},
}

var getFilterCmd = &cobra.Command{
	Use:        "filter ID",
	Short:      "Get details about a filter",
	Long:       `Get details about a filter`,
	Args:       cobra.ExactArgs(1),
	ArgAliases: []string{"ID"},
	Run: func(cmd *cobra.Command, args []string) {
		key := args[0]
		filter, err := getClientGQL().GetFilter(opslevel.ID(key))
		cobra.CheckErr(err)
		common.PrettyPrint(filter)
	},
}

var listFilterCmd = &cobra.Command{
	Use:     "filter",
	Aliases: []string{"filters"},
	Short:   "Lists filters",
	Long:    `Lists filters`,
	Run: func(cmd *cobra.Command, args []string) {
		resp, err := getClientGQL().ListFilters(nil)
		cobra.CheckErr(err)
		list := resp.Nodes
		if isJsonOutput() {
			common.JsonPrint(json.MarshalIndent(list, "", "    "))
		} else {
			w := common.NewTabWriter("NAME", "ALIAS", "ID")
			for _, item := range list {
				fmt.Fprintf(w, "%s\t%s\t%s\t\n", item.Name, item.Alias(), item.Id)
			}
			w.Flush()
		}
	},
}

var updateFilterCmd = &cobra.Command{
	Use:   "filter ID",
	Short: "Update a filter",
	Long:  `Update a filter`,
	Example: `
cat << EOF | opslevel update filter Z2lkOi8vb3BzbGV2ZWwvRmlsdGVyLzIzNTk -f -
name: "Tier 2 apps using RDS"
connective: "and"
predicates:
  - key: "tier_index"
    type: "equals"
    value: "2"
  - key: "tags"
    keyData: "db"
    type: "equals"
    value: "dynamo"
EOF`,
	Args:       cobra.ExactArgs(1),
	ArgAliases: []string{"ID"},
	Run: func(cmd *cobra.Command, args []string) {
		input, err := readResourceInput[opslevel.FilterCreateInput]()
		cobra.CheckErr(err)

		// hack: in the future all ObjectUpdateInput and ObjectCreateInput
		// will be merged into ObjectInput. for now, create an update input
		// by adding in the first argument.
		updateInput := &opslevel.FilterUpdateInput{
			Id:         *opslevel.NewID(args[0]),
			Name:       opslevel.NewNullableFrom(input.Name),
			Predicates: input.Predicates,
			Connective: input.Connective,
		}
		filter, err := getClientGQL().UpdateFilter(*updateInput)
		cobra.CheckErr(err)
		common.JsonPrint(json.MarshalIndent(filter, "", "    "))
	},
}

var deleteFilterCmd = &cobra.Command{
	Use:        "filter ID",
	Short:      "Delete a filter",
	Long:       `Delete a filter`,
	Args:       cobra.ExactArgs(1),
	ArgAliases: []string{"ID"},
	Run: func(cmd *cobra.Command, args []string) {
		key := args[0]
		err := getClientGQL().DeleteFilter(opslevel.ID(key))
		cobra.CheckErr(err)
		fmt.Printf("deleted '%s' filter\n", key)
	},
}

func init() {
	exampleCmd.AddCommand(exampleFilterCmd)
	createCmd.AddCommand(createFilterCmd)
	updateCmd.AddCommand(updateFilterCmd)
	getCmd.AddCommand(getFilterCmd)
	listCmd.AddCommand(listFilterCmd)
	deleteCmd.AddCommand(deleteFilterCmd)
}