Support ALTER GROUP ... ADD/DROP USER

Author: nene

src/cst/AlterAction.ts

@@ -153,7 +153,9 @@ export type AlterRoleAction =
   | AlterActionRename
   | AlterActionSetPostgresqlOption
   | AlterActionSetPostgresqlOptionFromCurrent
-  | AlterActionResetPostgresqlOption;
+  | AlterActionResetPostgresqlOption
+  | AlterActionAddUser
+  | AlterActionDropUser;
 
 export interface AlterActionRename extends BaseNode {
   type: "alter_action_rename";
@@ -541,6 +543,20 @@ export interface AlterActionWithRoleOptions extends BaseNode {
   options: RoleOption[];
 }
 
+// PostgreSQL
+export interface AlterActionAddUser extends BaseNode {
+  type: "alter_action_add_user";
+  addUserKw: [Keyword<"ADD">, Keyword<"USER">];
+  users: ListExpr<Identifier>;
+}
+
+// PostgreSQL
+export interface AlterActionDropUser extends BaseNode {
+  type: "alter_action_drop_user";
+  dropUserKw: [Keyword<"DROP">, Keyword<"USER">];
+  users: ListExpr<Identifier>;
+}
+
 export type AlterColumnAction =
   | AlterActionSetDefault
   | AlterActionDropDefault

src/parser.pegjs

@@ -4480,6 +4480,8 @@ alter_role_action
   / alter_action_set_postgresql_option
   / alter_action_set_postgresql_option_from_current
   / alter_action_reset_postgresql_option
+  / alter_action_add_user
+  / alter_action_drop_user
 
 alter_action_with_role_options
   = kw:WITH options:(__ role_option)+ {
@@ -4526,6 +4528,24 @@ alter_action_reset_postgresql_option
     });
   }
 
+alter_action_add_user
+  = kw:(ADD __ USER __) users:list$ident {
+    return loc({
+      type: "alter_action_add_user",
+      addUserKw: read(kw),
+      users,
+    });
+  }
+
+alter_action_drop_user
+  = kw:(DROP __ USER __) users:list$ident {
+    return loc({
+      type: "alter_action_drop_user",
+      dropUserKw: read(kw),
+      users,
+    });
+  }
+
 drop_role_stmt
   = kw:(DROP __ (ROLE / USER / GROUP) __) ifExistsKw:(if_exists __)? names:list$role_specification {
     return loc({

src/showNode/alter_action.ts

@@ -106,6 +106,8 @@ export const alterActionMap: FullTransformMap<string, AllAlterActionNodes> = {
       node.constraint,
       node.behaviorKw,
     ]),
+  alter_action_add_user: (node) => show([node.addUserKw, node.users]),
+  alter_action_drop_user: (node) => show([node.dropUserKw, node.users]),
 
   // ALTER COLUMN ...
   alter_action_alter_column: (node) =>

test/ddl/role.test.ts

@@ -163,6 +163,14 @@ describe("role", () => {
         testWc("ALTER GROUP my_user RENAME TO new_user");
         testWc(`ALTER GROUP ALL IN DATABASE foo SET something TO 128`);
       });
+
+      it("supports ADD USER", () => {
+        testWc("ALTER GROUP moderator ADD USER mary, john");
+      });
+
+      it("supports DROP USER", () => {
+        testWc("ALTER GROUP moderator DROP USER mary, john");
+      });
     });
 
     describe("DROP GROUP", () => {