feat(optimizer)!: annotate type for snowflake func TO_BOOLEAN (#6481)

* chore(optimizer)!: annotate type for snowflake func TO_BOOLEAN

* formatting

* remove _sql_names from ToBoolean expression

* remove TO_BOOLEAN from oracle function parsing

---------

Co-authored-by: Michael Lee <[email protected]>

Author: fivetran-MichaelLee

sqlglot/dialects/oracle.py

@@ -132,6 +132,7 @@ class Parser(parser.Parser):
                 unabbreviate=False,
             ),
         }
+        FUNCTIONS.pop("TO_BOOLEAN")
 
         NO_PAREN_FUNCTION_PARSERS = {
             **parser.Parser.NO_PAREN_FUNCTION_PARSERS,

sqlglot/expressions.py

@@ -5876,6 +5876,10 @@ class ToArray(Func):
     pass
 
 
+class ToBoolean(Func):
+    pass
+
+
 # https://materialize.com/docs/sql/types/list/
 class List(Func):
     arg_types = {"expressions": False}

sqlglot/typing/snowflake.py

@@ -239,6 +239,7 @@ def _annotate_math_with_float_decfloat(
             exp.IsNullValue,
             exp.Search,
             exp.SearchIp,
+            exp.ToBoolean,
         }
     },
     **{

tests/dialects/test_snowflake.py

@@ -85,6 +85,8 @@ def test_snowflake(self):
         self.validate_identity("SELECT BOOLAND(1, -2)")
         self.validate_identity("SELECT BOOLXOR(2, 0)")
         self.validate_identity("SELECT BOOLOR(1, 0)")
+        self.validate_identity("SELECT TO_BOOLEAN('true')")
+        self.validate_identity("SELECT TO_BOOLEAN(1)")
         self.validate_identity("SELECT IS_NULL_VALUE(GET_PATH(payload, 'field'))")
         self.validate_identity("SELECT RTRIMMED_LENGTH(' ABCD ')")
         self.validate_identity("SELECT HEX_DECODE_STRING('48656C6C6F')")

tests/fixtures/optimizer/annotate_functions.sql

@@ -1908,6 +1908,18 @@ BOOLEAN;
 BOOLOR_AGG(tbl.bool_col);
 BOOLEAN;
 
+# dialect: snowflake
+TO_BOOLEAN('true');
+BOOLEAN;
+
+# dialect: snowflake
+TO_BOOLEAN(1);
+BOOLEAN;
+
+# dialect: snowflake
+TO_BOOLEAN(tbl.varchar_col);
+BOOLEAN;
+
 # dialect: snowflake
 ARRAY_AGG(tbl.bin_col);
 ARRAY;