wip

Signed-off-by: Joe Lanford <[email protected]>

Author: joelanford

cmd/manager/main.go

@@ -161,7 +161,7 @@ func main() {
 	}
 
 	cl := mgr.GetClient()
-	catalogClient := catalogclient.New(cl, cache.NewFilesystemCache(cachePath, httpClient))
+	catalogClient := catalogclient.New(cache.NewFilesystemCache(cachePath, httpClient))
 
 	installNamespaceMapper := helmclient.ObjectToStringMapper(func(obj client.Object) (string, error) {
 		ext := obj.(*ocv1alpha1.ClusterExtension)
@@ -230,16 +230,16 @@ func main() {
 	}
 
 	if err = (&controllers.ClusterExtensionReconciler{
-		Client:                cl,
-		BundleProvider:        catalogClient,
-		ActionClientGetter:    acg,
-		Unpacker:              unpacker,
-		Storage:               localStorage,
-		InstalledBundleGetter: &controllers.DefaultInstalledBundleGetter{ActionClientGetter: acg},
-		Handler:               registryv1handler.HandlerFunc(registry.HandleBundleDeployment),
-		Finalizers:            clusterExtensionFinalizers,
-		CaCertDir:             caCertDir,
-		Preflights:            preflights,
+		Client:                 cl,
+		CatalogPackageProvider: catalogClient,
+		ActionClientGetter:     acg,
+		Unpacker:               unpacker,
+		Storage:                localStorage,
+		InstalledBundleGetter:  &controllers.DefaultInstalledBundleGetter{ActionClientGetter: acg},
+		Handler:                registryv1handler.HandlerFunc(registry.HandleBundleDeployment),
+		Finalizers:             clusterExtensionFinalizers,
+		CaCertDir:              caCertDir,
+		Preflights:             preflights,
 	}).SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "ClusterExtension")
 		os.Exit(1)

internal/bundleutil/bundle.go

@@ -0,0 +1,53 @@
+package bundleutil
+
+import (
+	"encoding/json"
+	"fmt"
+	"sync"
+
+	bsemver "github.com/blang/semver/v4"
+
+	"github.com/operator-framework/operator-registry/alpha/declcfg"
+	"github.com/operator-framework/operator-registry/alpha/property"
+
+	ocv1alpha1 "github.com/operator-framework/operator-controller/api/v1alpha1"
+)
+
+var VersionGetter = &versionGetter{cache: make(map[*declcfg.Bundle]*bsemver.Version)}
+
+type versionGetter struct {
+	mu    sync.Mutex
+	cache map[*declcfg.Bundle]*bsemver.Version
+}
+
+func (v *versionGetter) GetVersion(b *declcfg.Bundle) (*bsemver.Version, error) {
+	v.mu.Lock()
+	defer v.mu.Unlock()
+	if vers, ok := v.cache[b]; ok {
+		return &(*vers), nil
+	}
+	for _, p := range b.Properties {
+		if p.Type == property.TypePackage {
+			var pkg property.Package
+			if err := json.Unmarshal(p.Value, &pkg); err != nil {
+				return nil, fmt.Errorf("error unmarshalling package property: %w", err)
+			}
+			vers, err := bsemver.Parse(pkg.Version)
+			if err != nil {
+				return nil, err
+			}
+			v.cache[b] = &vers
+			return &(*(&vers)), nil
+		}
+	}
+	return nil, fmt.Errorf("no package property found in bundle %q", b.Name)
+}
+
+// MetadataFor returns a BundleMetadata for the given bundle. If the provided bundle is nil,
+// this function panics. It is up to the caller to ensure that the bundle is non-nil.
+func MetadataFor(bundleName string, bundleVersion *bsemver.Version) *ocv1alpha1.BundleMetadata {
+	return &ocv1alpha1.BundleMetadata{
+		Name:    bundleName,
+		Version: bundleVersion.String(),
+	}
+}

internal/catalogmetadata/cache/cache_test.go

@@ -68,7 +68,7 @@ func TestFilesystemCache(t *testing.T) {
 		catalog        *catalogd.ClusterCatalog
 		contents       fstest.MapFS
 		wantErr        bool
-		tripper        *MockTripper
+		tripper        *mockTripper
 		testCaching    bool
 		shouldHitCache bool
 	}
@@ -89,7 +89,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents: defaultFS,
-			tripper:  &MockTripper{},
+			tripper:  &mockTripper{},
 		},
 		{
 			name: "valid cached fetch",
@@ -107,7 +107,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents:       defaultFS,
-			tripper:        &MockTripper{},
+			tripper:        &mockTripper{},
 			testCaching:    true,
 			shouldHitCache: true,
 		},
@@ -127,7 +127,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents:       defaultFS,
-			tripper:        &MockTripper{},
+			tripper:        &mockTripper{},
 			testCaching:    true,
 			shouldHitCache: false,
 		},
@@ -147,7 +147,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents: defaultFS,
-			tripper:  &MockTripper{shouldError: true},
+			tripper:  &mockTripper{shouldError: true},
 			wantErr:  true,
 		},
 		{
@@ -166,14 +166,14 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents: defaultFS,
-			tripper:  &MockTripper{serverError: true},
+			tripper:  &mockTripper{serverError: true},
 			wantErr:  true,
 		},
 		{
 			name:     "nil catalog",
 			catalog:  nil,
 			contents: defaultFS,
-			tripper:  &MockTripper{serverError: true},
+			tripper:  &mockTripper{serverError: true},
 			wantErr:  true,
 		},
 		{
@@ -187,7 +187,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents: defaultFS,
-			tripper:  &MockTripper{serverError: true},
+			tripper:  &mockTripper{serverError: true},
 			wantErr:  true,
 		},
 		{
@@ -203,7 +203,7 @@ func TestFilesystemCache(t *testing.T) {
 				},
 			},
 			contents: defaultFS,
-			tripper:  &MockTripper{serverError: true},
+			tripper:  &mockTripper{serverError: true},
 			wantErr:  true,
 		},
 	} {
@@ -242,15 +242,15 @@ func TestFilesystemCache(t *testing.T) {
 	}
 }
 
-var _ http.RoundTripper = &MockTripper{}
+var _ http.RoundTripper = &mockTripper{}
 
-type MockTripper struct {
+type mockTripper struct {
 	content     fstest.MapFS
 	shouldError bool
 	serverError bool
 }
 
-func (mt *MockTripper) RoundTrip(_ *http.Request) (*http.Response, error) {
+func (mt *mockTripper) RoundTrip(_ *http.Request) (*http.Response, error) {
 	if mt.shouldError {
 		return nil, errors.New("mock tripper error")
 	}