clusterspec.go

package clusterspec

import (
	"context"
	"fmt"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
	"strconv"
)

type ClusterSpec struct {
	Name                      string
	ApiProvider               string
	CloudProvider             string
	Type                      string
	KubernetesVersion         string
	NodeType                  string
	NodeCount                 int
	MasterNodeCount           int
	Region                    string
	PodCidrBlock              string
	SshKeyName                string
	Tags                      string
	Description               string
	ClusterAutoscalerEnabled  bool
	ClusterAutoscalerMinNodes int
	ClusterAutoscalerMaxNodes int
}

const (
	ApiProviderKey               = "apiProvider"
	CloudProviderKey             = "cloudProvider"
	NodeTypeKey                  = "nodeType"
	ClusterTypeKey               = "type"
	KubernetesVersionKey         = "kubernetesVersion"
	NodeCountKey                 = "nodeCount"
	MasterNodeCountKey           = "masterNodeCount"
	RegionKey                    = "region"
	PodCidrBlockKey              = "podCidrBlock"
	SshKeyNameKey                = "sshKeyName"
	TagsKey                      = "tags"
	DescriptionKey               = "description"
	ClusterAutoscalerEnabledKey  = "clusterAutoscalerEnabled"
	ClusterAutoscalerMinNodesKey = "clusterAutoscalerMinNodes"
	ClusterAutoscalerMaxNodesKey = "clusterAutoscalerMaxNodes"

	defaultMasterNodeCount           = 3
	defaultClusterAutoscalerEnabled  = false
	defaultClusterAutoscalerMinNodes = 1
	defaultClusterAutoscalerMaxNodes = 9
)

var (
	ValidHelmParamKeys = []string{
		RegionKey,
		SshKeyNameKey,
		KubernetesVersionKey,
		PodCidrBlockKey,
		NodeCountKey,
		MasterNodeCountKey,
		NodeTypeKey,
		ClusterAutoscalerEnabledKey,
		ClusterAutoscalerMinNodesKey,
		ClusterAutoscalerMaxNodesKey,
	}
)

func Get(
	kubeClient *kubernetes.Clientset,
	arlonNs string,
	specName string,
) (cs *ClusterSpec, err error) {
	client := kubeClient.CoreV1()
	configMapApi := client.ConfigMaps(arlonNs)
	cm, err := configMapApi.Get(context.Background(), specName, metav1.GetOptions{})
	if err != nil {
		return nil, err
	}
	cs, err = FromConfigMap(cm)
	if err != nil {
		return nil, err
	}
	return
}

func FromConfigMap(cm *corev1.ConfigMap) (*ClusterSpec, error) {
	cs := &ClusterSpec{
		Name:              cm.Name,
		ApiProvider:       cm.Data[ApiProviderKey],
		CloudProvider:     cm.Data[CloudProviderKey],
		Type:              cm.Data[ClusterTypeKey],
		KubernetesVersion: cm.Data[KubernetesVersionKey],
		NodeType:          cm.Data[NodeTypeKey],
		Region:            cm.Data[RegionKey],
		PodCidrBlock:      cm.Data[PodCidrBlockKey],
		SshKeyName:        cm.Data[SshKeyNameKey],
		Tags:              cm.Data[TagsKey],
		Description:       cm.Data[DescriptionKey],
	}
	var err error
	cs.NodeCount, err = strconv.Atoi(cm.Data[NodeCountKey])
	if err != nil {
		return nil, fmt.Errorf("could not parse clusterspec nodeCount: %s", err)
	}
	cs.MasterNodeCount, err = strconv.Atoi(cm.Data[MasterNodeCountKey])
	if err != nil {
		cs.MasterNodeCount = defaultMasterNodeCount // for backward compatibility if not present
	}
	cs.ClusterAutoscalerEnabled, err = strconv.ParseBool(cm.Data[ClusterAutoscalerEnabledKey])
	if err != nil {
		cs.ClusterAutoscalerEnabled = defaultClusterAutoscalerEnabled
	}
	cs.ClusterAutoscalerMinNodes, err = strconv.Atoi(cm.Data[ClusterAutoscalerMinNodesKey])
	if err != nil {
		cs.ClusterAutoscalerMinNodes = defaultClusterAutoscalerMinNodes
	}
	cs.ClusterAutoscalerMaxNodes, err = strconv.Atoi(cm.Data[ClusterAutoscalerMaxNodesKey])
	if err != nil {
		cs.ClusterAutoscalerMaxNodes = defaultClusterAutoscalerMaxNodes
	}
	return cs, nil
}

func ToConfigMap(
	name string,
	apiProvider string,
	cloudProvider string,
	clusterType string,
	kubernetesVersion string,
	nodeType string,
	nodeCount int,
	masterNodeCount int,
	region string,
	podCidrBlock string,
	sshKeyName string,
	clusterAutoscalerEnabled bool,
	clusterAutoscalerMinNodes int,
	clusterAutoscalerMaxNodes int,
	tags string,
	description string,
) *corev1.ConfigMap {
	return &corev1.ConfigMap{
		ObjectMeta: metav1.ObjectMeta{
			Name: name,
			Labels: map[string]string{
				"managed-by": "arlon",
				"arlon-type": "clusterspec",
			},
		},
		Data: map[string]string{
			ApiProviderKey:               apiProvider,
			CloudProviderKey:             cloudProvider,
			ClusterTypeKey:               clusterType,
			KubernetesVersionKey:         kubernetesVersion,
			NodeTypeKey:                  nodeType,
			NodeCountKey:                 strconv.Itoa(nodeCount),
			MasterNodeCountKey:           strconv.Itoa(masterNodeCount),
			RegionKey:                    region,
			PodCidrBlockKey:              podCidrBlock,
			SshKeyNameKey:                sshKeyName,
			ClusterAutoscalerEnabledKey:  strconv.FormatBool(clusterAutoscalerEnabled),
			ClusterAutoscalerMinNodesKey: strconv.Itoa(clusterAutoscalerMinNodes),
			ClusterAutoscalerMaxNodesKey: strconv.Itoa(clusterAutoscalerMaxNodes),
			TagsKey:                      tags,
			DescriptionKey:               description,
		},
	}
}

func SubchartName(cm *corev1.ConfigMap) (string, error) {
	cs, err := FromConfigMap(cm)
	if err != nil {
		return "", err
	}
	return SubchartNameFromClusterSpec(cs)
}

func SubchartNameFromClusterSpec(cs *ClusterSpec) (string, error) {
	if err := ValidApiProvider(cs.ApiProvider); err != nil {
		return "", err
	}
	if err := ValidCloudProviderAndClusterType(cs.CloudProvider, cs.Type); err != nil {
		return "", err
	}
	return fmt.Sprintf("%s-%s-%s", cs.ApiProvider,
		cs.CloudProvider, cs.Type), nil
}

func ClusterAutoscalerSubchartNameFromClusterSpec(cs *ClusterSpec) (string, error) {
	if err := ValidApiProvider(cs.ApiProvider); err != nil {
		return "", err
	}
	return ClusterAutoscalerSubchartNameFromApiProvider(cs.ApiProvider), nil
}

func ClusterAutoscalerSubchartNameFromApiProvider(apiProvider string) string {
	return fmt.Sprintf("%s-cluster-autoscaler", apiProvider)
}