Resolution CLI POC: offline input

Signed-off-by: Mikalai Radchuk <[email protected]>

Author: Mikalai Radchuk

cmd/resolutioncli/entity_source.go

@@ -117,6 +117,12 @@ func (es *indexRefEntitySource) entities(ctx context.Context) (input.EntityList,
 	return es.entitiesCache, nil
 }
 
+// TODO: Reduce code duplication: share a function with catalogdEntitySource (see getEntities)
+// We don't want to maintain two functions performing conversion into input.EntityList.
+// For this we need some common format. So we need a package which will be able
+// to convert from declfcg structs into CRD structs directly or via model.Model.
+// One option would be to make this piece of code from catalogd reusable and exportable:
+// https://github.com/operator-framework/catalogd/blob/9fe45a628de2e74d9cd73c3650fa2582aaac5213/pkg/controllers/core/catalog_controller.go#L200-L360
 func modelToEntities(model model.Model) (input.EntityList, error) {
 	entities := input.EntityList{}
 

cmd/resolutioncli/input_manifests.go

@@ -0,0 +1,60 @@
+/*
+Copyright 2023.
+
+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 main
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"k8s.io/apimachinery/pkg/runtime"
+)
+
+func readManifestFiles(directory string) ([]runtime.Object, error) {
+	var objects []runtime.Object
+
+	err := filepath.Walk(directory, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if info.IsDir() {
+			return nil
+		}
+
+		fileContent, err := os.ReadFile(path)
+		if err != nil {
+			return fmt.Errorf("failed to read file %s: %w", path, err)
+		}
+
+		decoder := codecs.UniversalDecoder(scheme.PrioritizedVersionsAllGroups()...)
+		object, _, err := decoder.Decode(fileContent, nil, nil)
+		if err != nil {
+			return fmt.Errorf("failed to decode file %s: %w", path, err)
+		}
+
+		objects = append(objects, object)
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, fmt.Errorf("failed to read files: %w", err)
+	}
+
+	return objects, nil
+}

cmd/resolutioncli/main.go

@@ -25,11 +25,11 @@ import (
 	"github.com/operator-framework/deppy/pkg/deppy/solver"
 	rukpakv1alpha1 "github.com/operator-framework/rukpak/api/v1alpha1"
 	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/runtime/serializer"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
 	_ "k8s.io/client-go/plugin/pkg/client/auth"
-	"sigs.k8s.io/controller-runtime/pkg/client"
-	"sigs.k8s.io/controller-runtime/pkg/client/config"
+	"sigs.k8s.io/controller-runtime/pkg/client/fake"
 
 	catalogd "github.com/operator-framework/catalogd/api/core/v1alpha1"
 
@@ -49,10 +49,13 @@ const (
 	flagNamePackageVersion = "package-version"
 	flagNamePackageChannel = "package-channel"
 	flagNameIndexRef       = "index-ref"
+	flagNameInputDir       = "input-dir"
 )
 
 var (
 	scheme = runtime.NewScheme()
+
+	codecs = serializer.NewCodecFactory(scheme)
 )
 
 func init() {
@@ -71,11 +74,13 @@ func main() {
 	var packageVersion string
 	var packageChannel string
 	var indexRef string
+	var inputDir string
 	flag.StringVar(&packageName, flagNamePackageName, "", "Name of the package to resolve")
 	flag.StringVar(&packageVersion, flagNamePackageVersion, "", "Version of the package")
 	flag.StringVar(&packageChannel, flagNamePackageChannel, "", "Channel of the package")
 	// TODO: Consider adding support of multiple refs
 	flag.StringVar(&indexRef, flagNameIndexRef, "", "Index reference (FBC image or dir)")
+	flag.StringVar(&inputDir, flagNameInputDir, "", "Directory containing Kubernetes manifests (such as Operator) to be used as an input for resolution")
 	flag.Parse()
 
 	if err := validateFlags(packageName, indexRef); err != nil {
@@ -84,7 +89,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	err := run(ctx, packageName, packageVersion, packageChannel, indexRef)
+	err := run(ctx, packageName, packageVersion, packageChannel, indexRef, inputDir)
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
 		os.Exit(1)
@@ -103,17 +108,24 @@ func validateFlags(packageName, indexRef string) error {
 	return nil
 }
 
-func run(ctx context.Context, packageName, packageVersion, packageChannel, catalogRef string) error {
-	client, err := client.New(config.GetConfigOrDie(), client.Options{Scheme: scheme})
-	if err != nil {
-		return fmt.Errorf("failed to create client: %w", err)
+func run(ctx context.Context, packageName, packageVersion, packageChannel, catalogRef, inputDir string) error {
+	clientBuilder := fake.NewClientBuilder().WithScheme(scheme)
+
+	if inputDir != "" {
+		objects, err := readManifestFiles(inputDir)
+		if err != nil {
+			return err
+		}
+
+		clientBuilder.WithRuntimeObjects(objects...)
 	}
 
+	cl := clientBuilder.Build()
 	resolver := solver.NewDeppySolver(
 		newIndexRefEntitySourceEntitySource(catalogRef),
 		append(
 			variablesources.NestedVariableSource{newPackageVariableSource(packageName, packageVersion, packageChannel)},
-			controllers.NewVariableSource(client)...,
+			controllers.NewVariableSource(cl)...,
 		),
 	)