Support finding unused pvcs

cmd/kor/pvc.go

@@ -0,0 +1,23 @@
+package kor
+
+import (
+	"github.com/spf13/cobra"
+	"github.com/yonahd/kor/pkg/kor"
+)
+
+var pvcCmd = &cobra.Command{
+	Use:   "pvc",
+	Short: "Gets unused pvcs",
+	Args:  cobra.NoArgs,
+	Run: func(cmd *cobra.Command, args []string) {
+		if outputFormat == "json" {
+			kor.GetUnusedPvcsJson(namespace, kubeconfig)
+		} else {
+			kor.GetUnusedPvcs(namespace, kubeconfig)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(pvcCmd)
+}

pkg/kor/all.go

@@ -90,6 +90,15 @@ func getUnusedHpas(kubeClient *kubernetes.Clientset, namespace string) ResourceD
 	return namespaceHpaDiff
 }
 
+func getUnusedPvcs(kubeClient *kubernetes.Clientset, namespace string) ResourceDiff {
+	pvcDiff, err := processNamespacePvcs(kubeClient, namespace)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Failed to get %s namespace %s: %v\n", "pvcs", namespace, err)
+	}
+	namespacePvcDiff := ResourceDiff{"Pvc", pvcDiff}
+	return namespacePvcDiff
+}
+
 func GetUnusedAll(namespace string, kubeconfig string) {
 	var kubeClient *kubernetes.Clientset
 	var namespaces []string
@@ -115,6 +124,8 @@ func GetUnusedAll(namespace string, kubeconfig string) {
 		allDiffs = append(allDiffs, namespaceRoleDiff)
 		namespaceHpaDiff := getUnusedHpas(kubeClient, namespace)
 		allDiffs = append(allDiffs, namespaceHpaDiff)
+		namespacePvcDiff := getUnusedPvcs(kubeClient, namespace)
+		allDiffs = append(allDiffs, namespacePvcDiff)
 		output := FormatOutputAll(namespace, allDiffs)
 		fmt.Println(output)
 		fmt.Println()
@@ -156,6 +167,12 @@ func GetUnusedAllJSON(namespace string, kubeconfig string) (string, error) {
 		namespaceRoleDiff := getUnusedRoles(kubeClient, namespace)
 		allDiffs = append(allDiffs, namespaceRoleDiff)
 
+		namespaceHpaDiff := getUnusedHpas(kubeClient, namespace)
+		allDiffs = append(allDiffs, namespaceHpaDiff)
+
+		namespacePvcDiff := getUnusedPvcs(kubeClient, namespace)
+		allDiffs = append(allDiffs, namespacePvcDiff)
+
 		// Store the unused resources for each resource type in the JSON response
 		resourceMap := make(map[string][]string)
 		for _, diff := range allDiffs {

pkg/kor/pvc.go

@@ -0,0 +1,119 @@
+package kor
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	_ "k8s.io/client-go/plugin/pkg/client/auth/oidc"
+)
+
+func retreiveUsedPvcs(kubeClient *kubernetes.Clientset, namespace string) ([]string, error) {
+	pods, err := kubeClient.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		fmt.Printf("Failed to list Pods: %v\n", err)
+		os.Exit(1)
+	}
+	var usedPvcs []string
+	// Iterate through each Pod and check for PVC usage
+	for _, pod := range pods.Items {
+		for _, volume := range pod.Spec.Volumes {
+			if volume.PersistentVolumeClaim != nil {
+				usedPvcs = append(usedPvcs, volume.PersistentVolumeClaim.ClaimName)
+			}
+		}
+	}
+	return usedPvcs, err
+}
+
+func calculatePvcDifference(usedConfigMaps []string, configMapNames []string) []string {
+	difference := []string{}
+	for _, name := range configMapNames {
+		found := false
+		for _, usedName := range usedConfigMaps {
+			if name == usedName {
+				found = true
+				break
+			}
+		}
+		if !found {
+			difference = append(difference, name)
+		}
+	}
+	return difference
+}
+func processNamespacePvcs(kubeClient *kubernetes.Clientset, namespace string) ([]string, error) {
+	pvcs, err := kubeClient.CoreV1().PersistentVolumeClaims(namespace).List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		return nil, err
+	}
+	pvcNames := make([]string, 0, len(pvcs.Items))
+	for _, pvc := range pvcs.Items {
+		pvcNames = append(pvcNames, pvc.Name)
+	}
+
+	usedPvcs, err := retreiveUsedPvcs(kubeClient, namespace)
+	if err != nil {
+		return nil, err
+	}
+
+	diff := calculatePvcDifference(usedPvcs, pvcNames)
+	return diff, nil
+}
+
+func GetUnusedPvcs(namespace string, kubeconfig string) {
+	var kubeClient *kubernetes.Clientset
+	var namespaces []string
+
+	kubeClient = GetKubeClient(kubeconfig)
+
+	namespaces = SetNamespaceList(namespace, kubeClient)
+
+	for _, namespace := range namespaces {
+		diff, err := processNamespacePvcs(kubeClient, namespace)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Failed to process namespace %s: %v\n", namespace, err)
+			continue
+		}
+		output := FormatOutput(namespace, diff, "Pvcs")
+		fmt.Println(output)
+		fmt.Println()
+	}
+
+}
+
+func GetUnusedPvcsJson(namespace string, kubeconfig string) (string, error) {
+	var kubeClient *kubernetes.Clientset
+	var namespaces []string
+
+	kubeClient = GetKubeClient(kubeconfig)
+
+	namespaces = SetNamespaceList(namespace, kubeClient)
+	response := make(map[string]map[string][]string)
+
+	for _, namespace := range namespaces {
+		diff, err := processNamespaceHpas(kubeClient, namespace)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Failed to process namespace %s: %v\n", namespace, err)
+			continue
+		}
+		if len(diff) > 0 {
+			if response[namespace] == nil {
+				response[namespace] = make(map[string][]string)
+			}
+			response[namespace]["Pvc"] = diff
+		}
+	}
+
+	jsonResponse, err := json.MarshalIndent(response, "", "  ")
+	if err != nil {
+		return "", err
+	}
+
+	log.Println(string(jsonResponse))
+	return string(jsonResponse), nil
+}