mirror of
https://github.com/rancher/k3k.git
synced 2026-05-11 03:47:46 +00:00
ba1648e560
* first commit * refactor: update golangci configuration and reorder imports in namespace.go * refactor: update golangci configuration for linters and formatters * check restart for logs fetch * rverted a bit the structure * requested changes * WriteLogs rename
106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package container
|
|
|
|
import (
|
|
"maps"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"helm.sh/helm/v3/pkg/action"
|
|
"helm.sh/helm/v3/pkg/chart/loader"
|
|
|
|
fwclient "github.com/rancher/k3k/tests/framework/client"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
// HelmInstaller provides configuration for Helm chart installation.
|
|
type HelmInstaller struct {
|
|
ChartPath string
|
|
Namespace string
|
|
ReleaseName string
|
|
Timeout time.Duration
|
|
Wait bool
|
|
ControllerImage string
|
|
KubeletImage string
|
|
KubeconfigPath string
|
|
}
|
|
|
|
// InstallK3kChart installs the k3k Helm chart with the specified configuration.
|
|
// It uses the provided RESTClientGetter for authentication and returns the Helm action configuration.
|
|
func (h *HelmInstaller) InstallK3kChart(restClientGetter *fwclient.RESTClientGetter) *action.Configuration {
|
|
GinkgoHelper()
|
|
|
|
// Load chart
|
|
k3kChart, err := loader.Load(h.ChartPath)
|
|
Expect(err).To(Not(HaveOccurred()))
|
|
|
|
// Initialize Helm action configuration
|
|
helmActionConfig := new(action.Configuration)
|
|
|
|
err = helmActionConfig.Init(restClientGetter, h.Namespace, os.Getenv("HELM_DRIVER"), func(format string, v ...any) {
|
|
GinkgoWriter.Printf("[Helm] "+format+"\n", v...)
|
|
})
|
|
Expect(err).To(Not(HaveOccurred()))
|
|
|
|
// Create install action
|
|
iCli := action.NewInstall(helmActionConfig)
|
|
iCli.ReleaseName = h.ReleaseName
|
|
iCli.Namespace = h.Namespace
|
|
iCli.CreateNamespace = true
|
|
iCli.Timeout = h.Timeout
|
|
iCli.Wait = h.Wait
|
|
|
|
// Configure controller image
|
|
controllerMap, _ := k3kChart.Values["controller"].(map[string]any)
|
|
|
|
extraEnvArray, _ := controllerMap["extraEnv"].([]map[string]any)
|
|
extraEnvArray = append(extraEnvArray, map[string]any{
|
|
"name": "DEBUG",
|
|
"value": "true",
|
|
})
|
|
controllerMap["extraEnv"] = extraEnvArray
|
|
|
|
imageMap, _ := controllerMap["image"].(map[string]any)
|
|
maps.Copy(imageMap, map[string]any{
|
|
"repository": h.ControllerImage,
|
|
"tag": "dev",
|
|
"pullPolicy": "IfNotPresent",
|
|
})
|
|
|
|
// Configure agent image
|
|
agentMap, _ := k3kChart.Values["agent"].(map[string]any)
|
|
sharedAgentMap, _ := agentMap["shared"].(map[string]any)
|
|
sharedAgentImageMap, _ := sharedAgentMap["image"].(map[string]any)
|
|
maps.Copy(sharedAgentImageMap, map[string]any{
|
|
"repository": h.KubeletImage,
|
|
"tag": "dev",
|
|
})
|
|
|
|
// Install chart
|
|
release, err := iCli.Run(k3kChart, k3kChart.Values)
|
|
Expect(err).To(Not(HaveOccurred()))
|
|
|
|
GinkgoWriter.Printf("Helm release '%s' installed in '%s' namespace\n", release.Name, release.Namespace)
|
|
|
|
return helmActionConfig
|
|
}
|
|
|
|
// NewHelmInstaller creates a new HelmInstaller with default values.
|
|
func NewHelmInstaller(controllerImage, kubeletImage, kubeconfigPath string) *HelmInstaller {
|
|
pwd, err := os.Getwd()
|
|
Expect(err).To(Not(HaveOccurred()))
|
|
|
|
return &HelmInstaller{
|
|
ChartPath: path.Join(pwd, "../../charts/k3k"),
|
|
Namespace: "k3k-system",
|
|
ReleaseName: "k3k",
|
|
Timeout: time.Minute,
|
|
Wait: true,
|
|
ControllerImage: controllerImage,
|
|
KubeletImage: kubeletImage,
|
|
KubeconfigPath: kubeconfigPath,
|
|
}
|
|
}
|