fix formatter to recursively format objects inside function call arguments

Author: kkozik-amplify

hcl2/formatter.py

@@ -17,6 +17,7 @@
     TupleRule,
 )
 from hcl2.rules.expressions import ExprTermRule
+from hcl2.rules.functions import FunctionCallRule
 from hcl2.rules.for_expressions import (
     ForTupleExprRule,
     ForObjectExprRule,
@@ -175,9 +176,19 @@ def format_expression(self, rule: ExprTermRule, indent_level: int = 0):
         elif isinstance(rule.expression, ForObjectExprRule):
             self.format_forobjectexpr(rule.expression, indent_level)
 
+        elif isinstance(rule.expression, FunctionCallRule):
+            self.format_function_call(rule.expression, indent_level)
+
         elif isinstance(rule.expression, ExprTermRule):
             self.format_expression(rule.expression, indent_level)
 
+    def format_function_call(self, rule: FunctionCallRule, indent_level: int = 0):
+        """Format a function call by recursively formatting its arguments."""
+        if rule.arguments is not None:
+            for arg in rule.arguments.arguments:
+                if isinstance(arg, ExprTermRule):
+                    self.format_expression(arg, indent_level)
+
     def format_fortupleexpr(self, expression: ForTupleExprRule, indent_level: int = 0):
         """Format a for-tuple expression with newlines around clauses."""
         for child in expression.children:

test/integration/hcl2_original/function_objects.tf

@@ -0,0 +1,22 @@
+variable "object" {
+  type = object({
+    key   = string
+    value = string
+  })
+}
+
+variable "nested" {
+  type = map(object({
+    name    = string
+    enabled = bool
+  }))
+}
+
+variable "multi_arg" {
+  default = merge({
+    a = 1
+    b = 2
+  }, {
+    c = 3
+  })
+}

test/integration/hcl2_reconstructed/function_objects.tf

@@ -0,0 +1,24 @@
+variable "object" {
+  type = object({
+    key   = string,
+    value = string
+  })
+}
+
+
+variable "nested" {
+  type = map(object({
+    name    = string,
+    enabled = bool
+  }))
+}
+
+
+variable "multi_arg" {
+  default = merge({
+    a = 1,
+    b = 2
+  }, {
+    c = 3
+  })
+}

test/integration/json_reserialized/function_objects.json

@@ -0,0 +1,22 @@
+{
+  "variable": [
+    {
+      "\"object\"": {
+        "type": "${object({key = string, value = string})}",
+        "__is_block__": true
+      }
+    },
+    {
+      "\"nested\"": {
+        "type": "${map(object({name = string, enabled = bool}))}",
+        "__is_block__": true
+      }
+    },
+    {
+      "\"multi_arg\"": {
+        "default": "${merge({a = 1, b = 2}, {c = 3})}",
+        "__is_block__": true
+      }
+    }
+  ]
+}

test/integration/json_serialized/function_objects.json

@@ -0,0 +1,22 @@
+{
+  "variable": [
+    {
+      "\"object\"": {
+        "type": "${object({key = string, value = string})}",
+        "__is_block__": true
+      }
+    },
+    {
+      "\"nested\"": {
+        "type": "${map(object({name = string, enabled = bool}))}",
+        "__is_block__": true
+      }
+    },
+    {
+      "\"multi_arg\"": {
+        "default": "${merge({a = 1, b = 2}, {c = 3})}",
+        "__is_block__": true
+      }
+    }
+  ]
+}