Add images command

pkg/cmd/cmd.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"github.com/spf13/cobra"
 
+	"github.com/openshift-eng/openshift-tests-extension/pkg/cmd/cmdimages"
 	"github.com/openshift-eng/openshift-tests-extension/pkg/cmd/cmdinfo"
 	"github.com/openshift-eng/openshift-tests-extension/pkg/cmd/cmdlist"
 	"github.com/openshift-eng/openshift-tests-extension/pkg/cmd/cmdrun"
@@ -17,5 +18,6 @@ func DefaultExtensionCommands(registry *extension.Registry) []*cobra.Command {
 		cmdlist.NewListCommand(registry),
 		cmdinfo.NewInfoCommand(registry),
 		cmdupdate.NewUpdateCommand(registry),
+		cmdimages.NewImagesCommand(registry),
 	}
 }

pkg/cmd/cmdimages/cmdimages.go

@@ -0,0 +1,36 @@
+package cmdimages
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+
+	"github.com/openshift-eng/openshift-tests-extension/pkg/extension"
+	"github.com/openshift-eng/openshift-tests-extension/pkg/flags"
+)
+
+func NewImagesCommand(registry *extension.Registry) *cobra.Command {
+	componentFlags := flags.NewComponentFlags()
+
+	cmd := &cobra.Command{
+		Use:          "images",
+		Short:        "List test images",
+		SilenceUsage: true,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			extension := registry.Get(componentFlags.Component)
+			if extension == nil {
+				return fmt.Errorf("couldn't find the component %q", componentFlags.Component)
+			}
+			images, err := json.Marshal(extension.Images)
+			if err != nil {
+				return err
+			}
+			fmt.Fprintf(os.Stdout, "%s\n", images)
+			return nil
+		},
+	}
+	componentFlags.BindFlags(cmd.Flags())
+	return cmd
+}

pkg/extension/extension.go

@@ -125,6 +125,11 @@ func (e *Extension) AddSuite(suite Suite) *Extension {
 	return e
 }
 
+func (e *Extension) RegisterImage(image Image) *Extension {
+	e.Images = append(e.Images, image)
+	return e
+}
+
 func (e *Extension) FindSpecsByName(names ...string) (et.ExtensionTestSpecs, error) {
 	var specs et.ExtensionTestSpecs
 	var notFound []string

pkg/extension/types.go

@@ -17,6 +17,8 @@ type Extension struct {
 	// Suites that the extension wants to advertise/participate in.
 	Suites []Suite `json:"suites"`
 
+	Images []Image `json:"images"`
+
 	// Private data
 	specs         extensiontests.ExtensionTestSpecs
 	obsoleteTests sets.Set[string]
@@ -53,3 +55,10 @@ type Suite struct {
 	// Qualifiers are CEL expressions that are OR'd together for test selection that are members of the suite.
 	Qualifiers []string `json:"qualifiers,omitempty"`
 }
+
+type Image struct {
+	Index    int    `json:"index"`
+	Registry string `json:"registry"`
+	Name     string `json:"name"`
+	Version  string `json:"version"`
+}