first commit

app/bot/middlewares/__init__.py

@@ -0,0 +1,4 @@
+from app.bot.middlewares.database import DbSessionMiddleware
+from app.bot.middlewares.services import ServicesMiddleware
+
+__all__ = ("DbSessionMiddleware", "ServicesMiddleware")

app/bot/middlewares/database.py

@@ -0,0 +1,28 @@
+from typing import Any, Awaitable, Callable
+
+from aiogram import BaseMiddleware
+from aiogram.types import TelegramObject
+from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
+
+
+class DbSessionMiddleware(BaseMiddleware):
+    """Opens one SQLAlchemy async session per update; commits on success."""
+
+    def __init__(self, session_factory: async_sessionmaker[AsyncSession]) -> None:
+        self._session_factory = session_factory
+
+    async def __call__(
+        self,
+        handler: Callable[[TelegramObject, dict[str, Any]], Awaitable[Any]],
+        event: TelegramObject,
+        data: dict[str, Any],
+    ) -> Any:
+        async with self._session_factory() as session:
+            data["session"] = session
+            try:
+                result = await handler(event, data)
+                await session.commit()
+                return result
+            except Exception:
+                await session.rollback()
+                raise

app/bot/middlewares/services.py

@@ -0,0 +1,25 @@
+from typing import Any, Awaitable, Callable
+
+from aiogram import BaseMiddleware
+from aiogram.types import TelegramObject
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.repositories.user import UserRepository
+from app.services.referral_service import ReferralService
+from app.services.user_service import UserService
+
+
+class ServicesMiddleware(BaseMiddleware):
+    """Builds application services from the request-scoped session (dependency injection)."""
+
+    async def __call__(
+        self,
+        handler: Callable[[TelegramObject, dict[str, Any]], Awaitable[Any]],
+        event: TelegramObject,
+        data: dict[str, Any],
+    ) -> Any:
+        session: AsyncSession = data["session"]
+        users = UserRepository(session)
+        referrals = ReferralService(users)
+        data["user_service"] = UserService(users, referrals)
+        return await handler(event, data)