From bdeea59c73bc254890d2ba79088f64a11080feb0 Mon Sep 17 00:00:00 2001
From: Michael Samuel <mik@miknet.net>
Date: Wed, 14 Aug 2019 02:41:40 +1000
Subject: [PATCH] fleetctl: add --yaml flag to get command for labels, packs,
 and queries (#2092)

Closes #1813
---
 cmd/fleetctl/get.go | 68 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/cmd/fleetctl/get.go b/cmd/fleetctl/get.go
index 70f46d34fd..ed8b7ffda5 100644
--- a/cmd/fleetctl/get.go
+++ b/cmd/fleetctl/get.go
@@ -11,6 +11,10 @@ import (
 	"github.com/urfave/cli"
 )
 
+const (
+	yamlFlagName = "yaml"
+)
+
 type specGeneric struct {
 	Kind    string      `json:"kind"`
 	Version string      `json:"apiVersion"`
@@ -23,6 +27,13 @@ func defaultTable() *tablewriter.Table {
 	return table
 }
 
+func yamlFlag() cli.BoolFlag {
+	return cli.BoolFlag{
+		Name:  yamlFlagName,
+		Usage: "Output packs in yaml format",
+	}
+}
+
 func getQueriesCommand() cli.Command {
 	return cli.Command{
 		Name:    "queries",
@@ -31,6 +42,7 @@ func getQueriesCommand() cli.Command {
 		Flags: []cli.Flag{
 			configFlag(),
 			contextFlag(),
+			yamlFlag(),
 		},
 		Action: func(c *cli.Context) error {
 			fleet, err := clientFromCLI(c)
@@ -47,6 +59,24 @@ func getQueriesCommand() cli.Command {
 					return errors.Wrap(err, "could not list queries")
 				}
 
+				if c.Bool(yamlFlagName) {
+					for _, query := range queries {
+						spec := specGeneric{
+							Kind:    "query",
+							Version: kolide.ApiVersion,
+							Spec:    query,
+						}
+
+						b, err := yaml.Marshal(spec)
+						if err != nil {
+							return err
+						}
+
+						fmt.Printf("---\n%s", string(b))
+					}
+					return nil
+				}
+
 				if len(queries) == 0 {
 					fmt.Println("no queries found")
 					return nil
@@ -100,6 +130,7 @@ func getPacksCommand() cli.Command {
 		Flags: []cli.Flag{
 			configFlag(),
 			contextFlag(),
+			yamlFlag(),
 		},
 		Action: func(c *cli.Context) error {
 			fleet, err := clientFromCLI(c)
@@ -116,6 +147,24 @@ func getPacksCommand() cli.Command {
 					return errors.Wrap(err, "could not list packs")
 				}
 
+				if c.Bool(yamlFlagName) {
+					for _, pack := range packs {
+						spec := specGeneric{
+							Kind:    "pack",
+							Version: kolide.ApiVersion,
+							Spec:    pack,
+						}
+
+						b, err := yaml.Marshal(spec)
+						if err != nil {
+							return err
+						}
+
+						fmt.Printf("---\n%s", string(b))
+					}
+					return nil
+				}
+
 				if len(packs) == 0 {
 					fmt.Println("no packs found")
 					return nil
@@ -169,6 +218,7 @@ func getLabelsCommand() cli.Command {
 		Flags: []cli.Flag{
 			configFlag(),
 			contextFlag(),
+			yamlFlag(),
 		},
 		Action: func(c *cli.Context) error {
 			fleet, err := clientFromCLI(c)
@@ -185,6 +235,24 @@ func getLabelsCommand() cli.Command {
 					return errors.Wrap(err, "could not list labels")
 				}
 
+				if c.Bool(yamlFlagName) {
+					for _, label := range labels {
+						spec := specGeneric{
+							Kind:    "label",
+							Version: kolide.ApiVersion,
+							Spec:    label,
+						}
+
+						b, err := yaml.Marshal(spec)
+						if err != nil {
+							return err
+						}
+
+						fmt.Printf("---\n%s", string(b))
+					}
+					return nil
+				}
+
 				if len(labels) == 0 {
 					fmt.Println("no labels found")
 					return nil