Fix Snowflake TEXT(n) cast parsing

Author: yoavcloud

src/parser/mod.rs

@@ -11909,6 +11909,7 @@ impl<'a> Parser<'a> {
         self.advance_token();
         let next_token = self.get_current_token();
         let next_token_index = self.get_current_index();
+        let next_token_span = next_token.span;
 
         let mut trailing_bracket: MatchedTrailingBracket = false.into();
         let mut data = match &next_token.token {
@@ -12170,7 +12171,14 @@ impl<'a> Parser<'a> {
                     self.expect_token(&Token::RParen)?;
                     Ok(DataType::FixedString(character_length))
                 }
-                Keyword::TEXT => Ok(DataType::Text),
+                Keyword::TEXT => {
+                    let type_name = w.to_ident(next_token_span);
+                    if let Some(modifiers) = self.parse_optional_type_modifiers()? {
+                        Ok(DataType::Custom(type_name.into(), modifiers))
+                    } else {
+                        Ok(DataType::Text)
+                    }
+                }
                 Keyword::TINYTEXT => Ok(DataType::TinyText),
                 Keyword::MEDIUMTEXT => Ok(DataType::MediumText),
                 Keyword::LONGTEXT => Ok(DataType::LongText),

tests/sqlparser_snowflake.rs

@@ -5010,3 +5010,40 @@ fn test_select_dollar_column_from_stage() {
     // With table function args, without alias
     snowflake().verified_stmt("SELECT $1, $2 FROM @mystage1(file_format => 'myformat')");
 }
+
+#[test]
+fn test_text_cast_with_length_modifier() {
+    let select = snowflake().verified_only_select("SELECT col::TEXT(16777216) AS col FROM t");
+
+    match &select.projection[0] {
+        SelectItem::ExprWithAlias {
+            expr: Expr::Cast { data_type, .. },
+            alias,
+        } => {
+            assert_eq!(alias, &Ident::new("col"));
+            assert_eq!(
+                data_type,
+                &DataType::Custom(
+                    ObjectName::from(vec![Ident::new("TEXT")]),
+                    vec!["16777216".to_string()],
+                )
+            );
+        }
+        _ => unreachable!(),
+    }
+
+    snowflake().one_statement_parses_to(
+        "SELECT col::TEXT(16777216) AS col FROM t",
+        "SELECT col::TEXT(16777216) AS col FROM t",
+    );
+}
+
+#[test]
+fn test_plain_text_data_type_still_parses_as_text() {
+    match snowflake().verified_stmt("CREATE TABLE t (c TEXT)") {
+        Statement::CreateTable(CreateTable { columns, .. }) => {
+            assert_eq!(columns[0].data_type, DataType::Text);
+        }
+        _ => unreachable!(),
+    }
+}