Files
truecharts/clustertool/pkg/gencmd/upgrade.go
T
2024-10-19 15:18:05 +02:00

54 lines
1.5 KiB
Go

package gencmd
import (
"io"
"os"
"strings"
"github.com/rs/zerolog/log"
talhelperCfg "github.com/budimanjojo/talhelper/v3/pkg/config"
"github.com/budimanjojo/talhelper/v3/pkg/generate"
"github.com/truecharts/public/clustertool/embed"
"github.com/truecharts/public/clustertool/pkg/helper"
)
func GenUpgrade(node string, extraFlags []string) []string {
cfg, err := talhelperCfg.LoadAndValidateFromFile(helper.TalConfigFile, []string{helper.ClusterEnvFile}, false)
if err != nil {
log.Fatal().Err(err).Msgf("failed to parse talconfig or talenv file: %s", err)
}
upgradeStdout := os.Stdout
r, w, _ := os.Pipe()
os.Stdout = w
extraFlags = append(extraFlags, "--preserve")
err = generate.GenerateUpgradeCommand(cfg, helper.TalosGenerated, node, extraFlags)
w.Close()
out, _ := io.ReadAll(r)
os.Stdout = upgradeStdout
sliceOut := strings.Split(string(out), ";\n")
talosPath := embed.GetTalosExec()
var slice []string
for _, str := range sliceOut {
if str != "" {
str = strings.ReplaceAll(str, "talosctl", talosPath)
slice = append(slice, str)
}
}
if err != nil {
log.Fatal().Err(err).Msgf("failed to generate talosctl upgrade command: %s", err)
}
return slice
}
func GenKubeUpgrade(node string) string {
talosPath := embed.GetTalosExec()
strout := talosPath + " upgrade-k8s --talosconfig " + helper.TalosConfigFile + " -n " + node
return strout
}