Handle Snowflake TEXT(n) casts

Author: yoavcloud

src/dialect/snowflake.rs

@@ -1925,3 +1925,64 @@ fn parse_multi_table_insert_when_clauses(
 
     Ok((when_clauses, else_clause))
 }
+
+#[cfg(test)]
+mod tests {
+    use super::SnowflakeDialect;
+    use crate::ast::{DataType, ObjectName};
+    use crate::parser::Parser;
+
+    #[test]
+    fn parses_cast_to_text_with_size_in_snowflake() {
+        Parser::parse_sql(
+            &SnowflakeDialect,
+            r#"SELECT
+_ID::TEXT(16777216) AS _ID,
+TO_TIMESTAMP_NTZ(CREATED_ON) AS CREATED_ON,
+DEPARTMENT::TEXT(16777216) AS DEPARTMENT,
+DOB::TEXT(16777216) AS DOB,
+FIRST_NAME::TEXT(16777216) AS FIRST_NAME,
+GENDER::TEXT(16777216) AS GENDER,
+IS_ACTIVE::BOOLEAN AS IS_ACTIVE,
+LAST_NAME::TEXT(16777216) AS LAST_NAME,
+MIDDLE_NAME::TEXT(16777216) AS MIDDLE_NAME,
+ORGANIZATION::TEXT(16777216) AS ORGANIZATION,
+STATUS::TEXT(16777216) AS STATUS,
+TO_TIMESTAMP_NTZ(UPDATED_ON) AS UPDATED_ON,
+USER_ID::TEXT(16777216) AS USER_ID,
+EMAIL::TEXT(16777216) AS EMAIL,
+JOB_TITLE::TEXT(16777216) AS JOB_TITLE
+FROM INCARE_ANALYTICS.USER_DETAILS;"#,
+        )
+        .unwrap();
+    }
+
+    #[test]
+    fn parses_text_with_size_as_custom_data_type() {
+        let data_type = Parser::new(&SnowflakeDialect)
+            .try_with_sql("TEXT(16777216)")
+            .unwrap()
+            .parse_data_type()
+            .unwrap();
+
+        assert_eq!(
+            data_type,
+            DataType::Custom(
+                ObjectName::from(vec!["TEXT".into()]),
+                vec!["16777216".to_string()],
+            )
+        );
+        assert_eq!(data_type.to_string(), "TEXT(16777216)");
+    }
+
+    #[test]
+    fn preserves_bare_text_data_type() {
+        let data_type = Parser::new(&SnowflakeDialect)
+            .try_with_sql("TEXT")
+            .unwrap()
+            .parse_data_type()
+            .unwrap();
+
+        assert_eq!(data_type, DataType::Text);
+    }
+}

src/parser/mod.rs

@@ -12170,7 +12170,17 @@ 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.value.clone();
+                    if let Some(modifiers) = self.parse_optional_type_modifiers()? {
+                        Ok(DataType::Custom(
+                            ObjectName::from(Ident::new(type_name)),
+                            modifiers,
+                        ))
+                    } else {
+                        Ok(DataType::Text)
+                    }
+                }
                 Keyword::TINYTEXT => Ok(DataType::TinyText),
                 Keyword::MEDIUMTEXT => Ok(DataType::MediumText),
                 Keyword::LONGTEXT => Ok(DataType::LongText),