config.go

package config

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

// Supported config keys
const (
	AsyncKey = "async"
	OutputFormatKey = "output_format"
	ProjectIdKey = "project_id"
	RegionKey = "region"
	SessionTimeLimitKey = "session_time_limit"
	VerbosityKey = "verbosity"
	AssumeYesKey = "assume_yes"

	IdentityProviderCustomWellKnownConfigurationKey = "identity_provider_custom_well_known_configuration"
	IdentityProviderCustomClientIdKey = "identity_provider_custom_client_id"
	AllowedUrlDomainKey = "allowed_url_domain"

	AuthorizationCustomEndpointKey = "authorization_custom_endpoint"
	AlbCustomEndpoint = "alb_custom _endpoint"
	DNSCustomEndpointKey = "dns_custom_endpoint"
	EdgeCustomEndpointKey = "edge_custom_endpoint"
	LoadBalancerCustomEndpointKey = "load_balancer_custom_endpoint"
	LogMeCustomEndpointKey = "logme_custom_endpoint"
	MariaDBCustomEndpointKey = "mariadb_custom_endpoint"
	MongoDBFlexCustomEndpointKey = "mongodbflex_custom_endpoint"
	ObjectStorageCustomEndpointKey = "object_storage_custom_endpoint"
	ObservabilityCustomEndpointKey = "observability_custom_endpoint"
	OpenSearchCustomEndpointKey = "opensearch_custom_endpoint"
	PostgresFlexCustomEndpointKey = "postgresflex_custom_endpoint"
	RabbitMQCustomEndpointKey = "rabbitmq_custom_endpoint"
	RedisCustomEndpointKey = "redis_custom_endpoint"
	ResourceManagerEndpointKey = "resource_manager_custom_endpoint"
	SecretsManagerCustomEndpointKey = "secrets_manager_custom_endpoint"
	KMSCustomEndpointKey = "kms_custom_endpoint"
	ServiceAccountCustomEndpointKey = "service_account_custom_endpoint"
	ServiceEnablementCustomEndpointKey = "service_enablement_custom_endpoint"
	ServerBackupCustomEndpointKey = "serverbackup_custom_endpoint"
	ServerOsUpdateCustomEndpointKey = "serverosupdate_custom_endpoint"
	RunCommandCustomEndpointKey = "runcommand_custom_endpoint"
	SfsCustomEndpointKey = "sfs_custom_endpoint"
	SKECustomEndpointKey = "ske_custom_endpoint"
	SQLServerFlexCustomEndpointKey = "sqlserverflex_custom_endpoint"
	IaaSCustomEndpointKey = "iaas_custom_endpoint"
	TokenCustomEndpointKey = "token_custom_endpoint"
	GitCustomEndpointKey = "git_custom_endpoint"
	CDNCustomEndpointKey = "cdn_custom_endpoint"
	IntakeCustomEndpointKey = "intake_custom_endpoint"
	LogsCustomEndpointKey = "logs_custom_endpoint"
	VPNCustomEndpointKey = "vpn_custom_endpoint"

	ProjectNameKey = "project_name"
	DefaultProfileName = "default"

	AsyncDefault = false
	AssumeYesDefault = false
	RegionDefault = "eu01"
	SessionTimeLimitDefault = "12h"

	AllowedUrlDomainDefault = "stackit.cloud"
)

const (
	configFolder = "stackit"

	configFileName = "cli-config"
	configFileExtension = "json"

	profileRootFolder = "profiles"
	profileFileName = "cli-profile"
	profileFileExtension = "txt"
)

var ConfigKeys = []string{
	AsyncKey,
	OutputFormatKey,
	ProjectIdKey,
	RegionKey,
	SessionTimeLimitKey,
	VerbosityKey,
	AssumeYesKey,

	IdentityProviderCustomWellKnownConfigurationKey,
	IdentityProviderCustomClientIdKey,
	AllowedUrlDomainKey,

	AlbCustomEndpoint,
	AuthorizationCustomEndpointKey,
	CDNCustomEndpointKey,
	DNSCustomEndpointKey,
	EdgeCustomEndpointKey,
	GitCustomEndpointKey,
	IaaSCustomEndpointKey,
	IntakeCustomEndpointKey,
	KMSCustomEndpointKey,
	LoadBalancerCustomEndpointKey,
	LogMeCustomEndpointKey,
	LogsCustomEndpointKey,
	MariaDBCustomEndpointKey,
	MongoDBFlexCustomEndpointKey,
	ObjectStorageCustomEndpointKey,
	ObservabilityCustomEndpointKey,
	OpenSearchCustomEndpointKey,
	PostgresFlexCustomEndpointKey,
	RabbitMQCustomEndpointKey,
	RedisCustomEndpointKey,
	ResourceManagerEndpointKey,
	RunCommandCustomEndpointKey,
	SKECustomEndpointKey,
	SQLServerFlexCustomEndpointKey,
	SecretsManagerCustomEndpointKey,
	ServerBackupCustomEndpointKey,
	ServerOsUpdateCustomEndpointKey,
	ServiceAccountCustomEndpointKey,
	ServiceEnablementCustomEndpointKey,
	SfsCustomEndpointKey,
	TokenCustomEndpointKey,
	VPNCustomEndpointKey,
}

var defaultConfigFolderPath string
var configFolderPath string
var profileFilePath string

func InitConfig() {
	initConfig(getInitialConfigDir())
}

func initConfig(configPath string) {
	defaultConfigFolderPath = configPath
	profileFilePath = getInitialProfileFilePath() // Profile file path is in the default config folder

	configProfile, err := GetProfile()
	cobra.CheckErr(err)

	configFolderPath = GetProfileFolderPath(configProfile)

	configFilePath := getConfigFilePath(configFolderPath)

	// This hack is required to allow creating the config file with `viper.WriteConfig`
	// see https://github.com/spf13/viper/issues/851#issuecomment-789393451
	viper.SetConfigFile(configFilePath)
	viper.SetConfigType(configFileExtension)

	f, err := os.Open(configFilePath)
	if !os.IsNotExist(err) {
		if err := viper.ReadConfig(f); err != nil {
			cobra.CheckErr(err)
		}
	}
	defer func() {
		if f != nil {
			if err := f.Close(); err != nil {
				cobra.CheckErr(err)
			}
		}
	}()

	setConfigDefaults()

	viper.AutomaticEnv()
	viper.SetEnvPrefix("stackit")
}

// Write saves the config file (wrapping `viper.WriteConfig`) and ensures that its directory exists
func Write() error {
	err := os.MkdirAll(configFolderPath, 0o750)
	if err != nil {
		return fmt.Errorf("create config directory: %w", err)
	}
	return viper.WriteConfig()
}

// All config keys should be set to a default value so that they can be set as an environment variable
// They will not show in the config list if they are empty
func setConfigDefaults() {
	viper.SetDefault(AsyncKey, AsyncDefault)
	viper.SetDefault(OutputFormatKey, "")
	viper.SetDefault(ProjectIdKey, "")
	viper.SetDefault(RegionKey, RegionDefault)
	viper.SetDefault(SessionTimeLimitKey, SessionTimeLimitDefault)
	viper.SetDefault(IdentityProviderCustomWellKnownConfigurationKey, "")
	viper.SetDefault(IdentityProviderCustomClientIdKey, "")
	viper.SetDefault(AllowedUrlDomainKey, AllowedUrlDomainDefault)
	viper.SetDefault(DNSCustomEndpointKey, "")
	viper.SetDefault(EdgeCustomEndpointKey, "")
	viper.SetDefault(ObservabilityCustomEndpointKey, "")
	viper.SetDefault(AuthorizationCustomEndpointKey, "")
	viper.SetDefault(MongoDBFlexCustomEndpointKey, "")
	viper.SetDefault(ObjectStorageCustomEndpointKey, "")
	viper.SetDefault(OpenSearchCustomEndpointKey, "")
	viper.SetDefault(PostgresFlexCustomEndpointKey, "")
	viper.SetDefault(ResourceManagerEndpointKey, "")
	viper.SetDefault(SecretsManagerCustomEndpointKey, "")
	viper.SetDefault(KMSCustomEndpointKey, "")
	viper.SetDefault(ServiceAccountCustomEndpointKey, "")
	viper.SetDefault(ServiceEnablementCustomEndpointKey, "")
	viper.SetDefault(ServerBackupCustomEndpointKey, "")
	viper.SetDefault(ServerOsUpdateCustomEndpointKey, "")
	viper.SetDefault(RunCommandCustomEndpointKey, "")
	viper.SetDefault(SKECustomEndpointKey, "")
	viper.SetDefault(SQLServerFlexCustomEndpointKey, "")
	viper.SetDefault(IaaSCustomEndpointKey, "")
	viper.SetDefault(TokenCustomEndpointKey, "")
	viper.SetDefault(GitCustomEndpointKey, "")
	viper.SetDefault(IntakeCustomEndpointKey, "")
	viper.SetDefault(AlbCustomEndpoint, "")
	viper.SetDefault(LogsCustomEndpointKey, "")
	viper.SetDefault(CDNCustomEndpointKey, "")
	viper.SetDefault(VPNCustomEndpointKey, "")
}

func getConfigFilePath(configFolder string) string {
	return filepath.Join(configFolder, fmt.Sprintf("%s.%s", configFileName, configFileExtension))
}

func getInitialConfigDir() string {
	configDir, err := os.UserConfigDir()
	cobra.CheckErr(err)

	return filepath.Join(configDir, configFolder)
}

func getInitialProfileFilePath() string {
	configFolderPath := defaultConfigFolderPath
	if configFolderPath == "" {
		configFolderPath = getInitialConfigDir()
	}
	return filepath.Join(configFolderPath, fmt.Sprintf("%s.%s", profileFileName, profileFileExtension))
}