enhance media protections

matrix-org · Jul 25, 2024 · a6abe18 · a6abe18
1 parent 1ffde55
commit a6abe18
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/protections/MessageIsMedia.ts b/src/protections/MessageIsMedia.ts
@@ -35,10 +35,11 @@ export class MessageIsMedia extends Protection {
 
     public async handleEvent(mjolnir: Mjolnir, roomId: string, event: any): Promise<any> {
         if (event['type'] === 'm.room.message') {
-            const content = event['content'] || {};
+            let content = event['content'] || {};
+            content = content?.["m.new_content"] ?? content;
             const msgtype = content['msgtype'] || 'm.text';
             const formattedBody = content['formatted_body'] || '';
-            const isMedia = msgtype === 'm.image' || msgtype === 'm.video' || msgtype === 'm.sticker' || formattedBody.toLowerCase().includes('<img');
+            let isMedia = msgtype === 'm.image' || msgtype === 'm.video' || msgtype === 'm.sticker' || formattedBody.toLowerCase().includes('<img');
             if (isMedia) {
                 await mjolnir.managementRoomOutput.logMessage(LogLevel.WARN, "MessageIsMedia", `Redacting event from ${event['sender']} for posting an image/video. ${Permalinks.forEvent(roomId, event['event_id'], [new UserID(await mjolnir.client.getUserId()).domain])}`);
                 // Redact the event