omserv 0.0.0.dev280__py3-none-any.whl → 0.0.0.dev282__py3-none-any.whl

omserv/apps/base.py

@@ -1,13 +1,13 @@
 import contextvars
 import typing as ta
 
-from omlish.http.asgi import AsgiScope
+from omlish.http import asgi
 
 
 ##
 
 
-SCOPE: contextvars.ContextVar[AsgiScope] = contextvars.ContextVar('scope')
+SCOPE: contextvars.ContextVar[asgi.Scope] = contextvars.ContextVar('scope')
 
 
 ##

omserv/apps/inject.py

@@ -1,18 +1,17 @@
 import typing as ta
 
 from omlish import inject as inj
+from omlish.http import asgi
 from omlish.http import sessions
-from omlish.http.asgi import AsgiApp
-from omlish.http.asgi import AsgiScope
 
 from .base import SCOPE
 from .markers import AppMarker
 from .markers import AppMarkerProcessor
 from .markers import NopAppMarkerProcessor
-from .markers import get_app_markers
-from .routes import Handler_
 from .routes import Route
+from .routes import RouteHandlerHolder
 from .routes import _HandlesAppMarker
+from .routes import build_route_handler_map
 from .sessions import SESSION
 from .sessions import _WithSessionAppMarker
 from .sessions import _WithSessionAppMarkerProcessor
@@ -20,10 +19,10 @@ from .templates import JinjaNamespace
 from .templates import JinjaTemplates
 
 
-def bind_handler(hc: type[Handler_]) -> inj.Elemental:
+def bind_route_handler_class(hc: type[RouteHandlerHolder]) -> inj.Elemental:
     return inj.as_elements(
         inj.bind(hc, singleton=True),
-        inj.set_binder[Handler_]().bind(hc),
+        inj.set_binder[RouteHandlerHolder]().bind(hc),
     )
 
 
@@ -35,20 +34,13 @@ def bind_app_marker_processor(mc: type[AppMarker], pc: type[AppMarkerProcessor])
 
 
 def _build_route_handler_map(
-        handlers: ta.AbstractSet[Handler_],
+        handler_holders: ta.AbstractSet[RouteHandlerHolder],
         processors: ta.Mapping[type[AppMarker], AppMarkerProcessor],
-) -> ta.Mapping[Route, AsgiApp]:
-    route_handlers: dict[Route, AsgiApp] = {}
-    for h in handlers:
-        for rh in h.get_route_handlers():
-            app = rh.handler
-            markers = get_app_markers(rh.handler)
-            for m in markers:
-                mp = processors[type(m)]
-                if mp is not None:
-                    app = mp(app)
-            route_handlers[rh.route] = app
-    return route_handlers
+) -> ta.Mapping[Route, asgi.App]:
+    return build_route_handler_map(
+        handler_holders,
+        processors,
+    )
 
 
 def bind_route_handler_map() -> inj.Elemental:
@@ -60,7 +52,7 @@ def bind_route_handler_map() -> inj.Elemental:
 
 def bind() -> inj.Elemental:
     return inj.as_elements(
-        inj.bind(ta.Callable[[], AsgiScope], to_const=SCOPE.get),
+        inj.bind(ta.Callable[[], asgi.Scope], to_const=SCOPE.get),
         inj.bind(ta.Callable[[], sessions.Session], to_const=SESSION.get),
 
         ##
@@ -72,7 +64,7 @@ def bind() -> inj.Elemental:
 
         ##
 
-        inj.set_binder[Handler_](),
+        inj.set_binder[RouteHandlerHolder](),
     )
 
 

omserv/apps/markers.py

@@ -2,7 +2,7 @@ import abc
 import typing as ta
 
 from omlish import lang
-from omlish.http.asgi import AsgiApp
+from omlish.http import asgi
 from omlish.metadata import ObjectMetadata
 from omlish.metadata import append_object_metadata
 from omlish.metadata import get_object_metadata
@@ -35,10 +35,16 @@ def get_app_markers(obj: ta.Any) -> ta.Sequence[AppMarker]:
 
 class AppMarkerProcessor(lang.Abstract):
     @abc.abstractmethod
-    def __call__(self, app: AsgiApp) -> AsgiApp:
+    def process_app(self, app: asgi.App) -> asgi.App:
         raise NotImplementedError
 
 
 class NopAppMarkerProcessor(AppMarkerProcessor, lang.Final):
-    def __call__(self, app: AsgiApp) -> AsgiApp:
+    def process_app(self, app: asgi.App) -> asgi.App:
         return app
+
+
+##
+
+
+AppMarkerProcessorMap: ta.TypeAlias = ta.Mapping[type[AppMarker], AppMarkerProcessor]

omserv/apps/routes.py

@@ -1,3 +1,8 @@
+"""
+TODO:
+ - include route in process_app?
+  - or, deduplicate handlers and only process apps once?
+"""
 import contextlib
 import dataclasses as dc
 import logging
@@ -5,18 +10,15 @@ import typing as ta
 
 from omlish import check
 from omlish import lang
-from omlish.http.asgi import AsgiApp
-from omlish.http.asgi import AsgiApp_
-from omlish.http.asgi import AsgiRecv
-from omlish.http.asgi import AsgiScope
-from omlish.http.asgi import AsgiSend
-from omlish.http.asgi import send_response
-from omlish.http.asgi import stub_lifespan
+from omlish.http import asgi
 
 from .base import BASE_SERVER_URL
 from .base import SCOPE
 from .base import BaseServerUrl
 from .markers import AppMarker
+from .markers import AppMarkerProcessor
+from .markers import AppMarkerProcessorMap
+from .markers import NopAppMarkerProcessor
 from .markers import append_app_marker
 from .markers import get_app_markers
 
@@ -50,7 +52,7 @@ class Route(ta.NamedTuple):
 
 class RouteHandler(ta.NamedTuple):
     route: Route
-    handler: AsgiApp
+    handler: asgi.App
 
 
 @dc.dataclass(frozen=True)
@@ -67,15 +69,20 @@ def handles(*routes: Route):
     return inner
 
 
+HANDLES_APP_MARKER_PROCESSORS: AppMarkerProcessorMap = {
+    _HandlesAppMarker: NopAppMarkerProcessor(),
+}
+
+
 ##
 
 
-class Handler_(lang.Abstract):  # noqa
+class RouteHandlerHolder(lang.Abstract):  # noqa
     def get_route_handlers(self) -> ta.Iterable[RouteHandler]:
         return get_marked_route_handlers(self)
 
 
-def get_marked_route_handlers(h: Handler_) -> ta.Sequence[RouteHandler]:
+def get_marked_route_handlers(h: RouteHandlerHolder) -> ta.Sequence[RouteHandler]:
     ret: list[RouteHandler] = []
 
     cdct: dict[str, ta.Any] = {}
@@ -96,21 +103,67 @@ def get_marked_route_handlers(h: Handler_) -> ta.Sequence[RouteHandler]:
     return ret
 
 
+@dc.dataclass()
+class DuplicateRouteError(Exception):
+    route_handlers: ta.Sequence[RouteHandler]
+
+
+def build_route_handler_map(
+        handlers: ta.Iterable[RouteHandler | RouteHandlerHolder],
+        processors: ta.Mapping[type[AppMarker], AppMarkerProcessor],
+) -> ta.Mapping[Route, asgi.App]:
+    rh_by_r: dict[Route, RouteHandler] = {}
+    for h in handlers:
+        if isinstance(h, RouteHandlerHolder):
+            rhs = list(h.get_route_handlers())
+        else:
+            rhs = [h]
+
+        for rh in rhs:
+            try:
+                ex = rh_by_r[rh.route]
+            except KeyError:
+                pass
+            else:
+                raise DuplicateRouteError([rh, ex])
+
+            rh_by_r[rh.route] = rh
+
+    app_by_r: dict[Route, asgi.App] = {}
+    for r, rh in rh_by_r.items():
+        app = rh.handler
+
+        markers = get_app_markers(rh.handler)
+        for m in markers:
+            mp = processors[type(m)]
+            if mp is not None:
+                app = mp.process_app(app)
+
+        app_by_r[r] = app
+
+    return app_by_r
+
+
 ##
 
 
 @dc.dataclass(frozen=True)
-class RouteHandlerApp(AsgiApp_):
-    route_handlers: ta.Mapping[Route, AsgiApp]
+class RouteHandlerApp(asgi.App_):
+    route_handlers: ta.Mapping[Route, asgi.App]
     base_server_url: BaseServerUrl | None = None
 
-    async def __call__(self, scope: AsgiScope, recv: AsgiRecv, send: AsgiSend) -> None:
+    URL_SCHEME_PORT_PAIRS: ta.ClassVar[ta.Collection[tuple[str, int]]] = (
+        ('http', 80),
+        ('https', 443),
+    )
+
+    async def __call__(self, scope: asgi.Scope, recv: asgi.Recv, send: asgi.Send) -> None:
         with contextlib.ExitStack() as es:
             es.enter_context(lang.context_var_setting(SCOPE, scope))  # noqa
 
             match scope_ty := scope['type']:
                 case 'lifespan':
-                    await stub_lifespan(scope, recv, send)
+                    await asgi.stub_lifespan(scope, recv, send)
                     return
 
                 case 'http':
@@ -119,7 +172,7 @@ class RouteHandlerApp(AsgiApp_):
                     else:
                         sch = scope['scheme']
                         h, p = scope['server']
-                        if (sch, p) not in (('http', 80), ('https', 443)):
+                        if (sch, p) not in self.URL_SCHEME_PORT_PAIRS:
                             ps = f':{p}'
                         else:
                             ps = ''
@@ -133,7 +186,7 @@ class RouteHandlerApp(AsgiApp_):
                         await handler(scope, recv, send)
 
                     else:
-                        await send_response(send, 404)
+                        await asgi.send_response(send, 404)
 
                 case _:
                     raise ValueError(f'Unhandled scope type: {scope_ty!r}')

omserv/apps/sessions.py

@@ -3,11 +3,8 @@ import dataclasses as dc
 import logging
 
 from omlish import lang
+from omlish.http import asgi
 from omlish.http import sessions
-from omlish.http.asgi import AsgiApp
-from omlish.http.asgi import AsgiRecv
-from omlish.http.asgi import AsgiScope
-from omlish.http.asgi import AsgiSend
 
 from .markers import AppMarker
 from .markers import AppMarkerProcessor
@@ -35,7 +32,7 @@ def with_session(fn):
 class _WithSessionAppMarkerProcessor(AppMarkerProcessor):
     _ss: sessions.CookieSessionStore
 
-    async def _wrap(self, fn: AsgiApp, scope: AsgiScope, recv: AsgiRecv, send: AsgiSend) -> None:
+    async def _wrap(self, fn: asgi.App, scope: asgi.Scope, recv: asgi.Recv, send: asgi.Send) -> None:
         async def _send(obj):
             if obj['type'] == 'http.response.start':
                 out_session = SESSION.get()
@@ -53,5 +50,5 @@ class _WithSessionAppMarkerProcessor(AppMarkerProcessor):
         with lang.context_var_setting(SESSION, in_session):
             await fn(scope, recv, _send)
 
-    def __call__(self, app: AsgiApp) -> AsgiApp:
+    def process_app(self, app: asgi.App) -> asgi.App:
         return lang.decorator(self._wrap)(app)  # noqa

omserv/server/types.py

@@ -4,13 +4,13 @@ import typing as ta
 ##
 
 
-AsgiReceiveEvent: ta.TypeAlias = dict[str, ta.Any]
-AsgiSendEvent: ta.TypeAlias = dict[str, ta.Any]
+AsgiReceiveEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
+AsgiSendEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
 
 AsgiReceiveCallable: ta.TypeAlias = ta.Callable[[], ta.Awaitable[AsgiReceiveEvent]]
 AsgiSendCallable: ta.TypeAlias = ta.Callable[[AsgiSendEvent], ta.Awaitable[None]]
 
-Scope: ta.TypeAlias = dict[str, ta.Any]
+Scope: ta.TypeAlias = ta.Mapping[str, ta.Any]
 AsgiFramework: ta.TypeAlias = ta.Callable[
     [
         Scope,
@@ -22,13 +22,13 @@ AsgiFramework: ta.TypeAlias = ta.Callable[
 
 LifespanScope: ta.TypeAlias = Scope
 
-HttpResponseStartEvent: ta.TypeAlias = dict[str, ta.Any]
+HttpResponseStartEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
 HttpScope: ta.TypeAlias = Scope
 WebsocketScope: ta.TypeAlias = Scope
 
-WebsocketAcceptEvent: ta.TypeAlias = dict[str, ta.Any]
-WebsocketResponseBodyEvent: ta.TypeAlias = dict[str, ta.Any]
-WebsocketResponseStartEvent: ta.TypeAlias = dict[str, ta.Any]
+WebsocketAcceptEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
+WebsocketResponseBodyEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
+WebsocketResponseStartEvent: ta.TypeAlias = ta.Mapping[str, ta.Any]
 
 
 ##

{omserv-0.0.0.dev280.dist-info → omserv-0.0.0.dev282.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omserv
-Version: 0.0.0.dev280
+Version: 0.0.0.dev282
 Summary: omserv
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev280
+Requires-Dist: omlish==0.0.0.dev282
 Provides-Extra: all
 Requires-Dist: h11~=0.14; extra == "all"
 Requires-Dist: h2~=4.2; extra == "all"

{omserv-0.0.0.dev280.dist-info → omserv-0.0.0.dev282.dist-info}/RECORD

@@ -2,11 +2,11 @@ omserv/.manifests.json,sha256=qn9Vc6VXiVo9RzpJdLzQi_x2DPZIP3cq4S2bkqvJqjY,260
 omserv/__about__.py,sha256=w4jEBHhEVT7VZlwqMAh-4SecngU8xM6rehzWqi5ULZM,765
 omserv/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omserv/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-omserv/apps/base.py,sha256=KSwxbC0_fY87_DucvpEOHG6gZ2wL78D-N3We8RTEU4Q,385
-omserv/apps/inject.py,sha256=xtcn-C3A4yCdHfnAaejxrC_UGMVBYmWtSxeESIkorP0,2697
-omserv/apps/markers.py,sha256=JyVEXIaZC01bYObZ08TCwmQgiqZSvPfwNFJwpt9PU-c,870
-omserv/apps/routes.py,sha256=xWVgMwMeRr1QJiCCB_LCtxsb41ajdTmbDua4X71Djo0,3778
-omserv/apps/sessions.py,sha256=glruQSbOSbCYLPp6nDRNSHCyp5hj4oiOPhh3R0F9BTM,1537
+omserv/apps/base.py,sha256=HYsWbDot7zOnsdSL8dF_ANrNXEYgWIGzlnBdCU7n8v8,376
+omserv/apps/inject.py,sha256=GxM-TT3Ln8YrYBedJF__wY-zs4G3ow7RlxT2afr0FDs,2416
+omserv/apps/markers.py,sha256=JE_px0vsJqoDIDRFu9xtXsKjtxYAshUsFB0aRsoeIcg,965
+omserv/apps/routes.py,sha256=GeP1fk-tdd8LQ2no_FO2bFi48gSqjmTXpLTtcpsvSa0,5096
+omserv/apps/sessions.py,sha256=4HFHra43S0TaM2OcWUZ06eQwAqEaXuhjK5X5jlI0_44,1423
 omserv/apps/templates.py,sha256=PBRZHIF9UbnFnq-4EC6RmPeRkeH8lCBbpJkSdseHs6A,2125
 omserv/nginx/__init__.py,sha256=2d63LCGFA2qS7gdl2nCvNPmQWXNICu19wZIihQJPHCs,48
 omserv/nginx/build.py,sha256=87o2IahwUC-xriDpqc3OLKwwx4DacrDtUTplNQe_36k,3203
@@ -33,7 +33,7 @@ omserv/server/server.py,sha256=hgUTfZAUwF7V-HdsdV98KbYmE1IRKmEO3B1LlKTiPWc,5223
 omserv/server/sockets.py,sha256=lwqNP7URlp605ibsjHzp0pc-lyjcyTu-hD-uyojLUYk,3389
 omserv/server/ssl.py,sha256=gmB5ecM8Mck-YtGYF8pb2dwFdjABVGzERFCDzM9lBck,1483
 omserv/server/taskspawner.py,sha256=ljzF26UPtnp7GLAY_BvjzuwCoCO9aL7TKLwRNTmUy1M,3008
-omserv/server/types.py,sha256=XXY5py8RYlEeD4FZrWNqSyX7DD-ffSlcG-T2s9BY9eI,2017
+omserv/server/types.py,sha256=02GcS_zDxBzkleaptwr7kT7viefNh7fhFJixPdDpL_M,2059
 omserv/server/workercontext.py,sha256=4rcLuGsyiU7URO7T_eHylOBPPNUS9C23QfEUVyJUtIY,1200
 omserv/server/protocols/__init__.py,sha256=Ryu2PDZ1TUI6F2l-HBEYgyzZ7wHqE6VmjqnS0tIvmQI,47
 omserv/server/protocols/h11.py,sha256=_q_paD-ff0AWJEPaNK-6MUsQVtYRiALnWGwFyM3D0KU,11976
@@ -46,9 +46,9 @@ omserv/server/streams/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 omserv/server/streams/httpstream.py,sha256=0DeiAPLGbEGNa0fHTs8lUpi_CFZs4M5_QB-TiS8mobQ,8015
 omserv/server/streams/utils.py,sha256=aMOrqWIg_Hht5W4kLg3y7oR5AEkVvMrZhyjzo6U5owE,1527
 omserv/server/streams/wsstream.py,sha256=3Vyzox7dCE1tDSXjb6xBubWo41ZF9d38Hrsrlj6h1J8,15482
-omserv-0.0.0.dev280.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omserv-0.0.0.dev280.dist-info/METADATA,sha256=P1USQSjnjruJHdPBkuxjwBFgf8SDOwa95DhY9Qg4uik,1005
-omserv-0.0.0.dev280.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-omserv-0.0.0.dev280.dist-info/entry_points.txt,sha256=ivSrdA_ahEbI-eVMu-XZS-z4VrnQISvpecIkOqC9zFM,35
-omserv-0.0.0.dev280.dist-info/top_level.txt,sha256=HXehpnxeKscKNULzKNzZ27oNawBrsh1PaNAirbX-XNA,7
-omserv-0.0.0.dev280.dist-info/RECORD,,
+omserv-0.0.0.dev282.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omserv-0.0.0.dev282.dist-info/METADATA,sha256=AohL-vW9TC4ir0IKC54vRSEJtJladp8TN-mAyMZVD5g,1005
+omserv-0.0.0.dev282.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+omserv-0.0.0.dev282.dist-info/entry_points.txt,sha256=ivSrdA_ahEbI-eVMu-XZS-z4VrnQISvpecIkOqC9zFM,35
+omserv-0.0.0.dev282.dist-info/top_level.txt,sha256=HXehpnxeKscKNULzKNzZ27oNawBrsh1PaNAirbX-XNA,7
+omserv-0.0.0.dev282.dist-info/RECORD,,