don't shutdown a room mjolnir is currently protecting

matrix-org · Oct 22, 2024 · 1b43be9 · 1b43be9
1 parent 93282da
commit 1b43be9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/src/Mjolnir.ts b/src/Mjolnir.ts
@@ -58,7 +58,7 @@ export class Mjolnir {
      */
     private unlistedUserRedactionQueue = new UnlistedUserRedactionQueue();
 
-    private protectedRoomsConfig: ProtectedRoomsConfig;
+    public protectedRoomsConfig: ProtectedRoomsConfig;
     public readonly protectedRoomsTracker: ProtectedRoomsSet;
     private webapis: WebAPIs;
     private openMetrics: OpenMetrics;

diff --git a/src/commands/ShutdownRoomCommand.ts b/src/commands/ShutdownRoomCommand.ts
@@ -15,7 +15,7 @@ limitations under the License.
 */
 
 import { Mjolnir } from "../Mjolnir";
-import { RichReply } from "@vector-im/matrix-bot-sdk";
+import { LogLevel, RichReply } from "@vector-im/matrix-bot-sdk";
 
 // !mjolnir shutdown room <room> [<message>]
 export async function execShutdownRoomCommand(roomId: string, event: any, mjolnir: Mjolnir, parts: string[]) {
@@ -31,6 +31,21 @@ export async function execShutdownRoomCommand(roomId: string, event: any, mjolni
         return;
     }
 
+    let protectedRooms;
+    if (mjolnir.config.protectAllJoinedRooms) {
+        protectedRooms = await mjolnir.client.getJoinedRooms();
+    } else {
+        protectedRooms = mjolnir.protectedRoomsConfig.getExplicitlyProtectedRooms();
+    }
+    if (protectedRooms.includes(target)) {
+        await mjolnir.managementRoomOutput.logMessage(
+            LogLevel.INFO,
+            "ShutdownRoomCommand",
+            "You are attempting to shutdown a room that mjolnir currently protects, aborting.",
+        );
+        return;
+    }
+
     await mjolnir.shutdownSynapseRoom(await mjolnir.client.resolveRoom(target), reason);
     await mjolnir.client.unstableApis.addReactionToEvent(roomId, event["event_id"], "✅");
 }