from aiogram import types as t
from aiogram.dispatcher.middlewares import BaseMiddleware

from shared.database import Message
from shared.instances import session
from shared.samples import samples


class MessageMiddleware(BaseMiddleware):
    async def on_pre_process_message(self, msg: t.Message, data: dict) -> None:
        text = msg.text or msg.caption
        if text is not None and not text.startswith("/"):
            with session.begin() as s:
                s.add(
                    Message(
                        chat_id=msg.chat.id,
                        message_id=msg.message_id,
                        user_id=msg.from_user.id,
                        message=text,
                    )
                )
            samples.get(msg.chat.id).feed(text)