from vkbottle.user import Message
from dataclasses import dataclass


@dataclass
class APIMethod:
    message: Message = object
    success: bool = False
    data: dict = dict

    async def edit_messages(
        self,
        text: str = None,
        message_id: int = None,
        attachments: str = None
    ) -> None:
        """Edit message.

        Returns:
            NoneType: None
        """

        if message_id is None:
            message_id = self.message.id

        try:
            await self.message.ctx_api.messages.edit(
                peer_id=self.message.peer_id,
                message_id=message_id,
                keep_forward_messages=True,
                message=text,
                attachment=attachments
            )
        except:
            await self.message.ctx_api.messages.delete(peer_id=self.message.peer_id, message_ids=self.message.id, delete_for_all=1)
            await self.send_messages(text=text, peer_id=self.message.peer_id, attachments=attachments)

    async def send_messages(self, text: str, peer_id: int | None = None, attachments: str = None) -> None:
        """Send message.

        Returns:
            NoneType: None
        """

        try:
            if peer_id is None:
                peer_id = self.message.peer_id

            await self.message.ctx_api.messages.send(
                peer_id=peer_id,
                message=text,
                attachment=attachments,
                random_id=0
            )
        except:
            await self.message.ctx_api.messages.delete(
                peer_id=peer_id,
                message_ids=self.message.id,
                delete_for_all=1
            )