feat(distros): add rke2/k3s setup; call setup func in install

Signed-off-by: Vaughn Dice <[email protected]>

Author: vdice

cmd/node-installer/install.go

@@ -45,8 +45,14 @@ var installCmd = &cobra.Command{
 		}
 
 		config.Runtime.ConfigPath = distro.ConfigPath
+		env := preset.Env{ConfigPath: distro.ConfigPath, HostFs: hostFs}
 
-		if err := RunInstall(config, rootFs, hostFs, distro.Restarter(preset.Env{ConfigPath: distro.ConfigPath, HostFs: hostFs})); err != nil {
+		if err = distro.Setup(env); err != nil {
+			slog.Error("failed to run distro setup", "error", err)
+			os.Exit(1)
+		}
+
+		if err := RunInstall(config, rootFs, hostFs, distro.Restarter(env)); err != nil {
 			slog.Error("failed to install", "error", err)
 			os.Exit(1)
 		}

internal/preset/preset.go

@@ -2,7 +2,9 @@ package preset
 
 import (
 	"errors"
+	"io"
 	"os"
+	"strings"
 
 	"github.com/spf13/afero"
 	"github.com/spinkube/runtime-class-manager/internal/containerd"
@@ -45,7 +47,28 @@ var RKE2 = Default.WithConfigPath("/var/lib/rancher/rke2/agent/etc/containerd/co
 		}
 
 		if errors.Is(err, os.ErrNotExist) {
-			// TODO: Copy file from original file to new config file
+			// Copy base config into .tmpl version
+			src, _ := strings.CutSuffix(env.ConfigPath, ".tmpl")
+			in, err := env.HostFs.Open(src)
+			if err != nil {
+				return err
+			}
+			defer in.Close()
+			out, err := env.HostFs.Create(env.ConfigPath)
+			if err != nil {
+				return err
+			}
+			defer func() {
+				cerr := out.Close()
+				if err == nil {
+					err = cerr
+				}
+			}()
+			if _, err = io.Copy(out, in); err != nil {
+				return err
+			}
+			err = out.Sync()
+
 			return nil
 		}