mcp-server-code-assist 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

mcp_server_code_assist/server.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import json
 from pathlib import Path
 from typing import Any
 from enum import Enum
@@ -9,18 +10,20 @@ from mcp.server.stdio import stdio_server
 from mcp.types import ClientCapabilities, TextContent, Tool, ListRootsResult, RootsCapability
 import git
 from mcp_server_code_assist.tools.models import (
-    FileCreate, FileDelete, FileModify, FileRewrite,
+    FileCreate, FileDelete, FileModify, FileRewrite, FileRead,
     GitBase, GitAdd, GitCommit, GitDiff, GitCreateBranch,
-    GitCheckout, GitShow, GitLog
+    GitCheckout, GitShow, GitLog, ListDirectory
 )
 from mcp_server_code_assist.tools.file_tools import FileTools
 from mcp_server_code_assist.tools.git_functions import git_status, git_diff_unstaged, git_diff_staged, git_diff, git_log, git_show
 
 class CodeAssistTools(str, Enum):
+    LIST_DIRECTORY = "list_directory"
     FILE_CREATE = "file_create"
     FILE_DELETE = "file_delete" 
     FILE_MODIFY = "file_modify"
     FILE_REWRITE = "file_rewrite"
+    FILE_READ = "file_read"
     GIT_STATUS = "git_status"
     GIT_DIFF_UNSTAGED = "git_diff_unstaged"
     GIT_DIFF_STAGED = "git_diff_staged"
@@ -81,6 +84,11 @@ async def serve(working_dir: Path | None) -> None:
     @server.list_tools()
     async def list_tools() -> list[Tool]:
         return [
+            Tool(
+                name=CodeAssistTools.LIST_DIRECTORY,
+                description="Lists contents of a directory",
+                inputSchema=ListDirectory.schema(),
+            ),
             Tool(
                 name=CodeAssistTools.FILE_CREATE,
                 description="Creates a new file with content",
@@ -101,6 +109,11 @@ async def serve(working_dir: Path | None) -> None:
                 description="Rewrites entire file content",
                 inputSchema=FileRewrite.schema(),
             ),
+            Tool(
+                name=CodeAssistTools.FILE_READ,
+                description="Reads contents of a file",
+                inputSchema=FileRead.schema(),
+            ),
             Tool(
                 name=CodeAssistTools.GIT_STATUS,
                 description="Shows the working tree status",
@@ -161,6 +174,14 @@ async def serve(working_dir: Path | None) -> None:
     @server.call_tool()
     async def call_tool(name: str, arguments: dict) -> list[TextContent]:
         match name:
+            case CodeAssistTools.LIST_DIRECTORY:
+                result = await FileTools.list_directory(
+                    arguments["path"],
+                    arguments.get("recursive", False),
+                    arguments.get("include_hidden", False)
+                )
+                return [TextContent(type="text", text=json.dumps(result, indent=2))]
+
             case CodeAssistTools.FILE_CREATE:
                 result = await FileTools.create_file(arguments["path"], arguments.get("content", ""))
                 return [TextContent(type="text", text=result)]
@@ -177,6 +198,10 @@ async def serve(working_dir: Path | None) -> None:
                 result = await FileTools.rewrite_file(arguments["path"], arguments["content"])
                 return [TextContent(type="text", text=result)]
 
+            case CodeAssistTools.FILE_READ:
+                content = await FileTools.read_file(arguments["path"])
+                return [TextContent(type="text", text=content)]
+
             case CodeAssistTools.GIT_STATUS:
                 repo = git.Repo(arguments["repo_path"])
                 status = git_status(repo)

mcp_server_code_assist/tools/file_tools.py

@@ -90,12 +90,51 @@ class FileTools:
         return ''.join(diff)
 
     @classmethod
-    async def list_directory(cls, path: str) -> list[str]:
+    async def list_directory(cls, path: str, recursive: bool = False, include_hidden: bool = False) -> list[dict]:
+        """List contents of a directory.
+        
+        Args:
+            path: Directory path to list
+            recursive: Whether to list subdirectories recursively
+            include_hidden: Whether to include hidden files/directories
+            
+        Returns:
+            List of dicts with file/directory info:
+            {
+                "name": str,
+                "path": str,
+                "type": "file"|"directory",
+                "size": int,  # for files only
+                "children": list  # for directories when recursive=True
+            }
+        """
         path = await cls.validate_path(path)
-        entries = []
-        for item in Path(path).iterdir():
-            entries.append(str(item))
-        return entries
+        if not os.path.isdir(path):
+            raise ValueError(f"Path {path} is not a directory")
+            
+        result = []
+        for entry in os.scandir(path):
+            if not include_hidden and entry.name.startswith('.'):
+                continue
+                
+            info = {
+                "name": entry.name,
+                "path": str(Path(entry.path).relative_to(path)),
+                "type": "directory" if entry.is_dir() else "file",
+            }
+            
+            if entry.is_file():
+                info["size"] = entry.stat().st_size
+            elif recursive and entry.is_dir():
+                info["children"] = await cls.list_directory(
+                    entry.path,
+                    recursive=True,
+                    include_hidden=include_hidden
+                )
+                
+            result.append(info)
+            
+        return result
 
     @classmethod
     async def create_directory(cls, path: str) -> None:

mcp_server_code_assist/tools/models.py

@@ -13,6 +13,9 @@ class FileModify(BaseModel):
     path: Union[str, Path]
     replacements: dict[str, str]
 
+class FileRead(BaseModel):
+    path: Union[str, Path]
+
 class FileRewrite(BaseModel):
     path: Union[str, Path]
     content: str
@@ -42,6 +45,11 @@ class GitShow(GitBase):
 class GitLog(GitBase):
     max_count: int = 10
 
+class ListDirectory(BaseModel):
+    path: Union[str, Path]
+    recursive: bool = False
+    include_hidden: bool = False
+
 class RepositoryOperation(BaseModel):
     path: str
     content: Optional[str] = None

{mcp_server_code_assist-0.1.1.dist-info → mcp_server_code_assist-0.1.3.dist-info}/METADATA

@@ -1,11 +1,14 @@
 Metadata-Version: 2.4
 Name: mcp-server-code-assist
-Version: 0.1.1
+Version: 0.1.3
 Summary: MCP Code Assist Server
 Requires-Python: >=3.11
 Requires-Dist: aiofiles>=24.0.0
+Requires-Dist: click>=8.1.7
 Requires-Dist: gitpython>=3.1.40
+Requires-Dist: mcp>=1.2.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: xmlschema>=3.4.3
 Provides-Extra: test
 Requires-Dist: pytest-asyncio>=0.25.0; extra == 'test'
 Requires-Dist: pytest>=8.0.0; extra == 'test'

{mcp_server_code_assist-0.1.1.dist-info → mcp_server_code_assist-0.1.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
 mcp_server_code_assist/__init__.py,sha256=Aai1h9YUGDDWQn9rsHq5gCZXO7QCQs2pJtyGC9mYV0o,661
 mcp_server_code_assist/__main__.py,sha256=lbLwix6a0gDF7YKoqbXF88oTdmr8KQGKWchLKEIn5SU,47
 mcp_server_code_assist/schema.xsd,sha256=vtfQQDzCHvbILT78zF_53N-68GrJRbELg1Xgq3Uu9DE,1235
-mcp_server_code_assist/server.py,sha256=qtz9pD3_JxIJGare5Sal25E8DafZY-DX4thrghulT1s,10012
+mcp_server_code_assist/server.py,sha256=X5TkxoDTWNNqVV_d9xjVoUolAUPPus5UqcOwCcQiHjQ,11041
 mcp_server_code_assist/xml_parser.py,sha256=FMmcAk7PbnTNidBGUynq6ZbUs1O9OHHI298REAQEMqw,1827
 mcp_server_code_assist/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mcp_server_code_assist/tools/file_tools.py,sha256=iGdVGP9phD5wMZ5HxbRv3gbUpNx-QF27xDLrUI9wUkU,7978
+mcp_server_code_assist/tools/file_tools.py,sha256=SZwZjN763D31zCZ4yz8IBJtaf8R5A4bRd9DwwY_exuA,9404
 mcp_server_code_assist/tools/git_functions.py,sha256=iGIhi8i7BxIrAL8JytWjfcZP7xgla0uVweazSKUNtB4,882
 mcp_server_code_assist/tools/invoke.py,sha256=lYIu5bJYBaKDDVYz6HRqodpLBMnfVpcvgAref10LO2s,313
-mcp_server_code_assist/tools/models.py,sha256=3ZUfzodSVbwSoIKG5oKiLbnjVeptT5QKLaQSwl7McJY,937
+mcp_server_code_assist/tools/models.py,sha256=fNCjX0klqa8awTHJH4wx0Z50HmRXDf0_AA8YHV90PwE,1113
 mcp_server_code_assist/tools/repository_tools.py,sha256=5OG4ecmqXVEeuX9U5jdegbQ2So4lxKkIjG3M3e5rgFI,2700
-mcp_server_code_assist-0.1.1.dist-info/METADATA,sha256=t4IWUcbj3HEkVsv5IyKNzRlEZEV7mzekO8yVpRX3TOM,340
-mcp_server_code_assist-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_server_code_assist-0.1.1.dist-info/entry_points.txt,sha256=n1CONsVBue7ckhZetmRyyoIIUpyD_gmHHNf7SlWQUCM,80
-mcp_server_code_assist-0.1.1.dist-info/RECORD,,
+mcp_server_code_assist-0.1.3.dist-info/METADATA,sha256=PhuoTxw1z1lRmkLe5GsfdkGi0Hy1FNtqTXCna8uPINU,426
+mcp_server_code_assist-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_server_code_assist-0.1.3.dist-info/entry_points.txt,sha256=n1CONsVBue7ckhZetmRyyoIIUpyD_gmHHNf7SlWQUCM,80
+mcp_server_code_assist-0.1.3.dist-info/RECORD,,