codegraphcontext 0.1.6__tar.gz → 0.1.7__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.6
+Version: 0.1.7
 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
@@ -47,9 +47,12 @@ Requires-Dist: python-dotenv>=1.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 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)
+
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 

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

@@ -1,4 +1,6 @@
 # CodeGraphContext
+[![Build Status](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml/badge.svg)](https://github.com/Shashankss1205/CodeGraphContext/actions/workflows/test.yml)
+
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 
@@ -96,4 +98,4 @@ Once the server is running, you can interact with it through your AI assistant u
     -   "Find all implementations of the `render` method."
 
 -   **Code Quality and Maintenance:**
-    -   "Is there any dead or unused code in this project?"
+    -   "Is there any dead or unused code in this project?"

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

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.6"
+version = "0.1.7"
 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"
@@ -36,6 +36,7 @@ cgc = "codegraphcontext.cli.main:app"
 dev = [
     "pytest>=7.4.0",
     "black>=23.11.0",
+    "pytest-asyncio>=0.21.0",
 ]
 
 [tool.setuptools]

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

@@ -229,20 +229,141 @@ def setup_local_db():
         setup_local_binary()
 
 def setup_docker():
-    """Creates Docker files and runs docker-compose."""
-    console.print("This will create a `docker-compose.yml` and `.env` file in your current directory.")
-    # Here you would write the file contents
-    console.print("[green]docker-compose.yml and .env created.[/green]")
-    console.print("Please set your NEO4J_PASSWORD in the .env file.")
-    confirm_q = [{"type": "confirm", "message": "Ready to launch Docker containers?", "name": "proceed"}]
-    if prompt(confirm_q).get("proceed"):
-        try:
-            # Using our run_command to handle the subprocess call
-            docker_process = run_command(["docker", "compose", "up", "-d"], console, check=True)
-            if docker_process:
-                console.print("[bold green]Docker containers started successfully![/bold green]")
-        except Exception as e:
-            console.print(f"[bold red]Failed to start Docker containers:[/bold red] {e}")
+    """Creates Docker files and runs docker-compose for Neo4j."""
+    console.print("\n[bold cyan]Setting up Neo4j with Docker...[/bold cyan]")
+
+    # Prompt for password first
+    console.print("Please set a secure password for your Neo4j database:")
+    password_questions = [
+        {"type": "password", "message": "Enter Neo4j password:", "name": "password"},
+        {"type": "password", "message": "Confirm password:", "name": "password_confirm"},
+    ]
+    
+    while True:
+        passwords = prompt(password_questions)
+        if not passwords:
+            return  # User cancelled
+        
+        password = passwords.get("password", "")
+        if password and password == passwords.get("password_confirm"):
+            break
+        console.print("[red]Passwords do not match or are empty. Please try again.[/red]")
+
+    # Create data directories
+    neo4j_dir = Path.cwd() / "neo4j_data"
+    for subdir in ["data", "logs", "conf", "plugins"]:
+        (neo4j_dir / subdir).mkdir(parents=True, exist_ok=True)
+
+    # Fixed docker-compose.yml content
+    docker_compose_content = f"""
+services:
+  neo4j:
+    image: neo4j:5.21
+    container_name: neo4j-cgc
+    restart: unless-stopped
+    ports:
+      - "7474:7474"
+      - "7687:7687"
+    environment:
+      - NEO4J_AUTH=neo4j/12345678
+      - NEO4J_ACCEPT_LICENSE_AGREEMENT=yes
+    volumes:
+      - neo4j_data:/data
+      - neo4j_logs:/logs
+
+volumes:
+  neo4j_data:
+  neo4j_logs:
+"""
+
+    # Write docker-compose.yml
+    compose_file = Path.cwd() / "docker-compose.yml"
+    with open(compose_file, "w") as f:
+        f.write(docker_compose_content)
+
+    console.print("[green]✅ docker-compose.yml created with secure password.[/green]")
+
+    # Check if Docker is running
+    docker_check = run_command(["docker", "--version"], console, check=False)
+    if not docker_check:
+        console.print("[red]❌ Docker is not installed or not running. Please install Docker first.[/red]")
+        return
+
+    # Check if docker-compose is available
+    compose_check = run_command(["docker", "compose", "version"], console, check=False)
+    if not compose_check:
+        console.print("[red]❌ Docker Compose is not available. Please install Docker Compose.[/red]")
+        return
+
+    confirm_q = [{"type": "confirm", "message": "Ready to launch Neo4j in Docker?", "name": "proceed", "default": True}]
+    if not prompt(confirm_q).get("proceed"):
+        return
+
+    try:
+        # Pull the image first
+        console.print("[cyan]Pulling Neo4j Docker image...[/cyan]")
+        pull_process = run_command(["docker", "pull", "neo4j:5.21"], console, check=True)
+        if not pull_process:
+            console.print("[yellow]⚠️ Could not pull image, but continuing anyway...[/yellow]")
+
+        # Start containers
+        console.print("[cyan]Starting Neo4j container...[/cyan]")
+        docker_process = run_command(["docker", "compose", "up", "-d"], console, check=True)
+        
+        if docker_process:
+            console.print("[bold green]🚀 Neo4j Docker container started successfully![/bold green]")
+            
+            # Wait for Neo4j to be ready
+            console.print("[cyan]Waiting for Neo4j to be ready (this may take 30-60 seconds)...[/cyan]")
+            
+            # Try to connect for up to 2 minutes
+            max_attempts = 24  # 24 * 5 seconds = 2 minutes
+            for attempt in range(max_attempts):
+                time.sleep(5)
+                
+                # Check if container is still running
+                status_check = run_command(["docker", "compose", "ps", "-q", "neo4j"], console, check=False)
+                if not status_check or not status_check.stdout.strip():
+                    console.print("[red]❌ Neo4j container stopped unexpectedly. Check logs with: docker compose logs neo4j[/red]")
+                    return
+                
+                # Try to connect
+                health_check = run_command([
+                    "docker", "exec", "neo4j-cgc", "cypher-shell", 
+                    "-u", "neo4j", "-p", password, 
+                    "RETURN 'Connection successful' as status"
+                ], console, check=False)
+                
+                if health_check and health_check.returncode == 0:
+                    console.print("[bold green]✅ Neo4j is ready and accepting connections![/bold green]")
+                    break
+                    
+                if attempt < max_attempts - 1:
+                    console.print(f"[yellow]Still waiting... (attempt {attempt + 1}/{max_attempts})[/yellow]")
+            else:
+                console.print("[red]❌ Neo4j did not become ready within 2 minutes. Check logs with: docker compose logs neo4j[/red]")
+                return
+
+            # Generate MCP configuration
+            creds = {
+                "uri": "neo4j://localhost:7687",  # Use neo4j:// protocol for Neo4j 5.x
+                "username": "neo4j",
+                "password": password
+            }
+            _generate_mcp_json(creds)
+            
+            console.print("\n[bold green]🎉 Setup complete![/bold green]")
+            console.print("Neo4j is running at:")
+            console.print("  • Web interface: http://localhost:7474")
+            console.print("  • Bolt connection: neo4j://localhost:7687")
+            console.print("\n[cyan]Useful commands:[/cyan]")
+            console.print("  • Stop: docker compose down")
+            console.print("  • Restart: docker compose restart")
+            console.print("  • View logs: docker compose logs neo4j")
+            
+    except Exception as e:
+        console.print(f"[bold red]❌ Failed to start Neo4j Docker container:[/bold red] {e}")
+        console.print("[cyan]Try checking the logs with: docker compose logs neo4j[/cyan]")
 
 def setup_local_binary():
     """Automates the installation and configuration of Neo4j on Ubuntu/Debian."""

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

@@ -125,7 +125,7 @@ class MCPServer:
             },
             "execute_cypher_query": {
                 "name": "execute_cypher_query",
-                "description": "Fallback tool to run a direct, read-only Cypher query against the code graph.",
+                "description": "Fallback tool to run a direct, read-only Cypher query against the code graph. Use this for complex questions not covered by other tools. The graph contains nodes representing code structures and relationships between them. **Schema Overview:**\n- **Nodes:** `Repository`, `File`, `Module`, `Class`, `Function`.\n- **Properties:** Nodes have properties like `name`, `path`, `cyclomatic_complexity` (on Function nodes), and `code`.\n- **Relationships:** `CONTAINS` (e.g., File-[:CONTAINS]->Function), `CALLS` (Function-[:CALLS]->Function or File-[:CALLS]->Function), `IMPORTS` (File-[:IMPORTS]->Module), `INHERITS` (Class-[:INHERITS]->Class).",
                 "inputSchema": {
                     "type": "object",
                     "properties": { "cypher_query": {"type": "string", "description": "The read-only Cypher query to execute."} },

{codegraphcontext-0.1.6 → codegraphcontext-0.1.7}/src/codegraphcontext/tools/import_extractor.py

@@ -1,4 +1,5 @@
 # src/codegraphcontext/tools/import_extractor.py
+import ast
 import logging
 import re
 from pathlib import Path
@@ -13,23 +14,24 @@ class ImportExtractor:
 
     @staticmethod
     def extract_python_imports(file_path: str) -> Set[str]:
-        """Extract imports from a Python file"""
+        """Extract imports from a Python file using AST parsing."""
         imports = set()
         try:
             with open(file_path, 'r', encoding='utf-8') as f:
-                content = f.read()
-            patterns = [
-                r'^\s*import\s+([a-zA-Z_][a-zA-Z0-9_]*(?:\.[a-zA-Z_][a-zA-Z0-9_]*)*)',
-                r'^\s*from\s+([a-zA-Z_][a-zA-Z0-9_]*(?:\.[a-zA-Z_][a-zA-Z0-9_]*)*)\s+import',
-            ]
-            for line in content.split('\n'):
-                for pattern in patterns:
-                    match = re.match(pattern, line.strip())
-                    if match:
-                        import_name = match.group(1).split('.')[0]
-                        imports.add(import_name)
+                tree = ast.parse(f.read(), filename=file_path)
+
+            for node in ast.walk(tree):
+                if isinstance(node, ast.Import):
+                    for alias in node.names:
+                        imports.add(alias.name.split('.')[0]) # Get top-level package
+                elif isinstance(node, ast.ImportFrom):
+                    if node.level > 0: # Relative import
+                        # As per the user's request, do not list relative imports
+                        pass
+                    elif node.module:
+                        imports.add(node.module.split('.')[0]) # Get top-level package
         except Exception as e:
-            logger.warning(f"Error reading {file_path}: {e}")
+            logger.warning(f"Error parsing or reading {file_path}: {e}")
         return imports
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.6
+Version: 0.1.7
 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
@@ -47,9 +47,12 @@ Requires-Dist: python-dotenv>=1.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 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)
+
 
 An MCP server that indexes local code into a graph database to provide context to AI assistants.
 

{codegraphcontext-0.1.6 → codegraphcontext-0.1.7}/src/codegraphcontext.egg-info/requires.txt

@@ -10,3 +10,4 @@ python-dotenv>=1.0.0
 [dev]
 pytest>=7.4.0
 black>=23.11.0
+pytest-asyncio>=0.21.0