Handle library images

Signed-off-by: Tamal Saha <[email protected]>
kmodules · Sep 30, 2024 · f798cb2 · f798cb2
1 parent 7da7969
commit f798cb2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/cmds/scripts.go b/pkg/cmds/scripts.go
@@ -198,7 +198,11 @@ CMD="./crane"
 		buf.WriteString(" ")
 		buf.WriteString("images/" + strings.ReplaceAll(ref.Repository, "/", "-") + "-" + ref.Tag + ".tar")
 		buf.WriteString(" ")
-		buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository + ":" + ref.Tag)
+		if strings.HasPrefix(ref.Repository, "library/") {
+			buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository[len("library/"):] + ":" + ref.Tag)
+		} else {
+			buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository + ":" + ref.Tag)
+		}
 		buf.WriteRune('\n')
 	}
 	err = os.WriteFile(filepath.Join(outdir, "import-images.sh"), buf.Bytes(), 0o755)
@@ -251,7 +255,11 @@ CMD="./crane"
 		buf.WriteString(" ")
 		buf.WriteString(img)
 		buf.WriteString(" ")
-		buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository + ":" + ref.Tag)
+		if strings.HasPrefix(ref.Repository, "library/") {
+			buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository[len("library/"):] + ":" + ref.Tag)
+		} else {
+			buf.WriteString("$IMAGE_REGISTRY/" + ref.Repository + ":" + ref.Tag)
+		}
 		buf.WriteRune('\n')
 	}
 	err = os.WriteFile(filepath.Join(outdir, "copy-images.sh"), buf.Bytes(), 0o755)