nonsensebot/app/modules/misskey_preview.py

"""
This module takes misskey links (e.g.) and provides a preview in a reply.
"""
import niobot
import httpx
import typing
import textwrap
from urllib.parse import urlparse
if typing.TYPE_CHECKING:
    from ..main import TortoiseIntegratedBot


class MisskeyPreviewModule(niobot.Module):
    bot: "TortoiseIntegratedBot"

    @niobot.event("message")
    async def on_message(self, room: niobot.MatrixRoom, event: niobot.RoomMessage):
        supported_prefixes = self.bot.cfg.get("misskey_preview", {})
        supported_prefixes = supported_prefixes.get("urls", ["https://fedi.transgender.ing/notes"])
        if not isinstance(event, niobot.RoomMessageText):
            return

        sent = []
        async with httpx.AsyncClient() as client:
            for item in event.body.split():
                if not event.body.startswith(tuple(supported_prefixes)):
                    return
                parsed = urlparse(item)
                post_id = parsed.path.split("/")[-1]
                if post_id in sent:
                    continue
                elif len(sent) >= 5:
                    break
                resp = await client.get("https://%s/api/v1/statuses/%s" % (parsed.netloc, post_id))
                if resp.status_code != 200:
                    continue
                data = resp.json()
                username = data["user"]["fqn"]
                if not data.get("text"):
                    continue

                text = textwrap.shorten(data["text"], width=1000)
                rendered = self.bot._markdown_to_html(text)
                text_body = "<blockquote>%s</blockquote>" % rendered
                body = "<a href=\"%s\">@%s:</a><br>%s" % (
                    "https://%s/@%s" % (parsed.netloc, username),
                    username,
                    text_body,
                )
                await self.bot.send_message(
                    room,
                    body,
                    reply_to=event,
                    content_type="html.raw",
                    override={"body": f"@{username}: {data['text']!r}"}
                )
                sent.append(post_id)