Olares/cli/pkg/terminus/settings.go

package terminus

import (
	"context"
	"fmt"
	"path"
	"time"

	"github.com/beclab/Olares/cli/pkg/core/logger"
	corev1 "k8s.io/api/core/v1"

	"github.com/beclab/Olares/cli/pkg/common"
	cc "github.com/beclab/Olares/cli/pkg/core/common"
	"github.com/beclab/Olares/cli/pkg/core/connector"
	"github.com/beclab/Olares/cli/pkg/core/task"
	"github.com/beclab/Olares/cli/pkg/core/util"
	settingstemplates "github.com/beclab/Olares/cli/pkg/terminus/templates"
	"github.com/beclab/Olares/cli/pkg/utils"
	"github.com/pkg/errors"
	ctrl "sigs.k8s.io/controller-runtime"
)

type SetSettingsValues struct {
	common.KubeAction
}

func (p *SetSettingsValues) Execute(runtime connector.Runtime) error {
	s3SessionToken := "none"
	if p.KubeConf.Arg.Storage.StorageToken != "" {
		s3SessionToken = p.KubeConf.Arg.Storage.StorageToken
	}
	s3AccessKey := "none"
	if p.KubeConf.Arg.Storage.StorageAccessKey != "" {
		s3AccessKey = p.KubeConf.Arg.Storage.StorageAccessKey
	}
	s3SecretKey := "none"
	if p.KubeConf.Arg.Storage.StorageSecretKey != "" {
		s3SecretKey = p.KubeConf.Arg.Storage.StorageSecretKey
	}

	selfhosted := true
	if p.KubeConf.Arg.NetworkSettings.EnableReverseProxy != nil {
		selfhosted = *p.KubeConf.Arg.NetworkSettings.EnableReverseProxy
	} else {
		if p.KubeConf.Arg.NetworkSettings.CloudProviderPublicIP != nil {
			selfhosted = false
		}
	}

	terminusdInstalled := "0"
	if !runtime.GetSystemInfo().IsDarwin() {
		terminusdInstalled = "1"
	}

	var settingsFile = path.Join(runtime.GetInstallerDir(), "wizard", "config", "settings", settingstemplates.SettingsValue.Name())
	var data = util.Data{
		"UserName":           p.KubeConf.Arg.User.UserName,
		"S3SessionToken":     s3SessionToken,
		"S3AccessKey":        s3AccessKey,
		"S3SecretKey":        s3SecretKey,
		"ClusterID":          p.KubeConf.Arg.Storage.StorageClusterId,
		"DomainName":         p.KubeConf.Arg.User.DomainName,
		"SelfHosted":         selfhosted,
		"TerminusdInstalled": terminusdInstalled,
	}

	settingsStr, err := util.Render(settingstemplates.SettingsValue, data)
	if err != nil {
		return errors.Wrap(errors.WithStack(err), "render settings template failed")
	}

	if err := util.WriteFile(settingsFile, []byte(settingsStr), cc.FileMode0644); err != nil {
		return errors.Wrap(errors.WithStack(err), fmt.Sprintf("write settings %s failed", settingsFile))
	}

	return nil
}

type InstallSettings struct {
	common.KubeAction
}

func (t *InstallSettings) Execute(runtime connector.Runtime) error {
	config, err := ctrl.GetConfig()
	if err != nil {
		return err
	}
	ns := corev1.NamespaceDefault
	actionConfig, settings, err := utils.InitConfig(config, ns)
	if err != nil {
		return err
	}

	var ctx, cancel = context.WithTimeout(context.Background(), 3*time.Minute)
	defer cancel()

	var settingsPath = path.Join(runtime.GetInstallerDir(), "wizard", "config", "settings")
	if !util.IsExist(settingsPath) {
		return fmt.Errorf("settings not exists")
	}

	if err := utils.UpgradeCharts(ctx, actionConfig, settings, common.ChartNameSettings, settingsPath, "", ns, nil, false); err != nil {
		return err
	}

	return nil
}

type InstallSettingsModule struct {
	common.KubeModule
}

func (m *InstallSettingsModule) Init() {
	logger.InfoInstallationProgress("Installing settings ...")
	m.Name = "InstallSettings"

	detectPublicIPAddress := &task.LocalTask{
		Name:   "DetectPublicIPAddress",
		Action: new(DetectPublicIPAddress),
		Retry:  3,
	}

	setSettingsValues := &task.LocalTask{
		Name:   "SetSettingsValues",
		Action: new(SetSettingsValues),
		Retry:  1,
	}

	installSettings := &task.LocalTask{
		Name:   "InstallSettings",
		Action: new(InstallSettings),
		Retry:  1,
	}

	m.Tasks = []task.Interface{
		detectPublicIPAddress,
		setSettingsValues,
		installSettings,
	}
}