Fix(snowflake): treat $ as part of the json path key identifier (#4604)

sqlglot/dialects/snowflake.py

@@ -2,7 +2,7 @@
 
 import typing as t
 
-from sqlglot import exp, generator, parser, tokens, transforms
+from sqlglot import exp, generator, jsonpath, parser, tokens, transforms
 from sqlglot.dialects.dialect import (
     Dialect,
     NormalizationStrategy,
@@ -375,6 +375,10 @@ def quote_identifier(self, expression: E, identify: bool = True) -> E:
 
         return super().quote_identifier(expression, identify=identify)
 
+    class JSONPathTokenizer(jsonpath.JSONPathTokenizer):
+        SINGLE_TOKENS = jsonpath.JSONPathTokenizer.SINGLE_TOKENS.copy()
+        SINGLE_TOKENS.pop("$")
+
     class Parser(parser.Parser):
         IDENTIFY_PIVOT_STRINGS = True
         DEFAULT_SAMPLING_METHOD = "BERNOULLI"

tests/dialects/test_snowflake.py

@@ -77,6 +77,7 @@ def test_snowflake(self):
         self.validate_identity("SELECT MATCH_CONDITION")
         self.validate_identity("SELECT * REPLACE (CAST(col AS TEXT) AS scol) FROM t")
         self.validate_identity("1 /* /* */")
+        self.validate_identity("TO_TIMESTAMP(col, fmt)")
         self.validate_identity(
             "SELECT * FROM table AT (TIMESTAMP => '2024-07-24') UNPIVOT(a FOR b IN (c)) AS pivot_table"
         )
@@ -104,7 +105,10 @@ def test_snowflake(self):
         self.validate_identity(
             "SELECT * FROM DATA AS DATA_L ASOF JOIN DATA AS DATA_R MATCH_CONDITION (DATA_L.VAL > DATA_R.VAL) ON DATA_L.ID = DATA_R.ID"
         )
-        self.validate_identity("TO_TIMESTAMP(col, fmt)")
+        self.validate_identity(
+            "GET_PATH(json_data, '$id')",
+            """GET_PATH(json_data, '["$id"]')""",
+        )
         self.validate_identity(
             "CAST(x AS GEOGRAPHY)",
             "TO_GEOGRAPHY(x)",