resource_metakube_cluster_structure.go

package metakube

import (
	"github.com/syseleven/go-metakube/models"
)

// flatteners

func metakubeResourceClusterFlattenSpec(values clusterPreserveValues, in *models.ClusterSpec) []interface{} {
	if in == nil {
		return []interface{}{}
	}

	att := make(map[string]interface{})

	if in.Version != nil {
		att["version"] = in.Version
	}

	if in.UpdateWindow != nil && (in.UpdateWindow.Start != "" || in.UpdateWindow.Length != "") {
		att["update_window"] = flattenUpdateWindow(in.UpdateWindow)
	}

	att["enable_ssh_agent"] = in.EnableUserSSHKeyAgent

	if len(in.MachineNetworks) > 0 {
		att["machine_networks"] = flattenMachineNetworks(in.MachineNetworks)
	}

	att["audit_logging"] = false
	if in.AuditLogging != nil {
		att["audit_logging"] = in.AuditLogging.Enabled
	}

	att["pod_security_policy"] = in.UsePodSecurityPolicyAdmissionPlugin

	att["pod_node_selector"] = in.UsePodNodeSelectorAdmissionPlugin

	if network := in.ClusterNetwork; network != nil {
		if network.DNSDomain != "" {
			att["domain_name"] = network.DNSDomain
		}
		if v := network.Pods; len(v.CIDRBlocks) > 0 && v.CIDRBlocks[0] != "" {
			att["pods_cidr"] = v.CIDRBlocks[0]
		}
		if v := network.Services; len(v.CIDRBlocks) > 0 && v.CIDRBlocks[0] != "" {
			att["services_cidr"] = v.CIDRBlocks[0]
		}
	}

	if in.Cloud != nil {
		att["cloud"] = flattenClusterCloudSpec(values, in.Cloud)
	}

	if in.Sys11auth != nil {
		att["syseleven_auth"] = flattenClusterSys11Auth(in.Sys11auth)
	}

	return []interface{}{att}
}

func metakubeResourceClusterFlattenSys11AuthClusterRoleBindings(in []models.ClusterRoleBinding) []interface{} {
	return nil
}

func metakubeResourceClusterFlattenSys11AuthRoleBindings(in []models.ClusterRoleBinding) []interface{} {
	return nil
}

func flattenUpdateWindow(in *models.UpdateWindow) []interface{} {
	m := make(map[string]interface{})
	m["start"] = in.Start
	m["length"] = in.Length
	return []interface{}{m}
}

func flattenMachineNetworks(in []*models.MachineNetworkingConfig) []interface{} {
	if len(in) < 1 {
		return []interface{}{}
	}

	att := make([]interface{}, len(in))

	for i, v := range in {
		m := make(map[string]interface{})

		if v.CIDR != "" {
			m["cidr"] = v.CIDR
		}
		if v.Gateway != "" {
			m["gateway"] = v.Gateway
		}
		if l := len(v.DNSServers); l > 0 {
			ds := make([]interface{}, l)
			for i, s := range v.DNSServers {
				ds[i] = s
			}
			m["dns_servers"] = ds
		}
		att[i] = m
	}

	return att
}

func flattenClusterCloudSpec(values clusterPreserveValues, in *models.CloudSpec) []interface{} {
	if in == nil {
		return []interface{}{}
	}

	att := make(map[string]interface{})

	if in.Aws != nil {
		att["aws"] = flattenAWSCloudSpec(values.aws)
	}

	if in.Openstack != nil {
		att["openstack"] = flattenOpenstackSpec(values.openstack, in.Openstack)
	}

	if in.Azure != nil {
		att["azure"] = flattenAzureSpec(values.azure)
	}

	return []interface{}{att}
}

func flattenClusterSys11Auth(in *models.Sys11AuthSettings) []interface{} {
	if in == nil || in.Realm == "" {
		return nil
	}

	return []interface{}{map[string]interface{}{
		"realm": in.Realm,
	}}
}

func flattenAWSCloudSpec(in *models.AWSCloudSpec) []interface{} {
	if in == nil {
		return []interface{}{}
	}

	att := make(map[string]interface{})

	if in.AccessKeyID != "" {
		att["access_key_id"] = in.AccessKeyID
	}

	if in.SecretAccessKey != "" {
		att["secret_access_key"] = in.SecretAccessKey
	}

	if in.VPCID != "" {
		att["vpc_id"] = in.VPCID
	}

	if in.SecurityGroupID != "" {
		att["security_group_id"] = in.SecurityGroupID
	}

	if in.InstanceProfileName != "" {
		att["instance_profile_name"] = in.InstanceProfileName
	}

	if in.ControlPlaneRoleARN != "" {
		att["role_arn"] = in.ControlPlaneRoleARN
	}

	if in.OpenstackBillingTenant != "" {
		att["openstack_billing_tenant"] = in.OpenstackBillingTenant
	}

	if in.RouteTableID != "" {
		att["route_table_id"] = in.RouteTableID
	}

	return []interface{}{att}
}

func flattenOpenstackSpec(values *clusterOpenstackPreservedValues, in *models.OpenstackCloudSpec) []interface{} {
	if in == nil {
		return []interface{}{}
	}

	att := make(map[string]interface{})

	if in.FloatingIPPool != "" {
		att["floating_ip_pool"] = in.FloatingIPPool
	}

	if in.SecurityGroups != "" {
		att["security_group"] = in.SecurityGroups
	}

	if in.Network != "" {
		att["network"] = in.Network
	}

	if in.SubnetID != "" {
		att["subnet_id"] = in.SubnetID
	}

	if in.SubnetCIDR != "" {
		att["subnet_cidr"] = in.SubnetCIDR
	}

	if in.ServerGroupID != "" {
		att["server_group_id"] = in.ServerGroupID
	}

	if values != nil {
		if _, ok := att["server_group_id"]; !ok && values.openstackServerGroupID != nil {
			att["server_group_id"] = values.openstackServerGroupID
		}
		if values.openstackTenant != nil {
			att["tenant"] = values.openstackTenant
		}
		if values.openstackUsername != nil {
			att["username"] = values.openstackUsername
		}
		if values.openstackPassword != nil {
			att["password"] = values.openstackPassword
		}
		if values.openstackApplicationCredentialsID != nil {
			att["application_credentials_id"] = values.openstackApplicationCredentialsID
		}
		if values.openstackApplicationCredentialsSecret != nil {
			att["application_credentials_secret"] = values.openstackApplicationCredentialsSecret
		}
	}

	return []interface{}{att}
}

func flattenAzureSpec(in *models.AzureCloudSpec) []interface{} {
	if in == nil {
		return []interface{}{}
	}

	// API returns empty spec for Azure clusters, so we just preserve values used for creation

	att := make(map[string]interface{})

	if in.AvailabilitySet != "" {
		att["availability_set"] = in.AvailabilitySet
	}

	if in.ClientID != "" {
		att["client_id"] = in.ClientID
	}

	if in.ClientSecret != "" {
		att["client_secret"] = in.ClientSecret
	}

	if in.SubscriptionID != "" {
		att["subscription_id"] = in.SubscriptionID
	}

	if in.TenantID != "" {
		att["tenant_id"] = in.TenantID
	}

	if in.ResourceGroup != "" {
		att["resource_group"] = in.ResourceGroup
	}

	if in.RouteTableName != "" {
		att["route_table"] = in.RouteTableName
	}

	if in.OpenstackBillingTenant != "" {
		att["openstack_billing_tenant"] = in.OpenstackBillingTenant
	}

	if in.SecurityGroup != "" {
		att["security_group"] = in.SecurityGroup
	}

	if in.SubnetName != "" {
		att["subnet"] = in.SubnetName
	}

	if in.VNetName != "" {
		att["vnet"] = in.VNetName
	}

	return []interface{}{att}
}

// expanders

func metakubeResourceClusterExpandSpec(p []interface{}, dcName string) *models.ClusterSpec {
	if len(p) < 1 {
		return nil
	}
	obj := &models.ClusterSpec{}
	if p[0] == nil {
		return obj
	}
	in := p[0].(map[string]interface{})

	if v, ok := in["version"]; ok {
		if vv, ok := v.(string); ok {
			obj.Version = vv
		}
	}

	if v, ok := in["update_window"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.UpdateWindow = expandUpdateWindow(vv)
		}
	}

	if v, ok := in["enable_ssh_agent"]; ok {
		if vv, ok := v.(bool); ok {
			obj.EnableUserSSHKeyAgent = vv
		}
	}

	if v, ok := in["machine_networks"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.MachineNetworks = expandMachineNetworks(vv)
		}
	}

	if v, ok := in["audit_logging"]; ok {
		if vv, ok := v.(bool); ok {
			obj.AuditLogging = expandAuditLogging(vv)
		}
	}

	if v, ok := in["pod_security_policy"]; ok {
		if vv, ok := v.(bool); ok {
			obj.UsePodSecurityPolicyAdmissionPlugin = vv
		}
	}

	if v, ok := in["pod_node_selector"]; ok {
		if vv, ok := v.(bool); ok {
			obj.UsePodNodeSelectorAdmissionPlugin = vv
		}
	}

	if v, ok := in["services_cidr"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			if obj.ClusterNetwork == nil {
				obj.ClusterNetwork = &models.ClusterNetworkingConfig{}
			}
			obj.ClusterNetwork.Services = &models.NetworkRanges{
				CIDRBlocks: []string{vv},
			}
		}
	}

	if v, ok := in["pods_cidr"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			if obj.ClusterNetwork == nil {
				obj.ClusterNetwork = &models.ClusterNetworkingConfig{}
			}
			obj.ClusterNetwork.Pods = &models.NetworkRanges{
				CIDRBlocks: []string{vv},
			}
		}
	}

	if v, ok := in["domain_name"]; ok {
		if vv, ok := v.(string); ok && v != "" {
			if obj.ClusterNetwork == nil {
				obj.ClusterNetwork = &models.ClusterNetworkingConfig{}
			}
			obj.ClusterNetwork.DNSDomain = vv
		}
	}

	if v, ok := in["cloud"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.Cloud = expandClusterCloudSpec(vv, dcName)
		}
	}

	if v, ok := in["syseleven_auth"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.Sys11auth = expandClusterSys11Auth(vv)
		}
	}

	return obj
}

func expandUpdateWindow(p []interface{}) *models.UpdateWindow {
	if len(p) < 1 {
		return nil
	}

	m := p[0].(map[string]interface{})
	ret := new(models.UpdateWindow)
	if v, ok := m["start"]; ok {
		ret.Start = v.(string)
	}
	if v, ok := m["length"]; ok {
		ret.Length = v.(string)
	}
	return ret
}

func expandMachineNetworks(p []interface{}) []*models.MachineNetworkingConfig {
	if len(p) < 1 {
		return nil
	}
	var machines []*models.MachineNetworkingConfig
	for _, elem := range p {
		in := elem.(map[string]interface{})
		obj := &models.MachineNetworkingConfig{}

		if v, ok := in["cidr"]; ok {
			if vv, ok := v.(string); ok && v != "" {
				obj.CIDR = vv
			}
		}

		if v, ok := in["gateway"]; ok {
			if vv, ok := v.(string); ok && v != "" {
				obj.Gateway = vv
			}
		}

		if v, ok := in["dns_servers"]; ok {
			if vv, ok := v.([]interface{}); ok {
				for _, s := range vv {
					if ss, ok := s.(string); ok && s != "" {
						obj.DNSServers = append(obj.DNSServers, ss)
					}
				}
			}
		}

		machines = append(machines, obj)
	}

	return machines
}

func expandAuditLogging(enabled bool) *models.AuditLoggingSettings {
	return &models.AuditLoggingSettings{
		Enabled: enabled,
	}
}

func expandClusterCloudSpec(p []interface{}, dcName string) *models.CloudSpec {
	if len(p) < 1 {
		return nil
	}
	obj := &models.CloudSpec{}
	if p[0] == nil {
		return obj
	}
	in := p[0].(map[string]interface{})

	obj.DatacenterName = dcName

	if v, ok := in["aws"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.Aws = expandAWSCloudSpec(vv)
		}
	}

	if v, ok := in["openstack"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.Openstack = expandOpenstackCloudSpec(vv)
		}
	}

	if v, ok := in["azure"]; ok {
		if vv, ok := v.([]interface{}); ok {
			obj.Azure = expandAzureCloudSpec(vv)
		}
	}

	return obj
}

func expandClusterSys11Auth(p []interface{}) *models.Sys11AuthSettings {
	if len(p) < 1 {
		return nil
	}
	if p[0] == nil {
		return nil
	}
	in := p[0].(map[string]interface{})
	if v := in["realm"].(string); v != "" {
		return &models.Sys11AuthSettings{Realm: v}
	}
	return nil
}

func expandAWSCloudSpec(p []interface{}) *models.AWSCloudSpec {
	if len(p) < 1 {
		return nil
	}
	obj := &models.AWSCloudSpec{}
	if p[0] == nil {
		return obj
	}
	in := p[0].(map[string]interface{})

	if v, ok := in["access_key_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.AccessKeyID = vv
		}
	}

	if v, ok := in["secret_access_key"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SecretAccessKey = vv
		}
	}

	if v, ok := in["vpc_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.VPCID = vv
		}
	}

	if v, ok := in["security_group_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SecurityGroupID = vv
		}
	}

	if v, ok := in["instance_profile_name"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.InstanceProfileName = vv
		}
	}

	if v, ok := in["role_arn"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ControlPlaneRoleARN = vv
		}
	}

	if v, ok := in["openstack_billing_tenant"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.OpenstackBillingTenant = vv
		}
	}

	if v, ok := in["route_table_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.RouteTableID = vv
		}
	}

	return obj
}

func expandOpenstackCloudSpec(p []interface{}) *models.OpenstackCloudSpec {
	if len(p) < 1 {
		return nil
	}

	obj := &models.OpenstackCloudSpec{}
	if p[0] == nil {
		return obj
	}
	in := p[0].(map[string]interface{})

	if v, ok := in["tenant"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.Tenant = vv
		}
	}

	if v, ok := in["floating_ip_pool"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.FloatingIPPool = vv
		}
	}

	if v, ok := in["security_group"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SecurityGroups = vv
		}
	}

	if v, ok := in["network"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.Network = vv
		}
	}

	if v, ok := in["subnet_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SubnetID = vv
		}
	}

	if v, ok := in["subnet_cidr"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SubnetCIDR = vv
		}
	}

	if v, ok := in["server_group_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ServerGroupID = vv
		}
	}

	if v, ok := in["application_credentials_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ApplicationCredentialID = vv
		}
	}

	if v, ok := in["application_credentials_secret"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ApplicationCredentialSecret = vv
		}
	}

	if v, ok := in["username"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.Username = vv
		}
	}

	if v, ok := in["password"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.Password = vv
		}
	}

	// HACK(furkhat): API doesn't return domain for cluster. Use 'Default' all the time.
	obj.Domain = "Default"

	return obj
}

func expandAzureCloudSpec(p []interface{}) *models.AzureCloudSpec {
	if len(p) < 1 {
		return nil
	}

	obj := &models.AzureCloudSpec{}

	if p[0] == nil {
		return obj
	}

	in := p[0].(map[string]interface{})

	if v, ok := in["availability_set"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.AvailabilitySet = vv
		}
	}

	if v, ok := in["client_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ClientID = vv
		}
	}

	if v, ok := in["client_secret"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ClientSecret = vv
		}
	}

	if v, ok := in["subscription_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SubscriptionID = vv
		}
	}

	if v, ok := in["tenant_id"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.TenantID = vv
		}
	}

	if v, ok := in["resource_group"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.ResourceGroup = vv
		}
	}

	if v, ok := in["route_table"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.RouteTableName = vv
		}
	}

	if v, ok := in["openstack_billing_tenant"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.OpenstackBillingTenant = vv
		}
	}

	if v, ok := in["security_group"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SecurityGroup = vv
		}
	}

	if v, ok := in["subnet"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.SubnetName = vv
		}
	}

	if v, ok := in["vnet"]; ok {
		if vv, ok := v.(string); ok && vv != "" {
			obj.VNetName = vv
		}
	}

	return obj
}