add serialize helper func for debugging

Author: Emyrk

hclext/serialize.go

@@ -0,0 +1,39 @@
+package hclext
+
+import (
+	"bytes"
+
+	"github.com/aquasecurity/trivy/pkg/iac/terraform"
+	"github.com/hashicorp/hcl/v2/hclwrite"
+)
+
+// Serialize makes a best effort to serialize the blocks back to HCL.
+// This is not guaranteed to be the same as the original HCL, and should
+// only be used for debugging.
+func Serialize(blocks terraform.Blocks) string {
+	var buf bytes.Buffer
+
+	f := hclwrite.NewFile()
+	for _, b := range blocks {
+		f.Body().AppendBlock(hclwriteBlock(b))
+	}
+
+	_, _ = f.WriteTo(&buf)
+	return buf.String()
+}
+
+func hclwriteBlock(block *terraform.Block) *hclwrite.Block {
+	b := hclwrite.NewBlock(block.TypeLabel(), block.Labels())
+	for _, attr := range block.Attributes() {
+		// If the value is "null", it would be better to write the
+		// underlying expression. That gets a bit complicated with
+		// hclwrite, so just keep them as null for now.
+		b.Body().SetAttributeValue(attr.Name(), attr.Value())
+	}
+
+	for _, subBlock := range block.AllBlocks() {
+		b.Body().AppendBlock(hclwriteBlock(subBlock))
+	}
+
+	return b
+}

preview_test.go

@@ -306,7 +306,11 @@ func Test_Extract(t *testing.T) {
 				Owner:           types.WorkspaceOwner{},
 			},
 			unknownTags: []string{},
-			params:      map[string]assertParam{},
+			params: map[string]assertParam{
+				"ref": ap().value("Index 2").
+					optVals("Index 0", "Index 1", "Index 2"),
+				"ref_count": ap().value("Index 2"),
+			},
 		},
 		{
 			name:    "defexpression",

testdata/count/main.tf

@@ -1,28 +1,51 @@
 terraform {
   required_providers {
     coder = {
-      source  = "coder/coder"
+      source = "coder/coder"
+    }
+    null = {
+      source  = "hashicorp/null"
+      version = "3.2.2"
     }
   }
 }
 
-data "coder_parameter" "one" {
-    count = 1
-    name  = "one"
-    type  = "number"
-    default = 1
+data "null_data_source" "exists" {
+  # for_each = toset(["one", "two", "three"])
+  count = 3
+  inputs = {
+    foo = "Index ${count.index}"
+  }
 }
 
-data "coder_parameter" "two" {
-    count = data.coder_parameter.one[0].value
-    name  = "two"
-    type  = "string"
-    default = "two"
+data "coder_parameter" "ref" {
+  name = "ref"
+  type = "string"
+  default = data.null_data_source.exists[2].inputs.foo
+  count = 1
+
+  option {
+    name  = "exists 1"
+    value = data.null_data_source.exists[0].inputs.foo
+  }
+  option {
+    name  = "exists 2"
+    value = data.null_data_source.exists[1].inputs.foo
+  }
+  option {
+    name  = "exists 3"
+    value = data.null_data_source.exists[2].inputs.foo
+  }
 }
 
-data "coder_parameter" "three" {
-    count = 1
-    name  = "three"
-    type  = "number"
-    default = data.coder_parameter.one[0].value
-}
+data "coder_parameter" "ref_count" {
+  name = "ref_count"
+  type = "string"
+  default = data.coder_parameter.ref[0].default
+  count = 1
+
+  option {
+    name  = "Only"
+    value = data.coder_parameter.ref[0].default
+  }
+}