mirror of
https://github.com/hibobmaster/matrix_chatgpt_bot.git
synced 2024-06-29 04:42:10 +00:00
68 lines
2.0 KiB
Python
68 lines
2.0 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)
|