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

mrok/agent/devtools/__main__.py

@@ -1,34 +0,0 @@
-#!/usr/bin/env python3
-"""mrok agent devtools CLI entrypoint.
-
-Provides a small CLI that accepts an optional subscriber port and
-invokes the `run` function with that port.
-"""
-
-import argparse
-from typing import Any
-
-
-def run(port: int) -> None:
-    """Run the devtools agent using the given subscriber port.
-
-    This is a stub for the runtime function. Implementation goes here.
-    """
-    pass
-
-
-def main(argv: Any = None) -> None:
-    parser = argparse.ArgumentParser(description="mrok devtools agent")
-    parser.add_argument(
-        "-p",
-        "--subscriber-port",
-        type=int,
-        default=50001,
-        help="Port for subscriber (default: 50001)",
-    )
-    args = parser.parse_args(argv)
-    run(args.subscriber_port)
-
-
-if __name__ == "__main__":
-    main()

mrok/cli/commands/agent/utils.py

@@ -1,5 +0,0 @@
-import multiprocessing
-
-
-def number_of_workers() -> int:
-    return (multiprocessing.cpu_count() * 2) + 1

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/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