Add an r8.im clone command

this is used to create a copy of a public image for debugging purposes

Author: anotherjesse

pkg/cli/clone.go

@@ -0,0 +1,52 @@
+package cli
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/google/go-containerregistry/pkg/authn"
+	"github.com/spf13/cobra"
+
+	"github.com/anotherjesse/r8im/pkg/auth"
+	"github.com/anotherjesse/r8im/pkg/images"
+)
+
+func newCloneCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:    "clone",
+		Short:  "copy existing image to a new image",
+		Hidden: false,
+		RunE:   cloneCommmand,
+	}
+
+	cmd.Flags().StringVarP(&sToken, "token", "t", "", "replicate cog token")
+	cmd.Flags().StringVarP(&sRegistry, "registry", "r", "r8.im", "registry host")
+	cmd.Flags().StringVarP(&baseRef, "base", "b", "", "base image reference - include tag: r8.im/username/modelname@sha256:hexdigest")
+	cmd.MarkFlagRequired("base")
+	cmd.Flags().StringVarP(&dest, "dest", "d", "", "destination image reference: r8.im/username/modelname")
+	cmd.MarkFlagRequired("dest")
+
+	return cmd
+}
+
+func cloneCommmand(cmd *cobra.Command, args []string) error {
+	if sToken == "" {
+		sToken = os.Getenv("COG_TOKEN")
+	}
+
+	u, err := auth.VerifyCogToken(sRegistry, sToken)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, "authentication error, invalid token or registry host error")
+		return err
+	}
+	auth := authn.FromConfig(authn.AuthConfig{Username: u, Password: sToken})
+
+	image_id, err := images.Affix(baseRef, dest, "", auth)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println(image_id)
+
+	return nil
+}

pkg/cli/root.go

@@ -18,6 +18,7 @@ func NewRootCommand() (*cobra.Command, error) {
 
 	rootCmd.AddCommand(
 		newAffixCommand(),
+		newCloneCommand(),
 		newLayerCommand(),
 		newExtractCommand(),
 		newRemixCommand(),

pkg/images/affix.go

@@ -32,15 +32,30 @@ func Affix(baseRef string, dest string, newLayer string, auth authn.Authenticato
 
 	// --- adding new layer ontop of existing image
 
-	fmt.Fprintln(os.Stderr, "appending as new layer", newLayer)
-
-	start = time.Now()
-	img, err := appendLayer(base, newLayer)
-	if err != nil {
-		return "", fmt.Errorf("appending %v: %w", newLayer, err)
+	var img v1.Image
+
+	if newLayer != "" {
+		fmt.Fprintln(os.Stderr, "appending as new layer", newLayer)
+
+		start = time.Now()
+		img, err = appendLayer(base, newLayer)
+		if err != nil {
+			return "", fmt.Errorf("appending %v: %w", newLayer, err)
+		}
+		fmt.Fprintln(os.Stderr, "appending took", time.Since(start))
+	} else {
+		cfg, err := base.ConfigFile()
+		if err != nil {
+			return "", fmt.Errorf("getting config file: %w", err)
+		}
+
+		cfg.Config.Labels["cloned"] = "true"
+
+		img, err = mutate.ConfigFile(base, cfg)
+		if err != nil {
+			return "", fmt.Errorf("mutating config file: %w", err)
+		}
 	}
-	fmt.Fprintln(os.Stderr, "appending took", time.Since(start))
-
 	// --- pushing image
 
 	start = time.Now()