codegraphcontext 0.3.3__tar.gz → 0.3.4__tar.gz

{codegraphcontext-0.3.3/src/codegraphcontext.egg-info → codegraphcontext-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.3.3
+Version: 0.3.4
 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
@@ -45,6 +45,7 @@ Requires-Dist: inquirerpy>=0.3.4
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: tree-sitter>=0.21.0
 Requires-Dist: tree-sitter-language-pack>=0.6.0
+Requires-Dist: tree-sitter-c-sharp>=0.21.0
 Requires-Dist: pyyaml
 Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
@@ -56,6 +57,7 @@ Requires-Dist: uvicorn>=0.22.0
 Provides-Extra: parsing
 Requires-Dist: tree-sitter>=0.21.0; extra == "parsing"
 Requires-Dist: tree-sitter-language-pack>=0.6.0; extra == "parsing"
+Requires-Dist: tree-sitter-c-sharp>=0.21.0; extra == "parsing"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
@@ -160,7 +162,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.3
+- **Version:** 0.3.4
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.3.3 → codegraphcontext-0.3.4}/README.md

@@ -96,7 +96,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.3
+- **Version:** 0.3.4
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.3.3 → codegraphcontext-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.3.3"
+version = "0.3.4"
 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"
@@ -24,6 +24,7 @@ dependencies = [
     "python-dotenv>=1.0.0",
     "tree-sitter>=0.21.0",
     "tree-sitter-language-pack>=0.6.0",
+    "tree-sitter-c-sharp>=0.21.0",
     "pyyaml",
     "nbformat",
     "nbconvert>=7.16.6",
@@ -38,6 +39,7 @@ dependencies = [
 parsing = [
     "tree-sitter>=0.21.0",
     "tree-sitter-language-pack>=0.6.0",
+    "tree-sitter-c-sharp>=0.21.0",
 ]
 dev = [
     "pytest>=7.4.0",

{codegraphcontext-0.3.3 → codegraphcontext-0.3.4}/src/codegraphcontext/cli/cli_helpers.py

@@ -359,16 +359,40 @@ def visualize_helper(repo_path: Optional[str] = None, port: int = 8000):
     
     # Determine the static directory (built React app)
     # This points to src/codegraphcontext/viz/dist where we build the website
-    static_dir = Path(__file__).parent.parent / "viz" / "dist"
+    # (relative to src/codegraphcontext/cli/cli_helpers.py)
+    # Using .resolve() is more robust for path comparison and existence checks
+    this_file = Path(__file__).resolve()
+    package_root = this_file.parent.parent
+    static_dir = package_root / "viz" / "dist"
     
     # Fallback for development if not yet built in viz/dist
     if not static_dir.exists():
-        dev_static_dir = Path.cwd() / "website" / "dist"
+        # Look for website/dist in the project root (3 levels up from cli/cli_helpers.py, 4 parents)
+        # 1: cli/, 2: codegraphcontext/, 3: src/, 4: project_root/
+        project_root = this_file.parent.parent.parent.parent
+        dev_static_dir = project_root / "website" / "dist"
+        
+        # Also try one level up from package_root just in case of different layouts
+        alt_dev_dir = package_root.parent.parent / "website" / "dist"
+        
         if dev_static_dir.exists():
             static_dir = dev_static_dir
+        elif alt_dev_dir.exists():
+            static_dir = alt_dev_dir
         else:
-            console.print("[yellow]Warning: Visualization assets not found. Please run 'cd website && npm run build' first.[/yellow]")
-            # We continue anyway to let the server start (helpful for dev)
+            # Last resort: try current working directory
+            cwd_static_dir = Path.cwd() / "website" / "dist"
+            if cwd_static_dir.exists():
+                static_dir = cwd_static_dir
+            else:
+                console.print(f"[yellow]Warning: Visualization assets not found.[/yellow]")
+                console.print(f"[dim]Checked paths:[/dim]")
+                console.print(f"  [dim]- {static_dir}[/dim]")
+                console.print(f"  [dim]- {dev_static_dir}[/dim]")
+                console.print(f"  [dim]- {alt_dev_dir}[/dim]")
+                console.print(f"  [dim]- {cwd_static_dir}[/dim]")
+                console.print("[dim]Please run 'cd website && npm run build' first.[/dim]")
+                # We continue anyway to let the server start (helpful for dev)
     
     # Construct the URL
     backend_url = f"http://localhost:{port}"

codegraphcontext-0.3.4/src/codegraphcontext/viz/server.py

@@ -0,0 +1,230 @@
+from fastapi import FastAPI, HTTPException, Query, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse, FileResponse
+from pathlib import Path
+import uvicorn
+import json
+import os
+import sys
+from typing import Optional, List, Dict, Any
+
+from ..core.database import DatabaseManager
+from ..utils.debug_log import debug_log
+
+app = FastAPI()
+
+# Enable CORS for development
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Global database manager (will be initialized when server starts)
+db_manager: Optional[DatabaseManager] = None
+# Path to static directory
+_static_dir: Optional[str] = None
+
+def set_db_manager(manager: DatabaseManager):
+    global db_manager
+    db_manager = manager
+
+@app.get("/api/graph")
+async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str] = None):
+    if not db_manager:
+        raise HTTPException(status_code=500, detail="Database not initialized")
+    
+    def get_eid(element):
+        if not element: return None
+        if isinstance(element, (int, str)):
+            return str(element)
+        # Try various ways to get ID (Neo4j, FalkorDB, etc.)
+        for attr in ['element_id', 'id', '_id']:
+            if hasattr(element, attr):
+                val = getattr(element, attr)
+                if val is not None: return str(val)
+        return str(id(element))
+
+    try:
+        nodes_dict = {}
+        edges = []
+
+        print(f"DEBUG: Starting get_graph with repo_path={repo_path}", flush=True)
+        nodes_dict = {}
+        edges = []
+
+        with db_manager.get_driver().session() as session:
+            if cypher_query:
+                print(f"DEBUG: Executing custom query: {cypher_query}", flush=True)
+                result = session.run(cypher_query)
+            elif repo_path:
+                repo_path = str(Path(repo_path).resolve())
+                print(f"DEBUG: Fetching subgraph for: {repo_path}", flush=True)
+                query = """
+                MATCH (r:Repository {path: $repo_path})
+                OPTIONAL MATCH (r)-[:CONTAINS*0..]->(n)
+                WITH DISTINCT n
+                WHERE n IS NOT NULL
+                OPTIONAL MATCH (n)-[rel]->(m)
+                RETURN n, rel, m
+                """
+                result = session.run(query, repo_path=repo_path)
+            else:
+                print("DEBUG: Fetching global graph", flush=True)
+                query = "MATCH (n) OPTIONAL MATCH (n)-[rel]->(m) RETURN n, rel, m LIMIT 5000"
+                result = session.run(query)
+
+            record_count = 0
+            for record in result:
+                record_count += 1
+                # Use .get() to avoid KeyError if the query doesn't return all fields (n, rel, m)
+                for key in ['n', 'm']:
+                    try:
+                        node = record.get(key)
+                        if node:
+                            eid = get_eid(node)
+                            if eid and eid not in nodes_dict:
+                                # Extract labels
+                                labels = []
+                                for label_attr in ['_labels', 'labels']:
+                                    if hasattr(node, label_attr):
+                                        attr_val = getattr(node, label_attr)
+                                        if attr_val:
+                                            labels = list(attr_val)
+                                            break
+                                
+                                # Extract properties
+                                props = {}
+                                for prop_attr in ['properties', '_properties']:
+                                    if hasattr(node, prop_attr):
+                                        attr_val = getattr(node, prop_attr)
+                                        if attr_val:
+                                            props = dict(attr_val)
+                                            break
+                                            
+                                # Fallback if props still empty but node acts like dict
+                                if not props and hasattr(node, 'items'):
+                                    try:
+                                        props = dict(node.items())
+                                    except: pass
+                                
+                                # Extract name/label for frontend
+                                # Prefer 'name' property, fallback to 'label', then 'path' or 'Unknown'
+                                display_name = str(props.get('name', props.get('label', props.get('path', 'Unknown'))))
+                                
+                                nodes_dict[eid] = {
+                                    "id": eid,
+                                    "name": display_name,
+                                    "label": display_name,
+                                    "type": str(labels[0]).capitalize() if labels else "Other",
+                                    "file": str(props.get('path', props.get('file', ''))),
+                                    "val": 4 if (labels and labels[0] in ['Repository', 'Class', 'Interface', 'Trait']) else 2,
+                                    "properties": props
+                                }
+                    except Exception as e:
+                        print(f"DEBUG: Error parsing node: {e}", file=sys.stderr, flush=True)
+                        continue
+                
+                try:
+                    rel = record.get('rel')
+                    if rel:
+                        rid = get_eid(rel)
+                        
+                        # Try various ways to get start/end nodes
+                        start_node = None
+                        end_node = None
+                        for src_attr in ['start_node', 'src_node', '_src_node']:
+                            if hasattr(rel, src_attr):
+                                start_node = getattr(rel, src_attr)
+                                break
+                        for dest_attr in ['end_node', 'dest_node', '_dest_node']:
+                            if hasattr(rel, dest_attr):
+                                end_node = getattr(rel, dest_attr)
+                                break
+                        
+                        source = get_eid(start_node) if start_node is not None else None
+                        target = get_eid(end_node) if end_node is not None else None
+                        
+                        if source and target:
+                            # Extract relationship type
+                            rel_type = "RELATED"
+                            for rel_attr in ['type', 'relation', '_relation']:
+                                if hasattr(rel, rel_attr):
+                                    rel_type = getattr(rel, rel_attr)
+                                    break
+                                    
+                            edges.append({
+                                "id": rid,
+                                "source": source,
+                                "target": target,
+                                "type": str(rel_type).upper()
+                            })
+                except Exception as e:
+                    print(f"DEBUG: Error parsing relationship: {e}", file=sys.stderr, flush=True)
+                    pass
+
+        print(f"DEBUG: Processed {record_count} records. extracted {len(nodes_dict)} nodes and {len(edges)} edges.", file=sys.stderr, flush=True)
+
+        # Build a list of unique file paths from File-type nodes for the tree
+        file_paths = []
+        for n in nodes_dict.values():
+            if n.get("file") and str(n.get("type", "")).lower() == "file":
+                file_paths.append(str(n["file"]))
+        file_paths = sorted(list(set(file_paths)))
+
+        response_data = {
+            "nodes": list(nodes_dict.values()), 
+            "links": edges,
+            "files": file_paths,
+        }
+        
+        print(f"API SUCCESS: Returning graph with {len(response_data['nodes'])} nodes and {len(response_data['links'])} links.", file=sys.stderr, flush=True)
+        return response_data
+
+    except Exception as e:
+        debug_log(f"Error fetching graph: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        # Still return a valid structure so the frontend doesn't crash, but with 500 status if raised
+        # Actually, let's just return a 500 error but with JSON body if possible
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/api/file")
+async def get_file(path: str):
+    file_path = Path(path)
+    if not file_path.exists():
+        raise HTTPException(status_code=404, detail="File not found")
+    
+    try:
+        with open(file_path, "r", encoding="utf-8") as f:
+            return {"content": f.read()}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# SPA fallback handler
+@app.get("/{full_path:path}")
+async def spa_fallback(request: Request, full_path: str):
+    global _static_dir
+    if not _static_dir:
+        return HTMLResponse("Static directory not configured", status_code=500)
+    
+    # Filesystem path
+    file_path = Path(_static_dir) / full_path
+    
+    # If the file exists and is a file, serve it normally
+    if file_path.exists() and file_path.is_file():
+        return FileResponse(file_path)
+    
+    # Otherwise serve index.html (Standard SPA routing)
+    index_path = Path(_static_dir) / "index.html"
+    if index_path.exists():
+        return FileResponse(index_path)
+    
+    return HTMLResponse("Not Found (Built UI not found in viz/dist)", status_code=404)
+
+def run_server(host: str = "127.0.0.1", port: int = 8000, static_dir: Optional[str] = None):
+    global _static_dir
+    _static_dir = static_dir
+    uvicorn.run(app, host=host, port=port)

{codegraphcontext-0.3.3 → codegraphcontext-0.3.4/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.3.3
+Version: 0.3.4
 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
@@ -45,6 +45,7 @@ Requires-Dist: inquirerpy>=0.3.4
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: tree-sitter>=0.21.0
 Requires-Dist: tree-sitter-language-pack>=0.6.0
+Requires-Dist: tree-sitter-c-sharp>=0.21.0
 Requires-Dist: pyyaml
 Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
@@ -56,6 +57,7 @@ Requires-Dist: uvicorn>=0.22.0
 Provides-Extra: parsing
 Requires-Dist: tree-sitter>=0.21.0; extra == "parsing"
 Requires-Dist: tree-sitter-language-pack>=0.6.0; extra == "parsing"
+Requires-Dist: tree-sitter-c-sharp>=0.21.0; extra == "parsing"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
@@ -160,7 +162,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.3
+- **Version:** 0.3.4
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.3.3 → codegraphcontext-0.3.4}/src/codegraphcontext.egg-info/requires.txt

@@ -7,6 +7,7 @@ inquirerpy>=0.3.4
 python-dotenv>=1.0.0
 tree-sitter>=0.21.0
 tree-sitter-language-pack>=0.6.0
+tree-sitter-c-sharp>=0.21.0
 pyyaml
 nbformat
 nbconvert>=7.16.6
@@ -26,3 +27,4 @@ pytest-asyncio>=0.21.0
 [parsing]
 tree-sitter>=0.21.0
 tree-sitter-language-pack>=0.6.0
+tree-sitter-c-sharp>=0.21.0

codegraphcontext-0.3.3/src/codegraphcontext/viz/server.py

@@ -1,179 +0,0 @@
-from fastapi import FastAPI, HTTPException, Query, Request
-from fastapi.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse, FileResponse
-from pathlib import Path
-import uvicorn
-import json
-import os
-from typing import Optional, List, Dict, Any
-
-from ..core.database import DatabaseManager
-from ..utils.debug_log import debug_log
-
-app = FastAPI()
-
-# Enable CORS for development
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Global database manager (will be initialized when server starts)
-db_manager: Optional[DatabaseManager] = None
-# Path to static directory
-_static_dir: Optional[str] = None
-
-def set_db_manager(manager: DatabaseManager):
-    global db_manager
-    db_manager = manager
-
-@app.get("/api/graph")
-async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str] = None):
-    if not db_manager:
-        raise HTTPException(status_code=500, detail="Database not initialized")
-    
-    def get_eid(element):
-        if not element: return None
-        if isinstance(element, (int, str)):
-            return str(element)
-        # Try various ways to get ID (Neo4j, FalkorDB, etc.)
-        for attr in ['element_id', 'id', '_id']:
-            if hasattr(element, attr):
-                val = getattr(element, attr)
-                if val is not None: return str(val)
-        return str(id(element))
-
-    try:
-        nodes_dict = {}
-        edges = []
-
-        with db_manager.get_driver().session() as session:
-            if cypher_query:
-                # Direct user query (filtered view)
-                result = session.run(cypher_query)
-            elif repo_path:
-                repo_path = str(Path(repo_path).resolve())
-                # Optimized subgraph query
-                query = """
-                MATCH (r:Repository {path: $repo_path})
-                OPTIONAL MATCH (r)-[:CONTAINS*0..]->(n)
-                WITH DISTINCT n
-                WHERE n IS NOT NULL
-                OPTIONAL MATCH (n)-[rel]->(m)
-                RETURN n, rel, m
-                """
-                result = session.run(query, repo_path=repo_path)
-            else:
-                query = "MATCH (n) OPTIONAL MATCH (n)-[rel]->(m) RETURN n, rel, m LIMIT 5000"
-                result = session.run(query)
-
-            for record in result:
-                for key in ['n', 'm']:
-                    node = record[key]
-                    if node:
-                        eid = get_eid(node)
-                        if eid not in nodes_dict:
-                            # FalkorDB / Neo4j labels compatibility
-                            labels = []
-                            if hasattr(node, 'labels'):
-                                labels = list(node.labels)
-                            
-                            # FalkorDB / Neo4j properties compatibility
-                            props = {}
-                            if hasattr(node, 'properties'):
-                                props = node.properties
-                            elif hasattr(node, 'items'):
-                                props = dict(node.items())
-                                
-                            nodes_dict[eid] = {
-                                "id": eid,
-                                "label": props.get('name', props.get('label', 'Unknown')),
-                                "type": labels[0].capitalize() if labels else "Other",
-                                "file": props.get('path', props.get('file', '')),
-                                "properties": props
-                            }
-                
-                rel = record['rel']
-                if rel:
-                    rid = get_eid(rel)
-                    
-                    # FalkorDB / Neo4j compatibility for source/target nodes
-                    start_node = getattr(rel, 'start_node', getattr(rel, 'src_node', None))
-                    end_node = getattr(rel, 'end_node', getattr(rel, 'dest_node', None))
-                    
-                    source = get_eid(start_node)
-                    target = get_eid(end_node)
-                    
-                    if source and target:
-                        # relationship type/relation
-                        rel_type = "related"
-                        if hasattr(rel, 'type'):
-                            rel_type = rel.type
-                        elif hasattr(rel, 'relation'):
-                            rel_type = rel.relation
-                            
-                        edges.append({
-                            "id": rid,
-                            "source": source,
-                            "target": target,
-                            "type": str(rel_type).upper()
-                        })
-
-        # Build a list of unique file paths from File-type nodes for the tree
-        file_paths = sorted(set(
-            n["file"] for n in nodes_dict.values()
-            if n.get("file") and n.get("type", "").lower() == "file"
-        ))
-
-        return {
-            "nodes": list(nodes_dict.values()), 
-            "edges": edges,
-            "files": file_paths,
-        }
-
-    except Exception as e:
-        debug_log(f"Error fetching graph: {str(e)}")
-        import traceback
-        traceback.print_exc()
-        raise HTTPException(status_code=500, detail=str(e))
-
-@app.get("/api/file")
-async def get_file(path: str):
-    file_path = Path(path)
-    if not file_path.exists():
-        raise HTTPException(status_code=404, detail="File not found")
-    
-    try:
-        with open(file_path, "r", encoding="utf-8") as f:
-            return {"content": f.read()}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-# SPA fallback handler
-@app.get("/{full_path:path}")
-async def spa_fallback(request: Request, full_path: str):
-    global _static_dir
-    if not _static_dir:
-        return HTMLResponse("Static directory not configured", status_code=500)
-    
-    # Filesystem path
-    file_path = Path(_static_dir) / full_path
-    
-    # If the file exists and is a file, serve it normally
-    if file_path.exists() and file_path.is_file():
-        return FileResponse(file_path)
-    
-    # Otherwise serve index.html (Standard SPA routing)
-    index_path = Path(_static_dir) / "index.html"
-    if index_path.exists():
-        return FileResponse(index_path)
-    
-    return HTMLResponse("Not Found (Built UI not found in viz/dist)", status_code=404)
-
-def run_server(host: str = "127.0.0.1", port: int = 8000, static_dir: Optional[str] = None):
-    global _static_dir
-    _static_dir = static_dir
-    uvicorn.run(app, host=host, port=port)