mirror of
https://github.com/hibobmaster/matrix_chatgpt_bot.git
synced 2024-10-01 05:35:36 -04:00
45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
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)
|