alemi/aioappsrv
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Convert discord markdown code blocks

Browse source
This commit is contained in:
git-bruh 2021-01-11 20:20:20 +05:30
parent 8ce7c7b3f3
commit 61a60ea4c7
No known key found for this signature in database
GPG key ID: E1475C50075ADCE6
1 changed files with 37 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
main.py
View file

@ -79,52 +79,60 @@ class MatrixClient(nio.AsyncClient):
await self.close() await self.close()
async def process_emotes(self, message, emotes): async def upload_emote(self, emote_id):
formatted_body = message if emote_id in self.uploaded_emotes.keys():
return self.uploaded_emotes[emote_id]
async def upload_emote(emote_id): emote_url = f"https://cdn.discordapp.com/emojis/{emote_id}"
if emote_id in self.uploaded_emotes.keys():
return self.uploaded_emotes[emote_id]
emote_url = f"https://cdn.discordapp.com/emojis/{emote_id}" emote_file = f"/tmp/{str(uuid.uuid4())}"
emote_file = f"/tmp/{str(uuid.uuid4())}" async with aiohttp.ClientSession() as session:
async with session.get(emote_url) as resp:
emote = await resp.read()
content_type = resp.content_type
async with aiohttp.ClientSession() as session: async with aiofiles.open(emote_file, "wb") as f:
async with session.get(emote_url) as resp: await f.write(emote)
emote = await resp.read()
content_type = resp.content_type
async with aiofiles.open(emote_file, "wb") as f: async with aiofiles.open(emote_file, "rb") as f:
await f.write(emote) resp, maybe_keys = await self.upload(
f, content_type=content_type
)
async with aiofiles.open(emote_file, "rb") as f: await aiofiles.os.remove(emote_file)
resp, maybe_keys = await self.upload(
f, content_type=content_type
)
await aiofiles.os.remove(emote_file) if type(resp) != nio.UploadResponse:
self.logger.warning(
f"Failed to upload emote {emote_id}"
)
return
if type(resp) != nio.UploadResponse: self.uploaded_emotes[emote_id] = resp.content_uri
self.logger.warning(
f"Failed to upload emote {emote_id}"
)
return
self.uploaded_emotes[emote_id] = resp.content_uri return resp.content_uri
return resp.content_uri async def get_fmt_body(self, body, emotes):
# Markdown code blocks
replace = "```"
for i in range(body.count(replace)):
i += 1
if i % 2:
body = body.replace(replace, "<pre><code>", 1)
else:
body = body.replace(replace, "</code></pre>", 1)
for emote in emotes.keys(): for emote in emotes.keys():
emote_ = await upload_emote(emotes[emote]) emote_ = await self.upload_emote(emotes[emote])
if emote_: if emote_:
emote = f":{emote}:" emote = f":{emote}:"
formatted_body = formatted_body.replace( body = body.replace(
emote, f"""<img alt=\"{emote}\" title=\"{emote}\" emote, f"""<img alt=\"{emote}\" title=\"{emote}\"
height=\"32\" src=\"{emote_}\" data-mx-emoticon />""" height=\"32\" src=\"{emote_}\" data-mx-emoticon />"""
) )
return formatted_body return body
async def message_send(self, message, channel_id, emotes, async def message_send(self, message, channel_id, emotes,
reply_id=None, edit_id=None): reply_id=None, edit_id=None):
@ -133,7 +141,7 @@ height=\"32\" src=\"{emote_}\" data-mx-emoticon />"""
content = { content = {
"body": message, "body": message,
"format": "org.matrix.custom.html", "format": "org.matrix.custom.html",
"formatted_body": await self.process_emotes(message, emotes), "formatted_body": await self.get_fmt_body(message, emotes),
"msgtype": "m.text" "msgtype": "m.text"
} }