feat(snowflake)!: annotation support for CURRENT_ROLE. (#6478)

* feat(snowflake)!: annotation support for CURRENT_ROLE. Return type VARCHAR

* feat(snowflake)!: adding tests to mysql and postgres.annotation support for CURRENT_ROLE. Return type VARCHAR

* feat(snowflake)!: adding tests to mysql and postgres.annotation support for CURRENT_ROLE. Return type VARCHAR
added current_role to no_paren list

---------

Co-authored-by: Jo <46752250+georgesittas@users.noreply.github.com>

Author: fivetran-ashashankar

sqlglot/expressions.py

@@ -6360,6 +6360,10 @@ class CurrentRegion(Func):
     arg_types = {}
 
 
+class CurrentRole(Func):
+    arg_types = {}
+
+
 class CurrentRoleType(Func):
     arg_types = {}
 

sqlglot/parser.py

@@ -291,6 +291,7 @@ class Parser(metaclass=_Parser):
         TokenType.CURRENT_USER: exp.CurrentUser,
         TokenType.LOCALTIME: exp.Localtime,
         TokenType.LOCALTIMESTAMP: exp.Localtimestamp,
+        TokenType.CURRENT_ROLE: exp.CurrentRole,
     }
 
     STRUCT_TYPE_TOKENS = {

sqlglot/tokens.py

@@ -276,6 +276,7 @@ class TokenType(AutoName):
     CURRENT_TIME = auto()
     CURRENT_TIMESTAMP = auto()
     CURRENT_USER = auto()
+    CURRENT_ROLE = auto()
     DECLARE = auto()
     DEFAULT = auto()
     DELETE = auto()

sqlglot/typing/snowflake.py

@@ -375,6 +375,7 @@ def _annotate_math_with_float_decfloat(
             exp.Collation,
             exp.CurrentOrganizationUser,
             exp.CurrentRegion,
+            exp.CurrentRole,
             exp.CurrentRoleType,
             exp.CurrentOrganizationName,
             exp.DecompressString,

tests/dialects/test_mysql.py

@@ -214,6 +214,7 @@ def test_identity(self):
         self.validate_identity("""SELECT 'ab' MEMBER OF('[23, "abc", 17, "ab", 10]')""")
         self.validate_identity("""SELECT * FROM foo WHERE 'ab' MEMBER OF(content)""")
         self.validate_identity("SELECT CURRENT_TIMESTAMP(6)")
+        self.validate_identity("SELECT CURRENT_ROLE()")
         self.validate_identity("x ->> '$.name'")
         self.validate_identity("SELECT CAST(`a`.`b` AS CHAR) FROM foo")
         self.validate_identity("SELECT TRIM(LEADING 'bla' FROM ' XXX ')")

tests/dialects/test_postgres.py

@@ -79,6 +79,7 @@ def test_postgres(self):
         self.validate_identity("EXEC AS myfunc @id = 123", check_command_warning=True)
         self.validate_identity("SELECT CURRENT_SCHEMA")
         self.validate_identity("SELECT CURRENT_USER")
+        self.validate_identity("SELECT CURRENT_ROLE")
         self.validate_identity("SELECT * FROM ONLY t1")
         self.validate_identity("SELECT INTERVAL '-1 MONTH'")
         self.validate_identity("SELECT INTERVAL '4.1 DAY'")

tests/dialects/test_snowflake.py

@@ -243,6 +243,7 @@ def test_snowflake(self):
         self.validate_identity("SELECT OBJECT_CONSTRUCT()")
         self.validate_identity("SELECT CURRENT_ORGANIZATION_USER()")
         self.validate_identity("SELECT CURRENT_REGION()")
+        self.validate_identity("SELECT CURRENT_ROLE()")
         self.validate_identity("SELECT CURRENT_ROLE_TYPE()")
         self.validate_identity("SELECT YEAR(CURRENT_TIMESTAMP())")
         self.validate_identity("SELECT YEAROFWEEK(CURRENT_TIMESTAMP())")

tests/fixtures/optimizer/annotate_functions.sql

@@ -2284,6 +2284,10 @@ VARCHAR;
 CURRENT_REGION();
 VARCHAR;
 
+# dialect: snowflake
+CURRENT_ROLE();
+VARCHAR;
+
 # dialect: snowflake
 CURRENT_ROLE_TYPE();
 VARCHAR;