github/k8sgpt
1
0
Fork 0
mirror of https://github.com/k8sgpt-ai/k8sgpt.git synced 2025-04-28 03:20:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #17 from k8sgpt-ai/chore/tidying

Browse Source 
chore: tidying up packages
This commit is contained in:
Alex Jones 2023-03-22 09:39:21 +00:00 committed by GitHub
commit d08831278e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/find/problems.go
View File

@ -8,9 +8,9 @@ import (
"os"
"github.com/fatih/color"
"github.com/k8sgpt-ai/k8sgpt/pkg/ai"
"github.com/k8sgpt-ai/k8sgpt/pkg/analyzer"
"github.com/k8sgpt-ai/k8sgpt/pkg/client"
"github.com/k8sgpt-ai/k8sgpt/pkg/openai"
"github.com/k8sgpt-ai/k8sgpt/pkg/kubernetes"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
@ -26,7 +26,7 @@ var problemsCmd = &cobra.Command{
Run: func(cmd *cobra.Command, args []string) {
// Initialise the openAI client
openAIClient, err := openai.NewClient()
openAIClient, err := ai.NewClient()
if err != nil {
color.Red("Error: %v", err)
os.Exit(1)
@ -34,9 +34,12 @@ var problemsCmd = &cobra.Command{
ctx := context.Background()
// Get kubernetes client from viper
client := viper.Get("kubernetesClient").(*client.Client)
client := viper.Get("kubernetesClient").(*kubernetes.Client)
analyzer.RunAnalysis(ctx, client, openAIClient, explain)
if err := analyzer.RunAnalysis(ctx, client, openAIClient, explain); err != nil {
color.Red("Error: %v", err)
os.Exit(1)
}
},
}

4
cmd/root.go
View File

@ -9,7 +9,7 @@ import (
"github.com/fatih/color"
"github.com/k8sgpt-ai/k8sgpt/cmd/auth"
"github.com/k8sgpt-ai/k8sgpt/cmd/find"
"github.com/k8sgpt-ai/k8sgpt/pkg/client"
"github.com/k8sgpt-ai/k8sgpt/pkg/kubernetes"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
@ -56,7 +56,7 @@ func init() {
rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
//Initialise the kubeconfig
kubernetesClient, err := client.NewClient(masterURL, kubeconfig)
kubernetesClient, err := kubernetes.NewClient(masterURL, kubeconfig)
if err != nil {
color.Red("Error initialising kubernetes client: %v", err)
}

2
pkg/openai/openai.go → pkg/ai/ai.go
View File

@ -1,4 +1,4 @@
package openai
package ai
import (
"context"

17
pkg/analyzer/analyzer.go
View File

@ -3,24 +3,22 @@ package analyzer
import (
"context"
"fmt"
"os"
"strings"
"time"
"github.com/briandowns/spinner"
"github.com/fatih/color"
"github.com/k8sgpt-ai/k8sgpt/pkg/client"
ai "github.com/k8sgpt-ai/k8sgpt/pkg/openai"
"github.com/k8sgpt-ai/k8sgpt/pkg/ai"
"github.com/k8sgpt-ai/k8sgpt/pkg/kubernetes"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
func RunAnalysis(ctx context.Context, client *client.Client, openAIClient *ai.Client, explain bool) {
func RunAnalysis(ctx context.Context, client *kubernetes.Client, aiClient *ai.Client, explain bool) error {
// search all namespaces for pods that are not running
list, err := client.GetClient().CoreV1().Pods("").List(ctx, metav1.ListOptions{})
if err != nil {
color.Red("Error: %v", err)
os.Exit(1)
return err
}
var brokenPods = map[string][]string{}
@ -59,14 +57,15 @@ func RunAnalysis(ctx context.Context, client *client.Client, openAIClient *ai.Cl
s := spinner.New(spinner.CharSets[35], 100*time.Millisecond) // Build our new spinner
s.Start()
response, err := openAIClient.GetCompletion(ctx, strings.Join(value, " "))
response, err := aiClient.GetCompletion(ctx, strings.Join(value, " "))
s.Stop()
if err != nil {
color.Red("Error: %v", err)
return
return err
}
fmt.Printf("%s\n", color.GreenString(response))
}
}
return nil
}

2
pkg/client/client.go → pkg/kubernetes/kubernetes.go
View File

@ -1,4 +1,4 @@
package client
package kubernetes
import (
"k8s.io/client-go/kubernetes"