from aiogram import BaseMiddleware
from typing import Any, Callable

from services.orders import OrderService


class DIMiddleware(BaseMiddleware):
    def __init__(self, order_service: OrderService):
        self._order_service = order_service

    async def __call__(
        self,
        handler: Callable,
        event: Any,
        data: dict[str, Any],
    ) -> Any:
        data["order_service"] = self._order_service

        return await handler(event, data)