Created new DB column for role removal mute

backend/src/data/entities/Mute.ts

@@ -15,4 +15,6 @@ export class Mute {
   @Column() expires_at: string;
 
   @Column() case_id: number;
+
+  @Column() roles_to_restore: string[];
 }

backend/src/migrations/1589701998928-CreateRestoredRolesColumn.ts

@@ -0,0 +1,18 @@
+import { MigrationInterface, QueryRunner, TableColumn } from "typeorm";
+
+export class CreateRestoredRolesColumn1589701998928 implements MigrationInterface {
+  public async up(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.addColumn(
+      "mutes",
+      new TableColumn({
+        name: "roles_to_restore",
+        type: "text",
+        isNullable: true,
+      }),
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<any> {
+    await queryRunner.dropColumn("mutes", "roles_to_restore");
+  }
+}

backend/src/plugins/Mutes.ts

@@ -45,6 +45,8 @@ const ConfigSchema = t.type({
   mute_message: tNullable(t.string),
   timed_mute_message: tNullable(t.string),
   update_mute_message: tNullable(t.string),
+  remove_roles_on_mute: t.union([t.boolean, t.array(t.string)]),
+  role_restore_blacklist: t.union([t.boolean, t.array(t.string)]),
 
   can_view_list: t.boolean,
   can_cleanup: t.boolean,
@@ -103,6 +105,8 @@ export class MutesPlugin extends ZeppelinPlugin<TConfigSchema> {
         mute_message: "You have been muted on the {guildName} server. Reason given: {reason}",
         timed_mute_message: "You have been muted on the {guildName} server for {time}. Reason given: {reason}",
         update_mute_message: "Your mute on the {guildName} server has been updated to {time}.",
+        remove_roles_on_mute: false,
+        role_restore_blacklist: false,
 
         can_view_list: false,
         can_cleanup: false,