Allow certain or all roles to be removed upon mute and readded on unmute (#140)

backend/src/data/GuildMutes.ts

@@ -34,7 +34,7 @@ export class GuildMutes extends BaseGuildRepository {
     return mute != null;
   }
 
-  async addMute(userId, expiryTime): Promise<Mute> {
+  async addMute(userId, expiryTime, rolesToRestore?: string[]): Promise<Mute> {
     const expiresAt = expiryTime
       ? moment
           .utc()
@@ -46,12 +46,13 @@ export class GuildMutes extends BaseGuildRepository {
       guild_id: this.guildId,
       user_id: userId,
       expires_at: expiresAt,
+      roles_to_restore: rolesToRestore ?? [],
     });
 
     return (await this.mutes.findOne({ where: result.identifiers[0] }))!;
   }
 
-  async updateExpiryTime(userId, newExpiryTime) {
+  async updateExpiryTime(userId, newExpiryTime, rolesToRestore?: string[]) {
     const expiresAt = newExpiryTime
       ? moment
           .utc()
@@ -59,15 +60,28 @@ export class GuildMutes extends BaseGuildRepository {
           .format("YYYY-MM-DD HH:mm:ss")
       : null;
 
-    return this.mutes.update(
-      {
-        guild_id: this.guildId,
-        user_id: userId,
-      },
-      {
-        expires_at: expiresAt,
-      },
-    );
+    if (rolesToRestore && rolesToRestore.length) {
+      return this.mutes.update(
+        {
+          guild_id: this.guildId,
+          user_id: userId,
+        },
+        {
+          expires_at: expiresAt,
+          roles_to_restore: rolesToRestore,
+        },
+      );
+    } else {
+      return this.mutes.update(
+        {
+          guild_id: this.guildId,
+          user_id: userId,
+        },
+        {
+          expires_at: expiresAt,
+        },
+      );
+    }
   }
 
   async getActiveMutes(): Promise<Mute[]> {

backend/src/data/entities/Mute.ts

@@ -15,4 +15,6 @@ export class Mute {
   @Column({ type: String, nullable: true }) expires_at: string | null;
 
   @Column() case_id: number;
+
+  @Column("simple-array") roles_to_restore: string[];
 }