feat: add PULUMI_STACK for compiler run enviroment

CHANGELOG_PENDING.md

@@ -1,3 +1,5 @@
 ### Improvements
 
+- Pass `PULUMI_STACK`, `PULUMI_ORGANIZATION`, `PULUMI_PROJECT` and `PULUMI_CONFIG` as environment variable to compiler process. 
+
 ### Bug Fixes

pkg/pulumiyaml/codegen/eject.go

@@ -114,7 +114,7 @@ func LoadTemplate(dir string) (*workspace.Project, *ast.TemplateDecl, hcl.Diagno
 		if !ok {
 			return nil, nil, nil, fmt.Errorf("compiler option must be a string, got %v", reflect.TypeOf(compilerOpt))
 		}
-		t, diags, err = pulumiyaml.LoadFromCompiler(compiler, main)
+		t, diags, err = pulumiyaml.LoadFromCompiler(compiler, main, nil)
 	} else {
 		t, diags, err = pulumiyaml.LoadDir(main)
 	}

pkg/pulumiyaml/run.go

@@ -46,7 +46,7 @@ func Load() (*ast.TemplateDecl, syntax.Diagnostics, error) {
 	return LoadDir(".")
 }
 
-func LoadFromCompiler(compiler string, workingDirectory string) (*ast.TemplateDecl, syntax.Diagnostics, error) {
+func LoadFromCompiler(compiler string, workingDirectory string, env []string) (*ast.TemplateDecl, syntax.Diagnostics, error) {
 	var diags syntax.Diagnostics
 	var stdout bytes.Buffer
 	var stderr bytes.Buffer
@@ -60,6 +60,11 @@ func LoadFromCompiler(compiler string, workingDirectory string) (*ast.TemplateDe
 	cmd.Dir = workingDirectory
 	cmd.Stdout = &stdout
 	cmd.Stderr = &stderr
+	cmd.Env = append(env, os.Environ()...)
+	for _, duplicate := range conflictingEnvVars(cmd.Env) {
+		diags = append(diags, syntax.Warning(nil, fmt.Sprintf("environment variable %v is already set: compiler %v will not override conflicting environment variables", duplicate, name), ""))
+	}
+
 	err = cmd.Run()
 	if err != nil {
 		return nil, nil, fmt.Errorf("error running compiler %v: %v, stderr follows: %v", name, err, stderr.String())
@@ -74,6 +79,21 @@ func LoadFromCompiler(compiler string, workingDirectory string) (*ast.TemplateDe
 	return template, tdiags, err
 }
 
+func conflictingEnvVars(env []string) []string {
+	envMap := make(map[string]uint)
+	var duplicates []string
+	for _, e := range env {
+		key := strings.Split(e, "=")[0]
+		if cnt, exists := envMap[key]; exists && cnt <= 1 {
+			duplicates = append(duplicates, key)
+		}
+
+		envMap[key]++
+	}
+
+	return duplicates
+}
+
 // Load a template from the current working directory.
 func LoadDir(cwd string) (*ast.TemplateDecl, syntax.Diagnostics, error) {
 	// Read in the template file - search first for Main.json, then Main.yaml, then Pulumi.yaml.

pkg/pulumiyaml/run_test.go

@@ -2412,3 +2412,50 @@ func TestUnknownsDuringPreviewNotUpdate(t *testing.T) {
 	assert.NoError(t, runProgram(true))
 	assert.Error(t, runProgram(false))
 }
+
+func TestConflictingEnvVarsNoDuplicates(t *testing.T) {
+	t.Parallel()
+
+	env := []string{"FOO=bar", "BAZ=qux"}
+	conflicts := conflictingEnvVars(env)
+	assert.Empty(t, conflicts)
+}
+
+func TestConflictingEnvVarsWithDuplicates(t *testing.T) {
+	t.Parallel()
+
+	env := []string{"FOO=bar", "FOO=baz"}
+	conflicts := conflictingEnvVars(env)
+	assert.Equal(t, []string{"FOO"}, conflicts)
+}
+
+func TestConflictingEnvVarsEmptyEnv(t *testing.T) {
+	t.Parallel()
+
+	env := []string{}
+	conflicts := conflictingEnvVars(env)
+	assert.Empty(t, conflicts)
+}
+
+func TestConflictingEnvVarsNilEnv(t *testing.T) {
+	t.Parallel()
+
+	conflicts := conflictingEnvVars(nil)
+	assert.Empty(t, conflicts)
+}
+
+func TestConflictingEnvVarsInvalidFormat(t *testing.T) {
+	t.Parallel()
+
+	env := []string{"FOO", "BAR=qux"}
+	conflicts := conflictingEnvVars(env)
+	assert.Empty(t, conflicts)
+}
+
+func TestConflictingEnvVarsMultipleDuplicates(t *testing.T) {
+	t.Parallel()
+
+	env := []string{"FOO=bar", "FOO=baz", "BAR=qux", "BAR=quux", "FOO=foobar"}
+	conflicts := conflictingEnvVars(env)
+	assert.ElementsMatch(t, []string{"FOO", "BAR"}, conflicts)
+}

pkg/server/server.go

@@ -18,6 +18,8 @@ package server
 
 import (
 	"context"
+	"encoding/json"
+	"fmt"
 	"os"
 
 	pbempty "github.com/golang/protobuf/ptypes/empty"
@@ -53,8 +55,8 @@ func NewLanguageHost(engineAddress, tracing string, compiler string) pulumirpc.L
 	}
 }
 
-func (host *yamlLanguageHost) loadTemplate() (*ast.TemplateDecl, syntax.Diagnostics, error) {
-	if host.template != nil {
+func (host *yamlLanguageHost) loadTemplate(compilerEnv []string) (*ast.TemplateDecl, syntax.Diagnostics, error) {
+	if host.template != nil && host.compiler == "" {
 		return host.template, host.diags, nil
 	}
 
@@ -64,7 +66,7 @@ func (host *yamlLanguageHost) loadTemplate() (*ast.TemplateDecl, syntax.Diagnost
 	if host.compiler == "" {
 		template, diags, err = pulumiyaml.Load()
 	} else {
-		template, diags, err = pulumiyaml.LoadFromCompiler(host.compiler, "")
+		template, diags, err = pulumiyaml.LoadFromCompiler(host.compiler, "", compilerEnv)
 	}
 	if err != nil {
 		return nil, diags, err
@@ -82,7 +84,7 @@ func (host *yamlLanguageHost) loadTemplate() (*ast.TemplateDecl, syntax.Diagnost
 func (host *yamlLanguageHost) GetRequiredPlugins(ctx context.Context,
 	req *pulumirpc.GetRequiredPluginsRequest,
 ) (*pulumirpc.GetRequiredPluginsResponse, error) {
-	template, diags, err := host.loadTemplate()
+	template, diags, err := host.loadTemplate(nil)
 	if err != nil {
 		return nil, err
 	}
@@ -121,7 +123,20 @@ func (host *yamlLanguageHost) Run(ctx context.Context, req *pulumirpc.RunRequest
 		}
 	}
 
-	template, diags, err := host.loadTemplate()
+	configValue := req.GetConfig()
+	jsonConfigValue, err := json.Marshal(configValue)
+	if err != nil {
+		return nil, err
+	}
+
+	compilerEnv := []string{
+		fmt.Sprintf(`PULUMI_STACK=%s`, req.GetStack()),
+		fmt.Sprintf(`PULUMI_ORGANIZATION=%s`, req.GetOrganization()),
+		fmt.Sprintf(`PULUMI_PROJECT=%s`, req.GetProject()),
+		fmt.Sprintf(`PULUMI_CONFIG=%s`, jsonConfigValue),
+	}
+
+	template, diags, err := host.loadTemplate(compilerEnv)
 	if err != nil {
 		return &pulumirpc.RunResponse{Error: err.Error()}, nil
 	}

pkg/tests/integration_test.go

@@ -79,3 +79,45 @@ func TestProjectConfigWithSecretDecrypted(t *testing.T) {
 	}
 	integration.ProgramTest(t, &testOptions)
 }
+
+//nolint:paralleltest // uses parallel programtest
+func TestEnvVarsPassedToExecCommand(t *testing.T) {
+	testOptions := integration.ProgramTestOptions{
+		Dir:             filepath.Join(getCwd(t), "testdata", "env-vars"),
+		Env:             []string{"TEST_ENV_VAR=foobar"},
+		PrepareProject:  prepareYamlProject,
+		StackName:       "dev",
+		SecretsProvider: "default",
+		ExtraRuntimeValidation: func(t *testing.T, stack integration.RuntimeValidationStackInfo) {
+			assert.Equal(t, "foobar", stack.Outputs["TEST_ENV_VAR"])
+			assert.Equal(t, `dev`, stack.Outputs["PULUMI_STACK"])
+			assert.Equal(t, `project-env-vars`, stack.Outputs["PULUMI_PROJECT"])
+			assert.Equal(t, `organization`, stack.Outputs["PULUMI_ORGANIZATION"])
+			assert.EqualValues(t, map[string]interface{}{"project-env-vars:foo": "hello world"}, stack.Outputs["PULUMI_CONFIG"])
+		},
+	}
+	integration.ProgramTest(t, &testOptions)
+}
+
+//nolint:paralleltest // uses parallel programtest
+func TestEnvVarsKeepConflictingValues(t *testing.T) {
+	testOptions := integration.ProgramTestOptions{
+		Dir: filepath.Join(getCwd(t), "testdata", "env-vars"),
+		Env: []string{
+			"PULUMI_STACK=foo",
+			"PULUMI_PROJECT=bar",
+			"PULUMI_ORGANIZATION=foobar",
+			"PULUMI_CONFIG=bazz",
+		},
+		PrepareProject:  prepareYamlProject,
+		StackName:       "dev",
+		SecretsProvider: "default",
+		ExtraRuntimeValidation: func(t *testing.T, stack integration.RuntimeValidationStackInfo) {
+			assert.Equal(t, `foo`, stack.Outputs["PULUMI_STACK"])
+			assert.Equal(t, `bar`, stack.Outputs["PULUMI_PROJECT"])
+			assert.Equal(t, `foobar`, stack.Outputs["PULUMI_ORGANIZATION"])
+			assert.EqualValues(t, "bazz", stack.Outputs["PULUMI_CONFIG"])
+		},
+	}
+	integration.ProgramTest(t, &testOptions)
+}

pkg/tests/testdata/env-vars/Pulumi.yaml

@@ -0,0 +1,16 @@
+name: project-env-vars
+runtime:
+  name: yaml
+  options:
+    compiler: |
+      sh -c "cat Pulumi.yaml;
+      echo \"\";
+      echo \"outputs:\";
+      echo \"  TEST_ENV_VAR: $TEST_ENV_VAR\";
+      echo \"  PULUMI_STACK: $PULUMI_STACK\";
+      echo \"  PULUMI_PROJECT: $PULUMI_PROJECT\";
+      echo \"  PULUMI_ORGANIZATION: $PULUMI_ORGANIZATION\";
+      echo \"  PULUMI_CONFIG: $PULUMI_CONFIG\";
+      echo \"\";"
+config:
+  foo: "hello world"