matrix_chatgpt_bot/send_message.py

from nio import AsyncClient
import re
import markdown


async def send_room_message(
    client: AsyncClient,
    room_id: str,
    reply_message: str,
    sender_id: str = "",
    user_message: str = "",
    reply_to_event_id: str = "",
    markdown_formatted: bool = False,
) -> None:
    NORMAL_BODY = content = {
        "msgtype": "m.text",
        "body": reply_message,
    }
    if reply_to_event_id == "":
        if markdown_formatted:
            # only format message contains multiline codes, *, |
            if re.search(r"```|\*|\|", reply_message) is not None:
                content = {
                    "msgtype": "m.text",
                    "body": reply_message,
                    "format": "org.matrix.custom.html",
                    "formatted_body": markdown.markdown(
                        reply_message, extensions=["nl2br", "tables", "fenced_code"]
                    ),
                }
            else:
                content = NORMAL_BODY

        else:
            content = NORMAL_BODY
    else:
        body = r"> <" + sender_id + r"> " + user_message + r"\n\n" + reply_message
        format = r"org.matrix.custom.html"
        formatted_body = (
            r'<mx-reply><blockquote><a href="https://matrix.to/#/'
            + room_id
            + r"/"
            + reply_to_event_id
            + r'">In reply to</a> <a href="https://matrix.to/#/'
            + sender_id
            + r'">'
            + sender_id
            + r"</a><br>"
            + user_message
            + r"</blockquote></mx-reply>"
            + reply_message
        )

        content = {
            "msgtype": "m.text",
            "body": body,
            "format": format,
            "formatted_body": formatted_body,
            "m.relates_to": {"m.in_reply_to": {"event_id": reply_to_event_id}},
        }
    await client.room_send(
        room_id,
        message_type="m.room.message",
        content=content,
        ignore_unverified_devices=True,
    )
    await client.room_typing(room_id, typing_state=False)