From d386566a9c052de209f27f01648e4d009ba29c19 Mon Sep 17 00:00:00 2001
From: Venkat Ramaraju <venky2063@gmail.com>
Date: Fri, 18 Jun 2021 10:40:56 -0700
Subject: [PATCH] predicate for helm

---
 internal/cmd/helm-operator/run/cmd.go  |  1 +
 internal/helm/controller/controller.go | 12 +++++++++++-
 internal/helm/watches/watches.go       |  8 +++++---
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/internal/cmd/helm-operator/run/cmd.go b/internal/cmd/helm-operator/run/cmd.go
index 3eecfd48898..f72f121b13a 100644
--- a/internal/cmd/helm-operator/run/cmd.go
+++ b/internal/cmd/helm-operator/run/cmd.go
@@ -184,6 +184,7 @@ func run(cmd *cobra.Command, f *flags.Flags) {
 			WatchDependentResources: *w.WatchDependentResources,
 			OverrideValues:          w.OverrideValues,
 			MaxConcurrentReconciles: f.MaxConcurrentReconciles,
+			Selector:                w.Selector,
 		})
 		if err != nil {
 			log.Error(err, "Failed to add manager factory to controller.")
diff --git a/internal/helm/controller/controller.go b/internal/helm/controller/controller.go
index a3f5a5d0467..935cd634ae9 100644
--- a/internal/helm/controller/controller.go
+++ b/internal/helm/controller/controller.go
@@ -30,6 +30,7 @@ import (
 	crthandler "sigs.k8s.io/controller-runtime/pkg/handler"
 	logf "sigs.k8s.io/controller-runtime/pkg/log"
 	"sigs.k8s.io/controller-runtime/pkg/manager"
+	filterPredicate "sigs.k8s.io/controller-runtime/pkg/predicate"
 	"sigs.k8s.io/controller-runtime/pkg/source"
 	"sigs.k8s.io/yaml"
 
@@ -51,6 +52,7 @@ type WatchOptions struct {
 	WatchDependentResources bool
 	OverrideValues          map[string]string
 	MaxConcurrentReconciles int
+	Selector                metav1.LabelSelector
 }
 
 // Add creates a new helm operator controller and adds it to the manager
@@ -78,9 +80,17 @@ func Add(mgr manager.Manager, options WatchOptions) error {
 		return err
 	}
 
+	// Predicate that will filter events based on labels
+	filterPredicate, err := filterPredicate.LabelSelectorPredicate(options.Selector)
+
+	if err != nil {
+		log.Error(err, "Unable to create predicate from selector.")
+		return err
+	}
+
 	o := &unstructured.Unstructured{}
 	o.SetGroupVersionKind(options.GVK)
-	if err := c.Watch(&source.Kind{Type: o}, &libhandler.InstrumentedEnqueueRequestForObject{}); err != nil {
+	if err := c.Watch(&source.Kind{Type: o}, &libhandler.InstrumentedEnqueueRequestForObject{}, filterPredicate); err != nil {
 		return err
 	}
 
diff --git a/internal/helm/watches/watches.go b/internal/helm/watches/watches.go
index c02413abb32..8aa0e650832 100644
--- a/internal/helm/watches/watches.go
+++ b/internal/helm/watches/watches.go
@@ -22,6 +22,7 @@ import (
 	"os"
 
 	"helm.sh/helm/v3/pkg/chartutil"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"sigs.k8s.io/yaml"
 )
@@ -32,9 +33,10 @@ const WatchesFile = "watches.yaml"
 // custom resource.
 type Watch struct {
 	schema.GroupVersionKind `json:",inline"`
-	ChartDir                string            `json:"chart"`
-	WatchDependentResources *bool             `json:"watchDependentResources,omitempty"`
-	OverrideValues          map[string]string `json:"overrideValues,omitempty"`
+	ChartDir                string               `json:"chart"`
+	WatchDependentResources *bool                `json:"watchDependentResources,omitempty"`
+	OverrideValues          map[string]string    `json:"overrideValues,omitempty"`
+	Selector                metav1.LabelSelector `json:"selector"`
 }
 
 // UnmarshalYAML unmarshals an individual watch from the Helm watches.yaml file