codegraphcontext 0.1.7__tar.gz → 0.1.9__tar.gz

{codegraphcontext-0.1.7/src/codegraphcontext.egg-info → codegraphcontext-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.7
+Version: 0.1.9
 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
@@ -52,10 +52,18 @@ Dynamic: license-file
 
 # CodeGraphContext
 [![Build Status](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml/badge.svg)](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml)
-
+[![PyPI version](https://img.shields.io/pypi/v/codegraphcontext)](https://pypi.org/project/codegraphcontext/)
+[![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)
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
+## Project Details
+- **Version:** 0.1.8
+- **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
+- **License:** MIT License (See [LICENSE](LICENSE) for details)
+
 ## Features
 
 -   **Code Indexing:** Analyzes Python code and builds a knowledge graph of its components.
@@ -63,12 +71,33 @@ An MCP server that indexes local code into a graph database to provide context t
 -   **Live Updates:** Watches local files for changes and automatically updates the graph.
 -   **Interactive Setup:** A user-friendly command-line wizard for easy setup.
 
+## Used By
+
+CodeGraphContext is already being explored by developers and projects for:
+
+- **Static code analysis in AI assistants**
+- **Graph-based visualization of Python projects**
+- **Dead code and complexity detection**
+
+If you’re using CodeGraphContext in your project, feel free to open a PR and add it here! 🚀
+
+## Dependencies
+
+- `neo4j>=5.15.0`
+- `watchdog>=3.0.0`
+- `requests>=2.31.0`
+- `stdlibs>=2023.11.18`
+- `typer[all]>=0.9.0`
+- `rich>=13.7.0`
+- `inquirerpy>=0.3.4`
+- `python-dotenv>=1.0.0`
+
 ## Getting Started
 
 1.  **Install:** `pip install codegraphcontext`
 2.  **Setup:** `cgc setup`
 3.  **Start:** `cgc start`
-4.  **Index Code:** `cgc tool add-code-to-graph '{"path": "/path/to/your/project"}'`
+4.  **Index Code:** `cgc tool add-code-to-graph '{"path": "/path/to/your/project"}'` (Under active development)
 
 ## MCP Client Configuration
 
@@ -98,7 +127,11 @@ Add the following to your MCP client's configuration:
           "analyze_code_relationships",
           "watch_directory",
           "find_dead_code",
-          "execute_cypher_query"
+          "execute_cypher_query",
+          "calculate_cyclomatic_complexity",
+          "find_most_complex_functions",
+          "list_indexed_repositories",
+          "delete_repository"
         ],
         "disabled": false
       },
@@ -149,5 +182,29 @@ Once the server is running, you can interact with it through your AI assistant u
     -   "Which files import the `requests` library?"
     -   "Find all implementations of the `render` method."
 
+-   **Advanced Call Chain and Dependency Tracking (Spanning Hundreds of Files):**
+    The CodeGraphContext excels at tracing complex execution flows and dependencies across vast codebases. Leveraging the power of graph databases, it can identify direct and indirect callers and callees, even when a function is called through multiple layers of abstraction or across numerous files. This is invaluable for:
+    -   **Impact Analysis:** Understand the full ripple effect of a change to a core function.
+    -   **Debugging:** Trace the path of execution from an entry point to a specific bug.
+    -   **Code Comprehension:** Grasp how different parts of a large system interact.
+
+    -   "Show me the full call chain from the `main` function to `process_data`."
+    -   "Find all functions that directly or indirectly call `validate_input`."
+    -   "What are all the functions that `initialize_system` eventually calls?"
+    -   "Trace the dependencies of the `DatabaseManager` module."
+
 -   **Code Quality and Maintenance:**
     -   "Is there any dead or unused code in this project?"
+    -   "Calculate the cyclomatic complexity of the `process_data` function in `src/utils.py`."
+    -   "Find the 5 most complex functions in the codebase."
+
+-   **Repository Management:**
+    -   "List all currently indexed repositories."
+    -   "Delete the indexed repository at `/path/to/old-project`."
+
+## Contributing
+
+Contributions are welcome! 🎉  
+If you have ideas for new features, integrations, or improvements, open an [issue](https://github.com/Shashankss1205/CodeGraphContext/issues) or submit a PR.
+
+Join discussions and help shape the future of CodeGraphContext.

{codegraphcontext-0.1.7 → codegraphcontext-0.1.9}/README.md

@@ -1,9 +1,17 @@
 # CodeGraphContext
 [![Build Status](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml/badge.svg)](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml)
-
+[![PyPI version](https://img.shields.io/pypi/v/codegraphcontext)](https://pypi.org/project/codegraphcontext/)
+[![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)
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
+## Project Details
+- **Version:** 0.1.8
+- **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
+- **License:** MIT License (See [LICENSE](LICENSE) for details)
+
 ## Features
 
 -   **Code Indexing:** Analyzes Python code and builds a knowledge graph of its components.
@@ -11,12 +19,33 @@ An MCP server that indexes local code into a graph database to provide context t
 -   **Live Updates:** Watches local files for changes and automatically updates the graph.
 -   **Interactive Setup:** A user-friendly command-line wizard for easy setup.
 
+## Used By
+
+CodeGraphContext is already being explored by developers and projects for:
+
+- **Static code analysis in AI assistants**
+- **Graph-based visualization of Python projects**
+- **Dead code and complexity detection**
+
+If you’re using CodeGraphContext in your project, feel free to open a PR and add it here! 🚀
+
+## Dependencies
+
+- `neo4j>=5.15.0`
+- `watchdog>=3.0.0`
+- `requests>=2.31.0`
+- `stdlibs>=2023.11.18`
+- `typer[all]>=0.9.0`
+- `rich>=13.7.0`
+- `inquirerpy>=0.3.4`
+- `python-dotenv>=1.0.0`
+
 ## Getting Started
 
 1.  **Install:** `pip install codegraphcontext`
 2.  **Setup:** `cgc setup`
 3.  **Start:** `cgc start`
-4.  **Index Code:** `cgc tool add-code-to-graph '{"path": "/path/to/your/project"}'`
+4.  **Index Code:** `cgc tool add-code-to-graph '{"path": "/path/to/your/project"}'` (Under active development)
 
 ## MCP Client Configuration
 
@@ -46,7 +75,11 @@ Add the following to your MCP client's configuration:
           "analyze_code_relationships",
           "watch_directory",
           "find_dead_code",
-          "execute_cypher_query"
+          "execute_cypher_query",
+          "calculate_cyclomatic_complexity",
+          "find_most_complex_functions",
+          "list_indexed_repositories",
+          "delete_repository"
         ],
         "disabled": false
       },
@@ -97,5 +130,29 @@ Once the server is running, you can interact with it through your AI assistant u
     -   "Which files import the `requests` library?"
     -   "Find all implementations of the `render` method."
 
+-   **Advanced Call Chain and Dependency Tracking (Spanning Hundreds of Files):**
+    The CodeGraphContext excels at tracing complex execution flows and dependencies across vast codebases. Leveraging the power of graph databases, it can identify direct and indirect callers and callees, even when a function is called through multiple layers of abstraction or across numerous files. This is invaluable for:
+    -   **Impact Analysis:** Understand the full ripple effect of a change to a core function.
+    -   **Debugging:** Trace the path of execution from an entry point to a specific bug.
+    -   **Code Comprehension:** Grasp how different parts of a large system interact.
+
+    -   "Show me the full call chain from the `main` function to `process_data`."
+    -   "Find all functions that directly or indirectly call `validate_input`."
+    -   "What are all the functions that `initialize_system` eventually calls?"
+    -   "Trace the dependencies of the `DatabaseManager` module."
+
 -   **Code Quality and Maintenance:**
     -   "Is there any dead or unused code in this project?"
+    -   "Calculate the cyclomatic complexity of the `process_data` function in `src/utils.py`."
+    -   "Find the 5 most complex functions in the codebase."
+
+-   **Repository Management:**
+    -   "List all currently indexed repositories."
+    -   "Delete the indexed repository at `/path/to/old-project`."
+
+## Contributing
+
+Contributions are welcome! 🎉  
+If you have ideas for new features, integrations, or improvements, open an [issue](https://github.com/Shashankss1205/CodeGraphContext/issues) or submit a PR.
+
+Join discussions and help shape the future of CodeGraphContext.

{codegraphcontext-0.1.7 → codegraphcontext-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.7"
+version = "0.1.9"
 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"
@@ -31,6 +31,7 @@ dependencies = [
 
 [project.scripts]
 cgc = "codegraphcontext.cli.main:app"
+codegraphcontext = "codegraphcontext.cli.main:app"
 
 [project.optional-dependencies]
 dev = [

{codegraphcontext-0.1.7 → codegraphcontext-0.1.9}/src/codegraphcontext/cli/main.py

@@ -3,6 +3,10 @@ import typer
 from rich.console import Console
 import asyncio
 import logging
+import json
+import os
+from pathlib import Path
+from dotenv import load_dotenv, find_dotenv
 from codegraphcontext.server import MCPServer
 from .setup_wizard import run_setup_wizard
 
@@ -31,6 +35,44 @@ def start():
     Start the CodeGraphContext MCP server.
     """
     console.print("[bold green]Starting CodeGraphContext Server...[/bold green]")
+    
+    # 1. Prefer loading environment variables from mcp.json in the current directory
+    mcp_file_path = Path.cwd() / "mcp.json"
+    if mcp_file_path.exists():
+        try:
+            with open(mcp_file_path, "r") as f:
+                mcp_config = json.load(f)
+            
+            server_env = mcp_config.get("mcpServers", {}).get("CodeGraphContext", {}).get("env", {})
+            for key, value in server_env.items():
+                os.environ[key] = value
+            console.print("[green]Loaded Neo4j credentials from local mcp.json.[/green]")
+        except Exception as e:
+            console.print(f"[bold red]Error loading mcp.json:[/bold red] {e}")
+            console.print("[yellow]Attempting to start server without mcp.json environment variables.[/yellow]")
+    else:
+        # 2. If no local mcp.json, try to load from ~/.codegraphcontext/.env
+        global_env_path = Path.home() / ".codegraphcontext" / ".env"
+        if global_env_path.exists():
+            try:
+                load_dotenv(dotenv_path=global_env_path)
+                console.print(f"[green]Loaded Neo4j credentials from global .env file: {global_env_path}[/green]")
+            except Exception as e:
+                console.print(f"[bold red]Error loading global .env file from {global_env_path}:[/bold red] {e}")
+                console.print("[yellow]Attempting to start server without .env environment variables.[/yellow]")
+        else:
+            # Fallback: try to load from a .env file found by find_dotenv (searches up the tree)
+            try:
+                dotenv_path = find_dotenv(usecwd=True, raise_error_if_not_found=False)
+                if dotenv_path:
+                    load_dotenv(dotenv_path)
+                    console.print(f"[green]Loaded Neo4j credentials from global .env file: {dotenv_path}[/green]")
+                else:
+                    console.print("[yellow]No local mcp.json or global .env file found. Attempting to start server without explicit Neo4j credentials.[/yellow]")
+            except Exception as e:
+                console.print(f"[bold red]Error loading global .env file:[/bold red] {e}")
+                console.print("[yellow]Attempting to start server without .env environment variables.[/yellow]")
+
     server = None
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
@@ -79,15 +121,21 @@ def tool(
     console.print(f"Calling tool [bold cyan]{name}[/bold cyan] with args: {args}")
     console.print("[yellow]Note: This is a placeholder for direct tool invocation.[/yellow]")
 
+@app.command()
+def help(ctx: typer.Context):
+    """Show this message and exit."""
+    root_ctx = ctx.parent or ctx
+    typer.echo(root_ctx.get_help())
+
 
-@app.command(name="help")
-def show_help():
+@app.callback(invoke_without_command=True)
+def main(ctx: typer.Context):
     """
-    Show a list of available commands and their descriptions.
+    CodeGraphContext: An MCP server for AI-powered code analysis.
     """
-    console.print("[bold]CodeGraphContext CLI Commands:[/bold]")
-    for command in app.registered_commands:
-        # Get the first line of the docstring as a short description
-        description = command.help.split('\n')[0].strip() if command.help else "No description."
-        console.print(f"  [bold cyan]{command.name}[/bold cyan]: {description}")
-    console.print("\nFor more information on a specific command, run: [bold]cgc <command> --help[/bold]")
+    if ctx.invoked_subcommand is None:
+        console.print("[bold green]👋 Welcome to CodeGraphContext (cgc)![/bold green]\n")
+        console.print("👉 Run [cyan]cgc setup[/cyan] to configure the server and database.")
+        console.print("👉 Run [cyan]cgc start[/cyan] to launch the server.")
+        console.print("👉 Run [cyan]cgc help[/cyan] to see all available commands.\n")
+        console.print("👉 Running [green]codegraphcontext [white]works the same as using [green]cgc")

{codegraphcontext-0.1.7 → codegraphcontext-0.1.9}/src/codegraphcontext/cli/setup_wizard.py

@@ -30,7 +30,7 @@ def _generate_mcp_json(creds):
                 "args": args,
                 "env": {
                     "NEO4J_URI": creds.get("uri", ""),
-                    "NEO4J_USER": creds.get("username", "neo4j"),
+                    "NEO4J_USERNAME": creds.get("username", "neo4j"),
                     "NEO4J_PASSWORD": creds.get("password", "")
                 },
                 "tools": {
@@ -60,6 +60,16 @@ def _generate_mcp_json(creds):
         json.dump(mcp_config, f, indent=2)
     console.print(f"\n[cyan]For your convenience, the configuration has also been saved to: {mcp_file}[/cyan]")
 
+    # Also save to a .env file for convenience
+    env_file = Path.home() / ".codegraphcontext" / ".env"
+    env_file.parent.mkdir(parents=True, exist_ok=True)
+    with open(env_file, "w") as f:
+        f.write(f"NEO4J_URI={creds.get('uri', '')}\n")
+        f.write(f"NEO4J_USERNAME={creds.get('username', 'neo4j')}\n")
+        f.write(f"NEO4J_PASSWORD={creds.get('password', '')}\n")
+
+    console.print(f"[cyan]Neo4j credentials also saved to: {env_file}[/cyan]")
+
 
 def get_project_root() -> Path:
     """Always return the directory where the user runs `cgc` (CWD)."""
@@ -136,6 +146,7 @@ def find_latest_neo4j_creds_file():
 def setup_hosted_db():
     """Guides user to configure a remote Neo4j instance."""
     console.print("\nTo connect to a hosted Neo4j database, you'll need your connection credentials.")
+    console.print("[yellow]Warning: You are configuring to connect to a remote/hosted Neo4j database. Ensure your credentials are secure.[/yellow]")
     console.print("If you don't have a hosted database, you can create a free one at [bold blue]https://neo4j.com/product/auradb/[/bold blue] (click 'Start free').")
     
     questions = [
@@ -265,7 +276,7 @@ services:
       - "7474:7474"
       - "7687:7687"
     environment:
-      - NEO4J_AUTH=neo4j/12345678
+      - NEO4J_AUTH=neo4j/{password}
       - NEO4J_ACCEPT_LICENSE_AGREEMENT=yes
     volumes:
       - neo4j_data:/data

{codegraphcontext-0.1.7 → codegraphcontext-0.1.9}/src/codegraphcontext/server.py

@@ -103,11 +103,11 @@ class MCPServer:
             },
             "analyze_code_relationships": {
                 "name": "analyze_code_relationships",
-                "description": "Analyze code relationships like 'who calls this function' or 'class hierarchy'.",
+                "description": "Analyze code relationships like 'who calls this function' or 'class hierarchy'. Supported query types include: 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.",
                 "inputSchema": {
                     "type": "object",
                     "properties": {
-                        "query_type": {"type": "string", "description": "Type of relationship query to run."},
+                        "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."}
                     },
@@ -159,11 +159,12 @@ class MCPServer:
             },
             "find_dead_code": {
                 "name": "find_dead_code",
-                "description": "Find potentially unused functions (dead code) across the entire indexed codebase.",
+                "description": "Find potentially unused functions (dead code) across the entire indexed codebase, optionally excluding functions with specific decorators.",
                 "inputSchema": {
                     "type": "object",
-                    "properties": {},
-                    "additionalProperties": False
+                    "properties": {
+                        "exclude_decorated_with": {"type": "array", "items": {"type": "string"}, "description": "Optional: A list of decorator names (e.g., '@app.route') to exclude from dead code detection.", "default": []}
+                    }
                 }
             },
             "calculate_cyclomatic_complexity": {
@@ -207,7 +208,6 @@ class MCPServer:
                     "required": ["repo_path"]
                 }
             }
-            # Other tools like list_imports, add_package_to_graph can be added here following the same pattern
         }    
 
     def get_database_status(self) -> dict:
@@ -304,12 +304,12 @@ class MCPServer:
                 "details": str(e)
             }
     
-    def find_dead_code_tool(self) -> Dict[str, Any]:
+    def find_dead_code_tool(self, **args) -> Dict[str, Any]:
         """Tool to find potentially dead code across the entire project."""
+        exclude_decorated_with = args.get("exclude_decorated_with", [])
         try:
             debug_log("Finding dead code.")
-            # The target argument from the old tool is not needed.
-            results = self.code_finder.find_dead_code()
+            results = self.code_finder.find_dead_code(exclude_decorated_with=exclude_decorated_with)
             
             return {
                 "success": True,
@@ -446,17 +446,11 @@ class MCPServer:
             else:
                 return {"error": f"Path {path} does not exist"}
             
-            if language == 'python':
-                # Get the list of stdlib modules for the current Python version
-                stdlib_modules = set(stdlibs.module_names)
-                # stdlib_modules = {
-                #     'os', 'sys', 'json', 'time', 'datetime', 'math', 'random', 're', 'collections', 
-                #     'itertools', 'functools', 'operator', 'pathlib', 'urllib', 'http', 'logging', 
-                #     'threading', 'multiprocessing', 'asyncio', 'typing', 'dataclasses', 'enum', 
-                #     'abc', 'io', 'csv', 'sqlite3', 'pickle', 'base64', 'hashlib', 'hmac', 'secrets', 
-                #     'unittest', 'doctest', 'pdb', 'profile', 'cProfile', 'timeit'
-                # }
-                all_imports = all_imports - stdlib_modules
+            # Removed standard library filtering as per user request.
+            # if language == 'python':
+            #     # Get the list of stdlib modules for the current Python version
+            #     stdlib_modules = set(stdlibs.module_names)
+            #     all_imports = all_imports - stdlib_modules
             
             return {
                 "imports": sorted(list(all_imports)), "language": language,
@@ -642,8 +636,9 @@ class MCPServer:
             return {
                 "error": "Both 'query_type' and 'target' are required",
                 "supported_query_types": [
-                    "who_calls", "what_calls", "who_imports", "who_modifies",
-                    "class_hierarchy", "overrides", "dead_code"
+                    "find_callers", "find_callees", "find_importers", "who_modifies",
+                    "class_hierarchy", "overrides", "dead_code", "call_chain",
+                    "module_deps", "variable_scope", "find_complexity"
                 ]
             }