From de7e14f9caefd57c5f2ed0e1874cdb014b4e1248 Mon Sep 17 00:00:00 2001
From: miruka <miruka@disroot.org>
Date: Thu, 15 Apr 2021 07:27:54 -0400
Subject: [PATCH] Color reply prefix in room delegate last message

---
 src/backend/html_markdown.py      | 8 ++++----
 src/gui/MainPane/RoomDelegate.qml | 6 ++++--
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/backend/html_markdown.py b/src/backend/html_markdown.py
index c5822a39..0c9196c2 100644
--- a/src/backend/html_markdown.py
+++ b/src/backend/html_markdown.py
@@ -124,6 +124,7 @@ class HTMLProcessor:
 
     inline_tags = {
         "span", "font", "a", "sup", "sub", "b", "i", "s", "u", "code",
+        "mx-reply",
     }
 
     block_tags = {
@@ -479,7 +480,7 @@ class HTMLProcessor:
 
 
     def _reply_to_inline(self, el: HtmlElement) -> HtmlElement:
-        """Turn <mx-reply> into a plaintext inline form."""
+        """Shorten <mx-reply> to only include the replied to event's sender."""
 
         if el.tag != "mx-reply":
             return el
@@ -487,11 +488,10 @@ class HTMLProcessor:
         try:
             user_id = el.find("blockquote").findall("a")[1].text
             text    = f"↩ {user_id[1:].split(':')[0]}: "  # U+21A9 arrow
-        except (AttributeError, IndexError):
-            text = "↩ "  # U+21A9 arrow
+        except (AttributeError, IndexError) as e:
+            return el
 
         el.clear()
-        el.tag  = "span"
         el.text = text
         return el
 
diff --git a/src/gui/MainPane/RoomDelegate.qml b/src/gui/MainPane/RoomDelegate.qml
index 2d327fa6..e7b9aca0 100644
--- a/src/gui/MainPane/RoomDelegate.qml
+++ b/src/gui/MainPane/RoomDelegate.qml
@@ -127,8 +127,10 @@ HTile {
 
                     return text.replace(
                         /< *span +class=['"]?quote['"]? *>(.+?)<\/ *span *>/g,
-                        `<font color="${subColor}">` +
-                        `$1</font>`,
+                        `<font color="${subColor}">$1</font>`,
+                    ).replace(
+                        /< *mx-reply *>(.+?)<\/ *mx-reply *>/g,
+                        `<font color="${theme.colors.accentText}">$1</font>`,
                     )
                 }
             }