mrok 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

mrok/proxy/constants.py

@@ -1,22 +0,0 @@
-MAX_REQUEST_BODY_BYTES = 2 * 1024 * 1024
-MAX_RESPONSE_BODY_BYTES = 5 * 1024 * 1024
-
-BINARY_CONTENT_TYPES = {
-    "application/octet-stream",
-    "application/pdf",
-}
-
-BINARY_PREFIXES = (
-    "image/",
-    "video/",
-    "audio/",
-)
-
-TEXTUAL_CONTENT_TYPES = {
-    "application/json",
-    "application/xml",
-    "application/javascript",
-    "application/x-www-form-urlencoded",
-}
-
-TEXTUAL_PREFIXES = ("text/",)

mrok/proxy/lifespan.py

@@ -1,10 +0,0 @@
-import logging
-
-from uvicorn.config import Config
-from uvicorn.lifespan.on import LifespanOn
-
-
-class MrokLifespan(LifespanOn):
-    def __init__(self, config: Config) -> None:
-        super().__init__(config)
-        self.logger = logging.getLogger("mrok.proxy")

mrok/proxy/protocol.py

@@ -1,11 +0,0 @@
-import logging
-
-from uvicorn.protocols.http.httptools_impl import HttpToolsProtocol
-
-
-class MrokHttpToolsProtocol(HttpToolsProtocol):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.logger = logging.getLogger("mrok.proxy")
-        self.access_logger = logging.getLogger("mrok.access")
-        self.access_log = self.access_logger.hasHandlers()

mrok/proxy/server.py

@@ -1,14 +0,0 @@
-import logging
-import socket
-
-from uvicorn import server
-
-server.logger = logging.getLogger("mrok.proxy")
-
-
-class MrokServer(server.Server):
-    async def serve(self, sockets: list[socket.socket] | None = None) -> None:
-        if not sockets:
-            sockets = [self.config.bind_socket()]
-        with self.capture_signals():
-            await self._serve(sockets)

mrok/proxy/utils.py

@@ -1,90 +0,0 @@
-from collections.abc import Mapping
-
-from mrok.proxy.constants import (
-    BINARY_CONTENT_TYPES,
-    BINARY_PREFIXES,
-    MAX_REQUEST_BODY_BYTES,
-    MAX_RESPONSE_BODY_BYTES,
-    TEXTUAL_CONTENT_TYPES,
-    TEXTUAL_PREFIXES,
-)
-
-
-def is_binary(content_type: str) -> bool:
-    ct = content_type.lower()
-    if ct in BINARY_CONTENT_TYPES:
-        return True
-    if any(ct.startswith(p) for p in BINARY_PREFIXES):
-        return True
-    return False
-
-
-def is_textual(content_type: str) -> bool:
-    ct = content_type.lower()
-    if ct in TEXTUAL_CONTENT_TYPES:
-        return True
-    if any(ct.startswith(p) for p in TEXTUAL_PREFIXES):
-        return True
-    return False
-
-
-def must_capture_request(
-    method: str,
-    headers: Mapping,
-) -> bool:
-    method = method.upper()
-
-    # No body expected
-    if method in ("GET", "HEAD", "OPTIONS", "TRACE"):
-        return False
-
-    content_type = headers.get("content-type", "").lower()
-
-    content_length = None
-    if "content-length" in headers:
-        content_length = int(headers["content-length"])
-
-    if is_binary(content_type):
-        return False
-
-    if content_type.startswith("multipart/form-data"):
-        return False
-
-    if content_length is not None and content_length > MAX_REQUEST_BODY_BYTES:
-        return False
-
-    if is_textual(content_type):
-        return True
-
-    if content_length is None:
-        return True
-
-    return content_length <= MAX_REQUEST_BODY_BYTES
-
-
-def must_capture_response(
-    headers: Mapping,
-) -> bool:
-    content_type = headers.get("content-type", "").lower()
-    disposition = headers.get("content-disposition", "").lower()
-
-    content_length = None
-    if "content-length" in headers:
-        content_length = int(headers["content-length"])
-
-    if "attachment" in disposition:
-        return False
-
-    if is_binary(content_type):
-        return False
-
-    if content_length is not None and content_length > MAX_RESPONSE_BODY_BYTES:
-        return False
-
-    if is_textual(content_type):
-        return True
-
-    if content_length is None:
-        return True
-
-    return content_length <= MAX_RESPONSE_BODY_BYTES