From 59bb1acf2bfcd7420870c2557eeeae3ab49942e3 Mon Sep 17 00:00:00 2001
From: Ahmed Samir <asnweery@gmail.com>
Date: Fri, 6 Oct 2023 16:08:30 +0300
Subject: [PATCH] feat: added k8s namespaces support

Signed-off-by: Ahmed Samir <asnweery@gmail.com>
---
 providers/k8s/core/namespaces.go | 77 ++++++++++++++++++++++++++++++++
 providers/k8s/k8s.go             |  1 +
 2 files changed, 78 insertions(+)
 create mode 100644 providers/k8s/core/namespaces.go

diff --git a/providers/k8s/core/namespaces.go b/providers/k8s/core/namespaces.go
new file mode 100644
index 000000000..829c8b2f6
--- /dev/null
+++ b/providers/k8s/core/namespaces.go
@@ -0,0 +1,77 @@
+package core
+
+import (
+	"context"
+	"time"
+
+	log "github.com/sirupsen/logrus"
+
+	"github.com/tailwarden/komiser/models"
+	"github.com/tailwarden/komiser/providers"
+	oc "github.com/tailwarden/komiser/providers/k8s/opencost"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+func Namespaces(ctx context.Context, client providers.ProviderClient) ([]models.Resource, error) {
+	resources := make([]models.Resource, 0)
+
+	var config metav1.ListOptions
+
+	opencostEnabled := true
+	namespacesCost, err := oc.GetOpencostInfo(client.K8sClient.OpencostBaseUrl, "namespace")
+	if err != nil {
+		log.Errorf("ERROR: Couldn't get namespaces info from OpenCost: %v", err)
+		log.Warn("Opencost disabled")
+		opencostEnabled = false
+	}
+
+	for {
+		res, err := client.K8sClient.Client.CoreV1().Namespaces().List(ctx, config)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, namespace := range res.Items {
+			tags := make([]models.Tag, 0)
+
+			for key, value := range namespace.Labels {
+				tags = append(tags, models.Tag{
+					Key:   key,
+					Value: value,
+				})
+			}
+
+			cost := 0.0
+			if opencostEnabled {
+				cost = namespacesCost[namespace.Name].TotalCost
+			}
+
+			resources = append(resources, models.Resource{
+				Provider:   "Kubernetes",
+				Account:    client.Name,
+				Service:    "Namespace",
+				ResourceId: string(namespace.UID),
+				Name:       namespace.Name,
+				Region:     namespace.Namespace,
+				Cost:       cost,
+				CreatedAt:  namespace.CreationTimestamp.Time,
+				FetchedAt:  time.Now(),
+				Tags:       tags,
+			})
+		}
+
+		if res.GetContinue() == "" {
+			break
+		}
+
+		config.Continue = res.GetContinue()
+	}
+
+	log.WithFields(log.Fields{
+		"provider":  "Kubernetes",
+		"account":   client.Name,
+		"service":   "Namespace",
+		"resources": len(resources),
+	}).Info("Fetched resources")
+	return resources, nil
+}
diff --git a/providers/k8s/k8s.go b/providers/k8s/k8s.go
index 6d675361d..3b52335b7 100644
--- a/providers/k8s/k8s.go
+++ b/providers/k8s/k8s.go
@@ -21,6 +21,7 @@ func listOfSupportedServices() []providers.FetchDataFunction {
 		core.PersistentVolumeClaims,
 		core.ServiceAccounts,
 		core.Nodes,
+		core.Namespaces,
 	}
 }