config.go

package kube

import (
	"fmt"

	apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/client-go/dynamic"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/kubernetes/scheme"
	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"sigs.k8s.io/controller-runtime/pkg/client"

	"github.com/kudobuilder/kudo/pkg/apis"
	"github.com/kudobuilder/kudo/pkg/client/clientset/versioned"
	"github.com/kudobuilder/kudo/pkg/kudoctl/clog"
)

// Client provides access different K8S clients
type Client struct {
	Scheme        *runtime.Scheme
	KubeClient    kubernetes.Interface
	ExtClient     apiextensionsclient.Interface
	DynamicClient dynamic.Interface
	CtrlClient    client.Client
	KudoClient    versioned.Interface
}

// GetConfig returns a Kubernetes client config for a given kubeconfig.
func GetConfig(kubeconfig string) clientcmd.ClientConfig {
	rules := clientcmd.NewDefaultClientConfigLoadingRules()
	rules.DefaultClientConfig = &clientcmd.DefaultClientConfig

	overrides := &clientcmd.ConfigOverrides{ClusterDefaults: clientcmd.ClusterDefaults}

	if kubeconfig != "" {
		rules.ExplicitPath = kubeconfig
	}

	return clientcmd.NewNonInteractiveDeferredLoadingClientConfig(rules, overrides)
}

func getRestConfig(kubeconfig string) (*rest.Config, error) {
	config, err := GetConfig(kubeconfig).ClientConfig()
	if err != nil {
		return nil, fmt.Errorf("could not get Kubernetes config using configuration %q: %s", kubeconfig, err)
	}
	clog.V(4).Printf("configuration from %q finds host %v", kubeconfig, config.Host)
	return config, nil
}

// GetKubeClient provides k8s client for kubeconfig
func GetKubeClient(kubeconfig string) (*Client, error) {
	config, err := getRestConfig(kubeconfig)
	if err != nil {
		return nil, err
	}
	return GetKubeClientForConfig(config)
}

func GetKubeClientForConfig(config *rest.Config) (*Client, error) {
	kubeClient, err := kubernetes.NewForConfig(config)
	if err != nil {
		return nil, fmt.Errorf("could not get Kubernetes client: %s", err)
	}
	extClient, err := apiextensionsclient.NewForConfig(config)
	if err != nil {
		return nil, fmt.Errorf("could not get Kubernetes client: %s", err)
	}
	dynamicClient, err := dynamic.NewForConfig(config)
	if err != nil {
		return nil, fmt.Errorf("could not create Kubernetes dynamic client: %s", err)
	}
	ctrlClient, err := client.New(config, client.Options{})
	if err != nil {
		return nil, fmt.Errorf("could not create Controller Runtime client: %s", err)
	}
	kudoClient, err := versioned.NewForConfig(config)
	if err != nil {
		return nil, fmt.Errorf("could not create KUDO client: %v", err)
	}

	fullScheme := scheme.Scheme
	if err := apis.AddToScheme(fullScheme); err != nil {
		return nil, fmt.Errorf("failed to create runtime.Scheme: %v", err)
	}

	return &Client{
		KubeClient:    kubeClient,
		ExtClient:     extClient,
		DynamicClient: dynamicClient,
		CtrlClient:    ctrlClient,
		KudoClient:    kudoClient,
		Scheme:        fullScheme,
	}, nil
}