feat: Add dynamic type handling for variables.tf template generation

- Integrated `formatType` function into the template rendering pipeline.
- Updated `utils/file_utils.go` to include `formatType` in the function map.
- Added support for dynamic type handling in `formatType`, covering:
  - Primitive types: string, number, bool
  - Complex types: list, map, object, and tuple
- Modified `models.Variable` to include `Attributes` for object and tuple types.
- Ensured backward compatibility with existing variable definitions.
- Fixed template error: undefined function `formatType` in `variables.tf.tmpl`.

Tested with:
- Basic and complex variable types from `terraform-generator.json`.
- Terraform generation flow for `resource_group` module.

Author: Samyak Rout

infra_as_code/terraform_generator/backend/models/config.go

@@ -50,12 +50,13 @@ type ModuleOutput struct {
 }
 
 type Variable struct {
-	Type        string      `json:"type"`
-	Description string      `json:"description"`
-	Default     interface{} `json:"default,omitempty"`
-	Sensitive   bool        `json:"sensitive,omitempty"`
-	Value       interface{} `json:"value,omitempty"`
-	Validation  *Validation `json:"validation,omitempty"` // Add Validation field
+	Type        string                 `json:"type"`
+	Description string                 `json:"description"`
+	Default     interface{}            `json:"default,omitempty"`
+	Sensitive   bool                   `json:"sensitive,omitempty"`
+	Value       interface{}            `json:"value,omitempty"`
+	Attributes  map[string]interface{} `json:"attributes,omitempty"` // Add attributes for object/tuple types
+	Validation  *Validation            `json:"validation,omitempty"`
 }
 
 type Validation struct {

infra_as_code/terraform_generator/backend/templates/azure/resource_group/variables.tf.tmpl

@@ -1,12 +1,12 @@
 {{- range $name, $var := .Module.Variables }}
 variable "{{ $name }}" {
   description = "{{ or $var.Description "No description provided" }}"
-  type        = {{ $var.Type }}
+  type = {{ formatType $var.Type $var.Attributes }}
   {{- if $var.Default }}
-  default     = {{ formatDefault $var }}
+  default = {{ formatDefault $var }}
   {{- end }}
   {{- if $var.Sensitive }}
-  sensitive   = true
+  sensitive = true
   {{- end }}
   {{- if $var.Validation }}
   validation {

infra_as_code/terraform_generator/backend/templates/azure/vnet/variables.tf.tmpl

@@ -1,12 +1,12 @@
 {{- range $name, $var := .Module.Variables }}
 variable "{{ $name }}" {
   description = "{{ or $var.Description "No description provided" }}"
-  type        = {{ $var.Type }}
+  type = {{ formatType $var.Type $var.Attributes }}
   {{- if $var.Default }}
-  default     = {{ formatDefault $var }}
+  default = {{ formatDefault $var }}
   {{- end }}
   {{- if $var.Sensitive }}
-  sensitive   = true
+  sensitive = true
   {{- end }}
   {{- if $var.Validation }}
   validation {

infra_as_code/terraform_generator/backend/templates/generic/variables.tf.tmpl

@@ -1,12 +1,12 @@
 {{- range $name, $var := .Variables }}
 variable "{{ $name }}" {
   description = "{{ or $var.Description "No description provided" }}"
-  type        = {{ $var.Type }}
+  type = {{ formatType $var.Type $var.Attributes }}
   {{- if $var.Default }}
-  default     = {{ formatDefault $var }}
+  default = {{ formatDefault $var }}
   {{- end }}
   {{- if $var.Sensitive }}
-  sensitive   = true
+  sensitive = true
   {{- end }}
   {{- if $var.Validation }}
   validation {

infra_as_code/terraform_generator/backend/utils/file_utils.go

@@ -156,12 +156,11 @@ func GenerateFileFromTemplate(templatePath, destinationPath string, data interfa
 			case "bool", "number":
 				return fmt.Sprintf("%v", value)
 			case "string":
-				// Determine if value is an expression or a literal
 				expr, ok := value.(string)
 				if ok && strings.HasPrefix(expr, "var.") {
-					return expr // Expression
+					return expr
 				}
-				return fmt.Sprintf("\"%v\"", value) // Literal
+				return fmt.Sprintf("\"%v\"", value)
 			case "list(string)", "set(string)":
 				list, ok := value.([]interface{})
 				if !ok {
@@ -188,35 +187,12 @@ func GenerateFileFromTemplate(templatePath, destinationPath string, data interfa
 					}
 				}
 				return fmt.Sprintf("{ %s }", strings.Join(entries, ", "))
-			case "object({ provision_vm_agent = bool, enable_automatic_upgrades = bool })",
-				"object({ publisher = string, offer = string, sku = string, version = string })",
-				"object({ name = string, caching = string, create_option = string, managed_disk_type = string })":
-				// Assume value is an expression like var.os_profile_windows_config
-				expr, ok := value.(string)
-				if ok {
-					return expr
-				}
-				return "{}" // Default to empty object if not an expression
-			case "tuple":
-				tuple, ok := value.([]interface{})
-				if !ok {
-					return "[]"
-				}
-				var items []string
-				for _, item := range tuple {
-					switch item.(type) {
-					case string:
-						items = append(items, fmt.Sprintf("\"%v\"", item))
-					default:
-						items = append(items, fmt.Sprintf("%v", item))
-					}
-				}
-				return fmt.Sprintf("[%s]", strings.Join(items, ", "))
 			default:
 				return fmt.Sprintf("%v", value)
 			}
 		},
 		"formatDefault": FormatDefault,
+		"formatType":    formatType, // Add formatType to the funcMap
 	}
 
 	// Parse the template with the function map

infra_as_code/terraform_generator/backend/utils/terraform_utils.go

@@ -85,3 +85,32 @@ func ExtractModuleVariables(module models.Module) map[string]interface{} {
 	}
 	return variables
 }
+
+func formatType(varType string, attributes map[string]interface{}) string {
+	switch varType {
+	case "string", "number", "bool":
+		return varType
+	case "list(string)":
+		return "list(string)"
+	case "map(string)":
+		return "map(string)"
+	case "object":
+		// Construct the object structure
+		var fields []string
+		for name, attrType := range attributes {
+			fields = append(fields, fmt.Sprintf("%s = %s", name, attrType))
+		}
+		return fmt.Sprintf("object({ %s })", strings.Join(fields, ", "))
+	case "tuple":
+		// Construct the tuple structure
+		var elements []string
+		if tuple, ok := attributes["tuple_elements"].([]interface{}); ok {
+			for _, elem := range tuple {
+				elements = append(elements, fmt.Sprintf("%s", elem))
+			}
+		}
+		return fmt.Sprintf("tuple([%s])", strings.Join(elements, ", "))
+	default:
+		return "any"
+	}
+}