fix #500: handle athena map data type

src/sqlfmt/node_manager.py

@@ -54,6 +54,7 @@ def raise_on_mismatched_bracket(self, token: Token, last_bracket: Node) -> None:
             "[": "]",
             "case": "end",
             "array<": ">",
+            "map<": ">",
             "table<": ">",
             "struct<": ">",
         }

src/sqlfmt/rules/core.py

@@ -116,8 +116,8 @@
             r"\[",
             r"\(",
             r"\{",
-            # bq usese angle brackets for type definitions for compound types
-            r"(array|table|struct)\s*<",
+            # bq/athena uses angle brackets for type definitions for compound types
+            r"(array|map|table|struct)\s*<",
         ),
         action=partial(actions.add_node_to_buffer, token_type=TokenType.BRACKET_OPEN),
     ),

tests/data/unformatted/130_athena_data_types.sql

@@ -0,0 +1,9 @@
+-- source: https://github.com/tconbeer/sqlfmt/issues/500
+select 
+cast(
+json_parse(foo) as array<
+    map<varchar,varchar>>) 
+from dwh.table
+)))))__SQLFMT_OUTPUT__(((((
+-- source: https://github.com/tconbeer/sqlfmt/issues/500
+select cast(json_parse(foo) as array<map<varchar, varchar>>) from dwh.table

tests/functional_tests/test_general_formatting.py

@@ -46,6 +46,7 @@
         "unformatted/127_more_comments.sql",
         "unformatted/128_double_slash_comments.sql",
         "unformatted/129_duckdb_joins.sql",
+        "unformatted/130_athena_data_types.sql",
         "unformatted/200_base_model.sql",
         "unformatted/201_basic_snapshot.sql",
         "unformatted/202_unpivot_macro.sql",

tests/unit_tests/test_rule.py

@@ -190,6 +190,7 @@ def get_rule(ruleset: List[Rule], rule_name: str) -> Rule:
         (MAIN, "set_operator", "minus"),
         (MAIN, "set_operator", "except"),
         (CORE, "bracket_open", "array<"),
+        (CORE, "bracket_open", "map<"),
         (CORE, "bracket_open", "table\n<"),
         (CORE, "bracket_open", "struct<"),
         (MAIN, "explain", "explain"),