diff --git a/echo.py b/echo.py index f4cc194..045129b 100644 --- a/echo.py +++ b/echo.py @@ -2,7 +2,7 @@ from typing import Optional from time import time from html import escape -from mautrix.types import TextMessageEventContent, MessageType, Format +from mautrix.types import TextMessageEventContent, MessageType, Format, RelatesTo, RelationType from maubot import Plugin, MessageEvent from maubot.handlers import command @@ -46,18 +46,19 @@ class EchoBot(Plugin): body=f"{evt.sender}: Pong! (ping {text_message} {pretty_diff} to arrive)", formatted_body=f"{evt.sender}: Pong! " f"(ping {html_message} " - f"{pretty_diff} to arrive)") + f"{pretty_diff} to arrive)", + relates_to=RelatesTo( + rel_type=RelationType("xyz.maubot.pong"), + event_id=evt.event_id, + )) + pong_from = evt.sender.split(":", 1)[1] + content.relates_to["from"] = pong_from + content.relates_to["ms"] = diff content["pong"] = { "ms": diff, - "from": evt.sender.split(":", 1)[1], + "from": pong_from, "ping": evt.event_id, } - content["m.relates_to"] = { - "rel_type": "xyz.maubot.pong", - "event_id": evt.event_id, - "from": evt.sender.split(":", 1)[1], - "ms": diff, - } await evt.respond(content) @command.new("echo", help="Repeat a message")