Merge pull request #18 from wandb/platform-command

fix: allow platform specification

cmd/wsm/download.go

@@ -46,6 +46,8 @@ func downloadChartImages(
 }
 
 func DownloadCmd() *cobra.Command {
+	var platform string
+
 	cmd := &cobra.Command{
 		Use: "download",
 		Run: func(cmd *cobra.Command, args []string) {
@@ -93,7 +95,7 @@ func DownloadCmd() *cobra.Command {
 			cb := func(pkg string) {
 				path := "bundle/images/" + pkg
 				_ = os.MkdirAll(path, 0755)
-				err := images.Download(pkg, path+"/image.tgz")
+				err := images.Download(pkg, path+"/image.tgz", platform)
 				if err != nil {
 					fmt.Println(err)
 				}
@@ -106,5 +108,7 @@ func DownloadCmd() *cobra.Command {
 		},
 	}
 
+	cmd.Flags().StringVarP(&platform, "platform", "p", "linux/amd64", "Platform to download images for")
+
 	return cmd
 }

pkg/images/download.go

@@ -12,16 +12,16 @@ import (
 	"github.com/containers/image/v5/types"
 )
 
-func Download(image string, filename string) error {
+func Download(image string, filename string, platform string) error {
 	if _, err := exec.LookPath("docker"); err == nil {
-		return DownloadUsingDocker(image, filename)
+		return DownloadUsingDocker(image, filename, platform)
 	}
 
 	return fmt.Errorf("no supported container runtime found")
 }
 
-func DownloadUsingDocker(image string, filename string) error {
-	cmdPull := exec.Command("docker", "pull", image)
+func DownloadUsingDocker(image string, filename string, platform string) error {
+	cmdPull := exec.Command("docker", "pull", "--platform", platform, image)
 	err := cmdPull.Run()
 	if err != nil {
 		return fmt.Errorf("failed to pull image using docker %s: %v", image, err)