Add support for colors in MatrixParser

mautrix · Nov 25, 2021 · 9e9c293 · 9e9c293
1 parent b4f9dac
commit 9e9c293
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/mautrix/util/formatter/parser.py b/mautrix/util/formatter/parser.py
@@ -158,6 +158,10 @@ def room_pill_to_fstring(cls, msg: T, room_alias: RoomAlias) -> Optional[T]:
     def custom_node_to_fstring(cls, node: HTMLNode, ctx: RecursionContext) -> Optional[T]:
         return None
 
+    @classmethod
+    def color_to_fstring(cls, node: HTMLNode, ctx: RecursionContext, color: str) -> T:
+        return cls.tag_aware_parse_node(node, ctx)
+
     @classmethod
     def node_to_fstring(cls, node: HTMLNode, ctx: RecursionContext) -> T:
         custom = cls.custom_node_to_fstring(node, ctx)
@@ -181,6 +185,16 @@ def node_to_fstring(cls, node: HTMLNode, ctx: RecursionContext) -> T:
             return cls.link_to_fstring(node, ctx)
         elif node.tag == "p":
             return cls.tag_aware_parse_node(node, ctx).append("\n")
+        elif node.tag in ("font", "span"):
+            try:
+                color = node.attrib["color"]
+            except KeyError:
+                try:
+                    color = node.attrib["data-mx-color"]
+                except KeyError:
+                    color = None
+            if color:
+                return cls.color_to_fstring(node, ctx, color)
         elif node.tag == "pre":
             lang = ""
             try: