allow if and for_each keywords to be used as identifiers (#168)

amplify-education · Aug 6, 2024 · 92f2ee2 · 92f2ee2
1 parent c52be2d
commit 92f2ee2
Show file tree

Hide file tree

Showing 6 changed files with 15 additions and 25 deletions.
diff --git a/hcl2/hcl2.lark b/hcl2/hcl2.lark
@@ -5,9 +5,11 @@ block : identifier (identifier | STRING_LIT)* new_line_or_comment? "{" body "}"
 new_line_and_or_comma: new_line_or_comment | "," | "," new_line_or_comment
 new_line_or_comment: ( /\n/ | /#.*\n/ | /\/\/.*\n/ )+
 
-identifier : /[a-zA-Z_][a-zA-Z0-9_-]*/ | IN | FOR
+identifier : /[a-zA-Z_][a-zA-Z0-9_-]*/ | IN | FOR | IF | FOR_EACH
+IF : "if"
 IN : "in"
 FOR : "for"
+FOR_EACH : "for_each"
 
 ?expression : expr_term | operation | conditional
 

diff --git a/test/helpers/terraform-config-json/resource_in_attribute.json b/test/helpers/terraform-config-json/resource_in_attribute.json
diff --git a/...m-config-json/resource_for_attribute.json → ...nfig-json/resource_keyword_attribute.json b/...m-config-json/resource_for_attribute.json → ...nfig-json/resource_keyword_attribute.json
@@ -5,7 +5,10 @@
         "resource_name": {
           "name": "resource_name",
           "attribute": "attribute_value",
-          "for": "attribute_value2"
+          "if" : "attribute_value2",
+          "in" : "attribute_value3",
+          "for" : "attribute_value4",
+          "for_each" : "attribute_value5"
         }
       }
     }

diff --git a/test/helpers/terraform-config/resource_for_attribute.tf b/test/helpers/terraform-config/resource_for_attribute.tf
diff --git a/test/helpers/terraform-config/resource_in_attribute.tf b/test/helpers/terraform-config/resource_in_attribute.tf
diff --git a/test/helpers/terraform-config/resource_keyword_attribute.tf b/test/helpers/terraform-config/resource_keyword_attribute.tf
@@ -0,0 +1,8 @@
+resource "custom_provider_resource" "resource_name" {
+  name           = "resource_name"
+  attribute      = "attribute_value"
+  if             = "attribute_value2"
+  in             = "attribute_value3"
+  for            = "attribute_value4"
+  for_each       = "attribute_value5"
+}