codegraphcontext 0.3.1__tar.gz → 0.3.3__tar.gz

codegraphcontext-0.3.3/MANIFEST.in

@@ -0,0 +1,4 @@
+prune website
+prune images
+recursive-include src/codegraphcontext/viz/dist *
+recursive-include vscode-extension *

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.3.1
+Version: 0.3.3
 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
@@ -46,7 +46,6 @@ 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: pyyaml
-Requires-Dist: pytest
 Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
 Requires-Dist: pathspec>=0.12.1
@@ -70,6 +69,7 @@ Dynamic: license-file
 🌐 **Languages:**
 - 🇬🇧 [English](README.md)
 - 🇨🇳 [中文](README.zh-CN.md)
+- 🇰🇷 [한국어](README.kor.md)
 - 🇯🇵 日本語 (Soon)
 - 🇷🇺 Русский (Soon)
 - 🇪🇸 Español (Soon)
@@ -160,7 +160,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.1
+- **Version:** 0.3.3
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -365,7 +365,7 @@ cgc watch .
 cgc help
 ```
 
-**See the full [CLI Commands Guide](CLI_Commands.md) for all available commands and usage scenarios.**
+  **See the full [CLI Commands Guide](docs/CLI_COMPLETE_REFERENCE.md) for all available commands and usage scenarios.**
 
 ### 🎨 Premium Interactive Visualization
 CodeGraphContext can generate stunning, interactive knowledge graphs of your code. Unlike static diagrams, these are premium web-based explorers:

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

@@ -5,6 +5,7 @@
 🌐 **Languages:**
 - 🇬🇧 [English](README.md)
 - 🇨🇳 [中文](README.zh-CN.md)
+- 🇰🇷 [한국어](README.kor.md)
 - 🇯🇵 日本語 (Soon)
 - 🇷🇺 Русский (Soon)
 - 🇪🇸 Español (Soon)
@@ -95,7 +96,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.1
+- **Version:** 0.3.3
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -300,7 +301,7 @@ cgc watch .
 cgc help
 ```
 
-**See the full [CLI Commands Guide](CLI_Commands.md) for all available commands and usage scenarios.**
+  **See the full [CLI Commands Guide](docs/CLI_COMPLETE_REFERENCE.md) for all available commands and usage scenarios.**
 
 ### 🎨 Premium Interactive Visualization
 CodeGraphContext can generate stunning, interactive knowledge graphs of your code. Unlike static diagrams, these are premium web-based explorers:

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

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.3.1"
+version = "0.3.3"
 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"
@@ -25,7 +25,6 @@ dependencies = [
     "tree-sitter>=0.21.0",
     "tree-sitter-language-pack>=0.6.0",
     "pyyaml",
-    "pytest",
     "nbformat",
     "nbconvert>=7.16.6",
     "pathspec>=0.12.1",

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

@@ -340,7 +340,9 @@ def cypher_helper_visual(query: str):
         console.print(f"[bold red]An error occurred while executing query:[/bold red] {e}")
     finally:
         db_manager.close_driver()
-import webbrowser
+
+
+import uvicorn
 import urllib.parse
 from ..viz.server import run_server, set_db_manager
 
@@ -356,12 +358,18 @@ def visualize_helper(repo_path: Optional[str] = None, port: int = 8000):
     set_db_manager(db_manager)
     
     # 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"
+    
+    # Fallback for development if not yet built in viz/dist
     if not static_dir.exists():
-        console.print("[yellow]Warning: Visualizer UI assets not found in package. Using fallback static dir.[/yellow]")
-        # Fallback for development
-        static_dir = Path.cwd() / "website" / "dist"
-
+        dev_static_dir = Path.cwd() / "website" / "dist"
+        if dev_static_dir.exists():
+            static_dir = dev_static_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)
+    
     # Construct the URL
     backend_url = f"http://localhost:{port}"
     params = {"backend": backend_url}
@@ -369,7 +377,7 @@ def visualize_helper(repo_path: Optional[str] = None, port: int = 8000):
         params["repo_path"] = str(Path(repo_path).resolve())
     
     query_string = urllib.parse.urlencode(params)
-    visualization_url = f"{backend_url}/playground?{query_string}"
+    visualization_url = f"{backend_url}/explore?{query_string}"
     
     console.print(f"[green]Starting visualizer server on {backend_url}...[/green]")
     console.print(f"[cyan]Opening Playground UI:[/cyan] {visualization_url}")
@@ -377,6 +385,7 @@ def visualize_helper(repo_path: Optional[str] = None, port: int = 8000):
     # Open browser in a separate thread/process if possible, or just before starting server
     def open_browser():
         import time
+        import webbrowser
         time.sleep(1.5) # Give the server a moment to start
         webbrowser.open(visualization_url)
     
@@ -390,243 +399,6 @@ def visualize_helper(repo_path: Optional[str] = None, port: int = 8000):
     finally:
         db_manager.close_driver()
 
-def _visualize_falkordb(db_manager):
-    console.print("[dim]Generating FalkorDB visualization (showing up to 500 relationships)...[/dim]")
-    try:
-        data_nodes = []
-        data_edges = []
-        
-        with db_manager.get_driver().session() as session:
-            # Fetch nodes and edges
-            q = "MATCH (n)-[r]->(m) RETURN n, r, m LIMIT 500"
-            result = session.run(q)
-            
-            seen_nodes = set()
-            
-            for record in result:
-                # record values are Node/Relationship objects from falkordb client
-                n = record['n']
-                r = record['r']
-                m = record['m']
-                
-                # Process Node helper
-                def process_node(node):
-                    nid = getattr(node, 'id', -1)
-                    labels = getattr(node, 'labels', [])
-                    lbl = list(labels)[0] if labels else "Node"
-                    props = getattr(node, 'properties', {})
-                    name = props.get('name', str(nid))
-                    
-                    if nid not in seen_nodes:
-                        seen_nodes.add(nid)
-                        color = "#97c2fc" # Default blue
-                        if "Repository" in labels: color = "#ffb3ba" # Red
-                        elif "File" in labels: color = "#baffc9" # Green
-                        elif "Class" in labels: color = "#bae1ff" # Light Blue
-                        elif "Function" in labels: color = "#ffffba" # Yellow
-                        elif "Package" in labels: color = "#ffdfba" # Orange
-                        
-                        data_nodes.append({
-                            "id": nid, 
-                            "label": name, 
-                            "group": lbl, 
-                            "title": str(props),
-                            "color": color
-                        })
-                    return nid
-
-                nid = process_node(n)
-                mid = process_node(m)
-                
-                # Check Edge
-                e_type = getattr(r, 'relation', '') or getattr(r, 'type', 'REL')
-                data_edges.append({
-                    "from": nid,
-                    "to": mid,
-                    "label": e_type,
-                    "arrows": "to"
-                })
-        
-        filename = "codegraph_viz.html"
-        html_content = f"""
-<!DOCTYPE html>
-<html>
-<head>
-  <title>CodeGraphContext Visualization</title>
-  <script type="text/javascript" src="https://unpkg.com/vis-network/standalone/umd/vis-network.min.js"></script>
-  <style type="text/css">
-    #mynetwork {{
-      width: 100%;
-      height: 100vh;
-      border: 1px solid lightgray;
-    }}
-  </style>
-</head>
-<body>
-  <div id="mynetwork"></div>
-  <script type="text/javascript">
-    var nodes = new vis.DataSet({json.dumps(data_nodes)});
-    var edges = new vis.DataSet({json.dumps(data_edges)});
-    var container = document.getElementById('mynetwork');
-    var data = {{ nodes: nodes, edges: edges }};
-    var options = {{
-        nodes: {{ shape: 'dot', size: 16 }},
-        physics: {{ stabilization: false }},
-        layout: {{ improvedLayout: false }}
-    }};
-    var network = new vis.Network(container, data, options);
-  </script>
-</body>
-</html>
-"""
-        
-        out_path = Path(filename).resolve()
-        with open(out_path, "w") as f:
-            f.write(html_content)
-            
-        console.print(f"[green]Visualization generated at:[/green] {out_path}")
-        console.print("Opening in default browser...")
-        webbrowser.open(f"file://{out_path}")
-
-    except Exception as e:
-        console.print(f"[bold red]Visualization failed:[/bold red] {e}")
-        import traceback
-        traceback.print_exc()
-    finally:
-        db_manager.close_driver()
-
-
-def _visualize_kuzudb(db_manager):
-    console.print("[dim]Generating KùzuDB visualization (showing up to 500 relationships)...[/dim]")
-    try:
-        data_nodes = []
-        data_edges = []
-        
-        with db_manager.get_driver().session() as session:
-            # Fetch nodes and edges
-            # KùzuDB returns dicts for n, r, m in the result
-            q = "MATCH (n)-[r]->(m) RETURN n, r, m LIMIT 500"
-            result = session.run(q)
-            
-            seen_nodes = set()
-            
-            # Helper to extract Node ID and props
-            def process_node(node):
-                uid = None
-                lbl = 'Node'
-                props = {}
-                
-                # Handle Kuzu Node Object (processed by wrapper)
-                if hasattr(node, 'properties'):
-                    props = node.properties or {}
-                    if hasattr(node, 'labels') and node.labels:
-                        lbl = node.labels[0]
-                    if hasattr(node, 'id'):
-                        uid = str(node.id)
-                # Handle Dictionary (raw Kuzu result)
-                elif isinstance(node, dict):
-                    if '_id' in node:
-                        uid = f"{node['_id']['table']}_{node['_id']['offset']}"
-                    lbl = node.get('_label', 'Node')
-                    props = {k: v for k, v in node.items() if not k.startswith('_')}
-                
-                if not uid:
-                    uid = str(uuid.uuid4())
-                    
-                name = props.get('name', str(uid))
-                
-                if uid not in seen_nodes:
-                    seen_nodes.add(uid)
-                    color = "#97c2fc" # Default blue
-                    if "Repository" == lbl: color = "#ffb3ba"
-                    elif "File" == lbl: color = "#baffc9"
-                    elif "Class" == lbl: color = "#bae1ff"
-                    elif "Function" == lbl: color = "#ffffba"
-                    elif "Module" == lbl: color = "#ffdfba"
-                    
-                    data_nodes.append({
-                        "id": uid, 
-                        "label": name, 
-                        "group": lbl, 
-                        "title": str(props),
-                        "color": color
-                    })
-                return uid
-            
-            # Iterate results
-            for record in result:
-                # record is dict-like access to row items
-                n = record['n']
-                r = record['r']
-                m = record['m']
-                
-                nid = process_node(n)
-                mid = process_node(m)
-                
-                # Process Edge
-                e_type = 'REL'
-                if hasattr(r, 'type'):
-                    e_type = r.type
-                elif isinstance(r, dict):
-                    e_type = r.get('_label', 'REL')
-                elif hasattr(r, 'label'): # Some versions
-                     e_type = r.label
-                
-                data_edges.append({
-                    "from": nid,
-                    "to": mid,
-                    "label": e_type,
-                    "arrows": "to"
-                })
-        
-        filename = "codegraph_viz.html"
-        html_content = f"""
-<!DOCTYPE html>
-<html>
-<head>
-  <title>CodeGraphContext KùzuDB Visualization</title>
-  <script type="text/javascript" src="https://unpkg.com/vis-network/standalone/umd/vis-network.min.js"></script>
-  <style type="text/css">
-    #mynetwork {{
-      width: 100%;
-      height: 100vh;
-      border: 1px solid lightgray;
-    }}
-  </style>
-</head>
-<body>
-  <div id="mynetwork"></div>
-  <script type="text/javascript">
-    var nodes = new vis.DataSet({json.dumps(data_nodes)});
-    var edges = new vis.DataSet({json.dumps(data_edges)});
-    var container = document.getElementById('mynetwork');
-    var data = {{ nodes: nodes, edges: edges }};
-    var options = {{
-        nodes: {{ shape: 'dot', size: 16 }},
-        physics: {{ stabilization: false }},
-        layout: {{ improvedLayout: false }}
-    }};
-    var network = new vis.Network(container, data, options);
-  </script>
-</body>
-</html>
-"""
-        
-        out_path = Path(filename).resolve()
-        with open(out_path, "w") as f:
-            f.write(html_content)
-            
-        console.print(f"[green]Visualization generated at:[/green] {out_path}")
-        console.print("Opening in default browser...")
-        webbrowser.open(f"file://{out_path}")
-
-    except Exception as e:
-        console.print(f"[bold red]Visualization failed:[/bold red] {e}")
-        import traceback
-        traceback.print_exc()
-    finally:
-        db_manager.close_driver()
-
 
 def reindex_helper(path: str):
     """Force re-index by deleting and rebuilding the repository."""

{codegraphcontext-0.3.1 → codegraphcontext-0.3.3}/src/codegraphcontext/cli/main.py

@@ -999,7 +999,7 @@ def visualize(
     port: int = typer.Option(8000, "--port", "-p", help="Port to run the visualizer server on.")
 ):
     """
-    Launches the interactive Playground UI to visualize the code graph.
+    Launches the interactive UI to visualize the code graph.
     """
     _load_credentials()
     visualize_helper(repo, port)

codegraphcontext-0.3.3/src/codegraphcontext/cli/visualizer.py

@@ -0,0 +1,50 @@
+import urllib.parse
+from typing import Optional, List, Dict, Any, Set
+from pathlib import Path
+from .cli_helpers import visualize_helper
+
+def check_visual_flag(ctx, visual: bool, cypher_query: str = None):
+    """
+    Helper to check the --visual flag and launch the visualizer.
+    This is called from within analyze/find commands.
+    """
+    if visual and cypher_query:
+        # We start the visualizer on port 8000
+        # Passing empty repo handles showing just the query results
+        port = 8000
+        encoded_query = urllib.parse.quote(cypher_query)
+        visualization_url = f"http://localhost:{port}/explore?cypher_query={encoded_query}"
+        
+        from rich.console import Console
+        console = Console(stderr=True)
+        console.print(f"[green]Starting visualizer...[/green]")
+        console.print(f"[cyan]Visualizing results at:[/cyan] {visualization_url}")
+        
+        # Start the backend server and open the browser
+        visualize_helper(repo_path=None, port=port)
+        return True
+    return False
+
+def visualize_call_graph(cypher_query: str):
+    """Visualize a call graph result."""
+    visualize_helper(repo_path=None, port=8000)
+
+def visualize_call_chain(cypher_query: str):
+    """Visualize a call chain result."""
+    visualize_helper(repo_path=None, port=8000)
+
+def visualize_dependencies(cypher_query: str):
+    """Visualize code dependencies."""
+    visualize_helper(repo_path=None, port=8000)
+
+def visualize_inheritance_tree(cypher_query: str):
+    """Visualize class inheritance tree."""
+    visualize_helper(repo_path=None, port=8000)
+
+def visualize_overrides(cypher_query: str):
+    """Visualize method overrides."""
+    visualize_helper(repo_path=None, port=8000)
+
+def visualize_search_results(cypher_query: str):
+    """Visualize search results."""
+    visualize_helper(repo_path=None, port=8000)

{codegraphcontext-0.3.1 → codegraphcontext-0.3.3}/src/codegraphcontext/core/cgc_bundle.py

@@ -161,9 +161,13 @@ class CGCBundle:
             with tempfile.TemporaryDirectory() as temp_dir:
                 temp_path = Path(temp_dir)
                 
-                # Step 1: Extract ZIP
+                # Step 1: Extract ZIP (with Zip Slip protection)
                 info_logger("Extracting bundle...")
                 with zipfile.ZipFile(bundle_path, 'r') as zip_ref:
+                    for entry in zip_ref.namelist():
+                        resolved = (temp_path / entry).resolve()
+                        if not str(resolved).startswith(str(temp_path.resolve())):
+                            return False, f"Zip Slip detected: entry '{entry}' escapes target directory"
                     zip_ref.extractall(temp_path)
                 
                 # Step 2: Validate bundle