refactor: remove DocumentContext and related functionality

The DocumentContext class and all its related functionality has been removed from the codebase. This includes:
- Removing DocumentContext class and all its methods
- Removing DocumentContext parameter from all tool constructors
- Removing DocumentContext references from server initialization
- Removing DocumentContext from tool registration functions
- Removing DocumentContext from tests
- Updating filesystem tools to no longer track document contents

The permission management is now handled solely by PermissionManager. This simplifies the codebase by removing redundant path tracking functionality.

Author: SDGLBL

mcp_claude_code/server.py

@@ -6,7 +6,7 @@
 
 from mcp_claude_code.prompts import register_all_prompts
 from mcp_claude_code.tools import register_all_tools
-from mcp_claude_code.tools.common.context import DocumentContext
+
 from mcp_claude_code.tools.common.permissions import PermissionManager
 
 
@@ -47,15 +47,13 @@ def __init__(
         """
         self.mcp = mcp_instance if mcp_instance is not None else FastMCP(name)
 
-        # Initialize context, permissions, and command executor
-        self.document_context = DocumentContext()
+        # Initialize permissions and command executor
         self.permission_manager = PermissionManager()
 
         # Add allowed paths
         if allowed_paths:
             for path in allowed_paths:
                 self.permission_manager.add_allowed_path(path)
-                self.document_context.add_allowed_path(path)
 
         # Store project paths
         self.project_paths = project_paths
@@ -73,7 +71,6 @@ def __init__(
         # Register all tools
         register_all_tools(
             mcp_server=self.mcp,
-            document_context=self.document_context,
             permission_manager=self.permission_manager,
             agent_model=self.agent_model,
             agent_max_tokens=self.agent_max_tokens,
@@ -97,7 +94,6 @@ def run(self, transport: str = "stdio", allowed_paths: list[str] | None = None):
         allowed_paths_list = allowed_paths or []
         for path in allowed_paths_list:
             self.permission_manager.add_allowed_path(path)
-            self.document_context.add_allowed_path(path)
 
         # Run the server
         transport_type = cast(Literal["stdio", "sse"], transport)

mcp_claude_code/tools/__init__.py

@@ -14,7 +14,7 @@
 from mcp_claude_code.tools.agent import register_agent_tools
 from mcp_claude_code.tools.common import register_batch_tool, register_thinking_tool
 from mcp_claude_code.tools.common.base import BaseTool
-from mcp_claude_code.tools.common.context import DocumentContext
+
 from mcp_claude_code.tools.common.permissions import PermissionManager
 from mcp_claude_code.tools.filesystem import register_filesystem_tools
 from mcp_claude_code.tools.jupyter import register_jupyter_tools
@@ -24,7 +24,6 @@
 
 def register_all_tools(
     mcp_server: FastMCP,
-    document_context: DocumentContext,
     permission_manager: PermissionManager,
     agent_model: str | None = None,
     agent_max_tokens: int | None = None,
@@ -38,7 +37,6 @@ def register_all_tools(
 
     Args:
         mcp_server: The FastMCP server instance
-        document_context: Document context for tracking file contents
         permission_manager: Permission manager for access control
         agent_model: Optional model name for agent tool in LiteLLM format
         agent_max_tokens: Optional maximum tokens for agent responses
@@ -52,16 +50,12 @@ def register_all_tools(
     all_tools: dict[str, BaseTool] = {}
 
     # Register all filesystem tools
-    filesystem_tools = register_filesystem_tools(
-        mcp_server, document_context, permission_manager
-    )
+    filesystem_tools = register_filesystem_tools(mcp_server, permission_manager)
     for tool in filesystem_tools:
         all_tools[tool.name] = tool
 
     # Register all jupyter tools
-    jupyter_tools = register_jupyter_tools(
-        mcp_server, document_context, permission_manager
-    )
+    jupyter_tools = register_jupyter_tools(mcp_server, permission_manager)
     for tool in jupyter_tools:
         all_tools[tool.name] = tool
 
@@ -74,7 +68,6 @@ def register_all_tools(
     if enable_agent_tool:
         agent_tools = register_agent_tools(
             mcp_server,
-            document_context,
             permission_manager,
             agent_model=agent_model,
             agent_max_tokens=agent_max_tokens,

mcp_claude_code/tools/agent/__init__.py

@@ -8,13 +8,12 @@
 
 from mcp_claude_code.tools.agent.agent_tool import AgentTool
 from mcp_claude_code.tools.common.base import BaseTool, ToolRegistry
-from mcp_claude_code.tools.common.context import DocumentContext
+
 from mcp_claude_code.tools.common.permissions import PermissionManager
 
 
 def register_agent_tools(
     mcp_server: FastMCP,
-    document_context: DocumentContext,
     permission_manager: PermissionManager,
     agent_model: str | None = None,
     agent_max_tokens: int | None = None,
@@ -27,7 +26,7 @@ def register_agent_tools(
 
     Args:
         mcp_server: The FastMCP server instance
-        document_context: Document context for tracking file contents
+
         permission_manager: Permission manager for access control
         agent_model: Optional model name for agent tool in LiteLLM format
         agent_max_tokens: Optional maximum tokens for agent responses
@@ -41,7 +40,6 @@ def register_agent_tools(
     """
     # Create agent tool
     agent_tool = AgentTool(
-        document_context=document_context,
         permission_manager=permission_manager,
         model=agent_model,
         api_key=agent_api_key,

mcp_claude_code/tools/agent/agent_tool.py

@@ -29,7 +29,6 @@
 from mcp_claude_code.tools.common.base import BaseTool
 from mcp_claude_code.tools.common.batch_tool import BatchTool
 from mcp_claude_code.tools.common.context import (
-    DocumentContext,
     ToolContext,
     create_tool_context,
 )
@@ -104,7 +103,6 @@ def description(self) -> str:
 
     def __init__(
         self,
-        document_context: DocumentContext,
         permission_manager: PermissionManager,
         model: str | None = None,
         api_key: str | None = None,
@@ -116,7 +114,7 @@ def __init__(
         """Initialize the agent tool.
 
         Args:
-            document_context: Document context for tracking file contents
+
             permission_manager: Permission manager for access control
             model: Optional model name override in LiteLLM format (e.g., "openai/gpt-4o")
             api_key: Optional API key for the model provider
@@ -125,7 +123,7 @@ def __init__(
             max_iterations: Maximum number of iterations for agent (default: 10)
             max_tool_uses: Maximum number of total tool uses for agent (default: 30)
         """
-        self.document_context = document_context
+
         self.permission_manager = permission_manager
         self.model_override = model
         self.api_key_override = api_key
@@ -135,12 +133,10 @@ def __init__(
         self.max_tool_uses = max_tool_uses
         self.available_tools: list[BaseTool] = []
         self.available_tools.extend(
-            get_read_only_filesystem_tools(
-                self.document_context, self.permission_manager
-            )
+            get_read_only_filesystem_tools(self.permission_manager)
         )
         self.available_tools.extend(
-            get_read_only_jupyter_tools(self.document_context, self.permission_manager)
+            get_read_only_jupyter_tools(self.permission_manager)
         )
         self.available_tools.append(
             BatchTool({t.name: t for t in self.available_tools})

mcp_claude_code/tools/common/base.py

@@ -13,7 +13,7 @@
 from fastmcp import FastMCP
 from fastmcp import Context as MCPContext
 
-from mcp_claude_code.tools.common.context import DocumentContext
+
 from mcp_claude_code.tools.common.permissions import PermissionManager
 from mcp_claude_code.tools.common.validation import (
     ValidationResult,
@@ -122,16 +122,12 @@ class FileSystemTool(BaseTool, ABC):
     including permission checking and path validation.
     """
 
-    def __init__(
-        self, document_context: DocumentContext, permission_manager: PermissionManager
-    ) -> None:
+    def __init__(self, permission_manager: PermissionManager) -> None:
         """Initialize filesystem tool.
 
         Args:
-            document_context: Document context for tracking file contents
             permission_manager: Permission manager for access control
         """
-        self.document_context: DocumentContext = document_context
         self.permission_manager: PermissionManager = permission_manager
 
     def validate_path(self, path: str, param_name: str = "path") -> ValidationResult: