Add pods list for all namespace. Close #10. Open #20.

Author: JulienBreux

config.go

@@ -0,0 +1,53 @@
+package main
+
+import (
+	"flag"
+	"os"
+	"path/filepath"
+)
+
+type config struct {
+	homeDir    string
+	kubeConfig *string
+	frequency  int
+	setup      bool
+}
+
+// Check if configuration is initialized
+func (c *config) Initialized() bool {
+	return c != nil && c.setup
+}
+
+var CONFIG config
+
+// Get configuration
+func getConfig() config {
+	if CONFIG.Initialized() {
+		return CONFIG
+	}
+
+	c := config{}
+
+	// Home directory
+	c.homeDir = os.Getenv("USERPROFILE")
+	if h := os.Getenv("HOME"); h != "" {
+		c.homeDir = h
+	}
+
+	// Kubernetes configuration
+	if h := c.homeDir; h != "" {
+		c.kubeConfig = flag.String("kubeconfig", filepath.Join(h, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
+	} else {
+		c.kubeConfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
+	}
+
+	// Refreshing frequency
+	flag.IntVar(&c.frequency, "frequency", 3, "refreshing frequency in seconds (default: 5)")
+
+	flag.Parse()
+
+	c.setup = true
+	CONFIG = c
+
+	return c
+}

keys.go

@@ -8,16 +8,9 @@ type Key struct {
 	handler  func(*gocui.Gui, *gocui.View) error
 }
 
-// Configure globale keys
-var globalKeys []Key = []Key{
-	Key{"", gocui.KeyCtrlC, actionGlobalQuit},
-	Key{"", gocui.KeyCtrlD, actionGlobalToggleDebug},
-}
-
 // Define UI key bindings
 func uiKey(g *gocui.Gui) error {
-	// Glboal keys
-	for _, key := range globalKeys {
+	for _, key := range keys {
 		if err := g.SetKeybinding(key.viewname, key.key, gocui.ModNone, key.handler); err != nil {
 			return err
 		}

kubernetes.go

@@ -1,10 +1,7 @@
 package main
 
 import (
-	"flag"
 	"fmt"
-	"os"
-	"path/filepath"
 	"strconv"
 	"time"
 
@@ -16,46 +13,33 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+// Get Kubernetes client set
 func getClientSet() *kubernetes.Clientset {
+	c := getConfig()
+
 	// Use the current context in kubeconfig
-	c, err := clientcmd.BuildConfigFromFlags("", *getKubeConfig())
+	cc, err := clientcmd.BuildConfigFromFlags("", *c.kubeConfig)
 	if err != nil {
 		panic(err.Error())
 	}
 
 	// Create the client set
-	cs, err := kubernetes.NewForConfig(c)
+	cs, err := kubernetes.NewForConfig(cc)
 	if err != nil {
 		panic(err.Error())
 	}
 
 	return cs
 }
 
-func getKubeConfig() *string {
-	var kc *string
-	if h := getHomeDir(); h != "" {
-		kc = flag.String("kubeconfig", filepath.Join(h, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
-	} else {
-		kc = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
-	}
-	flag.Parse()
-
-	return kc
-}
-
-func getHomeDir() string {
-	if h := os.Getenv("HOME"); h != "" {
-		return h
-	}
-	return os.Getenv("USERPROFILE") // windows
-}
-
+// Get pods (use namespace)
 func getPods() (*v1.PodList, error) {
 	cs := getClientSet()
+
 	return cs.CoreV1().Pods(NAMESPACE).List(metav1.ListOptions{})
 }
 
+// Column helper: Restarts
 func columnHelperRestarts(cs []v1.ContainerStatus) string {
 	r := 0
 	for _, c := range cs {
@@ -64,29 +48,32 @@ func columnHelperRestarts(cs []v1.ContainerStatus) string {
 	return strconv.Itoa(r)
 }
 
+// Column helper: Age
 func columnHelperAge(t metav1.Time) string {
-	delta := time.Now().Sub(t.Time)
+	d := time.Now().Sub(t.Time)
 
-	if delta.Hours() > 1 {
-		if delta.Hours() > 24 {
-			d := float64(delta.Hours() / 24)
-			return fmt.Sprintf("%.0fd", d)
+	if d.Hours() > 1 {
+		if d.Hours() > 24 {
+			ds := float64(d.Hours() / 24)
+			return fmt.Sprintf("%.0fd", ds)
 		} else {
-			return fmt.Sprintf("%.0fh", delta.Hours())
+			return fmt.Sprintf("%.0fh", d.Hours())
 		}
-	} else if delta.Minutes() > 1 {
-		return fmt.Sprintf("%.0fm", delta.Minutes())
-	} else if delta.Seconds() > 1 {
-		return fmt.Sprintf("%.0fs", delta.Seconds())
+	} else if d.Minutes() > 1 {
+		return fmt.Sprintf("%.0fm", d.Minutes())
+	} else if d.Seconds() > 1 {
+		return fmt.Sprintf("%.0fs", d.Seconds())
 	}
 
 	return "?"
 }
 
-func columnHelperStatus(status v1.PodStatus) string {
-	return fmt.Sprintf("%s", status.Phase)
+// Column helper: Status
+func columnHelperStatus(s v1.PodStatus) string {
+	return fmt.Sprintf("%s", s.Phase)
 }
 
+// Column helper: Ready
 func columnHelperReady(cs []v1.ContainerStatus) string {
 	cr := 0
 	for _, c := range cs {

main.go

@@ -8,7 +8,13 @@ import (
 	"github.com/jroimartin/gocui"
 )
 
-var NAMESPACE string = "default"
+var NAMESPACE string = ""
+
+// Configure globale keys
+var keys []Key = []Key{
+	Key{"", gocui.KeyCtrlC, actionGlobalQuit},
+	Key{"", gocui.KeyCtrlD, actionGlobalToggleDebug},
+}
 
 func main() {
 	g, err := gocui.NewGui(gocui.OutputNormal)
@@ -42,6 +48,7 @@ func uiLayout(g *gocui.Gui) error {
 	return nil
 }
 
+// Useful to debug application (display with CTRL+D)
 func debug(g *gocui.Gui, output interface{}) {
 	v, err := g.View("debug")
 	if err == nil {

version.go

@@ -10,18 +10,22 @@ const APP = "Pody"
 const AUTHOR = "@JulienBreux"
 const VERSION = "0.1.0"
 
+// Get full banner of version
 func versionFull() string {
 	return fmt.Sprintf("%s %s - By %s", APP, VERSION, AUTHOR)
 }
 
+// Get only banner (used in title bar view)
 func versionBanner() string {
 	return fmt.Sprintf("  %s %s", APP, VERSION)
 }
 
+// Get only author (used in title bar view)
 func versionAuthor() string {
 	return fmt.Sprintf("By %s  ", AUTHOR)
 }
 
+// Prepare version title (used in title bar view)
 func versionTitle(width int) string {
 	return versionBanner() + pad.Left(versionAuthor(), width-len(versionBanner()), " ")
 }

views.go

@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/jroimartin/gocui"
 	"github.com/willf/pad"
@@ -72,48 +73,79 @@ func viewPods(g *gocui.Gui, lMaxX int, lMaxY int) error {
 		// Set as current view
 		g.SetCurrentView(v.Name())
 
-		// Content: Add column
-		//viewPodsAddLine(v, lMaxX, "NAME", "CPU", "MEMORY", "READY", "STATUS", "RESTARTS", "AGE") // TODO CPU + Memory
-		viewPodsAddLine(v, lMaxX, "NAME", "READY", "STATUS", "RESTARTS", "AGE")
-		fmt.Fprintln(v, strings.Repeat("─", lMaxX))
+		// Content
+		go viewPodsActualize(g, lMaxX)
+		go viewPodsAutoRefresh(g, lMaxX)
+	}
+
+	return nil
+}
+
+// Auto refresh view pods
+func viewPodsAutoRefresh(g *gocui.Gui, lMaxX int) {
+	c := getConfig()
+	t := time.NewTicker(time.Duration(c.frequency) * time.Second)
+	for {
+		select {
+		case <-t.C:
+			debug(g, fmt.Sprintf("View pods: Refreshing (%ds)", c.frequency))
+			go viewPodsActualize(g, lMaxX)
+		}
+	}
+}
+
+// Actualize pods view
+func viewPodsActualize(g *gocui.Gui, lMaxX int) {
+	g.Execute(func(g *gocui.Gui) error {
+		debug(g, "View pods: Actualize")
+		v, err := g.View("pods")
+		if err != nil {
+			return err
+		}
 
-		// Content: Add lines
-		// TODO Use goroutine
 		pods, err := getPods()
 		if err != nil {
 			panic(err.Error())
 		}
+
+		v.Clear()
+
+		// Content: Add column
+		//viewPodsAddLine(v, lMaxX, "NAME", "CPU", "MEMORY", "READY", "STATUS", "RESTARTS", "AGE") // TODO CPU + Memory #20
+		viewPodsAddLine(v, lMaxX, "NAME", "READY", "STATUS", "RESTARTS", "AGE")
+		fmt.Fprintln(v, strings.Repeat("─", lMaxX))
+
 		if len(pods.Items) > 0 {
-			debug(g, fmt.Sprintf("There are %d pods in the cluster", len(pods.Items)))
+			debug(g, fmt.Sprintf("View pods: %d pods found", len(pods.Items)))
 			for _, pod := range pods.Items {
 				n := pod.GetName()
-				//c := "?" // TODO CPU + Memory
-				//m := "?" // TODO CPU + Memory
+				//c := "?" // TODO CPU + Memory #20
+				//m := "?" // TODO CPU + Memory #20
 				s := columnHelperStatus(pod.Status)
 				r := columnHelperRestarts(pod.Status.ContainerStatuses)
 				a := columnHelperAge(pod.CreationTimestamp)
 				cr := columnHelperReady(pod.Status.ContainerStatuses)
 				viewPodsAddLine(v, lMaxX, n, cr, s, r, a)
-				//viewPodsAddLine(v, lMaxX, n, c, m, cr, s, r, a) // TODO CPU + Memory
+				//viewPodsAddLine(v, lMaxX, n, c, m, cr, s, r, a) // TODO CPU + Memory #20
 			}
 		} else {
-			debug(g, "Pods not found.")
+			debug(g, "View pods: Pods not found")
 		}
-	}
 
-	return nil
+		return nil
+	})
 }
 
 // Add line to view pods
-//func viewPodsAddLine(v *gocui.View, maxX int, name, cpu, memory, ready, status, restarts, age string) { // TODO CPU + Memory
+//func viewPodsAddLine(v *gocui.View, maxX int, name, cpu, memory, ready, status, restarts, age string) { // TODO CPU + Memory #20
 func viewPodsAddLine(v *gocui.View, maxX int, name, ready, status, restarts, age string) {
-	wN := maxX - 34 // 54 // TODO CPU + Memory
+	wN := maxX - 34 // 54 // TODO CPU + Memory #20
 	if wN < 45 {
 		wN = 45
 	}
 	line := pad.Right(name, wN, " ") +
-		//pad.Right(cpu, 10, " ") + // TODO CPU + Memory
-		//pad.Right(memory, 10, " ") + // TODO CPU + Memory
+		//pad.Right(cpu, 10, " ") + // TODO CPU + Memory #20
+		//pad.Right(memory, 10, " ") + // TODO CPU + Memory #20
 		pad.Right(ready, 10, " ") +
 		pad.Right(status, 10, " ") +
 		pad.Right(restarts, 10, " ") +