bells17/main.go

package main

import (
	"bytes"
	"flag"
	"fmt"
	"html/template"
	"os"
	"path/filepath"

	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
)

func main() {
	// creates the in-cluster config
	config, err := rest.InClusterConfig()
	if err != nil {
		// creates the out-of-cluster config
		var kubeconfig *string
		if home := os.Getenv("HOME"); home != "" {
			kubeconfig = flag.String("kubeconfig", filepath.Join(home, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
		} else {
			kubeconfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
		}
		flag.Parse()

		// use the current context in kubeconfig
		config, err = clientcmd.BuildConfigFromFlags("", *kubeconfig)
		if err != nil {
			panic(err.Error())
		}
	}

	// creates the clientset
	clientset, err := kubernetes.NewForConfig(config)
	if err != nil {
		panic(err.Error())
	}

	// output, err := outputPods(clientset, "docker-desktop", Tmpl)
	output, err := outputAllPodsPerEachNode(clientset, Tmpl2)
	if err != nil {
		panic(err.Error())
	}
	fmt.Printf(output)
}

var Tmpl = `
{{range .Items}}Node Name: {{.Spec.NodeName}}
Pod  Name: {{.ObjectMeta.Name}}
Pod Namespace: {{.ObjectMeta.Namespace}}

{{end}}
`

func outputPods(clientset *kubernetes.Clientset, nodeName string, tmpl string) (string, error) {
	pods, err := clientset.CoreV1().Pods("default").List(metav1.ListOptions{
		FieldSelector: "spec.nodeName=" + nodeName,
	})
	if err != nil {
		return "", err
	}

	t, err := template.New("output").Parse(tmpl)
	if err != nil {
		return "", err
	}

	b := new(bytes.Buffer)
	err = t.Execute(b, pods)
	return b.String(), err
}

var Tmpl2 = `
{{ range $key, $value := . }}
Node Name: {{$key}}

{{ range $k, $pod := $value }}
Pod  Name: {{$pod.ObjectMeta.Name}}
Pod Namespace: {{$pod.ObjectMeta.Namespace}}
{{ end }}

{{end}}
`

func outputAllPodsPerEachNode(clientset *kubernetes.Clientset, tmpl string) (string, error) {
	pods, err := clientset.CoreV1().Pods("default").List(metav1.ListOptions{})
	if err != nil {
		return "", err
	}

	podsPerNode := map[string][]corev1.Pod{}
	for _, pod := range pods.Items {
		name := pod.Spec.NodeName
		if _, ok := podsPerNode[name]; !ok {
			podsPerNode[name] = []corev1.Pod{}
		}
		podsPerNode[name] = append(podsPerNode[name], pod)
	}

	t, err := template.New("output").Parse(tmpl)
	if err != nil {
		return "", err
	}

	b := new(bytes.Buffer)
	err = t.Execute(b, podsPerNode)
	return b.String(), err
}