feat: labels vars

Signed-off-by: Vladislav Sukhin <vladislav@kubeshop.io>

cmd/tcl/testworkflow-toolkit/spawn/utils.go

@@ -197,7 +197,8 @@ func CreateExecutionMachine(prefix string, index int64) (string, expressions.Mac
 	}
 	return id, expressions.NewMachine().
 		Register("workflow", map[string]string{
-			"name": env.WorkflowName(),
+			"name":   env.WorkflowName(),
+			"labels": env.Config().Execution.Labels,
 		}).
 		Register("resource", map[string]string{
 			"root":     env.ExecutionId(),
@@ -293,6 +294,12 @@ func CreateBaseMachine() expressions.Machine {
 		dashboardUrl = fmt.Sprintf("%s/organization/%s/environment/%s/dashboard",
 			env.Config().Cloud.UiUrl, env.Config().Cloud.OrgId, env.Config().Cloud.EnvId)
 	}
+
+	var labelMap map[string]string
+	if labels := env.Config().Execution.Labels; labels != "" {
+		json.Unmarshal([]byte(labels), &labelMap)
+	}
+
 	return expressions.CombinedMachines(
 		data.GetBaseTestWorkflowMachine(),
 		expressions.NewMachine().RegisterStringMap("internal", map[string]string{
@@ -341,7 +348,8 @@ func CreateBaseMachine() expressions.Machine {
 			}).
 			Register("environment", map[string]string{
 				"id": env.Config().Cloud.EnvId,
-			}),
+			}).
+			RegisterStringMap("labels", labelMap),
 	)
 }
 

cmd/testworkflow-toolkit/env/config.go

@@ -48,6 +48,7 @@ type envExecutionConfig struct {
 	FSPrefix        string `envconfig:"TK_FS"`
 	DisableWebhooks bool   `envconfig:"TK_DWH"`
 	Tags            string `envconfig:"TK_TAG"`
+	Labels          string `envconfig:"TK_LBL"`
 }
 
 type envSystemConfig struct {

pkg/testworkflows/testworkflowexecutor/executor.go

@@ -3,10 +3,12 @@ package testworkflowexecutor
 import (
 	"bufio"
 	"context"
+	"encoding/json"
 	"fmt"
 	"io"
 	"os"
 	"strconv"
+	"strings"
 	"sync"
 	"time"
 
@@ -408,6 +410,16 @@ func (e *executor) Execute(ctx context.Context, workflow testworkflowsv1.TestWor
 	// Build the basic Execution data
 	id := primitive.NewObjectID().Hex()
 	now := time.Now()
+	labels := make(map[string]string)
+	for key, value := range workflow.Labels {
+		labels[strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(key, ".", "_"), "-", "_"), "/", "_")] = value
+	}
+
+	labelMap, err := json.Marshal(labels)
+	if err != nil {
+		return execution, errors.Wrap(err, "marsalling labels error")
+	}
+
 	machine := expressions.NewMachine().
 		RegisterStringMap("internal", map[string]string{
 			"storage.url":        os.Getenv("STORAGE_ENDPOINT"),
@@ -448,7 +460,8 @@ func (e *executor) Execute(ctx context.Context, workflow testworkflowsv1.TestWor
 			"images.cache.ttl":           common.GetOr(os.Getenv("TESTKUBE_IMAGE_CREDENTIALS_CACHE_TTL"), "30m"),
 		}).
 		Register("workflow", map[string]string{
-			"name": workflow.Name,
+			"name":   workflow.Name,
+			"labels": string(labelMap),
 		}).
 		Register("resource", map[string]string{
 			"id":       id,
@@ -463,7 +476,9 @@ func (e *executor) Execute(ctx context.Context, workflow testworkflowsv1.TestWor
 		}).
 		Register("environment", map[string]string{
 			"id": cloudEnvId,
-		})
+		}).
+		RegisterStringMap("labels", labels)
+
 	mockExecutionMachine := expressions.NewMachine().Register("execution", map[string]interface{}{
 		"id":              id,
 		"name":            "<mock_name>",

pkg/testworkflows/testworkflowprocessor/stage/container.go

@@ -494,6 +494,7 @@ func (c *container) EnableToolkit(ref string) Container {
 			"TK_IMG_P":                  "{{internal.images.persistence.enabled}}",
 			"TK_IMG_PK":                 "{{internal.images.persistence.key}}",
 			"TK_IMG_CRED_TTL":           "{{internal.images.cache.ttl}}",
+			"TK_LBL":                    "{{workflow.labels}}",
 		})
 }