from __future__ import annotations
from abc import ABC, abstractmethod
from dataclasses import dataclass
from typing import Optional, Any, Callable


class Handler(ABC):
    def __init__(self, nxt: Optional["Handler"] = None):
        self._next = nxt

    def set_next(self, nxt: "Handler") -> "Handler":
        self._next = nxt
        return nxt  # позволяет строить цепочку «в линию»

    def handle(self, request: Any) -> Any:
        # Базовая реализация: попытаться обработать здесь,
        # иначе передать дальше.
        result = self._handle_here(request)
        if result is not None:
            return result
        if self._next:
            return self._next.handle(request)
        return None  # никто не справился

    @abstractmethod
    def _handle_here(self, request: Any) -> Optional[Any]:
        ...


@dataclass
class HttpRequest:
    path: str
    headers: dict
    user_id: Optional[int] = None
    payload: Optional[dict] = None


class AuthHandler(Handler):
    def _handle_here(self, req: HttpRequest) -> Optional[Any]:
        token = req.headers.get("Authorization")
        if not token:
            # Нет токена — «решение на месте»: отклоняем и НЕ передаём дальше
            return {"status": 401, "message": "Unauthorized"}
        # валидируем (упростим) и ставим идентификатор
        req.user_id = 42
        return None  # пропускаем дальше


class RateLimitHandler(Handler):
    def __init__(self, check: Callable[[HttpRequest], bool], nxt: Optional[Handler] = None):
        super().__init__(nxt)
        self.check = check

    def _handle_here(self, req: HttpRequest) -> Optional[Any]:
        if not self.check(req):
            return {"status": 429, "message": "Too Many Requests"}
        return None


class RouterHandler(Handler):
    def _handle_here(self, req: HttpRequest) -> Optional[Any]:
        if req.path == "/me" and req.user_id:
            return {"status": 200, "data": {"id": req.user_id}}
        if req.path == "/ping":
            return {"status": 200, "data": "pong"}
        # Не мой маршрут — пропускаю дальше (если есть)
        return None


class NotFoundHandler(Handler):
    def _handle_here(self, req: HttpRequest) -> Optional[Any]:
        # Терминальный обработчик: если дошли сюда — 404
        return {"status": 404, "message": f"Route {req.path} not found"}


# Сборка цепочки
def build_pipeline() -> Handler:
    auth = AuthHandler()
    rate = RateLimitHandler(check=lambda r: True)
    router = RouterHandler()
    notfound = NotFoundHandler()

    auth.set_next(rate).set_next(router).set_next(notfound)
    return auth


if __name__ == "__main__":
    pipeline = build_pipeline()
    print(pipeline.handle(HttpRequest(path="/ping", headers={"Authorization": "Bearer x"})))
    print(pipeline.handle(HttpRequest(path="/me", headers={"Authorization": "ok"})))
    print(pipeline.handle(HttpRequest(path="/unknown", headers={"Authorization": "ok"})))
    print(pipeline.handle(HttpRequest(path="/ping", headers={})))