Adding src files with ast-grep tool support

Author: VishwasSomasekhariah

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()

src/server.py

@@ -0,0 +1,14 @@
+"""Tool registration for the MCP server."""
+
+def register_all_tools(mcp):
+    """Register all available tools with the MCP server."""
+    # Import and register ast-grep tools
+    from src.tools.ast_grep.tools import register_tools as register_ast_grep_tools
+    register_ast_grep_tools(mcp)
+    
+    # In the future, you can add more tool registrations here:
+    # from src.tools.semgrep.tools import register_tools as register_semgrep_tools
+    # register_semgrep_tools(mcp)
+    
+    # from src.tools.pylint.tools import register_tools as register_pylint_tools
+    # register_pylint_tools(mcp)

src/tools/__init__.py

@@ -0,0 +1,9 @@
+"""
+MCP Server for ast-grep Tool using Anthropic's fastmcp Framework
+"""
+
+__version__ = "0.1.0"
+
+from .server import main
+
+__all__ = ["main"]