feat: add update to auth cmd (#450)

* added update.go + fixed typos

Signed-off-by: Alexandre Steppé <[email protected]>

* updated readme

Signed-off-by: Alexandre Steppé <[email protected]>

* fixed default values

Signed-off-by: Alexandre Steppé <[email protected]>

* fixed typo

Signed-off-by: Alexandre Steppé <[email protected]>

---------

Signed-off-by: Alexandre Steppé <[email protected]>
Co-authored-by: Alex Jones <[email protected]>

README.md

@@ -226,12 +226,19 @@ k8sgpt filters remove [filter(s)]
 <details>
 
 <summary> Additional commands </summary>
+
 _List configured backends_
 
 ```
 k8sgpt auth list
 ```
 
+_Update configured backends_
+
+```
+k8sgpt auth update $MY_BACKEND1,$MY_BACKEND2..
+```
+
 _Remove configured backends_
 
 ```

cmd/auth/auth.go

@@ -50,4 +50,6 @@ func init() {
 	AuthCmd.AddCommand(removeCmd)
 	// add subcommand to set default backend provider
 	AuthCmd.AddCommand(defaultCmd)
+	// add subcommand to update backend provider
+	AuthCmd.AddCommand(updateCmd)
 }

cmd/auth/list.go

@@ -44,7 +44,7 @@ var listCmd = &cobra.Command{
 			fmt.Printf("> %s\n", color.BlueString("openai"))
 		}
 
-		// Get list of all AI Backends and only print htem if they are not in the provider list
+		// Get list of all AI Backends and only print them if they are not in the provider list
 		fmt.Print(color.YellowString("Active: \n"))
 		for _, aiBackend := range ai.Backends {
 			providerExists := false

cmd/auth/remove.go

@@ -46,7 +46,7 @@ var removeCmd = &cobra.Command{
 				if b == provider.Name {
 					foundBackend = true
 					configAI.Providers = append(configAI.Providers[:i], configAI.Providers[i+1:]...)
-					color.Green("%s deleted to the AI backend provider list", b)
+					color.Green("%s deleted from the AI backend provider list", b)
 					break
 				}
 			}
@@ -64,8 +64,3 @@ var removeCmd = &cobra.Command{
 
 	},
 }
-
-func init() {
-
-}
-

cmd/auth/update.go

@@ -0,0 +1,106 @@
+/*
+Copyright 2023 The K8sGPT Authors.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package auth
+
+import (
+	"os"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var updateCmd = &cobra.Command{
+	Use:   "update",
+	Short: "Update a backend provider",
+	Long:  "The command to update an AI backend provider",
+	Args:  cobra.ExactArgs(1),
+	PreRun: func(cmd *cobra.Command, args []string) {
+		backend, _ := cmd.Flags().GetString("backend")
+		if strings.ToLower(backend) == "azureopenai" {
+			_ = cmd.MarkFlagRequired("engine")
+			_ = cmd.MarkFlagRequired("baseurl")
+		}
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+
+		// get ai configuration
+		err := viper.UnmarshalKey("ai", &configAI)
+		if err != nil {
+			color.Red("Error: %v", err)
+			os.Exit(1)
+		}
+
+		inputBackends := strings.Split(args[0], ",")
+
+		if len(inputBackends) == 0 {
+			color.Red("Error: backend must be set.")
+			os.Exit(1)
+		}
+
+		for _, b := range inputBackends {
+			foundBackend := false
+			for i, provider := range configAI.Providers {
+				if b == provider.Name {
+					foundBackend = true
+					if backend != "" {
+						configAI.Providers[i].Name = backend
+						color.Blue("Backend name updated successfully")
+					}
+					if model != "" {
+						configAI.Providers[i].Model = model
+						color.Blue("Model updated successfully")
+					}
+					if password != "" {
+						configAI.Providers[i].Password = password
+						color.Blue("Password updated successfully")
+					}
+					if baseURL != "" {
+						configAI.Providers[i].BaseURL = baseURL
+						color.Blue("Base URL updated successfully")
+					}
+					if engine != "" {
+						configAI.Providers[i].Engine = engine
+					}
+					color.Green("%s updated in the AI backend provider list", b)
+				}
+			}
+			if !foundBackend {
+				color.Red("Error: %s does not exist in configuration file. Please use k8sgpt auth new.", args[0])
+				os.Exit(1)
+			}
+
+		}
+
+		viper.Set("ai", configAI)
+		if err := viper.WriteConfig(); err != nil {
+			color.Red("Error writing config file: %s", err.Error())
+			os.Exit(1)
+		}
+	},
+}
+
+func init() {
+	// update flag for backend
+	updateCmd.Flags().StringVarP(&backend, "backend", "b", "", "Update backend AI provider")
+	// update flag for model
+	updateCmd.Flags().StringVarP(&model, "model", "m", "", "Update backend AI model")
+	// update flag for password
+	updateCmd.Flags().StringVarP(&password, "password", "p", "", "Update backend AI password")
+	// update flag for url
+	updateCmd.Flags().StringVarP(&baseURL, "baseurl", "u", "", "Update URL AI provider, (e.g `http://localhost:8080/v1`)")
+	// update flag for azure open ai engine/deployment name
+	updateCmd.Flags().StringVarP(&engine, "engine", "e", "", "Update Azure AI deployment name")
+}