codegraphcontext 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

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/viz/server.py

@@ -6,6 +6,7 @@ from pathlib import Path
 import uvicorn
 import json
 import os
+import sys
 from typing import Optional, List, Dict, Any
 
 from ..core.database import DatabaseManager
@@ -50,13 +51,17 @@ async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str]
         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:
-                # Direct user query (filtered view)
+                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())
-                # Optimized subgraph query
+                print(f"DEBUG: Fetching subgraph for: {repo_path}", flush=True)
                 query = """
                 MATCH (r:Repository {path: $repo_path})
                 OPTIONAL MATCH (r)-[:CONTAINS*0..]->(n)
@@ -67,77 +72,123 @@ async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str]
                 """
                 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']:
-                    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())
+                    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
                                 
-                            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
-                            }
+                                # 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
                 
-                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()
-                        })
+                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 = sorted(set(
-            n["file"] for n in nodes_dict.values()
-            if n.get("file") and n.get("type", "").lower() == "file"
-        ))
+        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)))
 
-        return {
+        response_data = {
             "nodes": list(nodes_dict.values()), 
-            "edges": edges,
+            "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")

{codegraphcontext-0.3.3.dist-info → codegraphcontext-0.3.4.dist-info}/METADATA

@@ -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.dist-info → codegraphcontext-0.3.4.dist-info}/RECORD

@@ -4,7 +4,7 @@ codegraphcontext/prompts.py,sha256=E5P55paM0oHfBcNVfxkxpXRGZnya1kr_mKDg9i49FwM,6
 codegraphcontext/server.py,sha256=gcb6V4x0Oh8haBOCm2WBjTCO9FDukrSalAMMApL-oc8,12806
 codegraphcontext/tool_definitions.py,sha256=_0ahezQSURX_ewWydT2sFcvC6OFGdMoPA7KwgWNVcks,11482
 codegraphcontext/cli/__init__.py,sha256=v6CMDVKM5d_sXn3S5nZNf0phXn0IdrnhLazUoen9k9w,38
-codegraphcontext/cli/cli_helpers.py,sha256=8T7oMy6bBvQtfDjhQik5kKKkkyqWxeG4sXzeM8n-uxI,28374
+codegraphcontext/cli/cli_helpers.py,sha256=QcpEV9VRFuPiiIEUwiXCr_nm7Z7amzphXv9zQY0aWIU,29646
 codegraphcontext/cli/config_manager.py,sha256=MK7GMGZ4hd8-ZOShXBd_KDtNqHQ3ngdbef5KM_naPrQ,15899
 codegraphcontext/cli/main.py,sha256=89jWaO8Gr3nRdLQtfwF434wpU77d5bJwt016f_lLnGg,86486
 codegraphcontext/cli/registry_commands.py,sha256=30rJm4SeS0n1jax4JVuhuv4zYLzyMmlHcCSnsDDhgc8,19964
@@ -71,10 +71,10 @@ codegraphcontext/tools/query_tool_languages/typescript_toolkit.py,sha256=3S4hpmO
 codegraphcontext/utils/debug_log.py,sha256=Qg7jwyeg7x2h3Ur_2S34bdMCkHdlk_ngHfPwa97A9vE,2836
 codegraphcontext/utils/tree_sitter_manager.py,sha256=bIuKYN1aj1Zi6BnksGjZSLZzgBwuFRselZzyUpbToAU,9180
 codegraphcontext/utils/visualize_graph.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codegraphcontext/viz/server.py,sha256=f2xjgkjMOBgrZd5fZ6kCEO-lzH9hNfPPCcSwfitH4io,6719
-codegraphcontext-0.3.3.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-codegraphcontext-0.3.3.dist-info/METADATA,sha256=pw9ykqFiZCaX1PioazQoU9S_Kmb7BRRAuPjveWyykk4,21579
-codegraphcontext-0.3.3.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codegraphcontext-0.3.3.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.3.3.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.3.3.dist-info/RECORD,,
+codegraphcontext/viz/server.py,sha256=gBl6yfTDa4F7FW4JxHV9QtG1kVSt8iSKvhCXSKB0lIw,10017
+codegraphcontext-0.3.4.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+codegraphcontext-0.3.4.dist-info/METADATA,sha256=PHNdbGKtvsw3S0hhTHgCoanKx_T8DMPPdLZ3iGEWe4o,21685
+codegraphcontext-0.3.4.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+codegraphcontext-0.3.4.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
+codegraphcontext-0.3.4.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
+codegraphcontext-0.3.4.dist-info/RECORD,,