124 lines
4.6 KiB
Python
124 lines
4.6 KiB
Python
"""Launch codebase-memory-mcp's OWN HTTP graph UI (``--ui=true``) as a
|
|
background process for the GraphRAG tab to embed/open — this is the richer
|
|
"Graph / Projects / Control" visualization the binary ships with, distinct
|
|
from this app's own D3 fallback graph.
|
|
|
|
Not every distributed build of the binary includes that UI (some are built
|
|
"without the embedded UI" — the CLI prints that exact message and never opens
|
|
the port). :meth:`CodebaseMemoryUiServer.start` tells the two failure modes
|
|
apart so the caller can show the right guidance instead of a generic timeout.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import subprocess
|
|
import threading
|
|
import time
|
|
from typing import List, Optional
|
|
|
|
import requests
|
|
|
|
from .codebase_memory import resolve_binary
|
|
|
|
DEFAULT_PORT = 9749
|
|
_STARTUP_TIMEOUT = 8.0
|
|
_POLL_INTERVAL = 0.25
|
|
# Substring of the binary's own message when it was built without the UI
|
|
# (see the DeusData/codebase-memory-mcp --help output) — matched case-insensitively.
|
|
_NO_UI_MARKER = "built without the embedded ui"
|
|
|
|
|
|
class CmemUiError(RuntimeError):
|
|
"""The UI server could not be reached. ``no_ui_build`` is True when the
|
|
binary itself reported it has no embedded UI (needs the ``-ui`` release
|
|
asset) — a different remedy than a generic startup/timeout failure."""
|
|
|
|
def __init__(self, message: str, no_ui_build: bool = False):
|
|
super().__init__(message)
|
|
self.no_ui_build = no_ui_build
|
|
|
|
|
|
class CodebaseMemoryUiServer:
|
|
"""One ``codebase-memory-mcp --ui`` process, started on demand."""
|
|
|
|
def __init__(self, binary_path: str = "", port: int = DEFAULT_PORT):
|
|
self.binary = resolve_binary(binary_path)
|
|
self.port = port
|
|
self._proc: Optional[subprocess.Popen] = None
|
|
|
|
@property
|
|
def url(self) -> str:
|
|
return f"http://127.0.0.1:{self.port}/"
|
|
|
|
@property
|
|
def running(self) -> bool:
|
|
return self._proc is not None and self._proc.poll() is None
|
|
|
|
def start(self, repo_path: str = "") -> str:
|
|
"""Start the UI server (no-op — just returns the URL — if one is
|
|
already running) and block briefly until it answers. Raises
|
|
:class:`CmemUiError` with a clear reason on failure/timeout."""
|
|
if self.running:
|
|
return self.url
|
|
if not self.binary:
|
|
raise CmemUiError("codebase-memory-mcp chưa được cài đặt.")
|
|
cmd = [self.binary, "--ui=true", f"--port={self.port}"]
|
|
try:
|
|
self._proc = subprocess.Popen(
|
|
cmd, cwd=repo_path or None, stdin=subprocess.DEVNULL,
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True,
|
|
bufsize=1,
|
|
)
|
|
except OSError as exc:
|
|
raise CmemUiError(f"Không khởi chạy được codebase-memory-mcp: {exc}")
|
|
|
|
lines: List[str] = []
|
|
no_ui_event = threading.Event()
|
|
|
|
def _reader() -> None:
|
|
try:
|
|
stream = self._proc.stdout
|
|
if stream is None:
|
|
return
|
|
for line in stream:
|
|
lines.append(line)
|
|
if _NO_UI_MARKER in line.lower():
|
|
no_ui_event.set()
|
|
except (OSError, ValueError):
|
|
pass
|
|
|
|
threading.Thread(target=_reader, daemon=True).start()
|
|
|
|
deadline = time.monotonic() + _STARTUP_TIMEOUT
|
|
while time.monotonic() < deadline:
|
|
try:
|
|
resp = requests.get(self.url, timeout=1)
|
|
if resp.status_code < 500:
|
|
return self.url
|
|
except requests.RequestException:
|
|
pass
|
|
if no_ui_event.is_set():
|
|
self.stop()
|
|
raise CmemUiError(
|
|
"".join(lines).strip()[:400] or "Bản build này không có UI đồ thị nhúng.",
|
|
no_ui_build=True,
|
|
)
|
|
if self._proc.poll() is not None:
|
|
self.stop()
|
|
raise CmemUiError(
|
|
"".join(lines).strip()[:400] or "codebase-memory-mcp thoát ngay lập tức.")
|
|
time.sleep(_POLL_INTERVAL)
|
|
self.stop()
|
|
raise CmemUiError(f"Hết thời gian chờ UI trên cổng {self.port}.")
|
|
|
|
def stop(self) -> None:
|
|
proc, self._proc = self._proc, None
|
|
if proc is not None and proc.poll() is None:
|
|
try:
|
|
proc.terminate()
|
|
proc.wait(timeout=3)
|
|
except Exception: # noqa: BLE001 - best-effort cleanup, never raise on shutdown
|
|
try:
|
|
proc.kill()
|
|
except Exception: # noqa: BLE001
|
|
pass
|