Merge pull request #1 from VishwasSomasekhariah/feature-add-ast-grep-tool

Feature add ast grep tool

Author: VishwasSomasekhariah

Dockerfile

@@ -0,0 +1,41 @@
+# Use Python 3.12.8-slim as the base image
+FROM python:3.12.8-slim
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=off \
+    PIP_DISABLE_PIP_VERSION_CHECK=on \
+    PROJECT_PATH="/project"
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    curl \
+    git \
+    build-essential \
+    && curl -LO https://github.com/ast-grep/ast-grep/releases/latest/download/ast-grep-linux-x64.tar.gz \
+    && tar -xzf ast-grep-linux-x64.tar.gz \
+    && mv sg /usr/local/bin/ast-grep \
+    && chmod +x /usr/local/bin/ast-grep \
+    && rm ast-grep-linux-x64.tar.gz \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create a directory for the project mount
+RUN mkdir -p /project
+
+# Set working directory
+WORKDIR /app
+
+# Clone the repository (replace with your actual repository URL)
+RUN git clone https://github.com/yourusername/code-analysis-mcp-server.git . \
+    && pip install --no-cache-dir -e .
+
+# Expose the port for the MCP server
+EXPOSE 8000
+
+# Create a volume for the project
+VOLUME ["/project"]
+
+# Command to run the server
+ENTRYPOINT ["code-analysis-server"]

README.md

@@ -1,2 +1,192 @@
-# code-analysis-mcp-server
-MCP server for code analysis tools like ast-grep integration. Provides a standardized interface for AI assistants to perform structural code queries and transformations.
+# Code Analysis MCP Server
+A modular Model Context Protocol (MCP) server for code analysis tools, with a primary focus on ast-grep integration. This server provides a standardized interface for AI assistants and other clients to perform structural code queries and transformations.
+
+## Features
+- **Structural Code Analysis**: Find patterns in code using ast-grep's powerful pattern matching capabilities
+
+- **Code Transformations**: Replace patterns in code with new implementations
+
+- **YAML Rule Support**: Apply custom lint and transformation rules defined in YAML
+
+- **Project-Wide Scanning**: Scan entire projects for patterns or rule violations
+
+- **Containerized Deployment**: Easy deployment with Docker
+
+- **Modular Architecture**: Designed for easy extension with additional code analysis tools
+
+## Requirements
+- Python 3.12.8
+
+- ast-grep CLI tool
+
+- Docker (for containerized deployment)
+
+## Installation
+**Using pip**
+
+```bash
+pip install code-analysis-mcp-server
+```
+
+**From source**
+```bash
+git clone https://github.com/yourusername/code-analysis-mcp-server.git
+cd code-analysis-mcp-server
+pip install -e .
+```
+
+**Using Docker**
+```bash
+docker pull yourusername/code-analysis-mcp-server
+# or build locally
+docker build -t code-analysis-mcp-server .
+```
+
+## Usage
+### Starting the server
+```bash
+# Direct execution
+code-analysis-server
+
+# With Docker
+docker run -p 8000:8000 -v /path/to/your/project:/project code-analysis-mcp-server
+```
+
+### Using with Docker Compose  
+**Create a docker-compose.yml file:**
+```text
+version: '3'
+
+services:
+  code-analysis-mcp:
+    image: yourusername/code-analysis-mcp-server
+    ports:
+      - "8000:8000"
+    volumes:
+      - ${PROJECT_DIR:-./sample_project}:/project
+    environment:
+      - PROJECT_PATH=/project
+    restart: unless-stopped
+```
+**Then run:**
+```bash
+PROJECT_DIR=/path/to/your/project docker-compose up -d
+```
+
+**Client Example**
+```python
+import asyncio
+import json
+from fastmcp import Client
+from fastmcp.client.transports import SSETransport
+
+async def main():
+    async with Client(transport=SSETransport("http://localhost:8000/sse")) as client:
+        # Set project path
+        project_result = await client.call_tool(
+            "ast_grep_set_project_path",
+            {
+                "project_path": "/project"
+            }
+        )
+        
+        # Process the result correctly
+        content_item = project_result[0]
+        result_data = json.loads(content_item.text)
+        print("Project path result:", result_data)
+        
+        # Find patterns in a file
+        find_result = await client.call_tool(
+            "ast_grep_find_pattern",
+            {
+                "file_path": "src/example.cs",
+                "pattern": "public class $NAME"
+            }
+        )
+        
+        # Process results
+        content_item = find_result[0]
+        result_data = json.loads(content_item.text)
+        print(f"Found {result_data.get('count', 0)} matches")
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+## Available Tools
+The server provides the following tools:
+### ast-grep
+- **ast_grep_set_project_path:** Set the project path for subsequent operations
+- **ast_grep_parse_code:** Parse code into an AST
+- **ast_grep_find_pattern:** Find patterns in code files
+- **ast_grep_replace_pattern:** Replace patterns in code files
+- **ast_grep_run_yaml_rule:** Run custom YAML rules on code files
+- **ast_grep_scan_project:** Scan entire projects for patterns or rule violations
+- **ast_grep_initialize_project:** Initialize a new ast-grep project
+- **ast_grep_test_rule:** Test ast-grep rules
+
+## Supported Languages
+- Python
+- JavaScript
+- TypeScript
+- Rust
+- Go
+- Java
+- C
+- C++
+- C#
+
+## Project Structure
+```text
+code-analysis-mcp-server/
+├── src/
+│   ├── __init__.py
+│   ├── server.py                # Main MCP server setup
+│   ├── tools/
+│   │   ├── __init__.py          # Tool registration
+│   │   ├── ast_grep/
+│   │   │   ├── __init__.py
+│   │   │   ├── tools.py         # ast-grep tool implementations
+│   │   └── common/
+│   │       ├── __init__.py
+│   │       └── utils.py         # Shared utilities
+│   └── resources/
+│       ├── __init__.py
+│       └── status.py            # System status resource
+├── tests/
+├── pyproject.toml
+├── setup.py
+├── requirements.txt
+├── Dockerfile
+└── README.md
+```
+
+## Contributing
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+This project is licensed under the MIT License - see the LICENSE file for details.
+
+```text
+MIT License
+
+Copyright (c) 2025 Your Name
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+```

docker-compose.yml

@@ -0,0 +1,14 @@
+version: '3'
+
+services:
+  code-analysis-mcp:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    ports:
+      - "8000:8000"
+    volumes:
+      - ${PROJECT_DIR:-./sample_project}:/project
+    environment:
+      - PROJECT_PATH=/project
+    restart: unless-stopped

pyproject.toml

@@ -0,0 +1,29 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "code-analysis-mcp-server"
+version = "0.1.0"
+description = "MCP Server for Code Analysis Tools"
+readme = "README.md"
+requires-python = ">=3.12.8,<3.13"
+license = {text = "MIT"}
+authors = [
+    {name = "Your Name", email = "[email protected]"}
+]
+dependencies = [
+    "pip==25.0.1",
+    "uvicorn==0.34.1",
+    "starlette==0.46.2",
+    "PyYAML==6.0.2",
+    "ast-grep-py==0.37.0",
+    "fastmcp==2.1.2",
+]
+
+[project.scripts]
+code-analysis-server = "src.server:main"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["src*"]

requirements.txt

@@ -0,0 +1,6 @@
+fastmcp==2.1.2
+PyYAML==6.0.2
+uvicorn==0.34.1
+starlette==0.46.2
+pytest==7.4.0
+pytest-asyncio==0.21.1

setup.py

@@ -0,0 +1,3 @@
+from setuptools import setup
+
+setup()

src/__init__.py

@@ -0,0 +1,4 @@
+"""Resource implementations for the MCP server."""
+
+# This file makes the resources directory a proper Python package.
+# It can be used to export specific resources if needed.

src/resources/__init__.py

@@ -0,0 +1,49 @@
+"""System status resource for the MCP server."""
+
+from typing import Dict, Any
+
+def register_status_resource(mcp):
+    """Register the system status resource with the MCP server."""
+    
+    @mcp.resource("system://status")
+    async def get_system_status() -> Dict[str, Any]:
+        """
+        Provide system status information about the Code Analysis MCP server.
+        
+        Returns:
+            Status information including version details
+        """
+        try:
+            import ast_grep_py
+            import fastmcp
+            
+            # Get version information
+            ast_grep_version = getattr(ast_grep_py, "__version__", "unknown")
+            fastmcp_version = getattr(fastmcp, "__version__", "unknown")
+            
+            # Get available tools
+            available_tools = {
+                "ast_grep": [
+                    "ast_grep_parse_code",
+                    "ast_grep_find_pattern",
+                    "ast_grep_replace_pattern",
+                    "ast_grep_run_yaml_rule"
+                ]
+                # Future tools will be added here
+            }
+            
+            return {
+                "status": "operational",
+                "ast_grep_py_version": ast_grep_version,
+                "fastmcp_version": fastmcp_version,
+                "supported_languages": [
+                    "python", "javascript", "typescript", "rust", 
+                    "go", "java", "c", "cpp", "csharp"
+                ],
+                "available_tools": available_tools
+            }
+        except Exception as exc:
+            return {
+                "status": "degraded",
+                "error": str(exc)
+            }

src/resources/status.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+"""
+MCP Server for Code Analysis Tools using Anthropic's fastmcp Framework
+"""
+
+import importlib
+import os
+from fastmcp import FastMCP
+from starlette.applications import Starlette
+from starlette.routing import Mount
+import uvicorn
+
+# Create the MCP server instance
+mcp = FastMCP("Code Analysis Tools")
+
+# Import and register resources
+from src.resources.status import register_status_resource
+register_status_resource(mcp)
+
+# Import and register tools
+from src.tools import register_all_tools
+register_all_tools(mcp)
+
+# Create a Starlette application with the SSE endpoint
+app = Starlette(routes=[
+    # Mount the SSE app at the root
+    Mount("/", app=mcp.sse_app()),
+])
+
+def main():
+    """Run the MCP server."""
+    print("Starting Code Analysis MCP server...")
+    uvicorn.run(app, host="0.0.0.0", port=8000)
+
+if __name__ == "__main__":
+    main()