feat: support some magic env parsers

pkg/generator/appconfiguration/generator/workload/magic_env_var.go

@@ -0,0 +1,100 @@
+package workload
+
+import (
+	"strings"
+
+	corev1 "k8s.io/api/core/v1"
+)
+
+var (
+	SecretEnvParser MagicEnvParser = NewSecretEnvParser()
+	RawEnvParser                   = NewRawEnvParser()
+
+	supportedParsers = []MagicEnvParser{
+		SecretEnvParser,
+		// As the default parser, the RawEnvParser should be placed at
+		// the end.
+		RawEnvParser,
+	}
+)
+
+// MagicEnvVar generates a specialized EnvVar based on the key and
+// value of environment.
+//
+// Examples:
+//
+//	MagicEnvVar("secret_key", "secret://my_secret/my_key")
+//	MagicEnvVar("key", "value")
+func MagicEnvVar(k, v string) *corev1.EnvVar {
+	for _, p := range supportedParsers {
+		if p.Match(k, v) {
+			return p.Gen(k, v)
+		}
+	}
+	return nil
+}
+
+// MagicEnvParser is an interface for environment variable parsers.
+type MagicEnvParser interface {
+	Match(k, v string) (matched bool)
+	Gen(k, v string) *corev1.EnvVar
+}
+
+// rawEnvParser is a parser for raw environment variables.
+type rawEnvParser struct{}
+
+// NewRawEnvParser creates a new instance of RawEnvParser.
+func NewRawEnvParser() MagicEnvParser {
+	return &rawEnvParser{}
+}
+
+// Match checks if the value matches the raw parser.
+func (*rawEnvParser) Match(_ string, _ string) bool {
+	return true
+}
+
+// Gen generates a raw environment variable.
+func (*rawEnvParser) Gen(k string, v string) *corev1.EnvVar {
+	return &corev1.EnvVar{
+		Name:  k,
+		Value: v,
+	}
+}
+
+// secretEnvParser is a parser for secret-based environment variables.
+type secretEnvParser struct {
+	prefix string
+}
+
+// NewSecretEnvParser creates a new instance of SecretEnvParser.
+func NewSecretEnvParser() MagicEnvParser {
+	return &secretEnvParser{
+		prefix: "secret://",
+	}
+}
+
+// Match checks if the value matches the secret parser.
+func (p *secretEnvParser) Match(_ string, v string) bool {
+	return strings.HasPrefix(v, p.prefix)
+}
+
+// Gen generates a secret-based environment variable.
+func (p *secretEnvParser) Gen(k string, v string) *corev1.EnvVar {
+	vv := strings.TrimPrefix(v, p.prefix)
+	vs := strings.Split(vv, "/")
+	if len(vs) != 2 {
+		return nil
+	}
+
+	return &corev1.EnvVar{
+		Name: k,
+		ValueFrom: &corev1.EnvVarSource{
+			SecretKeyRef: &corev1.SecretKeySelector{
+				LocalObjectReference: corev1.LocalObjectReference{
+					Name: vs[0],
+				},
+				Key: vs[1],
+			},
+		},
+	}
+}

pkg/generator/appconfiguration/generator/workload/magic_env_var_test.go

@@ -0,0 +1,22 @@
+package workload
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMagicEnvVar(t *testing.T) {
+	// Test raw environment variable
+	rawEnv := MagicEnvVar("key", "value")
+	assert.NotNil(t, rawEnv, "Raw environment variable should not be nil")
+	assert.Equal(t, "key", rawEnv.Name, "Expected raw environment variable name to be 'key'")
+	assert.Equal(t, "value", rawEnv.Value, "Expected raw environment variable value to be 'value'")
+
+	// Test secret-based environment variable
+	secretEnv := MagicEnvVar("secret_key", "secret://my_secret/my_key")
+	assert.NotNil(t, secretEnv, "Secret-based environment variable should not be nil")
+	assert.Equal(t, "secret_key", secretEnv.Name, "Expected secret-based environment variable name to be 'secret_key'")
+	assert.Equal(t, "my_secret", secretEnv.ValueFrom.SecretKeyRef.LocalObjectReference.Name, "Expected secret name to be 'my_secret'")
+	assert.Equal(t, "my_key", secretEnv.ValueFrom.SecretKeyRef.Key, "Expected secret key to be 'my_key'")
+}

pkg/generator/appconfiguration/generator/workload/service_generator_test.go

@@ -8,7 +8,6 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
 	"kusionstack.io/kube-api/apps/v1alpha1"
-
 	"kusionstack.io/kusion/pkg/models"
 	"kusionstack.io/kusion/pkg/models/appconfiguration/workload"
 	"kusionstack.io/kusion/pkg/models/appconfiguration/workload/container"

pkg/generator/appconfiguration/generator/workload/workload_generator.go

@@ -79,10 +79,7 @@ func toOrderedContainers(appContainers map[string]container.Container) ([]corev1
 		// Create a slice of env vars based on the container's env vars.
 		var envs []corev1.EnvVar
 		for k, v := range c.Env {
-			envs = append(envs, corev1.EnvVar{
-				Name:  k,
-				Value: v,
-			})
+			envs = append(envs, *MagicEnvVar(k, v))
 		}
 
 		// Create a container object and append it to the containers slice.

pkg/generator/appconfiguration/util.go

@@ -5,7 +5,6 @@ import (
 
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
-
 	"kusionstack.io/kusion/pkg/models"
 )