Add option to use ConditionalSources in controller builder

Author: kevindelgado

pkg/builder/controller.go

@@ -23,6 +23,7 @@ import (
 	"github.com/go-logr/logr"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/client-go/discovery"
 
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
@@ -72,6 +73,7 @@ type ForInput struct {
 	predicates       []predicate.Predicate
 	objectProjection objectProjection
 	err              error
+	conditional      bool
 }
 
 // For defines the type of Object being *reconciled*, and configures the ControllerManagedBy to respond to create / delete /
@@ -97,6 +99,7 @@ type OwnsInput struct {
 	object           client.Object
 	predicates       []predicate.Predicate
 	objectProjection objectProjection
+	conditional      bool
 }
 
 // Owns defines types of Objects being *generated* by the ControllerManagedBy, and configures the ControllerManagedBy to respond to
@@ -118,6 +121,7 @@ type WatchesInput struct {
 	eventhandler     handler.EventHandler
 	predicates       []predicate.Predicate
 	objectProjection objectProjection
+	conditional      bool
 }
 
 // Watches exposes the lower-level ControllerManagedBy Watches functions through the builder.  Consider using
@@ -216,13 +220,52 @@ func (blder *Builder) project(obj client.Object, proj objectProjection) (client.
 	}
 }
 
+func (blder *Builder) generateConditionalSource(typeForSrc client.Object) (source.Source, error) {
+	gvk, err := getGvk(blder.forInput.object, blder.mgr.GetScheme())
+	if err != nil {
+		return nil, err
+	}
+	fmt.Printf("gvk = %+v\n", gvk)
+	dc, err := discovery.NewDiscoveryClientForConfig(blder.mgr.GetConfig())
+	if err != nil {
+		return nil, err
+	}
+	existsInDiscovery := func() bool {
+		resources, err := dc.ServerResourcesForGroupVersion(gvk.GroupVersion().String())
+		if err != nil {
+			fmt.Printf("NOT in discovery gvk = %+v\n", gvk)
+			return false
+		}
+		for _, res := range resources.APIResources {
+			if res.Kind == gvk.Kind {
+				fmt.Printf("YES in discovery gvk = %+v\n", gvk)
+				return true
+			}
+		}
+		fmt.Printf("NOT in discovery kind = %+v\n", gvk)
+		return false
+	}
+	return &source.ConditionalKind{Kind: source.Kind{Type: typeForSrc}, DiscoveryCheck: existsInDiscovery}, nil
+
+}
+
 func (blder *Builder) doWatch() error {
 	// Reconcile type
 	typeForSrc, err := blder.project(blder.forInput.object, blder.forInput.objectProjection)
 	if err != nil {
 		return err
 	}
-	src := &source.Kind{Type: typeForSrc}
+
+	var src source.Source
+	if blder.forInput.conditional {
+		var err error
+		src, err = blder.generateConditionalSource(typeForSrc)
+		if err != nil {
+			return err
+		}
+	} else {
+		src = &source.Kind{Type: typeForSrc}
+	}
 	hdler := &handler.EnqueueRequestForObject{}
 	allPredicates := append(blder.globalPredicates, blder.forInput.predicates...)
 	if err := blder.ctrl.Watch(src, hdler, allPredicates...); err != nil {
@@ -235,7 +278,17 @@ func (blder *Builder) doWatch() error {
 		if err != nil {
 			return err
 		}
-		src := &source.Kind{Type: typeForSrc}
+
+		var src source.Source
+		if own.conditional {
+			var err error
+			src, err = blder.generateConditionalSource(typeForSrc)
+			if err != nil {
+				return err
+			}
+		} else {
+			src = &source.Kind{Type: typeForSrc}
+		}
 		hdler := &handler.EnqueueRequestForOwner{
 			OwnerType:    blder.forInput.object,
 			IsController: true,

pkg/builder/options.go

@@ -99,6 +99,19 @@ func (p projectAs) ApplyToWatches(opts *WatchesInput) {
 	opts.objectProjection = objectProjection(p)
 }
 
+// Conditional confirms that it should configure a ConditionalSource
+// that can be started/stopped/restarted based on the existence of
+// the input's type in discovery.
+type Conditional struct{}
+
+func (s Conditional) ApplyToFor(opts *ForInput) {
+	opts.conditional = true
+}
+
+func (s Conditional) ApplyToOwns(opts *OwnsInput) {
+	opts.conditional = true
+}
+
 var (
 	// OnlyMetadata tells the controller to *only* cache metadata, and to watch
 	// the the API server in metadata-only form.  This is useful when watching