Add support for conditions

Author: cvhariharan

README.md

@@ -36,8 +36,11 @@ jobs:
   - name: Run tests
     stage: test
     image: "docker.io/golang:1.21.3"
+    variables:
+      - TEST: true
     script:
       - go test ./...
+    condition: TEST
 
   - name: Run checks
     stage: security

cmd/dot/root.go

@@ -8,6 +8,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/expr-lang/expr"
 	"github.com/go-playground/validator/v10"
 	"github.com/opnlabs/dot/pkg/artifacts"
 	"github.com/opnlabs/dot/pkg/models"
@@ -44,7 +45,7 @@ concurrently.`,
 					log.Fatalf("variables should be defined as KEY=VALUE: %s", v)
 				}
 
-				m := make(map[string]string)
+				m := make(map[string]any)
 				m[variables[0]] = variables[1]
 				environmentVariables = append(environmentVariables, m)
 			}
@@ -98,7 +99,36 @@ func run() {
 		if _, ok := stageMap[v.Stage]; !ok {
 			log.Fatalf("stage not defined: %s", v.Stage)
 		}
-		stageMap[v.Stage] = append(stageMap[v.Stage], v)
+
+		// Create expr program with the variables passed as env
+		if len(v.Condition) == 0 {
+			v.Condition = `true`
+		}
+
+		env := make(map[string]any)
+		for _, entries := range v.Variables {
+			if len(entries) > 1 {
+				log.Fatal("variables should be defined as a key value pair")
+			}
+			for k, value := range entries {
+				env[k] = value
+			}
+		}
+
+		p, err := expr.Compile(v.Condition, expr.Env(env), expr.AsBool())
+		if err != nil {
+			log.Fatalf("condition evaluation failed for job %s: %v", v.Name, err)
+		}
+		output, err := expr.Run(p, env)
+		if err != nil {
+			log.Fatalf("condition evaluation failed for job %s: %v", v.Name, err)
+		}
+
+		// Only append to stageMap if the condition evaluates to true
+		if output.(bool) {
+			stageMap[v.Stage] = append(stageMap[v.Stage], v)
+		}
+
 	}
 
 	dockerArtifactManager := artifacts.NewDockerArtifactsManager(".artifacts")

dot.yml

@@ -7,8 +7,11 @@ jobs:
   - name: Run tests
     stage: test
     image: "docker.io/golang:1.21.3"
+    variables:
+      - TEST: false
     script:
       - go test ./...
+    condition: TEST
 
   - name: Run checks
     stage: security

go.mod

@@ -21,6 +21,7 @@ require (
 	github.com/docker/distribution v2.8.3+incompatible // indirect
 	github.com/docker/go-connections v0.4.0 // indirect
 	github.com/docker/go-units v0.5.0 // indirect
+	github.com/expr-lang/expr v1.15.7 // indirect
 	github.com/gabriel-vasile/mimetype v1.4.2 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect

go.sum

@@ -16,6 +16,8 @@ github.com/docker/go-connections v0.4.0 h1:El9xVISelRB7BuFusrZozjnkIM5YnzCViNKoh
 github.com/docker/go-connections v0.4.0/go.mod h1:Gbd7IOopHjR8Iph03tsViu4nIes5XhDvyHbTtUxmeec=
 github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4=
 github.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk=
+github.com/expr-lang/expr v1.15.7 h1:BK0JcWUkoW6nrbLBo6xCKhz4BvH5DSOOu1Gx5lucyZo=
+github.com/expr-lang/expr v1.15.7/go.mod h1:uCkhfG+x7fcZ5A5sXHKuQ07jGZRl6J0FCAaf2k4PtVQ=
 github.com/fatih/color v1.15.0 h1:kOqh6YHBtK8aywxGerMG2Eq3H6Qgoqeo13Bk2Mv/nBs=
 github.com/fatih/color v1.15.0/go.mod h1:0h5ZqXfHYED7Bhv2ZJamyIOUej9KtShiJESRwBDUSsw=
 github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=

pkg/models/models.go

@@ -1,7 +1,7 @@
 package models
 
 type Stage string
-type Variable map[string]string
+type Variable map[string]any
 
 type JobFile struct {
 	Stages []Stage `yaml:"stages" validate:"required,dive"`
@@ -17,4 +17,5 @@ type Job struct {
 	Script     []string   `yaml:"script"`
 	Entrypoint []string   `yaml:"entrypoint"`
 	Artifacts  []string   `yaml:"artifacts"`
+	Condition  string     `yaml:"condition"`
 }

pkg/runner/docker.go

@@ -91,7 +91,7 @@ func (d *DockerRunner) WithEnv(env []models.Variable) *DockerRunner {
 			log.Fatal("variables should be defined as a key value pair")
 		}
 		for k, v := range v {
-			variables = append(variables, fmt.Sprintf("%s=%s", k, v))
+			variables = append(variables, fmt.Sprintf("%s=%s", k, fmt.Sprint(v)))
 		}
 	}
 	d.env = variables

pkg/runner/docker_test.go

@@ -72,7 +72,7 @@ func TestRun(t *testing.T) {
 			Manager: manager,
 			Image:   "docker.io/alpine",
 			Variables: []models.Variable{
-				map[string]string{
+				map[string]any{
 					"TESTING_VARIABLE": "TESTING",
 				},
 			},