misskey2telegram/defs/announcement.py

from datetime import datetime
from typing import Optional

from mipac import Route

from init import bot
from misskey_init import MisskeyBot

announcement_template = """<b>Misskey Announcement</b>

<b>{0}</b>

{1}"""


class Announcement:
    def __init__(self, data, misskey_bot: MisskeyBot):
        self.misskey_bot = misskey_bot
        self.id = data["id"]
        self.title = data["title"]
        self.text = data["text"]
        self.is_read = data["is_read"]
        self.image_url = data["image_url"]
        self._created_at = data["created_at"]
        self._updated_at = data["updated_at"]

    @property
    def created_at(self) -> datetime:
        return datetime.strptime(self._created_at, "%Y-%m-%dT%H:%M:%S.%fZ")

    @property
    def updated_at(self) -> Optional[datetime]:
        return datetime.strptime(self._updated_at, "%Y-%m-%dT%H:%M:%S.%fZ") if self._updated_at else None

    async def send_notice(self):
        if not self.image_url:
            await bot.send_message(
                self.misskey_bot.tg_user.chat_id,
                announcement_template.format(
                    self.title,
                    self.text[:1000],
                ),
                reply_to_message_id=self.misskey_bot.tg_user.notice_topic,
            )
        else:
            await bot.send_photo(
                self.misskey_bot.tg_user.chat_id,
                self.image_url,
                caption=announcement_template.format(
                    self.title,
                    self.text[:1000],
                ),
                reply_to_message_id=self.misskey_bot.tg_user.notice_topic,
            )

    async def mark_as_read(self):
        data = {
            "announcementId": self.id,
        }
        await self.misskey_bot.core.http.request(
            Route("POST", "/api/i/read-announcement"),
            json=data, auth=True, lower=True,
        )


async def get_unread_announcements(misskey_bot: MisskeyBot):
    data = {
        "limit": 10,
        "withUnreads": True,
    }
    req = await misskey_bot.core.http.request(
        Route("POST", "/api/announcements"),
        json=data, auth=True, lower=True,
    )
    return [Announcement(i, misskey_bot) for i in req]