jandre/formolcli
1
0
Fork 0
Code Issues 4 Pull Requests Projects Releases 1 Wiki Activity

Gather env variables from container.Env and container.EnvFrom

Browse Source
This commit is contained in:
Jean-Marc ANDRE 2023-02-15 13:55:34 +01:00
parent f3735e1409
commit fd8df677c2
1 changed files with 86 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
controllers/backupsession_controller_helpers.go
View File

@ -2,12 +2,95 @@ package controllers
import (
formolv1alpha1 "github.com/desmo999r/formol/api/v1alpha1"
corev1 "k8s.io/api/core/v1"
"os"
"sigs.k8s.io/controller-runtime/pkg/client"
)
func (r *BackupSessionReconciler) getSecretData(name string) map[string][]byte {
secret := corev1.Secret{}
namespace := os.Getenv(formolv1alpha1.POD_NAMESPACE)
if err := r.Get(r.Context, client.ObjectKey{
Namespace: namespace,
Name: name,
}, &secret); err != nil {
r.Log.Error(err, "unable to get Secret", "Secret", name)
return nil
}
return secret.Data
}
func (r *BackupSessionReconciler) getEnvFromSecretKeyRef(name string, key string) string {
if data := r.getSecretData(name); data != nil {
return string(data[key])
}
return ""
}
func (r *BackupSessionReconciler) getConfigMapData(name string) map[string]string {
configMap := corev1.ConfigMap{}
namespace := os.Getenv(formolv1alpha1.POD_NAMESPACE)
if err := r.Get(r.Context, client.ObjectKey{
Namespace: namespace,
Name: name,
}, &configMap); err != nil {
r.Log.Error(err, "unable to get ConfigMap", "configmap", name)
return nil
}
return configMap.Data
}
func (r *BackupSessionReconciler) getEnvFromConfigMapKeyRef(name string, key string) string {
if data := r.getConfigMapData(name); data != nil {
return string(data[key])
}
return ""
}
func (r *BackupSessionReconciler) getFuncEnv(vars map[string]string, envVars []corev1.EnvVar) {
for _, env := range envVars {
if env.ValueFrom != nil {
if env.ValueFrom.ConfigMapKeyRef != nil {
vars[env.Name] = r.getEnvFromConfigMapKeyRef(env.ValueFrom.ConfigMapKeyRef.LocalObjectReference.Name, env.ValueFrom.ConfigMapKeyRef.Key)
}
if env.ValueFrom.SecretKeyRef != nil {
vars[env.Name] = r.getEnvFromSecretKeyRef(env.ValueFrom.SecretKeyRef.LocalObjectReference.Name, env.ValueFrom.SecretKeyRef.Key)
}
}
}
}
func (r *BackupSessionReconciler) getEnvFromSecretEnvSource(vars map[string]string, name string) {
for key, value := range r.getSecretData(name) {
vars[key] = string(value)
}
}
func (r *BackupSessionReconciler) getEnvFromConfigMapEnvSource(vars map[string]string, name string) {
for key, value := range r.getConfigMapData(name) {
vars[key] = value
}
}
func (r *BackupSessionReconciler) getFuncEnvFrom(vars map[string]string, envVars []corev1.EnvFromSource) {
for _, env := range envVars {
if env.ConfigMapRef != nil {
r.getEnvFromConfigMapEnvSource(vars, env.ConfigMapRef.LocalObjectReference.Name)
}
if env.SecretRef != nil {
r.getEnvFromSecretEnvSource(vars, env.SecretRef.LocalObjectReference.Name)
}
}
}
func (r *BackupSessionReconciler) getFuncVars(function formolv1alpha1.Function, vars map[string]string) {
r.getFuncEnvFrom(vars, function.Spec.EnvFrom)
r.getFuncEnv(vars, function.Spec.Env)
}
func (r *BackupSessionReconciler) runInitBackupSteps(target formolv1alpha1.Target) error {
namespace := os.Getenv(formolv1alpha1.POD_NAMESPACE)
r.Log.V(0).Info("start to run the backup initializing steps it any")
for _, container := range target.Containers {
for _, step := range container.Steps {
if step.Finalize != nil && *step.Finalize == true {
@ -21,6 +104,9 @@ func (r *BackupSessionReconciler) runInitBackupSteps(target formolv1alpha1.Targe
r.Log.Error(err, "unable to get Function", "Function", step.Name)
return err
}
vars := make(map[string]string)
r.getFuncVars(function, vars)
}
}
return nil