Merge remote-tracking branch 'coder/stevenmasley/cwd' into dynamic-parameters

Author: Emyrk

pkg/iac/scanners/terraform/parser/option.go

@@ -34,6 +34,12 @@ func OptionWithLogger(log *log.Logger) Option {
 	}
 }
 
+func OptionWithWorkingDirectoryPath(cwd string) Option {
+	return func(p *Parser) {
+		p.cwd = cwd
+	}
+}
+
 func OptionsWithTfVars(vars map[string]cty.Value) Option {
 	return func(p *Parser) {
 		p.tfvars = vars

pkg/iac/scanners/terraform/parser/parser.go

@@ -53,6 +53,9 @@ type Parser struct {
 	configsFS         fs.FS
 	skipPaths         []string
 	stepHooks         []EvaluateStepHook
+	// cwd is optional, if left to empty string, 'os.Getwd'
+	// will be used for populating 'path.cwd' in terraform.
+	cwd string
 }
 
 // New creates a new Parser
@@ -295,9 +298,14 @@ func (p *Parser) Load(_ context.Context) (*evaluator, error) {
 		)
 	}
 
-	workingDir, err := os.Getwd()
-	if err != nil {
-		return nil, err
+	var workingDir string
+	if p.cwd != "" {
+		workingDir = p.cwd
+	} else {
+		workingDir, err = os.Getwd()
+		if err != nil {
+			return nil, err
+		}
 	}
 
 	p.logger.Debug("Working directory for module evaluation", log.FilePath(workingDir))

pkg/iac/scanners/terraform/parser/parser_test.go

@@ -3013,3 +3013,26 @@ func TestInstancedLogger(t *testing.T) {
 		t.Log(buf.String()) // Helpful for debugging
 	}
 }
+
+func TestProvidedWorkingDirectory(t *testing.T) {
+	const fakeCwd = "/some/path"
+	fsys := testutil.CreateFS(t, map[string]string{
+		"main.tf": `
+			resource "foo" "bar" {
+			  cwd = path.cwd
+			}
+		`,
+	})
+
+	parser := New(fsys, "", OptionWithWorkingDirectoryPath(fakeCwd))
+	err := parser.ParseFS(t.Context(), ".")
+	require.NoError(t, err)
+
+	modules, err := parser.EvaluateAll(t.Context())
+	require.NoError(t, err)
+
+	require.Len(t, modules, 1)
+	foo := modules[0].GetResourcesByType("foo")[0]
+	attr := foo.GetAttribute("cwd")
+	require.Equal(t, fakeCwd, attr.Value().AsString())
+}