feat: add collectioniterators to collection cr (#60)

krateoplatformops · Sep 10, 2024 · fc4fcf2 · fc4fcf2
1 parent 233cad7
commit fc4fcf2
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 3 deletions.
diff --git a/internal/resolvers/templates/refs/refs.go b/internal/resolvers/templates/refs/refs.go
@@ -76,6 +76,16 @@ func Resolve(ctx context.Context, opts ResolveOptions) *runtime.RawExtension {
 		}
 	}
 
+	if tools.IsCollectionIterator(gvr.GroupResource()) {
+		obj, err = cli.Templates().CollectionIterators(opts.WidgetRef.Namespace).
+			Get(ctx, opts.WidgetRef.Name, metav1.GetOptions{})
+		if err != nil {
+			klog.Errorf("unable to get collectioniterator '%s' %s/%s",
+				gvr.String(), opts.WidgetRef.Namespace, opts.WidgetRef.Name)
+			return nil
+		}
+	}
+
 	if tools.IsCollection(gvr.GroupResource()) {
 		obj, err = cli.Templates().Collections(opts.WidgetRef.Namespace).
 			Get(ctx, opts.WidgetRef.Name, metav1.GetOptions{})

diff --git a/internal/resolvers/templates/tools/tools.go b/internal/resolvers/templates/tools/tools.go
@@ -19,6 +19,12 @@ func IsWidget(gr schema.GroupResource) bool {
 	return ok
 }
 
+func IsCollectionIterator(gr schema.GroupResource) bool {
+	ok := (gr.Group == templates.SchemeGroupVersion.Group)
+	ok = ok && (gr.Resource == "collectioniterators")
+	return ok
+}
+
 func IsCollection(gr schema.GroupResource) bool {
 	ok := (gr.Group == templates.SchemeGroupVersion.Group)
 	ok = ok && (gr.Resource == "collections")

diff --git a/testdata/collectioniterator.sample.yaml b/testdata/collectioniterator.sample.yaml
@@ -1,20 +1,20 @@
 kind: ConfigMap
 apiVersion: v1
 metadata:
-  name: templates-row-column-1
+  name: collit
   namespace: demo-system
 data:
   width: "24"
 ---
 apiVersion: templates.krateo.io/v1alpha1
 kind: CollectionIterator
 metadata:
-  name: templates-row-column-1
+  name: collit
   namespace: demo-system
 spec:
   type: column
   propsRef:
-    name: templates-row-column-1
+    name: collit
     namespace: demo-system
   widgetsRefs:
   - iterator: .allPods.items[:3]

diff --git a/testdata/widgets.collections.sample.yaml b/testdata/widgets.collections.sample.yaml
@@ -26,6 +26,10 @@ spec:
     resource: widgets
     name: external-api
     namespace: demo-system
+  - apiVersion: templates.krateo.io/v1alpha1
+    resource: collectioniterators
+    name: collit
+    namespace: demo-system
 ---
 kind: ConfigMap 
 apiVersion: v1