Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Added moving spinner to context check, since it just kinda sat there #1

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Added moving spinner to context check, since it just kinda sat there #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 36 additions & 9 deletions main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,19 +6,18 @@ import (
"fmt"
"os"
"path/filepath"
"sync"
"time"

"github.com/alexellis/kubetrim/pkg"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
_ "k8s.io/client-go/plugin/pkg/client/auth/azure"
_ "k8s.io/client-go/plugin/pkg/client/auth/exec"

_ "k8s.io/client-go/plugin/pkg/client/auth/gcp" // Import for GCP auth
_ "k8s.io/client-go/plugin/pkg/client/auth/oidc" // Import for OIDC (often used with EKS)
"k8s.io/client-go/tools/clientcmd"
clientcmdapi "k8s.io/client-go/tools/clientcmd/api"

"k8s.io/client-go/util/homedir"
)

Expand All @@ -28,11 +27,8 @@ var (
)

func main() {

// set usage:

// Set usage
flag.Usage = func() {

fmt.Printf("kubetrim (%s %s) Copyright Alex Ellis (c) 2024\n\n", pkg.Version, pkg.GitCommit)
fmt.Print("Sponsor Alex on GitHub: https://github.com/sponsors/alexellis\n\n")
fmt.Fprintf(flag.CommandLine.Output(), "Usage of %s:\n", os.Args[0])
Expand Down Expand Up @@ -63,6 +59,16 @@ func main() {

fmt.Printf("kubetrim (%s %s) by Alex Ellis \n\nLoaded: %s. Checking..\n", pkg.Version, pkg.GitCommit, kubeconfig)

// Start the spinner in a goroutine
var wg sync.WaitGroup
stopSpinner := make(chan struct{})
wg.Add(1)

go func() {
defer wg.Done()
spinner(stopSpinner)
}()

st := time.Now()
// List of contexts to be deleted
var contextsToDelete []string
Expand Down Expand Up @@ -94,17 +100,21 @@ func main() {
fmt.Printf("❌ - (%v)\n", err)
contextsToDelete = append(contextsToDelete, contextName)
} else {
fmt.Println("/n")
fmt.Println("✅")
}
}

// Stop the spinner
close(stopSpinner)
wg.Wait()

// Delete the contexts that are not working
for _, contextName := range contextsToDelete {
deleteContextAndCluster(config, contextName)
}

if writeFile {

if len(contextsToDelete) == len(config.Contexts) && !force {
fmt.Println("No contexts are working, the Internet may be down, use --force to delete all contexts anyway.")
os.Exit(1)
Expand All @@ -116,7 +126,25 @@ func main() {
os.Exit(1)
}
}
fmt.Printf("Updated: %s (in %s).\n", kubeconfig, time.Since(st).Round(time.Millisecond))
fmt.Printf("\n Updated: %s (in %s).\n", kubeconfig, time.Since(st).Round(time.Millisecond))
}
}

// spinner runs a visual timer in the terminal
func spinner(stop <-chan struct{}) {
chars := []rune{'|', '/', '-', '\\'}
for {
for _, char := range chars {
select {
case <-stop:
fmt.Print("\r") // Clear spinner line
return
default:
fmt.Printf("")
fmt.Printf("\r%c Checking ... ", char)
time.Sleep(100 * time.Millisecond)
}
}
}
}

Expand All @@ -131,7 +159,6 @@ func checkCluster(clientset *kubernetes.Clientset) error {

// deleteContextAndCluster removes the context and its associated cluster from the config
func deleteContextAndCluster(config *clientcmdapi.Config, contextName string) {

// Get the cluster name associated with the context
clusterName := config.Contexts[contextName].Cluster

Expand Down