source: allow inline env vars in build

akupila · Apr 6, 2020 · 21b3ed0 · 21b3ed0
1 parent 811244a
commit 21b3ed0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/source/build.go b/source/build.go
@@ -30,8 +30,14 @@ func ParseBuildScript(str string) (BuildScript, error) {
 			continue
 		}
 		parts := strings.Split(line, " ")
-		if _, err := exec.LookPath(parts[0]); err != nil {
-			return nil, fmt.Errorf("line %d: %w", i, err)
+		for _, part := range parts {
+			if strings.Contains(part, "=") {
+				continue
+			}
+			if _, err := exec.LookPath(part); err != nil {
+				return nil, fmt.Errorf("line %d: %w", i, err)
+			}
+			break
 		}
 		out = append(out, BuildStep(line))
 	}

diff --git a/source/build_test.go b/source/build_test.go
@@ -54,6 +54,13 @@ func TestParseBuildScript(t *testing.T) {
 			input:   "thisfiledoesnotexist --xx --yy",
 			wantErr: true,
 		},
+		{
+			name:  "EnvVar",
+			input: "FOO=bar echo $FOO",
+			want: BuildScript{
+				"FOO=bar echo $FOO",
+			},
+		},
 		{
 			name:  "Empty",
 			input: "",