feat: refactor integration to use Failure object

Signed-off-by: Matthis Holleville <matthish29@gmail.com>

.github/workflows/release.yaml

@@ -25,7 +25,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@8f4b7f84864484a7bf31766abe9204da3cbe65b3 # v3
 
-      - uses: google-github-actions/release-please-action@f7edb9e61420a022c0f1123edaf342d7e127df92 # v3
+      - uses: google-github-actions/release-please-action@c078ea33917ab8cfa5300e48f4b7e6b16606aede # v3
         id: release
         with:
           command: manifest

cmd/analyze/analyze.go

@@ -67,7 +67,15 @@ var AnalyzeCmd = &cobra.Command{
 
 		ctx := context.Background()
 		// Get kubernetes client from viper
-		client := viper.Get("kubernetesClient").(*kubernetes.Client)
+
+		kubecontext := viper.GetString("kubecontext")
+		kubeconfig := viper.GetString("kubeconfig")
+		client, err := kubernetes.NewClient(kubecontext, kubeconfig)
+		if err != nil {
+			color.Red("Error initialising kubernetes client: %v", err)
+			os.Exit(1)
+		}
+
 		// AnalysisResult configuration
 		config := &analysis.Analysis{
 			Namespace: namespace,

cmd/filters/add.go

@@ -18,10 +18,9 @@ var addCmd = &cobra.Command{
 	Args:  cobra.ExactArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
 		inputFilters := strings.Split(args[0], ",")
-		coreFilters, additionalFilters := analyzer.ListFilters()
-
-		availableFilters := append(coreFilters, additionalFilters...)
+		coreFilters, additionalFilters, integrationFilters := analyzer.ListFilters()
 
+		availableFilters := append(append(coreFilters, additionalFilters...), integrationFilters...)
 		// Verify filter exist
 		invalidFilters := []string{}
 		for _, f := range inputFilters {

cmd/filters/list.go

@@ -16,23 +16,35 @@ var listCmd = &cobra.Command{
 	Long:  `The list command displays a list of available filters that can be used to analyze Kubernetes resources.`,
 	Run: func(cmd *cobra.Command, args []string) {
 		activeFilters := viper.GetStringSlice("active_filters")
-		coreFilters, additionalFilters := analyzer.ListFilters()
+		coreFilters, additionalFilters, integrationFilters := analyzer.ListFilters()
+
+		availableFilters := append(append(coreFilters, additionalFilters...), integrationFilters...)
 
-		availableFilters := append(coreFilters, additionalFilters...)
 		if len(activeFilters) == 0 {
 			activeFilters = coreFilters
 		}
-
 		inactiveFilters := util.SliceDiff(availableFilters, activeFilters)
 		fmt.Printf(color.YellowString("Active: \n"))
 		for _, filter := range activeFilters {
-			fmt.Printf("> %s\n", color.GreenString(filter))
+
+			// if the filter is an integration, mark this differently
+			if util.SliceContainsString(integrationFilters, filter) {
+				fmt.Printf("> %s\n", color.BlueString("%s (integration)", filter))
+			} else {
+				fmt.Printf("> %s\n", color.GreenString(filter))
+			}
 		}
+
 		// display inactive filters
 		if len(inactiveFilters) != 0 {
 			fmt.Printf(color.YellowString("Unused: \n"))
 			for _, filter := range inactiveFilters {
-				fmt.Printf("> %s\n", color.RedString(filter))
+				// if the filter is an integration, mark this differently
+				if util.SliceContainsString(integrationFilters, filter) {
+					fmt.Printf("> %s\n", color.BlueString("%s (integration)", filter))
+				} else {
+					fmt.Printf("> %s\n", color.RedString(filter))
+				}
 			}
 		}
 

cmd/filters/remove.go

@@ -21,7 +21,7 @@ var removeCmd = &cobra.Command{
 
 		// Get defined active_filters
 		activeFilters := viper.GetStringSlice("active_filters")
-		coreFilters, _ := analyzer.ListFilters()
+		coreFilters, _, _ := analyzer.ListFilters()
 
 		if len(activeFilters) == 0 {
 			activeFilters = coreFilters

cmd/integration/activate.go

@@ -0,0 +1,33 @@
+package integration
+
+import (
+	"github.com/fatih/color"
+	"github.com/k8sgpt-ai/k8sgpt/pkg/integration"
+	"github.com/spf13/cobra"
+)
+
+// activateCmd represents the activate command
+var activateCmd = &cobra.Command{
+	Use:   "activate [integration]",
+	Short: "Activate an integration",
+	Long:  ``,
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		integrationName := args[0]
+
+		integration := integration.NewIntegration()
+		// Check if the integation exists
+		err := integration.Activate(integrationName, namespace)
+		if err != nil {
+			color.Red("Error: %v", err)
+			return
+		}
+
+		color.Green("Activated integration %s", integrationName)
+	},
+}
+
+func init() {
+	IntegrationCmd.AddCommand(activateCmd)
+
+}

cmd/integration/deactivate.go

@@ -0,0 +1,35 @@
+/*
+Copyright © 2023 NAME HERE <EMAIL ADDRESS>
+*/
+package integration
+
+import (
+	"github.com/fatih/color"
+	"github.com/k8sgpt-ai/k8sgpt/pkg/integration"
+	"github.com/spf13/cobra"
+)
+
+// deactivateCmd represents the deactivate command
+var deactivateCmd = &cobra.Command{
+	Use:   "deactivate [integration]",
+	Short: "Deactivate an integration",
+	Args:  cobra.ExactArgs(1),
+	Long:  `For example e.g. k8sgpt integration deactivate trivy`,
+	Run: func(cmd *cobra.Command, args []string) {
+		integrationName := args[0]
+
+		integration := integration.NewIntegration()
+
+		if err := integration.Deactivate(integrationName, namespace); err != nil {
+			color.Red("Error: %v", err)
+			return
+		}
+
+		color.Green("Deactivated integration %s", integrationName)
+
+	},
+}
+
+func init() {
+	IntegrationCmd.AddCommand(deactivateCmd)
+}

cmd/integration/integration.go

@@ -0,0 +1,28 @@
+package integration
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var (
+	namespace string
+)
+
+// IntegrationCmd represents the integrate command
+var IntegrationCmd = &cobra.Command{
+	Use:     "integration",
+	Aliases: []string{"integrations"},
+	Short:   "Intergrate another tool into K8sGPT",
+	Long: `Intergrate another tool into K8sGPT. For example:
+	
+	k8sgpt integration activate trivy
+	
+	This would allow you to deploy trivy into your cluster and use a K8sGPT analyzer to parse trivy results.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		cmd.Help()
+	},
+}
+
+func init() {
+	IntegrationCmd.PersistentFlags().StringVarP(&namespace, "namespace", "n", "default", "The namespace to use for the integration")
+}

cmd/integration/list.go

@@ -0,0 +1,50 @@
+package integration
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/fatih/color"
+	"github.com/k8sgpt-ai/k8sgpt/pkg/integration"
+	"github.com/spf13/cobra"
+)
+
+// listCmd represents the list command
+var listCmd = &cobra.Command{
+	Use:   "list",
+	Short: "Lists built-in integrations",
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+		integrationProvider := integration.NewIntegration()
+		integrations := integrationProvider.List()
+
+		fmt.Println(color.YellowString("Active:"))
+		for _, i := range integrations {
+			b, err := integrationProvider.IsActivate(i)
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+			if b {
+				fmt.Printf("> %s\n", color.GreenString(i))
+			}
+		}
+
+		fmt.Println(color.YellowString("Unused: "))
+		for _, i := range integrations {
+			b, err := integrationProvider.IsActivate(i)
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+			if !b {
+				fmt.Printf("> %s\n", color.GreenString(i))
+			}
+		}
+	},
+}
+
+func init() {
+	IntegrationCmd.AddCommand(listCmd)
+
+}

cmd/root.go

@@ -4,16 +4,14 @@ import (
 	"os"
 	"path/filepath"
 
-	"github.com/k8sgpt-ai/k8sgpt/cmd/filters"
-	"github.com/k8sgpt-ai/k8sgpt/cmd/generate"
-	"k8s.io/client-go/util/homedir"
-
-	"github.com/fatih/color"
 	"github.com/k8sgpt-ai/k8sgpt/cmd/analyze"
 	"github.com/k8sgpt-ai/k8sgpt/cmd/auth"
-	"github.com/k8sgpt-ai/k8sgpt/pkg/kubernetes"
+	"github.com/k8sgpt-ai/k8sgpt/cmd/filters"
+	"github.com/k8sgpt-ai/k8sgpt/cmd/generate"
+	"github.com/k8sgpt-ai/k8sgpt/cmd/integration"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+	"k8s.io/client-go/util/homedir"
 )
 
 var (
@@ -54,13 +52,10 @@ func init() {
 	rootCmd.AddCommand(analyze.AnalyzeCmd)
 	rootCmd.AddCommand(filters.FiltersCmd)
 	rootCmd.AddCommand(generate.GenerateCmd)
+	rootCmd.AddCommand(integration.IntegrationCmd)
 	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.k8sgpt.yaml)")
 	rootCmd.PersistentFlags().StringVar(&kubecontext, "kubecontext", "", "Kubernetes context to use. Only required if out-of-cluster.")
 	rootCmd.PersistentFlags().StringVar(&kubeconfig, "kubeconfig", kubeconfigPath, "Path to a kubeconfig. Only required if out-of-cluster.")
-	// Cobra also supports local flags, which will only run
-	// when this action is called directly.
-	// rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
-
 }
 
 // initConfig reads in config file and ENV variables if set.
@@ -81,14 +76,8 @@ func initConfig() {
 		viper.SafeWriteConfig()
 	}
 
-	//Initialise the kubeconfig
-	kubernetesClient, err := kubernetes.NewClient(kubecontext, kubeconfig)
-	if err != nil {
-		color.Red("Error initialising kubernetes client: %v", err)
-		os.Exit(1)
-	}
-
-	viper.Set("kubernetesClient", kubernetesClient)
+	viper.Set("kubecontext", kubecontext)
+	viper.Set("kubeconfig", kubeconfig)
 
 	viper.AutomaticEnv() // read in environment variables that match