Add support for ASGI pathsend

blacksheep/contents.pxd

@@ -15,6 +15,10 @@ cdef class StreamedContent(Content):
     cdef readonly object generator
 
 
+cdef class FilepathContent(Content):
+    cdef readonly str path
+
+
 cdef class ASGIContent(Content):
     cdef readonly object receive
     cpdef void dispose(self)

blacksheep/contents.py

@@ -45,6 +45,14 @@ async def get_parts(self):
             yield chunk
 
 
+class FilepathContent(Content):
+    def __init__(self, content_type: bytes, path: str):
+        self.type = content_type
+        self.body = None
+        self.length = 0
+        self.path = path
+
+
 class ASGIContent(Content):
     def __init__(self, receive):
         self.type = None

blacksheep/contents.pyi

@@ -36,6 +36,17 @@ class StreamedContent(Content):
 
     async def get_parts(self) -> AsyncIterable[bytes]: ...
 
+class FilepathContent(Content):
+    def __init__(
+        self,
+        content_type: bytes,
+        path: str,
+    ) -> None:
+        self.type = content_type
+        self.body = None
+        self.length = 0
+        self.path = path
+
 class ASGIContent(Content):
     def __init__(self, receive: Callable[[], Awaitable[dict]]):
         self.type = None

blacksheep/contents.pyx

@@ -61,6 +61,19 @@ cdef class StreamedContent(Content):
             yield chunk
 
 
+cdef class FilepathContent(Content):
+
+    def __init__(
+        self,
+        bytes content_type,
+        str path
+    ):
+        self.type = content_type
+        self.body = None
+        self.length = 0
+        self.path = path
+
+
 cdef class ASGIContent(Content):
 
     def __init__(self, object receive):

blacksheep/messages.py

@@ -470,6 +470,12 @@ async def is_disconnected(self):
             self._is_disconnected = True
         return self._is_disconnected
 
+    @property
+    def accept_pathsend(self):
+        if not isinstance(self.content, ASGIContent) or self.scope is None:
+            return False
+        return "http.response.pathsend" in self.scope.get("extensions", {})
+
 
 class Response(Message):
     def __init__(self, status: int, headers=None, content: Content = None):

blacksheep/messages.pyi

@@ -147,6 +147,9 @@ class Request(Message):
         detect if the client closed the original connection.
         """
 
+    @property
+    def accept_pathsend(self) -> bool: ...
+
 class Response(Message):
     def __init__(
         self,

blacksheep/messages.pyx

@@ -543,6 +543,11 @@ cdef class Request(Message):
 
         return self._is_disconnected
 
+    @property
+    def accept_pathsend(self):
+        if not isinstance(self.content, ASGIContent) or self.scope is None:
+            return False
+        return "http.response.pathsend" in self.scope.get("extensions", {})
 
 cdef class Response(Message):
 

blacksheep/scribe.py

@@ -1,7 +1,7 @@
 import http
 import re
 
-from .contents import Content, StreamedContent
+from .contents import Content, FilepathContent, StreamedContent
 from .cookies import Cookie, write_cookie_for_response
 from .messages import Request, Response
 
@@ -288,6 +288,8 @@ async def send_asgi_response(response: Response, send):
                 await send(
                     {"type": "http.response.body", "body": b"", "more_body": False}
                 )
+        elif isinstance(content, FilepathContent):
+            await send({"type": "http.response.pathsend", "path": content.path})
         else:
             if content.length > MAX_RESPONSE_CHUNK_SIZE:
                 for chunk in get_chunks(content.body):

blacksheep/server/files/__init__.py

@@ -7,6 +7,7 @@
 from blacksheep.common.files.asyncfs import FilesHandler
 from blacksheep.common.files.info import FileInfo
 from blacksheep.common.files.pathsutils import get_mime_type_from_name
+from blacksheep.contents import FilepathContent
 from blacksheep.exceptions import BadRequest, InvalidArgument, RangeNotSatisfiable
 from blacksheep.ranges import InvalidRangeValue, Range, RangePart
 from blacksheep.server.headers.cache import CacheControlHeaderValue
@@ -289,6 +290,8 @@ def get_response_for_file(
                 file_type=file_type,
             ),
         )
+    elif request.accept_pathsend:
+        content = FilepathContent(mime, resource_path)
     else:
         content = StreamedContent(
             mime, get_file_getter(files_handler, resource_path, info.size)