feat(genconfig): allow node level installer image url

budimanjojo · Oct 19, 2023 · 6c09821 · 6c09821
1 parent 74e104d
commit 6c09821
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/docs/docs/reference/configuration.md b/docs/docs/reference/configuration.md
@@ -289,6 +289,18 @@ installDisk: /dev/sda
 <td markdown="1" align="center">:white_check_mark:</td>
 </tr>
 
+<tr markdown="1">
+<td markdown="1">`talosImageURL`</td>
+<td markdown="1">string</td>
+<td markdown="1">Allows for supplying the node level image used to perform the installation.<details><summary>*Show example*</summary>
+```yaml
+talosImageURL: factory.talos.dev/installer/e9c7ef96884d4fbc8c0a1304ccca4bb0287d766a8b4125997cb9dbe84262144e
+```
+</details></td>
+<td markdown="1" align="center">`""`</td>
+<td markdown="1" align="center">:negative_squared_cross_mark:</td>
+</tr>
+
 <tr markdown="1">
 <td markdown="1">`installDiskSelector`</td>
 <td markdown="1">[InstallDiskSelector](#installdiskselector)</td>

diff --git a/example/talconfig.yaml b/example/talconfig.yaml
@@ -26,6 +26,7 @@ patches:
 nodes:
   - hostname: kmaster1
     ipAddress: 192.168.200.11
+    talosImageURL: factory.talos.dev/installer/e9c7ef96884d4fbc8c0a1304ccca4bb0287d766a8b4125997cb9dbe84262144e
     extensions:
       - image: ghcr.io/siderolabs/tailscale:1.44.0
     machineFiles:

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -39,6 +39,7 @@ type Node struct {
 	ConfigPatches       []map[string]interface{}          `yaml:"configPatches,omitempty"`
 	InlinePatch         map[string]interface{}            `yaml:"inlinePatch,omitempty"`
 	Patches             []string                          `yaml:"patches,omitempty"`
+	TalosImageURL       string                            `yaml:"talosImageURL"`
 }
 
 type cniConfig struct {

diff --git a/pkg/talos/nodeconfig.go b/pkg/talos/nodeconfig.go
@@ -1,6 +1,8 @@
 package talos
 
 import (
+	"strings"
+
 	"github.com/budimanjojo/talhelper/pkg/config"
 	taloscfg "github.com/siderolabs/talos/pkg/machinery/config"
 	"github.com/siderolabs/talos/pkg/machinery/config/generate"
@@ -46,6 +48,11 @@ func GenerateNodeConfig(node *config.Node, input *generate.Input) (taloscfg.Prov
 func applyNodeOverride(node *config.Node, cfg taloscfg.Provider) taloscfg.Provider {
 	cfg.RawV1Alpha1().MachineConfig.MachineNetwork.NetworkHostname = node.Hostname
 
+	if node.TalosImageURL != "" {
+		version := strings.Split(cfg.Machine().Install().Image(), ":")
+		cfg.RawV1Alpha1().MachineConfig.MachineInstall.InstallImage = node.TalosImageURL + ":" + version[1]
+	}
+
 	if len(node.Nameservers) > 0 {
 		cfg.RawV1Alpha1().MachineConfig.MachineNetwork.NameServers = node.Nameservers
 	}