mtrx-cli 0.1.16 → 0.1.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mtrx-cli",
-  "version": "0.1.16",
+  "version": "0.1.17",
   "description": "MATRX CLI for routing Codex, Claude, and Cursor through Matrx",
   "homepage": "https://mtrx.so",
   "repository": {
@@ -31,6 +31,7 @@
     "src/matrx/cli/cursor_daemon.py",
     "src/matrx/cli/cursor_launcher.py",
     "src/matrx/cli/cursor_proxy.py",
+    "src/matrx/cli/cursor_reroute.py",
     "src/matrx/cli/cursor_service.py",
     "src/matrx/cli/launcher.py",
     "src/matrx/cli/main.py",

package/src/matrx/cli/cursor_proxy.py

@@ -31,7 +31,16 @@ from typing import Any
 import httpx
 
 from matrx.cli.cursor_ca import CertCache, load_ca
-from matrx.cli.cursor_reroute import is_ai_path, try_reroute_to_matrx
+
+try:
+    from matrx.cli.cursor_reroute import is_ai_path, try_reroute_to_matrx
+except ImportError:
+    # Stubs when cursor_reroute not available (e.g. npm package omit).
+    def is_ai_path(path: str) -> bool:
+        return False
+
+    async def try_reroute_to_matrx(*, path: str, method: str, **kwargs: Any) -> None:
+        return None
 
 logger = logging.getLogger(__name__)
 

package/src/matrx/cli/cursor_reroute.py

@@ -0,0 +1,76 @@
+"""
+Cursor → MTRX rerouting for live injection.
+
+When the MITM proxy intercepts Cursor's AI traffic (RunSSE, StreamCpp, etc.),
+this module reroutes it through MTRX instead of forwarding to Cursor's servers.
+That enables the full suite: memory injection, compression, loop guard, etc.
+
+Cursor uses Connect/gRPC protocol with binary protobuf. We parse the request,
+convert to OpenAI/Anthropic format, call MTRX, and convert the response back.
+
+Refs: cursor-tap (https://github.com/burpheart/cursor-tap), everestmz/cursor-rpc
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+import re
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+# Cursor AI RPC paths (Connect protocol). RunSSE = main chat, StreamCpp = code completion.
+_AI_PATH_PATTERNS = (
+    r"RunSSE",
+    r"StreamCpp",
+    r"BidiAppend",
+    r"AgentService",
+    r"AiService",
+)
+
+
+def is_ai_path(path: str) -> bool:
+    """Return True if this path is an AI/LLM endpoint we should reroute to MTRX."""
+    if not path:
+        return False
+    return any(re.search(p, path, re.IGNORECASE) for p in _AI_PATH_PATTERNS)
+
+
+def _cursor_model_to_openai(cursor_model: str) -> str:
+    """Map Cursor model names to OpenAI-style names MTRX expects."""
+    # Cursor uses names like "claude-sonnet-4" or "gpt-4o" - usually compatible
+    normalized = (cursor_model or "").strip().lower()
+    if not normalized:
+        return "gpt-4o"  # fallback
+    return cursor_model
+
+
+async def try_reroute_to_matrx(
+    *,
+    path: str,
+    method: str,
+    req_headers: dict[str, str],
+    req_body: bytes,
+    matrx_base_url: str,
+    matrx_key: str,
+    session_id: str | None = None,
+    group_id: str | None = None,
+    project_id: str | None = None,
+) -> tuple[bool, dict[str, str], bytes | None, bool] | None:
+    """
+    Attempt to reroute a Cursor AI request through MTRX.
+
+    Returns:
+        (success, response_headers, response_body, is_streaming) if handled,
+        None to fall back to normal forward.
+    """
+    if method != "POST" or not is_ai_path(path):
+        return None
+
+    # TODO: Full protobuf parsing. Cursor uses Connect/gRPC with binary frames.
+    # For now we don't have the proto conversion - fall back to forward.
+    # When implemented: parse req_body, extract messages+model, call MTRX,
+    # convert response back to Cursor's gRPC format.
+    logger.debug("cursor_reroute: path=%s would reroute (protobuf conversion not yet implemented)", path)
+    return None