20240319 ./pkg/clusterapi/localcontrolplane.go

localcontrolplane.go

package clusterapi

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

	"github.com/sirupsen/logrus"
	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
	"k8s.io/client-go/kubernetes/scheme"
	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/client-go/tools/clientcmd/api"
	"k8s.io/klog/v2"
	capav1beta1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta1"
	capav1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
	capzv1 "sigs.k8s.io/cluster-api-provider-azure/api/v1beta1"
	capgv1 "sigs.k8s.io/cluster-api-provider-gcp/api/v1beta1"
	capiv1 "sigs.k8s.io/cluster-api-provider-ibmcloud/api/v1beta2"
	capov1 "sigs.k8s.io/cluster-api-provider-openstack/api/v1alpha7"
	capvv1 "sigs.k8s.io/cluster-api-provider-vsphere/apis/v1beta1"
	clusterv1alpha4 "sigs.k8s.io/cluster-api/api/v1alpha4" //nolint:staticcheck
	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/envtest"
	"sigs.k8s.io/controller-runtime/pkg/log"

	"github.com/openshift/installer/cmd/openshift-install/command"
)

var (
	// Scheme is the scheme used by the local control plane.
	Scheme = scheme.Scheme
)

func init() {
	utilruntime.Must(clusterv1alpha4.AddToScheme(Scheme))
	utilruntime.Must(clusterv1.AddToScheme(Scheme))
	utilruntime.Must(capav1beta1.AddToScheme(Scheme))
	utilruntime.Must(capav1.AddToScheme(Scheme))
	utilruntime.Must(capzv1.AddToScheme(Scheme))
	utilruntime.Must(capgv1.AddToScheme(Scheme))
	utilruntime.Must(capvv1.AddToScheme(Scheme))
	utilruntime.Must(capov1.AddToScheme(Scheme))
	utilruntime.Must(capiv1.AddToScheme(Scheme))
}

// localControlPlane creates a local capi control plane
// to use as a management cluster.
type localControlPlane struct {
	Env            *envtest.Environment
	Client         client.Client
	Cfg            *rest.Config
	BinDir         string
	KubeconfigPath string
}

// Run launches the local control plane.
func (c *localControlPlane) Run(ctx context.Context) error {
	// Create a temporary directory to unpack the cluster-api binaries.
	c.BinDir = filepath.Join(command.RootOpts.Dir, "bin", "cluster-api")
	if err := UnpackClusterAPIBinary(c.BinDir); err != nil {
		return fmt.Errorf("failed to unpack cluster-api binary: %w", err)
	}
	if err := UnpackEnvtestBinaries(c.BinDir); err != nil {
		return fmt.Errorf("failed to unpack envtest binaries: %w", err)
	}

	log.SetLogger(klog.NewKlogr())
	logrus.Info("Started local control plane with envtest")
	c.Env = &envtest.Environment{
		Scheme:                   Scheme,
		AttachControlPlaneOutput: false,
		BinaryAssetsDirectory:    c.BinDir,
		ControlPlaneStartTimeout: 10 * time.Second,
		ControlPlaneStopTimeout:  10 * time.Second,
	}
	var err error
	c.Cfg, err = c.Env.Start()
	if err != nil {
		return err
	}

	kc := fromEnvTestConfig(c.Cfg)
	{
		dir := filepath.Join(command.RootOpts.Dir, "auth")
		kf, err := os.Create(filepath.Join(dir, "envtest.kubeconfig"))
		if err != nil {
			return err
		}
		if _, err := kf.Write(kc); err != nil {
			return err
		}
		if err := kf.Close(); err != nil {
			return err
		}
		c.KubeconfigPath, err = filepath.Abs(kf.Name())
		if err != nil {
			return err
		}
	}

	// Create a new client to interact with the cluster.
	cl, err := client.New(c.Cfg, client.Options{
		Scheme: c.Env.Scheme,
	})
	if err != nil {
		return err
	}
	c.Client = cl

	logrus.Infof("Stored kubeconfig for envtest in: %v", c.KubeconfigPath)
	return nil
}

func (c *localControlPlane) Stop() error {
	return c.Env.Stop()
}

// fromEnvTestConfig returns a new Kubeconfig in byte form when running in envtest.
func fromEnvTestConfig(cfg *rest.Config) []byte {
	clusterName := "envtest"
	contextName := fmt.Sprintf("%s@%s", cfg.Username, clusterName)
	c := api.Config{
		Clusters: map[string]*api.Cluster{
			clusterName: {
				Server:                   cfg.Host,
				CertificateAuthorityData: cfg.CAData,
			},
		},
		Contexts: map[string]*api.Context{
			contextName: {
				Cluster:  clusterName,
				AuthInfo: cfg.Username,
			},
		},
		AuthInfos: map[string]*api.AuthInfo{
			cfg.Username: {
				ClientKeyData:         cfg.KeyData,
				ClientCertificateData: cfg.CertData,
			},
		},
		CurrentContext: contextName,
	}
	data, err := clientcmd.Write(c)
	if err != nil {
		logrus.Fatalf("failed to write kubeconfig: %v", err)
	}
	return data
}