Ingress controller implemented

Author: rajsinghtech

PROJECT

@@ -17,4 +17,10 @@ resources:
   kind: Dashboard
   path: github.com/rajsinghtech/homer-operator.git/api/v1alpha1
   version: v1alpha1
+- controller: true
+  domain: k8s.io
+  group: networking
+  kind: Ingress
+  path: k8s.io/api/networking/v1
+  version: v1
 version: "3"

cmd/main.go

@@ -129,6 +129,13 @@ func main() {
 		setupLog.Error(err, "unable to create controller", "controller", "Dashboard")
 		os.Exit(1)
 	}
+	if err = (&controller.IngressReconciler{
+		Client: mgr.GetClient(),
+		Scheme: mgr.GetScheme(),
+	}).SetupWithManager(mgr); err != nil {
+		setupLog.Error(err, "unable to create controller", "controller", "Ingress")
+		os.Exit(1)
+	}
 	//+kubebuilder:scaffold:builder
 
 	if err := mgr.AddHealthzCheck("healthz", healthz.Ping); err != nil {

internal/controller/ingress_controller.go

@@ -0,0 +1,109 @@
+/*
+Copyright 2024 RajSingh.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package controller
+
+import (
+	"context"
+	homerv1alpha1 "github.com/rajsinghtech/homer-operator.git/api/v1alpha1"
+	networkingv1 "k8s.io/api/networking/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	homer "github.com/rajsinghtech/homer-operator.git/pkg/homer"
+	corev1 "k8s.io/api/core/v1"
+	"sigs.k8s.io/controller-runtime/pkg/log"
+)
+
+// IngressReconciler reconciles a Ingress object
+type IngressReconciler struct {
+	client.Client
+	Scheme *runtime.Scheme
+}
+
+//+kubebuilder:rbac:groups=networking.k8s.io,resources=ingresses,verbs=get;list;watch;create;update;patch;delete
+//+kubebuilder:rbac:groups=networking.k8s.io,resources=ingresses/status,verbs=get;update;patch
+//+kubebuilder:rbac:groups=networking.k8s.io,resources=ingresses/finalizers,verbs=update
+
+// Reconcile is part of the main kubernetes reconciliation loop which aims to
+// move the current state of the cluster closer to the desired state.
+// TODO(user): Modify the Reconcile function to compare the state specified by
+// the Ingress object against the actual cluster state, and then
+// perform operations to make the cluster state reflect the state specified by
+// the user.
+//
+// For more details, check Reconcile and its Result here:
+// - https://pkg.go.dev/sigs.k8s.io/[email protected]/pkg/reconcile
+func (r *IngressReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
+	log := log.FromContext(ctx)
+	var ingress networkingv1.Ingress
+	if err := r.Get(ctx, req.NamespacedName, &ingress); err != nil {
+		if client.IgnoreNotFound(err) != nil {
+			log.Error(err, "unable to fetch Ingress", "ingress", req.NamespacedName)
+			return ctrl.Result{}, client.IgnoreNotFound(err)
+		}
+	}
+	dashboardList, error := getAllDashboard(ctx, r)
+	if error != nil {
+		log.Error(error, "unable to fetch DashboardList")
+		return ctrl.Result{}, error
+	}
+	for _, dashboard := range dashboardList.Items {
+		// Check if dashboard annotations are a subset of the ingress annotations
+		delete(dashboard.Annotations, "kubectl.kubernetes.io/last-applied-configuration")
+		if isSubset(ingress.Annotations, dashboard.Annotations) {
+			configMap := corev1.ConfigMap{}
+			log.Info("Dashboard annotations are a subset of the ingress annotations", "dashboard", dashboard.Name)
+			if error := r.Get(ctx, client.ObjectKey{Namespace: dashboard.Namespace, Name: dashboard.Name}, &configMap); error != nil {
+				log.Error(error, "unable to fetch ConfigMap", "configmap", dashboard.Name)
+				return ctrl.Result{}, error
+			}
+			homer.UpdateConfigMapIngress(&configMap, ingress)
+			if error := r.Update(ctx, &configMap); error != nil {
+				log.Error(error, "unable to update ConfigMap", "configmap", dashboard.Name)
+				return ctrl.Result{}, error
+			}
+			log.Info("Updated ConfigMap", "configmap", dashboard.Name)
+		}
+	}
+
+	return ctrl.Result{}, nil
+}
+
+// isSubset checks if the first map is a subset of the second map
+func isSubset(map1, map2 map[string]string) bool {
+	for key, value := range map2 {
+		if map1[key] != value {
+			return false
+		}
+	}
+	return true
+}
+
+// SetupWithManager sets up the controller with the Manager.
+func (r *IngressReconciler) SetupWithManager(mgr ctrl.Manager) error {
+	return ctrl.NewControllerManagedBy(mgr).
+		For(&networkingv1.Ingress{}).
+		Complete(r)
+}
+
+func getAllDashboard(ctx context.Context, r *IngressReconciler) (*homerv1alpha1.DashboardList, error) {
+	var dashboardList homerv1alpha1.DashboardList
+	if err := r.List(ctx, &dashboardList); err != nil {
+		return nil, err
+	}
+	return &dashboardList, nil
+}

internal/controller/ingress_controller_test.go

@@ -0,0 +1,32 @@
+/*
+Copyright 2024 RajSingh.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package controller
+
+import (
+	. "github.com/onsi/ginkgo/v2"
+)
+
+var _ = Describe("Ingress Controller", func() {
+	Context("When reconciling a resource", func() {
+
+		It("should successfully reconcile the resource", func() {
+
+			// TODO(user): Add more specific assertions depending on your controller's reconciliation logic.
+			// Example: If you expect a certain status condition after reconciliation, verify it here.
+		})
+	})
+})

internal/controller/suite_test.go

@@ -32,6 +32,8 @@ import (
 	logf "sigs.k8s.io/controller-runtime/pkg/log"
 	"sigs.k8s.io/controller-runtime/pkg/log/zap"
 
+	networkingv1 "k8s.io/api/networking/v1"
+
 	homerv1alpha1 "github.com/rajsinghtech/homer-operator.git/api/v1alpha1"
 	//+kubebuilder:scaffold:imports
 )
@@ -75,6 +77,9 @@ var _ = BeforeSuite(func() {
 	err = homerv1alpha1.AddToScheme(scheme.Scheme)
 	Expect(err).NotTo(HaveOccurred())
 
+	err = networkingv1.AddToScheme(scheme.Scheme)
+	Expect(err).NotTo(HaveOccurred())
+
 	//+kubebuilder:scaffold:scheme
 
 	k8sClient, err = client.New(cfg, client.Options{Scheme: scheme.Scheme})

pkg/homer/config.go

@@ -236,3 +236,52 @@ func UpdateHomerConfig(config *HomerConfig, ingresses networkingv1.IngressList)
 	}
 	return nil
 }
+func UpdateHomerConfigIngress(homerConfig *HomerConfig, ingress networkingv1.Ingress) {
+	service := Service{}
+	item := Item{}
+	service.Name = ingress.ObjectMeta.Namespace
+	item.Name = ingress.ObjectMeta.Name
+	service.Logo = "https://raw.githubusercontent.com/kubernetes/community/master/icons/png/resources/labeled/ns-128.png"
+	if len(ingress.Spec.TLS) > 0 {
+		item.Url = "https://" + ingress.Spec.Rules[0].Host
+	} else {
+		item.Url = "http://" + ingress.Spec.Rules[0].Host
+	}
+	item.Logo = "https://raw.githubusercontent.com/kubernetes/community/master/icons/png/resources/labeled/ing-128.png"
+	item.Subtitle = ingress.Spec.Rules[0].Host
+	for key, value := range ingress.ObjectMeta.Annotations {
+		if strings.HasPrefix(key, "item.homer.rajsingh.info/") {
+			fieldName := strings.TrimPrefix(key, "item.homer.rajsingh.info/")
+			reflect.ValueOf(&item).Elem().FieldByName(fieldName).SetString(value)
+		}
+		if strings.HasPrefix(key, "service.homer.rajsingh.info/") {
+			fieldName := strings.TrimPrefix(key, "service.homer.rajsingh.info/")
+			reflect.ValueOf(&service).Elem().FieldByName(fieldName).SetString(value)
+		}
+	}
+	for sx, s := range homerConfig.Services {
+		if s.Name == service.Name {
+			for ix, i := range s.Items {
+				if i.Name == item.Name {
+					homerConfig.Services[sx].Items[ix] = item
+					return
+				}
+			}
+			homerConfig.Services[sx].Items = append(homerConfig.Services[sx].Items, item)
+		}
+	}
+}
+
+func UpdateConfigMapIngress(cm *corev1.ConfigMap, ingress networkingv1.Ingress) {
+	homerConfig := HomerConfig{}
+	err := yaml.Unmarshal([]byte(cm.Data["config.yml"]), &homerConfig)
+	if err != nil {
+		return
+	}
+	UpdateHomerConfigIngress(&homerConfig, ingress)
+	objYAML, err := yaml.Marshal(homerConfig)
+	if err != nil {
+		return
+	}
+	cm.Data["config.yml"] = string(objYAML)
+}