From bde03c2a756d7dd092f94a4ae0dab314484ef1f6 Mon Sep 17 00:00:00 2001
From: "H. Shay" <hillerys@element.io>
Date: Fri, 26 Jul 2024 11:17:05 -0700
Subject: [PATCH] stronger check that message is edit

---
 src/protections/MessageIsMedia.ts          | 7 +++++--
 test/integration/protectionSettingsTest.ts | 3 ++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/protections/MessageIsMedia.ts b/src/protections/MessageIsMedia.ts
index bf27aac..791db1a 100644
--- a/src/protections/MessageIsMedia.ts
+++ b/src/protections/MessageIsMedia.ts
@@ -36,10 +36,13 @@ export class MessageIsMedia extends Protection {
     public async handleEvent(mjolnir: Mjolnir, roomId: string, event: any): Promise<any> {
         if (event['type'] === 'm.room.message') {
             let content = event['content'] || {};
-            content = content?.["m.new_content"] ?? content;
+            const relation = content["m.relates_to"]
+            if (relation && relation["rel_type"] === "m.replace") {
+                content = content?.["m.new_content"] ?? content;
+            }
             const msgtype = content['msgtype'] || 'm.text';
             const formattedBody = content['formatted_body'] || '';
-            let isMedia = msgtype === 'm.image' || msgtype === 'm.video' || msgtype === 'm.sticker' || formattedBody.toLowerCase().includes('<img');
+            const 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
diff --git a/test/integration/protectionSettingsTest.ts b/test/integration/protectionSettingsTest.ts
index 5244e05..0afcb1c 100644
--- a/test/integration/protectionSettingsTest.ts
+++ b/test/integration/protectionSettingsTest.ts
@@ -179,7 +179,7 @@ describe("Test: Protection settings", function() {
         });
         await reply;
 
-        await client.sendMessage(room, {"m.new_content": {msgtype: "m.image", body: ""}, body: "", msgtype: "m.text"})
+        await client.sendMessage(room, {body: "", msgtype: "m.text", "m.new_content": {msgtype: "m.image", body: ""}, "m.relates_to": {"rel_type": "m.replace"}})
         let reply2 = () => new Promise((resolve, reject) => {
             client.on('room.message', noticeListener(this.mjolnir.managementRoomId, (event) => {
                 if (event.content.body.includes("Redacting event")) {
@@ -190,3 +190,4 @@ describe("Test: Protection settings", function() {
         await reply2;
     });
 });
+