149 lines
4.2 KiB
Python
149 lines
4.2 KiB
Python
import logging
|
|
import re
|
|
from collections.abc import Awaitable
|
|
from enum import Enum
|
|
from functools import wraps
|
|
|
|
LOG: logging.Logger = logging.getLogger(__name__)
|
|
|
|
|
|
# make a enum for route types
|
|
class RouteType(str, Enum):
|
|
HTTP = "http"
|
|
SOCKET = "socket"
|
|
|
|
@classmethod
|
|
def all(cls) -> list[str]:
|
|
return [member.value for member in cls]
|
|
|
|
|
|
class Route:
|
|
"""
|
|
A class to represent an route.
|
|
|
|
Attributes:
|
|
method (str): The HTTP method (GET, POST, etc.).
|
|
path (str): The path for the route.
|
|
name (str): The name of the route.
|
|
handler (Awaitable): The function that handles the route.
|
|
|
|
"""
|
|
|
|
def __init__(self, method: str, path: str, name: str, handler: Awaitable):
|
|
self.method: str = method.upper()
|
|
self.path: str = path
|
|
self.name: str = name
|
|
self.handler: Awaitable = handler
|
|
|
|
|
|
ROUTES: dict[str, dict[str, Route]] = {}
|
|
|
|
|
|
def make_route_name(method: str, path: str) -> str:
|
|
method = method.lower()
|
|
path = path.strip("/")
|
|
|
|
segments: list = []
|
|
for part in path.split("/"):
|
|
part = re.sub(r"[^\w]", "_", part) # remove invalid chars
|
|
if not part:
|
|
part = "part"
|
|
elif part[0].isdigit():
|
|
part = f"p_{part}"
|
|
segments.append(part)
|
|
|
|
return f"{method}:" + ".".join(segments or ["root"])
|
|
|
|
|
|
def route(method: RouteType | str, path: str, name: str | None = None, **kwargs) -> Awaitable:
|
|
"""
|
|
Decorator to mark a method as an HTTP route handler.
|
|
|
|
Args:
|
|
method (RouteType|str): The HTTP method.
|
|
path (str): The path to the route.
|
|
name (str): The name of the route.
|
|
kwargs: Additional keyword arguments.
|
|
|
|
Returns:
|
|
Awaitable: The decorated function.
|
|
|
|
"""
|
|
if not name:
|
|
name = make_route_name(method, path)
|
|
|
|
def decorator(func):
|
|
@wraps(func)
|
|
async def wrapper(*args, **kwargs):
|
|
return await func(*args, **kwargs)
|
|
|
|
route_type: str = RouteType.SOCKET if RouteType.SOCKET == method else RouteType.HTTP
|
|
if route_type not in ROUTES:
|
|
ROUTES[route_type] = {}
|
|
|
|
ROUTES[route_type][name] = Route(method=method.upper(), path=path, name=name, handler=wrapper)
|
|
if "http" == route_type and path.endswith("/") and "/" != path and not kwargs.get("no_slash", False):
|
|
ROUTES[route_type][f"{name}_no_slash"] = Route(
|
|
method=method.upper(), path=path[:-1], name=f"{name}_no_slash", handler=wrapper
|
|
)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|
|
|
|
|
|
def add_route(method: RouteType | str, path: str, handler: Awaitable, name: str | None = None, **kwargs):
|
|
"""
|
|
Decorator to mark a method as an HTTP route handler.
|
|
|
|
Args:
|
|
method (RouteType|str): The HTTP method.
|
|
path (str): The path to the route.
|
|
name (str): The name of the route.
|
|
handler (Awaitable): The function that handles the route.
|
|
kwargs: Additional keyword arguments.
|
|
|
|
"""
|
|
if not name:
|
|
name = make_route_name(method, path)
|
|
|
|
route_type: str = RouteType.SOCKET if RouteType.SOCKET == method else RouteType.HTTP
|
|
|
|
if route_type not in ROUTES:
|
|
ROUTES[route_type] = {}
|
|
|
|
ROUTES[route_type][name] = Route(method=method.upper(), path=path, name=name, handler=handler)
|
|
|
|
if "http" == route_type and path.endswith("/") and "/" != path and not kwargs.get("no_slash", False):
|
|
ROUTES[route_type][f"{name}_no_slash"] = Route(
|
|
method=method.upper(), path=path[:-1], name=f"{name}_no_slash", handler=handler
|
|
)
|
|
|
|
|
|
def get_route(route_type: RouteType, name: str) -> dict[str, Route] | None:
|
|
"""
|
|
Get the route information by name.
|
|
|
|
Args:
|
|
route_type (RouteType): The type of the route (e.g., RouteType.HTTP, RouteType.SOCKET).
|
|
name (str): The name of the route.
|
|
|
|
Returns:
|
|
dict: The route information, or None if not found.
|
|
|
|
"""
|
|
return ROUTES.get(route_type, {}).get(name, None)
|
|
|
|
|
|
def get_routes(route_type: RouteType) -> dict[str, Route]:
|
|
"""
|
|
Get all registered routes.
|
|
|
|
Args:
|
|
route_type (RouteType): The type of the route (e.g., RouteType.HTTP, RouteType.SOCKET).
|
|
|
|
Returns:
|
|
dict[str, dict]: A dictionary of all registered routes.
|
|
|
|
"""
|
|
return ROUTES.get(route_type, {})
|