codegraphcontext 0.1.10__tar.gz → 0.1.12__tar.gz

codegraphcontext-0.1.12/MANIFEST.in

@@ -0,0 +1,2 @@
+prune website
+prune images

{codegraphcontext-0.1.10/src/codegraphcontext.egg-info → codegraphcontext-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.10
+Version: 0.1.12
 Summary: An MCP server that indexes local code into a graph database to provide context to AI assistants.
 Author-email: Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 License: MIT License
@@ -33,7 +33,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: neo4j>=5.15.0
@@ -56,14 +56,26 @@ Dynamic: license-file
 [![PyPI downloads](https://img.shields.io/pypi/dm/codegraphcontext)](https://pypi.org/project/codegraphcontext/)
 [![GitHub stars](https://img.shields.io/github/stars/Shashankss1205/CodeGraphContext?style=social)](https://github.com/Shashankss1205/CodeGraphContext/stargazers)
 [![License](https://img.shields.io/github/license/Shashankss1205/CodeGraphContext)](LICENSE)
+[![Website](https://img.shields.io/badge/website-up-brightgreen)](http://codegraphcontext.vercel.app/)
+[![Watch on YouTube](https://img.shields.io/badge/YouTube-Watch%20Demo-red?logo=youtube)](https://youtu.be/KYYSdxhg1xU)
+[![Join Discord](https://img.shields.io/badge/Discord-Join%20Chat-7289da?logo=discord&logoColor=white)](https://discord.gg/dR4QY32uYQ)
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
+### Indexing a codebase
+![Indexing using an MCP client](images/Indexing.gif)
+
+### Using the MCP server
+![Using the MCP server](images/Usecase.gif)
+
 ## Project Details
-- **Version:** 0.1.10
+- **Version:** 0.1.11
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
-- **Website:** [CodeGraphContext](http://code-graph-context.vercel.app/)
+- **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
+
+## Star History
+[![Star History Chart](https://api.star-history.com/svg?repos=Shashankss1205/CodeGraphContext&type=Date)](https://www.star-history.com/#Shashankss1205/CodeGraphContext&Date)
 
 ## Features
 
@@ -97,19 +109,33 @@ If you’re using CodeGraphContext in your project, feel free to open a PR and a
 
 1.  **Install:** `pip install codegraphcontext`
 2.  **Setup:** `cgc setup`
-    This interactive command guides you through configuring your Neo4j database connection. It offers several options:
-    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker, which is the easiest way to get started.
-    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j directly on your machine.
+    This interactive command guides you through configuring your Neo4j database connection and automatically setting up your IDE.
+    
+      **Database Configuration:**
+    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker. Requires Docker and Docker Compose to be installed.
+    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j. Requires `sudo` privileges.
     *   **Hosted Setup:** Allows you to connect to an existing remote Neo4j database (e.g., Neo4j AuraDB).
-    Upon successful configuration, `cgc setup` will generate two important files:
-    *   `mcp.json`: Contains the MCP client configuration for CodeGraphContext.
-    *   `~/.codegraphcontext/.env`: Stores your Neo4j connection credentials securely.
+
+    **IDE/CLI Configuration:**
+    After setting up your database, the wizard will ask to configure your development environment. It can automatically detect and configure the following:
+    *   VS Code
+    *   Cursor
+    *   Claude
+    *   Gemini CLI
+
+    Upon successful configuration, `cgc setup` will generate and place the necessary configuration files:
+    *   It creates an `mcp.json` file in your current directory for reference.
+    *   It stores your Neo4j credentials securely in `~/.codegraphcontext/.env`.
+    *   It updates the settings file of your chosen IDE/CLI (e.g., `.claude.json` or VS Code's `settings.json`).
+
 3.  **Start:** `cgc start`
 
 
 ## MCP Client Configuration
 
-Add the following to your MCP client's configuration:
+The `cgc setup` command attempts to automatically configure your IDE/CLI. If you choose not to use the automatic setup, or if your tool is not supported, you can configure it manually.
+
+Add the following server configuration to your client's settings file (e.g., VS Code's `settings.json` or `.claude.json`):
 
 ```json
 {
@@ -120,9 +146,9 @@ Add the following to your MCP client's configuration:
         "start"
       ],
       "env": {
-        "NEO4J_URI": "************",
-        "NEO4J_USER": "************",
-        "NEO4J_PASSWORD": "**************"
+        "NEO4J_URI": "YOUR_NEO4J_URI",
+        "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
+        "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
       },
       "tools": {
         "alwaysAllow": [

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/README.md

@@ -4,14 +4,26 @@
 [![PyPI downloads](https://img.shields.io/pypi/dm/codegraphcontext)](https://pypi.org/project/codegraphcontext/)
 [![GitHub stars](https://img.shields.io/github/stars/Shashankss1205/CodeGraphContext?style=social)](https://github.com/Shashankss1205/CodeGraphContext/stargazers)
 [![License](https://img.shields.io/github/license/Shashankss1205/CodeGraphContext)](LICENSE)
+[![Website](https://img.shields.io/badge/website-up-brightgreen)](http://codegraphcontext.vercel.app/)
+[![Watch on YouTube](https://img.shields.io/badge/YouTube-Watch%20Demo-red?logo=youtube)](https://youtu.be/KYYSdxhg1xU)
+[![Join Discord](https://img.shields.io/badge/Discord-Join%20Chat-7289da?logo=discord&logoColor=white)](https://discord.gg/dR4QY32uYQ)
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
+### Indexing a codebase
+![Indexing using an MCP client](images/Indexing.gif)
+
+### Using the MCP server
+![Using the MCP server](images/Usecase.gif)
+
 ## Project Details
-- **Version:** 0.1.10
+- **Version:** 0.1.11
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
-- **Website:** [CodeGraphContext](http://code-graph-context.vercel.app/)
+- **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
+
+## Star History
+[![Star History Chart](https://api.star-history.com/svg?repos=Shashankss1205/CodeGraphContext&type=Date)](https://www.star-history.com/#Shashankss1205/CodeGraphContext&Date)
 
 ## Features
 
@@ -45,19 +57,33 @@ If you’re using CodeGraphContext in your project, feel free to open a PR and a
 
 1.  **Install:** `pip install codegraphcontext`
 2.  **Setup:** `cgc setup`
-    This interactive command guides you through configuring your Neo4j database connection. It offers several options:
-    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker, which is the easiest way to get started.
-    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j directly on your machine.
+    This interactive command guides you through configuring your Neo4j database connection and automatically setting up your IDE.
+    
+      **Database Configuration:**
+    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker. Requires Docker and Docker Compose to be installed.
+    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j. Requires `sudo` privileges.
     *   **Hosted Setup:** Allows you to connect to an existing remote Neo4j database (e.g., Neo4j AuraDB).
-    Upon successful configuration, `cgc setup` will generate two important files:
-    *   `mcp.json`: Contains the MCP client configuration for CodeGraphContext.
-    *   `~/.codegraphcontext/.env`: Stores your Neo4j connection credentials securely.
+
+    **IDE/CLI Configuration:**
+    After setting up your database, the wizard will ask to configure your development environment. It can automatically detect and configure the following:
+    *   VS Code
+    *   Cursor
+    *   Claude
+    *   Gemini CLI
+
+    Upon successful configuration, `cgc setup` will generate and place the necessary configuration files:
+    *   It creates an `mcp.json` file in your current directory for reference.
+    *   It stores your Neo4j credentials securely in `~/.codegraphcontext/.env`.
+    *   It updates the settings file of your chosen IDE/CLI (e.g., `.claude.json` or VS Code's `settings.json`).
+
 3.  **Start:** `cgc start`
 
 
 ## MCP Client Configuration
 
-Add the following to your MCP client's configuration:
+The `cgc setup` command attempts to automatically configure your IDE/CLI. If you choose not to use the automatic setup, or if your tool is not supported, you can configure it manually.
+
+Add the following server configuration to your client's settings file (e.g., VS Code's `settings.json` or `.claude.json`):
 
 ```json
 {
@@ -68,9 +94,9 @@ Add the following to your MCP client's configuration:
         "start"
       ],
       "env": {
-        "NEO4J_URI": "************",
-        "NEO4J_USER": "************",
-        "NEO4J_PASSWORD": "**************"
+        "NEO4J_URI": "YOUR_NEO4J_URI",
+        "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
+        "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
       },
       "tools": {
         "alwaysAllow": [

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.10"
+version = "0.1.12"
 description = "An MCP server that indexes local code into a graph database to provide context to AI assistants."
 authors = [{ name = "Shashank Shekhar Singh", email = "shashankshekharsingh1205@gmail.com" }]
 readme = "README.md"
 license = { file = "LICENSE" }
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: MIT License",

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/cli/main.py

@@ -31,7 +31,7 @@ app = typer.Typer(
     help="CodeGraphContext: An MCP server for AI-powered code analysis.",
     add_completion=False,
 )
-console = Console()
+console = Console(stderr=True)
 
 # Configure basic logging for the application.
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(name)s - %(message)s')

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/cli/setup_wizard.py

@@ -15,7 +15,7 @@ def _generate_mcp_json(creds):
     """Generates and prints the MCP JSON configuration."""
     cgc_path = shutil.which("cgc") or sys.executable
 
-    if "python" in Path(cgc_path).name:  
+    if "python" in Path(cgc_path).name:
         # fallback to running as module if no cgc binary is found
         command = cgc_path
         args = ["-m", "cgc", "start"]
@@ -49,11 +49,11 @@ def _generate_mcp_json(creds):
             }
         }
     }
-    
+
     console.print("\n[bold green]Configuration successful![/bold green]")
     console.print("Copy the following JSON and add it to your MCP server configuration file:")
     console.print(json.dumps(mcp_config, indent=2))
-    
+
     # Also save to a file for convenience
     mcp_file = Path.cwd() / "mcp.json"
     with open(mcp_file, "w") as f:
@@ -69,6 +69,109 @@ def _generate_mcp_json(creds):
         f.write(f"NEO4J_PASSWORD={creds.get('password', '')}\n")
 
     console.print(f"[cyan]Neo4j credentials also saved to: {env_file}[/cyan]")
+    _configure_ide(mcp_config)
+
+def _configure_ide(mcp_config):
+    """Asks user for their IDE and configures it automatically."""
+    questions = [
+        {
+            "type": "confirm",
+            "message": "Automatically configure your IDE/CLI (VS Code, Cursor, Claude, Gemini)?",
+            "name": "configure_ide",
+            "default": True,
+        }
+    ]
+    result = prompt(questions)
+    if not result or not result.get("configure_ide"):
+        console.print("\n[cyan]Skipping automatic IDE configuration. You can add the MCP server manually.[/cyan]")
+        return
+
+    ide_questions = [
+        {
+            "type": "list",
+            "message": "Choose your IDE/CLI to configure:",
+            "choices": ["VS Code", "Cursor", "Claude code", "Gemini CLI", "None of the above"],
+            "name": "ide_choice",
+        }
+    ]
+    ide_result = prompt(ide_questions)
+    ide_choice = ide_result.get("ide_choice")
+
+    if not ide_choice or ide_choice == "None of the above":
+        console.print("\n[cyan]You can add the MCP server manually to your IDE/CLI.[/cyan]")
+        return
+
+    if ide_choice in ["VS Code", "Cursor", "Claude code", "Gemini CLI"]:
+        console.print(f"\n[bold cyan]Configuring for {ide_choice}...[/bold cyan]")
+        
+        config_paths = {
+            "VS Code": [
+                Path.home() / ".config" / "Code" / "User" / "settings.json",
+                Path.home() / "Library" / "Application Support" / "Code" / "User" / "settings.json",
+                Path.home() / "AppData" / "Roaming" / "Code" / "User" / "settings.json"
+            ],
+            "Cursor": [
+                Path.home() / ".cursor" / "settings.json",
+                Path.home() / ".config" / "cursor" / "settings.json",
+                Path.home() / "Library" / "Application Support" / "cursor" / "settings.json",
+                Path.home() / "AppData" / "Roaming" / "cursor" / "settings.json",
+                Path.home() / ".config" / "Cursor" / "User" / "settings.json",
+            ],
+            "Claude code": [
+                Path.home() / ".claude.json"
+            ],
+            "Gemini CLI": [
+                Path.home() / ".gemini" / "settings.json"
+            ]
+        }
+
+        target_path = None
+        paths_to_check = config_paths.get(ide_choice, [])
+        for path in paths_to_check:
+            if path.exists():
+                target_path = path
+                break
+        
+        if not target_path:
+            # If file doesn't exist, check if parent directory exists
+            for path in paths_to_check:
+                if path.parent.exists():
+                    target_path = path
+                    break
+        
+        if not target_path:
+            console.print(f"[yellow]Could not automatically find or create the configuration directory for {ide_choice}.[/yellow]")
+            console.print("Please add the MCP configuration manually from the `mcp.json` file generated above.")
+            return
+
+        console.print(f"Using configuration file at: {target_path}")
+        
+        try:
+            with open(target_path, "r") as f:
+                try:
+                    settings = json.load(f)
+                except json.JSONDecodeError:
+                    settings = {}
+        except FileNotFoundError:
+            settings = {}
+
+        if not isinstance(settings, dict):
+            console.print(f"[red]Error: Configuration file at {target_path} is not a valid JSON object.[/red]")
+            return
+
+        if "mcpServers" not in settings:
+            settings["mcpServers"] = {}
+        
+        settings["mcpServers"].update(mcp_config["mcpServers"])
+
+        try:
+            with open(target_path, "w") as f:
+                json.dump(settings, f, indent=2)
+            console.print(f"[green]Successfully updated {ide_choice} configuration.[/green]")
+        except Exception as e:
+            console.print(f"[red]Failed to write to configuration file: {e}[/red]")
+
+
 
 
 def get_project_root() -> Path:

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/server.py

@@ -7,6 +7,7 @@ import stdlibs
 import sys
 import traceback
 import os
+import re
 from datetime import datetime
 from pathlib import Path
 from neo4j.exceptions import CypherSyntaxError
@@ -21,18 +22,10 @@ from .core.watcher import CodeWatcher
 from .tools.graph_builder import GraphBuilder
 from .tools.code_finder import CodeFinder
 from .tools.import_extractor import ImportExtractor
-
+from .utils.debug_log import debug_log
 
 logger = logging.getLogger(__name__)
 
-def debug_log(message):
-    """Write debug message to a file for development and testing."""
-    debug_file = os.path.expanduser("~/mcp_debug.log")
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    with open(debug_file, "a") as f:
-        f.write(f"[{timestamp}] {message}\n")
-        f.flush()
-
 class MCPServer:
     """
     The main MCP Server class.
@@ -132,7 +125,7 @@ class MCPServer:
                     "properties": {
                         "query_type": {"type": "string", "description": "Type of relationship query to run.", "enum": ["find_callers", "find_callees", "find_all_callers", "find_all_callees", "find_importers", "who_modifies", "class_hierarchy", "overrides", "dead_code", "call_chain", "module_deps", "variable_scope", "find_complexity", "find_functions_by_argument", "find_functions_by_decorator"]},
                         "target": {"type": "string", "description": "The function, class, or module to analyze."},
-                        "context": {"type": "string", "description": "Optional: specific file path for precise results."}
+                        "context": {"type": "string", "description": "Optional: specific file path for precise results."} 
                     },
                     "required": ["query_type", "target"]
                 }
@@ -197,7 +190,7 @@ class MCPServer:
                     "type": "object",
                     "properties": {
                         "function_name": {"type": "string", "description": "The name of the function to analyze."},
-                        "file_path": {"type": "string", "description": "Optional: The full path to the file containing the function for a more specific query."}
+                        "file_path": {"type": "string", "description": "Optional: The full path to the file containing the function for a more specific query."} 
                     },
                     "required": ["function_name"]
                 }
@@ -307,12 +300,21 @@ class MCPServer:
             return {"error": "Cypher query cannot be empty."}
 
         # Safety Check: Prevent any write operations to the database.
+        # This check first removes all string literals and then checks for forbidden keywords.
         forbidden_keywords = ['CREATE', 'MERGE', 'DELETE', 'SET', 'REMOVE', 'DROP', 'CALL apoc']
-        query_upper = cypher_query.upper()
-        if any(keyword in query_upper for keyword in forbidden_keywords):
-            return {
-                "error": "This tool only supports read-only queries. Prohibited keywords like CREATE, MERGE, DELETE, SET, etc., are not allowed."
-            }
+        
+        # Regex to match single or double quoted strings, handling escaped quotes.
+        string_literal_pattern = r'"(?:\\.|[^"\\])*"|\'(?:\\.|[^\'\\])*\''
+        
+        # Remove all string literals from the query.
+        query_without_strings = re.sub(string_literal_pattern, '', cypher_query)
+        
+        # Now, check for forbidden keywords in the query without strings.
+        for keyword in forbidden_keywords:
+            if re.search(r'\b' + keyword + r'\b', query_without_strings, re.IGNORECASE):
+                return {
+                    "error": "This tool only supports read-only queries. Prohibited keywords like CREATE, MERGE, DELETE, SET, etc., are not allowed."
+                }
 
         try:
             debug_log(f"Executing Cypher query: {cypher_query}")
@@ -785,7 +787,7 @@ class MCPServer:
                     response = {
                         "jsonrpc": "2.0", "id": request_id,
                         "result": {
-                            "protocolVersion": "2024-11-05",
+                            "protocolVersion": "2025-03-26",
                             "serverInfo": {
                                 "name": "CodeGraphContext", "version": "0.1.0",
                                 "systemPrompt": LLM_SYSTEM_PROMPT
@@ -804,24 +806,40 @@ class MCPServer:
                     tool_name = params.get('name')
                     args = params.get('arguments', {})
                     result = await self.handle_tool_call(tool_name, args)
-                    response = {
-                        "jsonrpc": "2.0", "id": request_id,
-                        "result": {"content": [{"type": "text", "text": json.dumps(result, indent=2)}]}
-                    }
+                    
+                    if "error" in result:
+                        response = {
+                            "jsonrpc": "2.0", "id": request_id,
+                            "error": {"code": -32000, "message": "Tool execution error", "data": result}
+                        }
+                    else:
+                        response = {
+                            "jsonrpc": "2.0", "id": request_id,
+                            "result": {"content": [{"type": "text", "text": json.dumps(result, indent=2)}]}
+                        }
+                elif method == 'notifications/initialized':
+                    # This is a notification, no response needed.
+                    pass
                 else:
                     # Handle unknown methods.
-                    response = {
-                        "jsonrpc": "2.0", "id": request_id,
-                        "error": {"code": -32601, "message": f"Method not found: {method}"}
-                    }
+                    if request_id is not None:
+                        response = {
+                            "jsonrpc": "2.0", "id": request_id,
+                            "error": {"code": -32601, "message": f"Method not found: {method}"}
+                        }
                 
-                # Send the response to standard output.
-                print(json.dumps(response), flush=True)
+                # Send the response to standard output if it's not a notification.
+                if request_id is not None and response:
+                    print(json.dumps(response), flush=True)
 
             except Exception as e:
                 logger.error(f"Error processing request: {e}\n{traceback.format_exc()}")
+                request_id = "unknown"
+                if 'request' in locals() and isinstance(request, dict):
+                    request_id = request.get('id', "unknown")
+
                 error_response = {
-                    "jsonrpc": "2.0", "id": request.get('id') if 'request' in locals() else None,
+                    "jsonrpc": "2.0", "id": request_id,
                     "error": {"code": -32603, "message": f"Internal error: {str(e)}", "data": traceback.format_exc()}
                 }
                 print(json.dumps(error_response), flush=True)

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/tools/graph_builder.py

@@ -9,6 +9,7 @@ from datetime import datetime
 
 from ..core.database import DatabaseManager
 from ..core.jobs import JobManager, JobStatus
+from ..utils.debug_log import debug_log
 
 logger = logging.getLogger(__name__)
 
@@ -16,15 +17,6 @@ logger = logging.getLogger(__name__)
 # Set to 1 to enable, 0 to disable.
 debug_mode = 0
 
-def debug_log(message):
-    """Write debug message to a file"""
-    debug_file = os.path.expanduser("~/mcp_debug.log")
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    with open(debug_file, "a") as f:
-        f.write(f"[{timestamp}] {message}\n")
-        f.flush()
-
-
 class CyclomaticComplexityVisitor(ast.NodeVisitor):
     """Calculates cyclomatic complexity for a given AST node."""
     def __init__(self):
@@ -985,16 +977,29 @@ class GraphBuilder:
                 """, path=path)
 
     def delete_repository_from_graph(self, repo_path: str):
-        """Deletes a repository and all its contents from the graph."""
+        """
+        Deletes a repository and all its contents from the graph, then cleans up
+        any orphaned Module nodes that are no longer referenced.
+        """
         repo_path_str = str(Path(repo_path).resolve())
         with self.driver.session() as session:
+            # Delete the repository and all its contained elements, including parameters
             session.run("""
                 MATCH (r:Repository {path: $path})
                 OPTIONAL MATCH (r)-[:CONTAINS*]->(e)
-                DETACH DELETE r, e
+                OPTIONAL MATCH (e)-[:HAS_PARAMETER]->(p)
+                DETACH DELETE r, e, p
             """, path=repo_path_str)
             logger.info(f"Deleted repository and its contents from graph: {repo_path_str}")
 
+            # Clean up orphaned Module nodes that are no longer imported by any file
+            session.run("""
+                MATCH (m:Module)
+                WHERE NOT ()-[:IMPORTS]->(m)
+                DETACH DELETE m
+            """)
+            logger.info("Cleaned up orphaned Module nodes.")
+
     def update_file_in_graph(self, file_path: Path, repo_path: Path, imports_map: dict):
         """
         Updates a single file's nodes in the graph and returns its new parsed data.

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/tools/import_extractor.py

@@ -13,15 +13,9 @@ import stdlibs
 import os
 from datetime import datetime
 
-logger = logging.getLogger(__name__)
+from ..utils.debug_log import debug_log
 
-def debug_log(message):
-    """Write debug message to a file for development and testing."""
-    debug_file = os.path.expanduser("~/mcp_debug.log")
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    with open(debug_file, "a") as f:
-        f.write(f"[{timestamp}] {message}\n")
-        f.flush()
+logger = logging.getLogger(__name__)
 
 class ImportExtractor:
     """

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext/tools/system.py

@@ -8,12 +8,10 @@ from neo4j.exceptions import CypherSyntaxError
 
 from ..core.database import DatabaseManager
 from ..core.jobs import JobManager, JobStatus
+from ..utils.debug_log import debug_log
 
 logger = logging.getLogger(__name__)
 
-def debug_log(message):
-    # This helper can be moved to a shared utils.py file later
-    ...
 
 class SystemTools:
     """Handles system-level tools like job management and direct DB queries."""

codegraphcontext-0.1.12/src/codegraphcontext/utils/debug_log.py

@@ -0,0 +1,10 @@
+import os
+from datetime import datetime
+
+def debug_log(message):
+    """Write debug message to a file"""
+    debug_file = os.path.expanduser("~/mcp_debug.log")
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    with open(debug_file, "a") as f:
+        f.write(f"[{timestamp}] {message}\n")
+        f.flush()

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.10
+Version: 0.1.12
 Summary: An MCP server that indexes local code into a graph database to provide context to AI assistants.
 Author-email: Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 License: MIT License
@@ -33,7 +33,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: neo4j>=5.15.0
@@ -56,14 +56,26 @@ Dynamic: license-file
 [![PyPI downloads](https://img.shields.io/pypi/dm/codegraphcontext)](https://pypi.org/project/codegraphcontext/)
 [![GitHub stars](https://img.shields.io/github/stars/Shashankss1205/CodeGraphContext?style=social)](https://github.com/Shashankss1205/CodeGraphContext/stargazers)
 [![License](https://img.shields.io/github/license/Shashankss1205/CodeGraphContext)](LICENSE)
+[![Website](https://img.shields.io/badge/website-up-brightgreen)](http://codegraphcontext.vercel.app/)
+[![Watch on YouTube](https://img.shields.io/badge/YouTube-Watch%20Demo-red?logo=youtube)](https://youtu.be/KYYSdxhg1xU)
+[![Join Discord](https://img.shields.io/badge/Discord-Join%20Chat-7289da?logo=discord&logoColor=white)](https://discord.gg/dR4QY32uYQ)
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
+### Indexing a codebase
+![Indexing using an MCP client](images/Indexing.gif)
+
+### Using the MCP server
+![Using the MCP server](images/Usecase.gif)
+
 ## Project Details
-- **Version:** 0.1.10
+- **Version:** 0.1.11
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
-- **Website:** [CodeGraphContext](http://code-graph-context.vercel.app/)
+- **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
+
+## Star History
+[![Star History Chart](https://api.star-history.com/svg?repos=Shashankss1205/CodeGraphContext&type=Date)](https://www.star-history.com/#Shashankss1205/CodeGraphContext&Date)
 
 ## Features
 
@@ -97,19 +109,33 @@ If you’re using CodeGraphContext in your project, feel free to open a PR and a
 
 1.  **Install:** `pip install codegraphcontext`
 2.  **Setup:** `cgc setup`
-    This interactive command guides you through configuring your Neo4j database connection. It offers several options:
-    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker, which is the easiest way to get started.
-    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j directly on your machine.
+    This interactive command guides you through configuring your Neo4j database connection and automatically setting up your IDE.
+    
+      **Database Configuration:**
+    *   **Local Setup (Docker Recommended):** Helps you set up a local Neo4j instance using Docker. Requires Docker and Docker Compose to be installed.
+    *   **Local Setup (Linux Binary):** For Debian-based Linux systems (like Ubuntu), `cgc setup` can automate the installation of Neo4j. Requires `sudo` privileges.
     *   **Hosted Setup:** Allows you to connect to an existing remote Neo4j database (e.g., Neo4j AuraDB).
-    Upon successful configuration, `cgc setup` will generate two important files:
-    *   `mcp.json`: Contains the MCP client configuration for CodeGraphContext.
-    *   `~/.codegraphcontext/.env`: Stores your Neo4j connection credentials securely.
+
+    **IDE/CLI Configuration:**
+    After setting up your database, the wizard will ask to configure your development environment. It can automatically detect and configure the following:
+    *   VS Code
+    *   Cursor
+    *   Claude
+    *   Gemini CLI
+
+    Upon successful configuration, `cgc setup` will generate and place the necessary configuration files:
+    *   It creates an `mcp.json` file in your current directory for reference.
+    *   It stores your Neo4j credentials securely in `~/.codegraphcontext/.env`.
+    *   It updates the settings file of your chosen IDE/CLI (e.g., `.claude.json` or VS Code's `settings.json`).
+
 3.  **Start:** `cgc start`
 
 
 ## MCP Client Configuration
 
-Add the following to your MCP client's configuration:
+The `cgc setup` command attempts to automatically configure your IDE/CLI. If you choose not to use the automatic setup, or if your tool is not supported, you can configure it manually.
+
+Add the following server configuration to your client's settings file (e.g., VS Code's `settings.json` or `.claude.json`):
 
 ```json
 {
@@ -120,9 +146,9 @@ Add the following to your MCP client's configuration:
         "start"
       ],
       "env": {
-        "NEO4J_URI": "************",
-        "NEO4J_USER": "************",
-        "NEO4J_PASSWORD": "**************"
+        "NEO4J_URI": "YOUR_NEO4J_URI",
+        "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
+        "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
       },
       "tools": {
         "alwaysAllow": [

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/src/codegraphcontext.egg-info/SOURCES.txt

@@ -1,4 +1,5 @@
 LICENSE
+MANIFEST.in
 README.md
 pyproject.toml
 src/codegraphcontext/__init__.py
@@ -23,6 +24,7 @@ src/codegraphcontext/tools/code_finder.py
 src/codegraphcontext/tools/graph_builder.py
 src/codegraphcontext/tools/import_extractor.py
 src/codegraphcontext/tools/system.py
+src/codegraphcontext/utils/debug_log.py
 tests/test_cgc_integration.py
 tests/test_imports.py
 tests/test_jsonrpc.py

{codegraphcontext-0.1.10 → codegraphcontext-0.1.12}/tests/test_tools.py

@@ -16,9 +16,18 @@ def call_tool(server, name, args):
         "params": {"name": name, "arguments": args}
     }
     response = server(request)
-    content = json.loads(response["result"]["content"][0]["text"])
-    return content
-
+    
+    # Handle both success and error responses
+    if "result" in response:
+        content = json.loads(response["result"]["content"][0]["text"])
+        return content
+    elif "error" in response:
+        # For error responses, return the error data
+        error_data = response["error"].get("data", {})
+        return error_data
+    else:
+        raise ValueError(f"Unexpected response format: {response}")
+        
 @pytest.fixture(scope="module")
 def server():
     """
@@ -371,3 +380,28 @@ def test_execute_cypher_query(indexed_project):
     assert len(results) > 0, "No results from Cypher query."
     assert "functionName" in results[0], "Cypher query result missing 'functionName' key."
     print("Successfully executed Cypher query.")
+
+def test_execute_cypher_query_with_forbidden_keyword_in_string(indexed_project):
+    """
+    Tests that a read-only query with a forbidden keyword inside a string literal is allowed.
+    """
+    server = indexed_project
+    print("\n--- Executing Cypher query with forbidden keyword in string ---")
+    # This query should not be blocked by the safety check
+    cypher_query = "MATCH (n:Function) WHERE n.name = 'create_user_function' RETURN n.name AS functionName"
+    query_result = call_tool(server, "execute_cypher_query", {"cypher_query": cypher_query})
+    assert query_result.get("success") is True, f"execute_cypher_query with forbidden keyword in string failed: {query_result.get('error')}"
+    print("Successfully executed Cypher query with forbidden keyword in string.")
+
+def test_execute_cypher_query_with_write_operation(indexed_project):
+    """
+    Tests that a query with a write operation is blocked.
+    """
+    server = indexed_project
+    print("\n--- Executing Cypher query with write operation ---")
+    cypher_query = "CREATE (n:TestNode) RETURN n"
+    query_result = call_tool(server, "execute_cypher_query", {"cypher_query": cypher_query})
+    assert query_result.get("success") is None, "execute_cypher_query with write operation should have failed"
+    assert "error" in query_result, "execute_cypher_query with write operation should have returned an error"
+    assert "read-only" in query_result.get("error", ""), "Error message should indicate that only read-only queries are supported"
+    print("Successfully blocked a write operation.")