mirror of
https://github.com/rancher/k3k.git
synced 2026-08-18 03:46:31 +00:00
Hussein GalalandGitHub
3abffae92d
* Adding new k3s function to get the config * Adding a new way to get the bootstrap data directly from the server Pod * Gate the bootstrap fetch if external datastore is in use --------- Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com>
95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"k8s.io/client-go/rest"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
|
|
"github.com/rancher/k3k/pkg/apis/k3k.io/v1beta1"
|
|
"github.com/rancher/k3k/pkg/controller"
|
|
"github.com/rancher/k3k/pkg/k3s"
|
|
)
|
|
|
|
// Fetch requests bootstrap data from k3s using the token and decodes it,
|
|
// to avoid double encoding when stored as secret.
|
|
func Fetch(ctx context.Context, k3sClient *k3s.Client, cluster *v1beta1.Cluster, restConfig *rest.Config) (*k3s.BootstrapData, error) {
|
|
log := ctrl.LoggerFrom(ctx)
|
|
|
|
config, err := k3sClient.GetServerConfig()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if config.ClusterInit {
|
|
log.V(1).Info("Fetching bootstrap data from K3s server API")
|
|
return k3sClient.GetServerBootstrap()
|
|
}
|
|
|
|
log.V(1).Info("Fetching bootstrap data from K3s server Pod")
|
|
|
|
return k3s.ReadBootstrapFromK3sPod(ctx, restConfig, cluster.Name, cluster.Namespace)
|
|
}
|
|
|
|
// SaveToSecret marshals the bootstrap data and stores it in a Secret owned by the cluster,
|
|
// creating the Secret if it does not exist or updating it otherwise.
|
|
func SaveToSecret(ctx context.Context, c client.Client, cluster *v1beta1.Cluster, data *k3s.BootstrapData) error {
|
|
bootstrapData, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
secret := &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: controller.SafeConcatNameWithPrefix(cluster.Name, "bootstrap"),
|
|
Namespace: cluster.Namespace,
|
|
},
|
|
}
|
|
|
|
_, err = controllerutil.CreateOrUpdate(ctx, c, secret, func() error {
|
|
if err := controllerutil.SetControllerReference(cluster, secret, c.Scheme()); err != nil {
|
|
return err
|
|
}
|
|
|
|
secret.Data = map[string][]byte{
|
|
"bootstrap": bootstrapData,
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
// LoadFromSecret reads the bootstrap data of a certain cluster and returns the decoded content.
|
|
func LoadFromSecret(ctx context.Context, client client.Client, cluster *v1beta1.Cluster) (*k3s.BootstrapData, error) {
|
|
key := types.NamespacedName{
|
|
Name: controller.SafeConcatNameWithPrefix(cluster.Name, "bootstrap"),
|
|
Namespace: cluster.Namespace,
|
|
}
|
|
|
|
var bootstrapSecret corev1.Secret
|
|
if err := client.Get(ctx, key, &bootstrapSecret); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
bootstrapData := bootstrapSecret.Data["bootstrap"]
|
|
if bootstrapData == nil {
|
|
return nil, errors.New("empty bootstrap")
|
|
}
|
|
|
|
var bootstrap k3s.BootstrapData
|
|
|
|
err := json.Unmarshal(bootstrapData, &bootstrap)
|
|
|
|
return &bootstrap, err
|
|
}
|