refactor: extract core logic of router

Author: JoJoJoJoJoJoJo

src/mcpm/commands/router.py

@@ -16,9 +16,10 @@
 from rich.prompt import Confirm
 
 from mcpm.clients.client_registry import ClientRegistry
+from mcpm.core.utils.log_manager import get_log_directory
 from mcpm.router.share import Tunnel
 from mcpm.utils.config import MCPM_AUTH_HEADER, MCPM_PROFILE_HEADER, ConfigManager
-from mcpm.utils.platform import get_log_directory, get_pid_directory
+from mcpm.utils.platform import get_pid_directory
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)

src/mcpm/router/client_connection.py renamed to src/mcpm/core/router/client_connection.py

@@ -1,13 +1,45 @@
 import os
+import sys
 from pathlib import Path
 from typing import Optional, TextIO
 
-from .platform import get_log_directory
+
+def get_log_directory(app_name: str = "mcpm") -> Path:
+    """
+    Return the appropriate log directory path based on the current operating system.
+
+    Args:
+        app_name: The name of the application, used in the path
+
+    Returns:
+        Path object representing the log directory
+    """
+    # macOS
+    if sys.platform == "darwin":
+        return Path.home() / "Library" / "Logs" / app_name / "logs"
+
+    # Windows
+    elif sys.platform == "win32":
+        localappdata = os.environ.get("LOCALAPPDATA")
+        if localappdata:
+            return Path(localappdata) / app_name / "logs"
+        return Path.home() / "AppData" / "Local" / app_name / "logs"
+
+    # Linux and other Unix-like systems
+    else:
+        # Check if XDG_DATA_HOME is defined
+        xdg_data_home = os.environ.get("XDG_DATA_HOME")
+        if xdg_data_home:
+            return Path(xdg_data_home) / app_name / "logs"
+
+        # Default to ~/.local/share if XDG_DATA_HOME is not defined
+        return Path.home() / ".local" / "share" / app_name / "logs"
+
 
 DEFAULT_ROOT_STDERR_LOG_DIR = get_log_directory("mcpm") / "errlogs"
 
 
-class ServerErrorLogManager:
+class ServerLogManager:
     """
     A manager for server error logs.
     """

src/mcpm/core/router/router.py

@@ -14,7 +14,7 @@
     TextContent,
 )
 
-from mcpm.utils.config import PROMPT_SPLITOR, RESOURCE_SPLITOR, TOOL_SPLITOR
+from mcpm.core.router.router import PROMPT_SPLITOR, RESOURCE_SPLITOR, TOOL_SPLITOR
 
 from .base import AccessEventType
 from .duckdb import DuckDBAccessMonitor

src/mcpm/utils/errlog_manager.py renamed to src/mcpm/core/utils/log_manager.py

@@ -1,11 +1,11 @@
 import logging
 import os
 
+from mcpm.core.utils.log_manager import get_log_directory
 from mcpm.monitor.event import monitor
 from mcpm.router.router import MCPRouter
 from mcpm.router.router_config import RouterConfig
 from mcpm.utils.config import ConfigManager
-from mcpm.utils.platform import get_log_directory
 
 LOG_DIR = get_log_directory("mcpm")
 LOG_DIR.mkdir(parents=True, exist_ok=True)

src/mcpm/monitor/event.py

@@ -12,12 +12,12 @@
 from starlette.routing import Mount, Route
 from starlette.types import Receive, Scope, Send
 
+from mcpm.core.utils.log_manager import get_log_directory
 from mcpm.monitor.base import AccessEventType
 from mcpm.monitor.event import monitor
 from mcpm.router.router import MCPRouter
 from mcpm.router.transport import RouterSseTransport
 from mcpm.utils.config import ConfigManager
-from mcpm.utils.platform import get_log_directory
 
 LOG_DIR = get_log_directory("mcpm")
 LOG_DIR.mkdir(parents=True, exist_ok=True)

src/mcpm/router/app.py

@@ -16,10 +16,6 @@
 DEFAULT_HOST = "localhost"
 DEFAULT_PORT = 6276  # 6276 represents MCPM on a T9 keypad (6=M, 2=C, 7=P, 6=M)
 # default splitor pattern
-TOOL_SPLITOR = "_t_"
-RESOURCE_SPLITOR = ":"
-RESOURCE_TEMPLATE_SPLITOR = ":"
-PROMPT_SPLITOR = "_p_"
 DEFAULT_SHARE_ADDRESS = f"share.mcpm.sh:{DEFAULT_PORT}"
 MCPM_AUTH_HEADER = "X-MCPM-SECRET"
 MCPM_PROFILE_HEADER = "X-MCPM-PROFILE"

src/mcpm/router/router.py

@@ -9,38 +9,6 @@
 from pathlib import Path
 
 
-def get_log_directory(app_name: str = "mcpm") -> Path:
-    """
-    Return the appropriate log directory path based on the current operating system.
-
-    Args:
-        app_name: The name of the application, used in the path
-
-    Returns:
-        Path object representing the log directory
-    """
-    # macOS
-    if sys.platform == "darwin":
-        return Path.home() / "Library" / "Logs" / app_name / "logs"
-
-    # Windows
-    elif sys.platform == "win32":
-        localappdata = os.environ.get("LOCALAPPDATA")
-        if localappdata:
-            return Path(localappdata) / app_name / "logs"
-        return Path.home() / "AppData" / "Local" / app_name / "logs"
-
-    # Linux and other Unix-like systems
-    else:
-        # Check if XDG_DATA_HOME is defined
-        xdg_data_home = os.environ.get("XDG_DATA_HOME")
-        if xdg_data_home:
-            return Path(xdg_data_home) / app_name / "logs"
-
-        # Default to ~/.local/share if XDG_DATA_HOME is not defined
-        return Path.home() / ".local" / "share" / app_name / "logs"
-
-
 def get_pid_directory(app_name: str = "mcpm") -> Path:
     """
     Return the appropriate PID directory path based on the current operating system.