makeshift/cmd/list.go

package cmd

import (
	"encoding/json"
	"fmt"
	"net/http"
	"os"

	makeshift "git.towk2.me/towk/makeshift/pkg"
	"git.towk2.me/towk/makeshift/pkg/client"
	"github.com/rs/zerolog/log"
	"github.com/spf13/cobra"
)

var listCmd = &cobra.Command{
	Use: "list",
	Example: `
  # list files in a remote data directory
  configurator list --path test
  configurator list --host http://localhost:5050 --path test

  # list files using 'curl'
  curl http://localhost:5050/list/test
`,
	Args:  cobra.NoArgs,
	Short: "List all files in a remote data directory",
	PreRun: func(cmd *cobra.Command, args []string) {
		setenv(cmd, "host", "MAKESHIFT_HOST")
		setenv(cmd, "path", "MAKESHIFT_PATH")
	},
	Run: func(cmd *cobra.Command, args []string) {
		var (
			host, _ = cmd.Flags().GetString("host")
			path, _ = cmd.Flags().GetString("path")

			c      = client.New(host)
			body   []byte
			output []string
			err    error
		)

		log.Debug().
			Str("host", host).
			Str("path", path).
			Send()

		// make request to /list endpoint
		_, body, err = c.MakeRequest(client.HTTPEnvelope{
			Path:   fmt.Sprintf("/list/%s", path),
			Method: http.MethodGet,
		})
		if err != nil {
			log.Error().Err(err).
				Str("host", host).
				Str("path", path).
				Msg("failed to make request")
			os.Exit(1)
		}

		err = json.Unmarshal(body, &output)
		if err != nil {
			log.Error().Err(err).Msg("failed to unmarshal response body")
			os.Exit(1)
		}

		// show the list of files and directories
		log.Info().Strs("output", output).Send()
	},
}

var listPluginsCmd = &cobra.Command{
	Use: "plugins",
	Example: `
  # show all plugins
  makeshift list plugins

  # show details for specific plugins
  makeshift list plugins smd jinja2
`,
	Short: "Show plugins information",
	Run: func(cmd *cobra.Command, args []string) {
		var (
			host, _ = cmd.Flags().GetString("host")
			path, _ = cmd.Flags().GetString("path")

			c       = client.New(host)
			res     *http.Response
			plugins []string
			body    []byte
			err     error
		)

		if len(args) == 0 {
			// make request to /list endpoint
			res, body, err = c.MakeRequest(client.HTTPEnvelope{
				Path:   "/plugins",
				Method: http.MethodGet,
			})
			if err != nil {
				log.Error().Err(err).
					Str("host", host).
					Str("path", path).
					Msg("failed to make request")
				os.Exit(1)
			}
			if res.StatusCode != http.StatusOK {
				log.Error().
					Any("status", map[string]any{
						"code":    res.StatusCode,
						"message": res.Status,
					}).
					Str("host", host).
					Str("path", path).
					Msg("response returned bad status")
				os.Exit(1)
			}
			err = json.Unmarshal(body, &plugins)
			if err != nil {
				log.Error().Err(err).
					Msg("failed to unmarshal plugins")
				return
			}
		} else {
			for _, pluginName := range args {
				// make request to /list endpoint
				res, body, err = c.MakeRequest(client.HTTPEnvelope{
					Path:   fmt.Sprintf("/plugins/%s/info", pluginName),
					Method: http.MethodGet,
				})
				if err != nil {
					log.Error().Err(err).
						Str("host", host).
						Str("path", path).
						Msg("failed to make request")
					os.Exit(1)
				}
				if res.StatusCode != http.StatusOK {
					log.Error().
						Any("status", map[string]any{
							"code":    res.StatusCode,
							"message": res.Status,
						}).
						Str("host", host).
						Str("path", path).
						Msg("response returned bad status")
					os.Exit(1)
				}

				plugins = append(plugins, string(body))
			}
		}
		log.Info().Strs("plugins", plugins).Send()
	},
}

var listProfilesCmd = &cobra.Command{
	Use:   "profiles",
	Short: "Show all available profiles",
	Run: func(cmd *cobra.Command, args []string) {
		var (
			host, _ = cmd.Flags().GetString("host")
			path, _ = cmd.Flags().GetString("path")

			c        = client.New(host)
			res      *http.Response
			profiles []makeshift.Profile
			body     []byte
			err      error
		)

		if len(args) == 0 {
			// make request to /list endpoint
			res, body, err = c.MakeRequest(client.HTTPEnvelope{
				Path:   "/profiles",
				Method: http.MethodGet,
			})
			if err != nil {
				log.Error().Err(err).
					Str("host", host).
					Str("path", path).
					Msg("failed to make request")
				os.Exit(1)
			}
			if res.StatusCode != http.StatusOK {
				log.Error().
					Any("status", map[string]any{
						"code":    res.StatusCode,
						"message": res.Status,
					}).
					Str("host", host).
					Str("path", path).
					Msg("response returned bad status")
				os.Exit(1)
			}

			err = json.Unmarshal(body, &profiles)
			if err != nil {
				log.Error().Err(err).
					Msg("failed to unmarshal plugins")
				return
			}
		} else {
			for _, profileID := range args {
				// make request to /list endpoint
				res, body, err = c.MakeRequest(client.HTTPEnvelope{
					Path:   fmt.Sprintf("/profiles/%s", profileID),
					Method: http.MethodGet,
				})
				if err != nil {
					log.Error().Err(err).
						Str("host", host).
						Str("path", path).
						Msg("failed to make request")
					os.Exit(1)
				}
				if res.StatusCode != http.StatusOK {
					log.Error().
						Any("status", map[string]any{
							"code":    res.StatusCode,
							"message": res.Status,
						}).
						Str("host", host).
						Str("path", path).
						Msg("response returned bad status")
					os.Exit(1)
				}
				var profile makeshift.Profile
				err = json.Unmarshal(body, &profile)
				if err != nil {
					log.Error().Err(err).
						Msg("failed to unmarshal plugin")
					continue
				}
				profiles = append(profiles, profile)
			}
		}
		log.Info().Any("plugins", profiles).Send()
	},
}

func init() {
	listCmd.PersistentFlags().String("host", "http://localhost:5050", "Set the configurator remote host (can be set with MAKESHIFT_HOST)")
	listCmd.Flags().StringP("path", "p", ".", "Set the path to list files (can be set with MAKESHIFT_PATH)")

	listCmd.AddCommand(listPluginsCmd, listProfilesCmd)
	rootCmd.AddCommand(listCmd)
}