agent.go

package cmd

import (
	"fmt"
	"os"

	"github.com/spf13/cobra"

	"github.com/jetstack/preflight/pkg/agent"
	"github.com/jetstack/preflight/pkg/permissions"
)

var agentCmd = &cobra.Command{
	Use:   "agent",
	Short: "start the preflight agent",
	Long: `The agent will periodically gather data for the configured data
	gatherers and send it to a remote backend for evaluation`,
	RunE: agent.Run,
}

var agentInfoCmd = &cobra.Command{
	Use:   "info",
	Short: "print several internal parameters of the agent",
	Long:  `Print several internal parameters of the agent, as the built-in OAuth2 client ID.`,
	Run: func(cmd *cobra.Command, args []string) {
		printVersion(true)
		fmt.Println()
		printOAuth2Config()
	},
}

var agentRBACCmd = &cobra.Command{
	Use:   "rbac",
	Short: "print the agent's minimal RBAC manifest",
	Long:  `Print RBAC string by reading GVRs`,
	RunE: func(cmd *cobra.Command, args []string) error {

		b, err := os.ReadFile(agent.Flags.ConfigFilePath)
		if err != nil {
			return fmt.Errorf("Failed to read config file: %s", err)
		}
		cfg, err := agent.ParseConfig(b)
		if err != nil {
			return fmt.Errorf("Failed to parse config file: %s", err)
		}

		err = agent.ValidateDataGatherers(cfg.DataGatherers)
		if err != nil {
			return fmt.Errorf("Failed to validate data gatherers: %s", err)
		}

		out := permissions.GenerateFullManifest(cfg.DataGatherers)
		fmt.Print(out)
		return nil
	},
}

func init() {
	rootCmd.AddCommand(agentCmd)
	agentCmd.AddCommand(agentInfoCmd)
	agentCmd.AddCommand(agentRBACCmd)
	agent.InitAgentCmdFlags(agentCmd, &agent.Flags)
}