add uvicorn to asgi middleware

Author: bitterpanda63

aikido_zen/context/__init__.py

@@ -21,7 +21,7 @@
 current_context = contextvars.ContextVar("current_context", default=None)
 
 WSGI_SOURCES = ["django", "flask"]
-ASGI_SOURCES = ["quart", "django_async", "starlette", "hypercorn_asgi"]
+ASGI_SOURCES = ["quart", "django_async", "starlette", "hypercorn_asgi", "uvicorn"]
 
 
 def get_current_context():

aikido_zen/sources/servers/uvicorn.py

@@ -1,14 +1,16 @@
 from aikido_zen.helpers.get_argument import get_argument
 from aikido_zen.sinks import before, on_import, patch_function
 from aikido_zen.helpers.logging import logger
+from aikido_zen.sources.functions.asgi_middleware import InternalASGIMiddleware
+
 
 @before
 def _init(func, instance, args, kwargs):
     config = get_argument(args, kwargs, 0, "config")
     if not config.app:
         return
-
     logger.debug("Server detected as: Uvicorn, Wrapping ASGI app with Zen.")
+    config.app = InternalASGIMiddleware(config.app, "uvicorn")
 
 @on_import("uvicorn.server", "uvicorn")
 def patch(m):