starlark/types: evaluate include main thread predeclared variables

Author: mcuadros

starlark/runtime/runtime.go

@@ -51,6 +51,16 @@ type Runtime struct {
 // NewRuntime returns a new Runtime for the given terraform.PluginManager.
 func NewRuntime(pm *terraform.PluginManager) *Runtime {
 	tf := types.NewTerraform(pm)
+	predeclared := starlark.StringDict{}
+	predeclared["tf"] = tf
+	predeclared["provisioner"] = types.BuiltinProvisioner()
+	predeclared["backend"] = types.BuiltinBackend()
+	predeclared["validate"] = types.BuiltinValidate()
+	predeclared["hcl"] = types.BuiltinHCL()
+	predeclared["fn"] = types.BuiltinFunctionAttribute()
+	predeclared["evaluate"] = types.BuiltinEvaluate(predeclared)
+	predeclared["struct"] = starlark.NewBuiltin("struct", starlarkstruct.Make)
+	predeclared["module"] = starlark.NewBuiltin("module", starlarkstruct.MakeModule)
 
 	return &Runtime{
 		Terraform:   tf,
@@ -70,16 +80,7 @@ func NewRuntime(pm *terraform.PluginManager) *Runtime {
 			"time":            time.LoadModule,
 			"http":            http.LoadModule,
 		},
-		predeclared: starlark.StringDict{
-			"tf":          tf,
-			"provisioner": types.BuiltinProvisioner(),
-			"backend":     types.BuiltinBackend(),
-			"hcl":         types.BuiltinHCL(),
-			"fn":          types.BuiltinFunctionAttribute(),
-			"evaluate":    types.BuiltinEvaluate(),
-			"struct":      starlark.NewBuiltin("struct", starlarkstruct.Make),
-			"module":      starlark.NewBuiltin("module", starlarkstruct.MakeModule),
-		},
+		predeclared: predeclared,
 	}
 }
 

starlark/types/evaluate.go

@@ -25,10 +25,9 @@ import (
 //             Defines the predeclared context for the execution. Execution does
 //             not modify this dictionary
 //
-func BuiltinEvaluate() starlark.Value {
+func BuiltinEvaluate(predeclared starlark.StringDict) starlark.Value {
 	return starlark.NewBuiltin("evaluate", func(t *starlark.Thread, _ *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
 		var raw starlark.String
-		predeclared := starlark.StringDict{}
 
 		switch len(args) {
 		case 2:

starlark/types/provider_test.go

@@ -13,6 +13,7 @@ import (
 	"github.com/mcuadros/ascode/terraform"
 	"go.starlark.net/resolve"
 	"go.starlark.net/starlark"
+	"go.starlark.net/starlarkstruct"
 )
 
 var id int
@@ -75,15 +76,16 @@ func doTestPrint(t *testing.T, filename string, print func(*starlark.Thread, str
 
 	test.SetReporter(thread, t)
 
-	predeclared := starlark.StringDict{
-		"provisioner": BuiltinProvisioner(),
-		"backend":     BuiltinBackend(),
-		"hcl":         BuiltinHCL(),
-		"fn":          BuiltinFunctionAttribute(),
-		"evaluate":    BuiltinEvaluate(),
-		"validate":    BuiltinValidate(),
-		"tf":          NewTerraform(pm),
-	}
+	predeclared := starlark.StringDict{}
+	predeclared["tf"] = NewTerraform(pm)
+	predeclared["provisioner"] = BuiltinProvisioner()
+	predeclared["backend"] = BuiltinBackend()
+	predeclared["hcl"] = BuiltinHCL()
+	predeclared["validate"] = BuiltinValidate()
+	predeclared["fn"] = BuiltinFunctionAttribute()
+	predeclared["evaluate"] = BuiltinEvaluate(predeclared)
+	predeclared["struct"] = starlark.NewBuiltin("struct", starlarkstruct.Make)
+	predeclared["module"] = starlark.NewBuiltin("module", starlarkstruct.MakeModule)
 
 	_, err := starlark.ExecFile(thread, filename, nil, predeclared)
 	if err != nil {

starlark/types/testdata/evaluate/test.star

@@ -1 +1,2 @@
-foo = bar
+foo = bar
+tf