Add package cli command

Author: DhairyaMajmudar

cmd/kro/commands/package.go

@@ -0,0 +1,115 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/google/go-containerregistry/pkg/name"
+	v1 "github.com/google/go-containerregistry/pkg/v1"
+	"github.com/google/go-containerregistry/pkg/v1/empty"
+	"github.com/google/go-containerregistry/pkg/v1/mutate"
+	"github.com/google/go-containerregistry/pkg/v1/static"
+	"github.com/google/go-containerregistry/pkg/v1/tarball"
+	"github.com/google/go-containerregistry/pkg/v1/types"
+	"github.com/spf13/cobra"
+	"sigs.k8s.io/yaml"
+
+	"github.com/kro-run/kro/api/v1alpha1"
+)
+
+type PackageConfig struct {
+	resourceGraphDefinitionFile string
+	outputFile                  string
+}
+
+var packageConfig = &PackageConfig{}
+
+func init() {
+	packageRGDCmd.PersistentFlags().StringVarP(&packageConfig.resourceGraphDefinitionFile, "file", "f", "",
+		"Path to the ResourceGraphDefinition file")
+	packageRGDCmd.PersistentFlags().StringVarP(&packageConfig.outputFile, "output", "o", "",
+		"Output file name for the OCI image tarball")
+}
+
+var packageRGDCmd = &cobra.Command{
+	Use:   "package",
+	Short: "Package ResourceGraphDefinition file into an OCI image",
+	Long:  "Package command packages the ResourceGraphDefinition file into an OCI image, which can be used for distribution and deployment.",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if packageConfig.resourceGraphDefinitionFile == "" {
+			return fmt.Errorf("ResourceGraphDefinition file is required")
+		}
+
+		if packageConfig.outputFile == "" {
+			return fmt.Errorf("output file name is required")
+		}
+
+		data, err := os.ReadFile(packageConfig.resourceGraphDefinitionFile)
+		if err != nil {
+			return fmt.Errorf("failed to read ResourceGraphDefinition file: %w", err)
+		}
+
+		var rgd v1alpha1.ResourceGraphDefinition
+		if err := yaml.Unmarshal(data, &rgd); err != nil {
+			return fmt.Errorf("failed to unmarshal ResourceGraphDefinition: %w", err)
+		}
+
+		fmt.Fprintf(os.Stderr, "Packaging ResourceGraphDefinition into %s...\n",
+			packageConfig.outputFile)
+
+		if err = packageRGD(data, &rgd); err != nil {
+			return fmt.Errorf("failed to package ResourceGraphDefinition: %w", err)
+		}
+
+		fmt.Fprintf(os.Stderr, "Successfully packaged ResourceGraphDefinition to %s\n",
+			packageConfig.outputFile)
+
+		return nil
+	},
+}
+
+func packageRGD(data []byte, rgd *v1alpha1.ResourceGraphDefinition) error {
+	layer := static.NewLayer(data, types.MediaType("application/vnd.kro.resourcegraphdefinition.v1alpha1+yaml"))
+
+	img := empty.Image
+
+	img, err := mutate.AppendLayers(img, layer)
+
+	if err != nil {
+		return fmt.Errorf("failed to append layer: %w", err)
+	}
+
+	configFile, err := img.ConfigFile()
+	if err != nil {
+		return fmt.Errorf("failed to get config file: %w", err)
+	}
+
+	now := time.Now()
+	configFile.Created = v1.Time{Time: now}
+
+	configFile.Config.Labels = map[string]string{
+		"kro.run/type": "resourcegraphdefinition",
+		"kro.run/name": rgd.Name,
+	}
+
+	img, err = mutate.ConfigFile(img, configFile)
+	if err != nil {
+		return fmt.Errorf("failed to update image config: %w", err)
+	}
+
+	ref, err := name.ParseReference(fmt.Sprintf("kro.run/rgd/%s:latest", rgd.Name))
+	if err != nil {
+		return fmt.Errorf("failed to parse image reference: %w", err)
+	}
+
+	if err := tarball.WriteToFile(packageConfig.outputFile, ref, img); err != nil {
+		return fmt.Errorf("failed to write image to file: %w", err)
+	}
+
+	return nil
+}
+
+func AddPackageCommand(rootCmd *cobra.Command) {
+	rootCmd.AddCommand(packageRGDCmd)
+}

cmd/kro/main.go

@@ -63,5 +63,6 @@ func NewRootCommand() *cobra.Command {
 	// TODO: Command groups
 	commands.AddValidateCommands(cmd)
 	commands.AddGenerateCommands(cmd)
+	commands.AddPackageCommand(cmd)
 	return cmd
 }