feat(postgres,trino,duckdb): added support for current_catalog

sqlglot/dialects/postgres.py

@@ -365,6 +365,7 @@ class Tokenizer(tokens.Tokenizer):
             "BIGSERIAL": TokenType.BIGSERIAL,
             "CONSTRAINT TRIGGER": TokenType.COMMAND,
             "CSTRING": TokenType.PSEUDO_TYPE,
+            "CURRENT_CATALOG": TokenType.CURRENT_CATALOG,
             "DECLARE": TokenType.COMMAND,
             "DO": TokenType.COMMAND,
             "EXEC": TokenType.COMMAND,
@@ -452,6 +453,7 @@ class Parser(parser.Parser):
         NO_PAREN_FUNCTIONS = {
             **parser.Parser.NO_PAREN_FUNCTIONS,
             TokenType.CURRENT_SCHEMA: exp.CurrentSchema,
+            TokenType.CURRENT_CATALOG: exp.CurrentCatalog,
         }
 
         FUNCTION_PARSERS = {
@@ -899,3 +901,7 @@ def arraycontains_sql(self, expression: exp.ArrayContains) -> str:
             )
 
             return self.sql(case_expr)
+
+        def currentcatalog_sql(self, expression: exp.CurrentCatalog) -> str:
+            this = expression.this
+            return self.func("CURRENT_CATALOG", this) if this else "CURRENT_CATALOG"

sqlglot/dialects/trino.py

@@ -20,6 +20,7 @@ class Tokenizer(Presto.Tokenizer):
         KEYWORDS = {
             **Presto.Tokenizer.KEYWORDS,
             "REFRESH": TokenType.REFRESH,
+            "CURRENT_CATALOG": TokenType.CURRENT_CATALOG,
         }
 
     class Parser(Presto.Parser):
@@ -31,6 +32,11 @@ class Parser(Presto.Parser):
             "LISTAGG": lambda self: self._parse_string_agg(),
         }
 
+        NO_PAREN_FUNCTIONS = {
+            **parser.Parser.NO_PAREN_FUNCTIONS,
+            TokenType.CURRENT_CATALOG: exp.CurrentCatalog,
+        }
+
         JSON_QUERY_OPTIONS: parser.OPTIONS_TYPE = {
             **dict.fromkeys(
                 ("WITH", "WITHOUT"),
@@ -120,3 +126,7 @@ def jsonextract_sql(self, expression: exp.JSONExtract) -> str:
                 expression.this,
                 json_path + option + quote + on_condition,
             )
+
+        def currentcatalog_sql(self, expression: exp.CurrentCatalog) -> str:
+            this = expression.this
+            return self.func("CURRENT_CATALOG", this) if this else "CURRENT_CATALOG"

sqlglot/expressions.py

@@ -6356,6 +6356,10 @@ class CurrentUser(Func):
     arg_types = {"this": False}
 
 
+class CurrentCatalog(Func):
+    arg_types = {"this": False}
+
+
 class CurrentRegion(Func):
     arg_types = {}
 

sqlglot/tokens.py

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

tests/dialects/test_postgres.py

@@ -1679,3 +1679,6 @@ def test_interval_span(self):
             day_time_str = "a > INTERVAL '1 00:00' AND TRUE"
             self.validate_identity(day_time_str, "a > INTERVAL '1 00:00' AND TRUE")
             self.assertIsInstance(self.parse_one(day_time_str), exp.And)
+
+    def test_current_catalog(self):
+        self.validate_identity("SELECT CURRENT_CATALOG")

tests/dialects/test_trino.py

@@ -159,3 +159,6 @@ def test_json_value(self):
             self.validate_identity(
                 f"""SELECT JSON_VALUE({json_doc}, 'lax $.price' RETURNING DECIMAL(4, 2) {on_option} ON EMPTY {on_option} ON ERROR) AS price"""
             )
+
+    def test_current_catalog(self):
+        self.validate_identity("SELECT CURRENT_CATALOG")