PostgreSQL: Support SCHEMA members in ALTER EXTENSION

fuzz/fuzz_seeds/postgres_alter_extension.sql

@@ -6,3 +6,4 @@ ALTER EXTENSION citext DROP VIEW public.v;
 ALTER EXTENSION citext ADD MATERIALIZED VIEW public.mv;
 ALTER EXTENSION citext DROP FOREIGN TABLE public.ft;
 ALTER EXTENSION citext ADD SEQUENCE public.seq;
+ALTER EXTENSION citext DROP SCHEMA other;

src/ast/ddl.rs

@@ -4930,6 +4930,8 @@ pub enum AlterExtensionMember {
     ForeignTable(ObjectName),
     /// `MATERIALIZED VIEW object_name`
     MaterializedView(ObjectName),
+    /// `SCHEMA object_name`
+    Schema(Ident),
     /// `SEQUENCE object_name`
     Sequence(ObjectName),
     /// `TABLE object_name`
@@ -4962,6 +4964,7 @@ impl fmt::Display for AlterExtensionMember {
         match self {
             AlterExtensionMember::ForeignTable(name) => write!(f, "FOREIGN TABLE {name}"),
             AlterExtensionMember::MaterializedView(name) => write!(f, "MATERIALIZED VIEW {name}"),
+            AlterExtensionMember::Schema(name) => write!(f, "SCHEMA {name}"),
             AlterExtensionMember::Sequence(name) => write!(f, "SEQUENCE {name}"),
             AlterExtensionMember::Table(name) => write!(f, "TABLE {name}"),
             AlterExtensionMember::View(name) => write!(f, "VIEW {name}"),

src/parser/mod.rs

@@ -12547,6 +12547,8 @@ impl<'a> Parser<'a> {
             Ok(AlterExtensionMember::MaterializedView(
                 self.parse_object_name(false)?,
             ))
+        } else if self.parse_keyword(Keyword::SCHEMA) {
+            Ok(AlterExtensionMember::Schema(self.parse_identifier()?))
         } else if self.parse_keyword(Keyword::SEQUENCE) {
             Ok(AlterExtensionMember::Sequence(
                 self.parse_object_name(false)?,

tests/sqlparser_postgres.rs

@@ -13531,6 +13531,7 @@ fn alter_extension_member_round_trip_forms(action: &str) {
         format!("ALTER EXTENSION ext_name {action} FOREIGN TABLE ft"),
         format!("ALTER EXTENSION ext_name {action} FOREIGN TABLE schema.ft"),
         format!("ALTER EXTENSION ext_name {action} MATERIALIZED VIEW mv"),
+        format!("ALTER EXTENSION ext_name {action} SCHEMA schema_name"),
         format!("ALTER EXTENSION ext_name {action} SEQUENCE seq"),
         format!("ALTER EXTENSION ext_name {action} TABLE t"),
         format!("ALTER EXTENSION ext_name {action} TABLE schema.t"),
@@ -13671,6 +13672,10 @@ fn parse_alter_extension_add_member() {
             "ALTER EXTENSION ext_name ADD FOO bar",
             "Expected: member object type after ADD or DROP, found: FOO",
         ),
+        (
+            "ALTER EXTENSION ext_name ADD SCHEMA a.b",
+            "Expected: end of statement, found: .",
+        ),
     ] {
         assert_eq!(
             pg_and_generic().parse_sql_statements(sql).unwrap_err(),
@@ -13682,6 +13687,7 @@ fn parse_alter_extension_add_member() {
     for member in [
         "FOREIGN TABLE",
         "MATERIALIZED VIEW",
+        "SCHEMA",
         "SEQUENCE",
         "TABLE",
         "VIEW",