cloudfoundry/stratos

package config

import (
    "errors"
    "fmt"
    "net/url"
    "reflect"

    "github.com/cloudfoundry-incubator/stratos/src/jetstream/repository/interfaces/config"

    "gopkg.in/yaml.v2"
)

type KubeConfigClusterDetail struct {
    Server string `yaml:"server"`
}

type KubeConfigCluster struct {
    Name    string `yaml:"name"`
    Cluster struct {
        Server string
    }
}

type KubeConfigUser struct {
    Name string `yaml:"name"`
    User struct {
        AuthProvider struct {
            Name   string                 `yaml:"name"`
            Config map[string]interface{} `yaml:"config"`
        } `yaml:"auth-provider,omitempty"`
        ClientCertificate string `yaml:"client-certificate-data,omitempty"`
        ClientKeyData     string `yaml:"client-key-data,omitempty"`
        Token             string `yaml:"token,omitempty"`
    }
}

//ExtraScopes string `yaml:"extra-scopes"`

type KubeConfigContexts struct {
    Name    string `yaml:"name"`
    Context struct {
        Cluster string
        User    string
    } `yaml:"context"`
}

type KubeConfigFile struct {
    ApiVersion     string               `yaml:"apiVersion"`
    Kind           string               `yaml:"kind"`
    Clusters       []KubeConfigCluster  `yaml:"clusters"`
    Users          []KubeConfigUser     `yaml:"users"`
    Contexts       []KubeConfigContexts `yaml:"contexts"`
    CurrentContext string               `yaml:"current-context"`
}

func (k *KubeConfigFile) GetClusterByAPIEndpoint(endpoint string) (*KubeConfigCluster, error) {
    for _, cluster := range k.Clusters {
        if compareURL(cluster.Cluster.Server, endpoint) {
            return &cluster, nil
        }
    }
    return nil, fmt.Errorf("Unable to find cluster")
}

func (k *KubeConfigFile) GetClusterByName(name string) (*KubeConfigCluster, error) {
    for _, cluster := range k.Clusters {
        if cluster.Name == name {
            return &cluster, nil
        }
    }
    return nil, fmt.Errorf("Unable to find cluster")
}

func (k *KubeConfigFile) GetClusterContext(clusterName string) (*KubeConfigContexts, error) {
    for _, context := range k.Contexts {
        if context.Context.Cluster == clusterName {
            return &context, nil
        }
    }
    return nil, fmt.Errorf("Unable to find context")
}

func (k *KubeConfigFile) GetContext(contextName string) (*KubeConfigContexts, error) {
    for _, context := range k.Contexts {
        if context.Name == contextName {
            return &context, nil
        }
    }
    return nil, fmt.Errorf("Unable to find context")
}

func (k *KubeConfigFile) GetUser(userName string) (*KubeConfigUser, error) {
    for _, user := range k.Users {
        if user.Name == userName {
            return &user, nil
        }
    }
    return nil, fmt.Errorf("Unable to find user")
}

func (k *KubeConfigFile) GetUserForCluster(clusterEndpoint string) (*KubeConfigUser, error) {

    var cluster *KubeConfigCluster
    var err error

    // Check to see if the current-context is for this endpoint, before going a search through all contexts
    if len(k.CurrentContext) > 0 {
        currentContext, err := k.GetContext(k.CurrentContext)
        if err == nil {
            c, err := k.GetClusterByName(currentContext.Context.Cluster)
            if err == nil {
                if compareURL(c.Cluster.Server, clusterEndpoint) {
                    // Cluster refrences the same Kube API Server
                    cluster = c
                }
            }
        }
    }

    if cluster == nil {
        cluster, err = k.GetClusterByAPIEndpoint(clusterEndpoint)
        if err != nil {
            return nil, errors.New("Unable to find cluster in kubeconfig")
        }
    }

    clusterName := cluster.Name
    if clusterName == "" {
        return nil, errors.New("Unable to find cluster")
    }

    context, err := k.GetClusterContext(clusterName)
    if err != nil {
        return nil, errors.New("Unable to find cluster context")
    }

    kubeConfigUser, err := k.GetUser(context.Context.User)
    if err != nil {
        return nil, errors.New("Can not find config for Kubernetes cluster")
    }

    return kubeConfigUser, nil
}

func ParseKubeConfig(kubeConfigData []byte) (*KubeConfigFile, error) {

    kubeConfig := &KubeConfigFile{}
    err := yaml.Unmarshal(kubeConfigData, &kubeConfig)
    if err != nil {
        return nil, err
    }
    if kubeConfig.ApiVersion != "v1" || kubeConfig.Kind != "Config" {
        return nil, errors.New("Not a valid Kubernetes Config file")
    }

    return kubeConfig, nil
}

func UnMarshalHelper(values map[string]interface{}, intf interface{}) error {

    value := reflect.ValueOf(intf)

    if value.Kind() != reflect.Ptr {
        return errors.New("config: must provide pointer to struct value")
    }

    value = value.Elem()
    if value.Kind() != reflect.Struct {
        return errors.New("config: must provide pointer to struct value")
    }

    nFields := value.NumField()
    typ := value.Type()

    for i := 0; i < nFields; i++ {
        field := value.Field(i)
        strField := typ.Field(i)
        tag := strField.Tag.Get("yaml")
        if tag == "" {
            continue
        }

        if tagValue, ok := values[tag].(string); ok {
            if err := config.SetStructFieldValue(value, field, tagValue); err != nil {
                return err
            }
        }
    }

    return nil
}

// Compare two URLs, taking into account default HTTP/HTTPS ports and ignoring query string
func compareURL(a, b string) bool {

    ua, err := url.Parse(a)
    if err != nil {
        return false
    }

    ub, err := url.Parse(b)
    if err != nil {
        return false
    }

    aPort := getPort(ua)
    bPort := getPort(ub)
    return ua.Scheme == ub.Scheme && ua.Hostname() == ub.Hostname() && aPort == bPort && ua.Path == ub.Path
}

func getPort(u *url.URL) string {
    port := u.Port()
    if len(port) == 0 {
        switch u.Scheme {
        case "http":
            port = "80"
        case "https":
            port = "443"
        default:
            port = ""
        }
    }

    return port
}