add entity source predicates

Signed-off-by: perdasilva <[email protected]>

Author: perdasilva

internal/resolution/variable_sources/utils/predicates/predicates.go

@@ -0,0 +1,58 @@
+package predicates
+
+import (
+	"github.com/blang/semver/v4"
+	"github.com/operator-framework/deppy/pkg/deppy/input"
+
+	olmentity "github.com/operator-framework/operator-controller/internal/resolution/variable_sources/entity"
+)
+
+func WithPackageName(packageName string) input.Predicate {
+	return func(entity *input.Entity) bool {
+		bundleEntity := olmentity.NewBundleEntity(entity)
+		name, err := bundleEntity.PackageName()
+		if err != nil {
+			return false
+		}
+		return name == packageName
+	}
+}
+
+func InSemverRange(semverRange semver.Range) input.Predicate {
+	return func(entity *input.Entity) bool {
+		bundleEntity := olmentity.NewBundleEntity(entity)
+		bundleVersion, err := bundleEntity.Version()
+		if err != nil {
+			return false
+		}
+		return semverRange(*bundleVersion)
+	}
+}
+
+func InChannel(channelName string) input.Predicate {
+	return func(entity *input.Entity) bool {
+		bundleEntity := olmentity.NewBundleEntity(entity)
+		bundleChannel, err := bundleEntity.ChannelName()
+		if err != nil {
+			return false
+		}
+		return channelName == bundleChannel
+	}
+}
+
+func ProvidesGVK(gvk *olmentity.GVK) input.Predicate {
+	return func(entity *input.Entity) bool {
+		bundleEntity := olmentity.NewBundleEntity(entity)
+		providedGVKs, err := bundleEntity.ProvidedGVKs()
+		if err != nil {
+			return false
+		}
+		for i := 0; i < len(providedGVKs); i++ {
+			providedGVK := &providedGVKs[i]
+			if providedGVK.String() == gvk.String() {
+				return true
+			}
+		}
+		return false
+	}
+}

internal/resolution/variable_sources/utils/predicates/predicates_test.go

@@ -0,0 +1,71 @@
+package predicates_test
+
+import (
+	"testing"
+
+	"github.com/blang/semver/v4"
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"github.com/operator-framework/deppy/pkg/deppy/input"
+	"github.com/operator-framework/operator-registry/alpha/property"
+
+	olmentity "github.com/operator-framework/operator-controller/internal/resolution/variable_sources/entity"
+	"github.com/operator-framework/operator-controller/internal/resolution/variable_sources/utils/predicates"
+)
+
+func TestPredicates(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Predicates Suite")
+}
+
+var _ = Describe("Predicates", func() {
+	Describe("WithPackageName", func() {
+		It("should return true when the entity has the same package name", func() {
+			entity := input.NewEntity("test", map[string]string{
+				property.TypePackage: `{"packageName": "mypackage", "version": "1.0.0"}`,
+			})
+			Expect(predicates.WithPackageName("mypackage")(entity)).To(BeTrue())
+			Expect(predicates.WithPackageName("notmypackage")(entity)).To(BeFalse())
+		})
+	})
+
+	Describe("InSemverRange", func() {
+		It("should return true when the entity has the has version in the right range", func() {
+			entity := input.NewEntity("test", map[string]string{
+				property.TypePackage: `{"packageName": "mypackage", "version": "1.0.0"}`,
+			})
+			inRange := semver.MustParseRange(">=1.0.0")
+			notInRange := semver.MustParseRange(">=2.0.0")
+			Expect(predicates.InSemverRange(inRange)(entity)).To(BeTrue())
+			Expect(predicates.InSemverRange(notInRange)(entity)).To(BeFalse())
+		})
+	})
+
+	Describe("InChannel", func() {
+		It("should return true when the entity has the has version in the right range", func() {
+			entity := input.NewEntity("test", map[string]string{
+				property.TypeChannel: `{"channelName":"stable","priority":0}`,
+			})
+			Expect(predicates.InChannel("stable")(entity)).To(BeTrue())
+			Expect(predicates.InChannel("unstable")(entity)).To(BeFalse())
+		})
+	})
+
+	Describe("ProvidesGVK", func() {
+		It("should return true when the entity provides the specified gvk", func() {
+			entity := input.NewEntity("test", map[string]string{
+				property.TypeGVK: `[{"group":"foo.io","kind":"Foo","version":"v1"},{"group":"bar.io","kind":"Bar","version":"v1"}]`,
+			})
+			Expect(predicates.ProvidesGVK(&olmentity.GVK{
+				Group:   "foo.io",
+				Version: "v1",
+				Kind:    "Foo",
+			})(entity)).To(BeTrue())
+			Expect(predicates.ProvidesGVK(&olmentity.GVK{
+				Group:   "baz.io",
+				Version: "v1alpha1",
+				Kind:    "Baz",
+			})(entity)).To(BeFalse())
+		})
+	})
+})