feat: upload aliases from config file

Adds a flag for specifying the path to a YAML file with extension name aliases.

Signed-off-by: Aleksandr Gamzin <[email protected]>

Author: Alex619829

cmd/image-factory/cmd/options.go

@@ -121,6 +121,9 @@ type Options struct { //nolint:govet
 	ExtensionManifestImage string
 	OverlayManifestImage   string
 	TalosctlImage          string
+
+	// Path to YAML file with extension name aliases.
+	ExtensionNameAlias string
 }
 
 // SecureBootOptions configures SecureBoot.
@@ -184,4 +187,6 @@ var DefaultOptions = Options{
 	ExtensionManifestImage: "siderolabs/extensions",
 	OverlayManifestImage:   "siderolabs/overlays",
 	TalosctlImage:          "siderolabs/talosctl-all",
+
+	ExtensionNameAlias: "/etc/image-factory/aliases.yaml",
 }

cmd/image-factory/flags.go

@@ -106,6 +106,8 @@ func initFlags() cmd.Options {
 	flag.StringVar(&opts.OverlayManifestImage, "overlay-manifest-image", cmd.DefaultOptions.OverlayManifestImage, "Name of image OverlayManifestImage")
 	flag.StringVar(&opts.TalosctlImage, "talosctl-image", cmd.DefaultOptions.TalosctlImage, "Name of image TalosctlImage")
 
+	flag.StringVar(&opts.ExtensionNameAlias, "extension-name-alias", cmd.DefaultOptions.ExtensionNameAlias, "Path to YAML file with extension name aliases")
+
 	opts.LogLevel = zap.LevelFlag("log-level", zap.InfoLevel, "set the logging level")
 
 	flag.Parse()

cmd/image-factory/main.go

@@ -17,6 +17,7 @@ import (
 
 	"github.com/siderolabs/image-factory/cmd/image-factory/cmd"
 	"github.com/siderolabs/image-factory/internal/artifacts"
+	"github.com/siderolabs/image-factory/internal/config"
 )
 
 func main() {
@@ -89,6 +90,12 @@ func runWithContext(ctx context.Context) error {
 		artifacts.TalosctlImage = opts.TalosctlImage
 	}
 
+	if opts.ExtensionNameAlias != "" {
+		if err := config.LoadAliases(opts.ExtensionNameAlias); err != nil {
+			log.Fatalf("failed to load extension aliases: %v", err)
+		}
+	}
+
 	cfg := zap.NewProductionConfig()
 	cfg.Level = zap.NewAtomicLevelAt(*opts.LogLevel)
 

internal/config/alias.go

@@ -0,0 +1,51 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package config
+
+import (
+	"fmt"
+	"os"
+
+	"gopkg.in/yaml.v3"
+)
+
+type aliasConfig struct {
+	ExtensionNameAlias []struct {
+		Name  string `yaml:"name"`
+		Alias string `yaml:"alias"`
+	} `yaml:"extensionNameAlias"`
+}
+
+var aliasMap map[string]string
+
+// Loads aliases from aliases.yaml.
+func LoadAliases(path string) error {
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return fmt.Errorf("failed to read alias config: %w", err)
+	}
+
+	var cfg aliasConfig
+	if err := yaml.Unmarshal(data, &cfg); err != nil {
+		return fmt.Errorf("failed to unmarshal alias config: %w", err)
+	}
+
+	aliasMap = make(map[string]string)
+	for _, item := range cfg.ExtensionNameAlias {
+		aliasMap[item.Name] = item.Alias
+	}
+
+	return nil
+}
+
+// GetAlias returns alias if exists.
+func GetAlias(name string) (string, bool) {
+	if aliasMap == nil {
+		return "", false
+	}
+	alias, ok := aliasMap[name]
+	return alias, ok
+}
+

internal/profile/profile.go

@@ -20,6 +20,7 @@ import (
 	"github.com/siderolabs/gen/xerrors"
 	"github.com/siderolabs/gen/xslices"
 	"github.com/siderolabs/go-pointer"
+	"github.com/siderolabs/image-factory/internal/config"
 	"github.com/siderolabs/talos/pkg/imager/profile"
 	"github.com/siderolabs/talos/pkg/machinery/config/merge"
 	"github.com/siderolabs/talos/pkg/machinery/constants"
@@ -393,7 +394,15 @@ func EnhanceFromSchematic(
 				}
 
 				if value.IsZero(extensionRef) {
-					return prof, xerrors.NewTaggedf[InvalidErrorTag]("official extension %q is not available for Talos version %s", extensionName, versionTag)
+					// try YAML-based aliases
+					if aliasedName, ok := config.GetAlias(extensionName); ok {
+						extensionRef = findExtension(availableExtensions, aliasedName)
+					} else {
+						// fallback to default aliases if no YAML config was loaded
+						if aliasedName, ok := extensionNameAlias(extensionName); ok {
+							extensionRef = findExtension(availableExtensions, aliasedName)
+						}
+					}
 				}
 
 				imagePath, err := artifactProducer.GetExtensionImage(ctx, artifacts.Arch(prof.Arch), extensionRef)