fix #235 - do not fail if env sh declared but not defined

Author: m.kindritskiy

config/config/env.go

@@ -36,12 +36,48 @@ func (e *Envs) UnmarshalYAML(node *yaml.Node) error {
 		keyNode := node.Content[i]
 		valueNode := node.Content[i+1]
 
-		var env Env
-		if err := valueNode.Decode(&env); err != nil {
+		envAsStr := ""
+
+		if err := valueNode.Decode(&envAsStr); err == nil {
+			e.Set(keyNode.Value, Env{Name: keyNode.Value, Value: envAsStr})
+			continue
+		}
+
+		envAsMap := struct {
+			Name     string
+			Value    string
+			Sh       *string
+			Checksum *Checksum
+		}{}
+
+		if err := valueNode.Decode(&envAsMap); err != nil {
 			return err
 		}
-		env.Name = keyNode.Value
-		e.Set(keyNode.Value, env)
+
+		env := Env{
+			Name:     keyNode.Value,
+			Value:    envAsMap.Value,
+			Sh:       "",
+			Checksum: Checksum{},
+		}
+
+		if env.Value == "" {
+			if envAsMap.Sh == nil && envAsMap.Checksum == nil {
+				return fmt.Errorf("lets: environment variable '%s' must have value or 'sh' or 'checksum'", keyNode.Value)
+			}
+
+			if envAsMap.Sh != nil && envAsMap.Checksum != nil {
+				return fmt.Errorf("lets: environment variable '%s' must have only 'sh' or 'checksum'", keyNode.Value)
+			}
+
+			if envAsMap.Sh != nil {
+				env.Sh = *envAsMap.Sh
+			} else if envAsMap.Checksum != nil {
+				env.Checksum = *envAsMap.Checksum
+			}
+		}
+
+		e.Set(env.Name, env)
 	}
 
 	return nil

executor/executor.go

@@ -37,8 +37,8 @@ func (e *ExecuteError) ExitCode() int {
 }
 
 type Executor struct {
-	cfg *config.Config
-	out io.Writer
+	cfg        *config.Config
+	out        io.Writer
 	initCalled bool
 }
 

tests/global_env.bats

@@ -9,10 +9,11 @@ setup() {
 @test "global_env: should provide env to command" {
     run lets global-env
     assert_success
-    assert_line --index 0 "ONE=1"
-    assert_line --index 1 "TWO=two"
-    assert_line --index 2 "THREE=3"
-    assert_line --index 3 "FOUR=4"
-    assert_line --index 4 "BAR=Bar"
-    assert_line --index 5 "FOO=bb1da47569d9fbe3b5f2216fdbd4c9b040ccb5c1"
+    assert_line --index 0 "INT=1"
+    assert_line --index 1 "STR=hi"
+    assert_line --index 2 "STR_INT=1"
+    assert_line --index 3 "BOOL=true"
+    assert_line --index 4 "ORIGINAL=b"
+    assert_line --index 5 "BAR=Bar"
+    assert_line --index 6 "FOO=bb1da47569d9fbe3b5f2216fdbd4c9b040ccb5c1"
 }

tests/global_env/lets.yaml

@@ -1,9 +1,11 @@
 shell: bash
 
 env:
-  ONE: 1
-  TWO: two
-  THREE: "3"
+  INT: 1
+  STR: "hi"
+  STR_INT: "1"
+  BOOL: true
+  ORIGINAL: "a"
   BAR:
     sh: echo Bar
   FOO:
@@ -13,11 +15,12 @@ commands:
   global-env:
     description: Test global env
     env:
-      FOUR: "4"
+      ORIGINAL: "b"
     cmd: |
-      echo ONE=${ONE}
-      echo TWO=${TWO}
-      echo THREE=${THREE}
-      echo FOUR=${FOUR}
+      echo INT=${INT}
+      echo STR=${STR}
+      echo STR_INT=${STR_INT}
+      echo BOOL=${BOOL}
+      echo ORIGINAL=${ORIGINAL}
       echo BAR=${BAR}
       echo FOO=${FOO}