Allow setting the arch of the image to be pulled

because when we are on amd64 host, the auroraboot image is amd64 but we
may want to pull and build artifacts for arm64 (e.g. when building rpi
raw images).

Signed-off-by: Dimitris Karakasilis <dimitris@karakasilis.me>

Author: jimmykarily

deployer/steps.go

@@ -97,7 +97,7 @@ func (d *Deployer) StepDumpSource() error {
 	// Ops to generate from container image
 	return d.Add(constants.OpDumpSource,
 		herd.EnableIf(d.fromImage),
-		herd.WithDeps(constants.OpPrepareDirs), herd.WithCallback(ops.DumpSource(d.Artifact.ContainerImage, d.tmpRootFs)))
+		herd.WithDeps(constants.OpPrepareDirs), herd.WithCallback(ops.DumpSource(d.Artifact.ContainerImage, d.tmpRootFs, d.Config.Arch)))
 }
 
 func (d *Deployer) StepGenISO() error {

pkg/ops/container.go

@@ -13,16 +13,20 @@ import (
 // or simply copies the directory to the destination.
 // Supports these prefixes:
 // https://github.com/kairos-io/kairos-agent/blob/1e81cdef38677c8a36cae50d3334559976f66481/pkg/types/v1/common.go#L30-L33
-func DumpSource(image string, dstFunc valueGetOnCall) func(ctx context.Context) error {
+func DumpSource(image string, dstFunc valueGetOnCall, arch string) func(ctx context.Context) error {
 	return func(ctx context.Context) error {
 		dst := dstFunc()
 		if image == "" {
 			return fmt.Errorf("image source is empty, cannot dump to %s", dst)
 		}
-		cfg := NewConfig(
+		opts := []GenericOptions{
 			WithImageExtractor(v1.OCIImageExtractor{}),
 			WithLogger(internal.Log),
-		)
+		}
+		if arch != "" {
+			opts = append(opts, WithArch(arch))
+		}
+		cfg := NewConfig(opts...)
 		e := elemental.NewElemental(cfg)
 
 		imgSource, err := v1.NewSrcFromURI(image)

pkg/ops/iso.go

@@ -756,3 +756,16 @@ func WithImageExtractor(extractor v1types.ImageExtractor) func(r *agentconfig.Co
 		return nil
 	}
 }
+
+func WithArch(arch string) func(r *agentconfig.Config) error {
+	return func(r *agentconfig.Config) error {
+		if arch != "" {
+			convertedArch, err := utils.GolangArchToArch(arch)
+			if err != nil {
+				return fmt.Errorf("invalid architecture %s: %w", arch, err)
+			}
+			r.Arch = convertedArch
+		}
+		return nil
+	}
+}

pkg/schema/config.go

@@ -27,6 +27,9 @@ type Config struct {
 
 	ListenAddr string `yaml:"listen_addr"`
 
+	// Architecture to use for container image pulling (e.g., "amd64", "arm64")
+	Arch string `yaml:"arch"`
+
 	// ISO block configuration
 	ISO ISO `yaml:"iso"`