feat!: impelement routers

Author: Damego

interactions_restful/abc.py

@@ -1,7 +1,13 @@
 from abc import ABC, abstractmethod
 from typing import Callable, Coroutine
 
-__all__ = ("BaseApi", )
+__all__ = ("BaseApi", "BaseRouter")
+
+
+class BaseRouter(ABC):
+    @abstractmethod
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+        pass
 
 
 class BaseApi(ABC):
@@ -10,9 +16,17 @@ def __init__(self, host: str, port: int, **kwargs):
         pass
 
     @abstractmethod
-    def add_route(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+        pass
+
+    @staticmethod
+    @abstractmethod
+    def create_router(**kwargs) -> BaseRouter:
+        pass
+
+    def add_router(self, router: BaseRouter):
         pass
 
     @abstractmethod
     async def run(self):
-        pass
+        pass

interactions_restful/backends/fast_api.py

@@ -1,13 +1,26 @@
 from typing import Callable, Coroutine
 
-from fastapi import FastAPI as _FastAPI
+from fastapi import FastAPI as _FastAPI, APIRouter
 from uvicorn import Config, Server
 
-from ..abc import BaseApi
+from ..abc import BaseApi, BaseRouter
 
 __all__ = ("FastAPI", )
 
 
+class FastApiRouter(BaseRouter):
+    def __init__(self, **kwargs):
+        self.api_router = APIRouter(**kwargs)
+
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+        self.api_router.add_api_route(
+            endpoint,
+            coro,
+            methods=[method],
+            **kwargs
+        )
+
+
 class FastAPI(BaseApi):
     def __init__(self, host: str, port: int, **kwargs):
         self.host = host
@@ -16,14 +29,21 @@ def __init__(self, host: str, port: int, **kwargs):
         self.app = _FastAPI(**kwargs)
         self._config = Config(self.app, host=self.host, port=self.port)
 
-    def add_route(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
         self.app.router.add_api_route(
             endpoint,
             coro,
             methods=[method],
             **kwargs
         )
 
+    @staticmethod
+    def create_router(**kwargs) -> BaseRouter:
+        return FastApiRouter(**kwargs)
+
+    def add_router(self, router: FastApiRouter):
+        self.app.include_router(router.api_router)
+
     async def run(self):
         server = Server(config=self._config)
         await server.serve()

interactions_restful/backends/flask_api.py

@@ -4,13 +4,21 @@
     from asgiref.wsgi import WsgiToAsgi
     from hypercorn.config import Config
     from hypercorn.asyncio import serve
-    from flask import Flask
+    from flask import Flask, Blueprint
 except ImportError as e:
     raise ImportError("Flask dependencies weren't installed. Please, install they with [flask] option") from e
 
-from ..abc import BaseApi
+from ..abc import BaseApi, BaseRouter
 
-__all__ = ("FlaskAPI", )
+__all__ = ("FlaskAPI", "FlaskRouter")
+
+
+class FlaskRouter(BaseRouter):
+    def __init__(self, name: str, **kwargs):
+        self.blueprint = Blueprint(name.lower(), name, **kwargs)
+
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+        self.blueprint.route(endpoint, methods=[method], **kwargs)(coro)
 
 
 class FlaskAPI(BaseApi):
@@ -22,8 +30,15 @@ def __init__(self, host: str, port: int, **kwargs):
         self._config = Config()
         self._config.bind = [f"{host}:{port}"]
 
-    def add_route(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
+    def add_endpoint_method(self, coro: Callable[..., Coroutine], endpoint: str, method: str, **kwargs):
         self.app.route(endpoint, methods=[method], **kwargs)(coro)
 
+    @staticmethod
+    def create_router(**kwargs):
+        return FlaskRouter(kwargs.pop("name"), **kwargs)
+
+    def add_router(self, router: FlaskRouter):
+        self.app.register_blueprint(router.blueprint)
+
     async def run(self):
         await serve(WsgiToAsgi(self.app), self._config)

interactions_restful/extension.py

@@ -5,7 +5,7 @@
 
 from interactions import Client
 
-from .abc import BaseApi
+from .abc import BaseApi, BaseRouter
 
 __all__ = ("APIClient", "setup")
 API_TYPE = TypeVar("API_TYPE", bound=BaseApi)
@@ -33,11 +33,20 @@ def _run_api_thread_worker(self):
 
     def _register_extension_routes(self):
         for extension in self.bot.ext.values():
+            has_methods = False
             for _, coro in getmembers(extension, predicate=asyncio.iscoroutinefunction):
                 if not hasattr(coro, "__api__"):
                     continue
+
+                has_methods = True
+
+                if getattr(extension, "router", None) is None:
+                    extension.router = self.client.create_router(name=extension.name)
                 data = coro.__api__
-                self.client.add_route(coro, data["endpoint"], data["method"], **data["kwargs"])
+                extension.router.add_endpoint_method(coro, data["endpoint"], data["method"], **data["kwargs"])
+
+            if has_methods:
+                self.client.add_router(extension.router)
 
 
 def setup(client: Client, api: Type[API_TYPE], host: str, port: int, **kwargs) -> APIClient: