codegraphcontext 0.3.2__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.2/src/codegraphcontext.egg-info → codegraphcontext-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.3.2
+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
@@ -160,7 +160,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.3.2
+- **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/)

{codegraphcontext-0.3.2 → codegraphcontext-0.3.3}/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.2
+- **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/)

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

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.3.2"
+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"

{codegraphcontext-0.3.2 → 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.2 → 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.2 → codegraphcontext-0.3.3}/src/codegraphcontext/tools/graph_builder.py

@@ -115,39 +115,54 @@ class GraphBuilder:
         self.loop = loop
         self.driver = self.db_manager.get_driver()
         self.parsers = {
-            '.py': TreeSitterParser('python'),
-            '.ipynb': TreeSitterParser('python'),
-            '.js': TreeSitterParser('javascript'),
-            '.jsx': TreeSitterParser('javascript'),
-            '.mjs': TreeSitterParser('javascript'),
-            '.cjs': TreeSitterParser('javascript'),
-            '.go': TreeSitterParser('go'),
-            '.ts': TreeSitterParser('typescript'),
-            '.tsx': TreeSitterParser('typescript'),
-            '.cpp': TreeSitterParser('cpp'),
-            '.h': TreeSitterParser('cpp'),
-            '.hpp': TreeSitterParser('cpp'),
-            '.hh': TreeSitterParser('cpp'),
-            '.rs': TreeSitterParser('rust'),
-            '.c': TreeSitterParser('c'),
-            # '.h': TreeSitterParser('c'), # Need to write an algo for distinguishing C vs C++ headers
-            '.java': TreeSitterParser('java'),
-            '.rb': TreeSitterParser('ruby'),
-            '.cs': TreeSitterParser('c_sharp'),
-            '.php': TreeSitterParser('php'),
-            '.kt': TreeSitterParser('kotlin'),
-            '.scala': TreeSitterParser('scala'),
-            '.sc': TreeSitterParser('scala'),
-            '.swift': TreeSitterParser('swift'),
-            '.hs': TreeSitterParser('haskell'),
-            '.dart': TreeSitterParser('dart'),
-            '.pl': TreeSitterParser('perl'),
-            '.pm': TreeSitterParser('perl'),
-            '.ex': TreeSitterParser('elixir'),
-            '.exs': TreeSitterParser('elixir'),
+            '.py': 'python',
+            '.ipynb': 'python',
+            '.js': 'javascript',
+            '.jsx': 'javascript',
+            '.mjs': 'javascript',
+            '.cjs': 'javascript',
+            '.go': 'go',
+            '.ts': 'typescript',
+            '.tsx': 'typescript',
+            '.cpp': 'cpp',
+            '.h': 'cpp',
+            '.hpp': 'cpp',
+            '.hh': 'cpp',
+            '.rs': 'rust',
+            '.c': 'c',
+            # '.h': 'c', # Need to write an algo for distinguishing C vs C++ headers
+            '.java': 'java',
+            '.rb': 'ruby',
+            '.cs': 'c_sharp',
+            '.php': 'php',
+            '.kt': 'kotlin',
+            '.scala': 'scala',
+            '.sc': 'scala',
+            '.swift': 'swift',
+            '.hs': 'haskell',
+            '.dart': 'dart',
+            '.pl': 'perl',
+            '.pm': 'perl',
+            '.ex': 'elixir',
+            '.exs': 'elixir',
         }
+        self._parsed_cache = {}
         self.create_schema()
 
+    def get_parser(self, extension: str) -> Optional[TreeSitterParser]:
+        """Gets or creates a TreeSitterParser for the given extension."""
+        lang_name = self.parsers.get(extension)
+        if not lang_name:
+            return None
+        
+        if lang_name not in self._parsed_cache:
+            try:
+                self._parsed_cache[lang_name] = TreeSitterParser(lang_name)
+            except Exception as e:
+                warning_logger(f"Failed to initialize parser for {lang_name}: {e}")
+                return None
+        return self._parsed_cache[lang_name]
+
     # A general schema creation based on common features across languages
     def create_schema(self):
         """Create constraints and indexes in Neo4j."""
@@ -175,6 +190,7 @@ class GraphBuilder:
                 session.run("CREATE INDEX function_lang IF NOT EXISTS FOR (f:Function) ON (f.lang)")
                 session.run("CREATE INDEX class_lang IF NOT EXISTS FOR (c:Class) ON (c.lang)")
                 session.run("CREATE INDEX annotation_lang IF NOT EXISTS FOR (a:Annotation) ON (a.lang)")
+                
                 is_falkordb = getattr(self.db_manager, 'get_backend_type', lambda: 'neo4j')() != 'neo4j'
                 if is_falkordb:
                     # FalkorDB uses db.idx.fulltext.createNodeIndex per label
@@ -242,85 +258,85 @@ class GraphBuilder:
 
         if '.py' in files_by_lang:
             from .languages import python as python_lang_module
-            imports_map.update(python_lang_module.pre_scan_python(files_by_lang['.py'], self.parsers['.py']))
+            imports_map.update(python_lang_module.pre_scan_python(files_by_lang['.py'], self.get_parser('.py')))
         if '.ipynb' in files_by_lang:
             from .languages import python as python_lang_module
-            imports_map.update(python_lang_module.pre_scan_python(files_by_lang['.ipynb'], self.parsers['.ipynb']))
+            imports_map.update(python_lang_module.pre_scan_python(files_by_lang['.ipynb'], self.get_parser('.ipynb')))
         if '.js' in files_by_lang:
             from .languages import javascript as js_lang_module
-            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.js'], self.parsers['.js']))
+            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.js'], self.get_parser('.js')))
         if '.jsx' in files_by_lang:
             from .languages import javascript as js_lang_module
-            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.jsx'], self.parsers['.jsx']))
+            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.jsx'], self.get_parser('.jsx')))
         if '.mjs' in files_by_lang:
             from .languages import javascript as js_lang_module
-            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.mjs'], self.parsers['.mjs']))
+            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.mjs'], self.get_parser('.mjs')))
         if '.cjs' in files_by_lang:
             from .languages import javascript as js_lang_module
-            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.cjs'], self.parsers['.cjs']))
+            imports_map.update(js_lang_module.pre_scan_javascript(files_by_lang['.cjs'], self.get_parser('.cjs')))
         if '.go' in files_by_lang:
              from .languages import go as go_lang_module
-             imports_map.update(go_lang_module.pre_scan_go(files_by_lang['.go'], self.parsers['.go']))
+             imports_map.update(go_lang_module.pre_scan_go(files_by_lang['.go'], self.get_parser('.go')))
         if '.ts' in files_by_lang:
             from .languages import typescript as ts_lang_module
-            imports_map.update(ts_lang_module.pre_scan_typescript(files_by_lang['.ts'], self.parsers['.ts']))
+            imports_map.update(ts_lang_module.pre_scan_typescript(files_by_lang['.ts'], self.get_parser('.ts')))
         if '.tsx' in files_by_lang:
             from .languages import typescriptjsx as tsx_lang_module
-            imports_map.update(tsx_lang_module.pre_scan_typescript(files_by_lang['.tsx'], self.parsers['.tsx']))
+            imports_map.update(tsx_lang_module.pre_scan_typescript(files_by_lang['.tsx'], self.get_parser('.tsx')))
         if '.cpp' in files_by_lang:
             from .languages import cpp as cpp_lang_module
-            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.cpp'], self.parsers['.cpp']))
+            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.cpp'], self.get_parser('.cpp')))
         if '.h' in files_by_lang:
             from .languages import cpp as cpp_lang_module
-            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.h'], self.parsers['.h']))
+            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.h'], self.get_parser('.h')))
         if '.hpp' in files_by_lang:
             from .languages import cpp as cpp_lang_module
-            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hpp'], self.parsers['.hpp']))
+            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hpp'], self.get_parser('.hpp')))
         if '.hh' in files_by_lang:
             from .languages import cpp as cpp_lang_module
-            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hh'], self.parsers['.hh']))
+            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hh'], self.get_parser('.hh')))
         if '.rs' in files_by_lang:
             from .languages import rust as rust_lang_module
-            imports_map.update(rust_lang_module.pre_scan_rust(files_by_lang['.rs'], self.parsers['.rs']))
+            imports_map.update(rust_lang_module.pre_scan_rust(files_by_lang['.rs'], self.get_parser('.rs')))
         if '.c' in files_by_lang:
             from .languages import c as c_lang_module
-            imports_map.update(c_lang_module.pre_scan_c(files_by_lang['.c'], self.parsers['.c']))
+            imports_map.update(c_lang_module.pre_scan_c(files_by_lang['.c'], self.get_parser('.c')))
         elif '.java' in files_by_lang:
             from .languages import java as java_lang_module
-            imports_map.update(java_lang_module.pre_scan_java(files_by_lang['.java'], self.parsers['.java']))
+            imports_map.update(java_lang_module.pre_scan_java(files_by_lang['.java'], self.get_parser('.java')))
         elif '.rb' in files_by_lang:
             from .languages import ruby as ruby_lang_module
-            imports_map.update(ruby_lang_module.pre_scan_ruby(files_by_lang['.rb'], self.parsers['.rb']))
+            imports_map.update(ruby_lang_module.pre_scan_ruby(files_by_lang['.rb'], self.get_parser('.rb')))
         elif '.cs' in files_by_lang:
             from .languages import csharp as csharp_lang_module
-            imports_map.update(csharp_lang_module.pre_scan_csharp(files_by_lang['.cs'], self.parsers['.cs']))
+            imports_map.update(csharp_lang_module.pre_scan_csharp(files_by_lang['.cs'], self.get_parser('.cs')))
         if '.kt' in files_by_lang:
             from .languages import kotlin as kotlin_lang_module
-            imports_map.update(kotlin_lang_module.pre_scan_kotlin(files_by_lang['.kt'], self.parsers['.kt']))
+            imports_map.update(kotlin_lang_module.pre_scan_kotlin(files_by_lang['.kt'], self.get_parser('.kt')))
         if '.scala' in files_by_lang:
             from .languages import scala as scala_lang_module
-            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.scala'], self.parsers['.scala']))
+            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.scala'], self.get_parser('.scala')))
         if '.sc' in files_by_lang:
             from .languages import scala as scala_lang_module
-            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.sc'], self.parsers['.sc']))
+            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.sc'], self.get_parser('.sc')))
         if '.swift' in files_by_lang:
             from .languages import swift as swift_lang_module
-            imports_map.update(swift_lang_module.pre_scan_swift(files_by_lang['.swift'], self.parsers['.swift']))
+            imports_map.update(swift_lang_module.pre_scan_swift(files_by_lang['.swift'], self.get_parser('.swift')))
         if '.dart' in files_by_lang:
             from .languages import dart as dart_lang_module
-            imports_map.update(dart_lang_module.pre_scan_dart(files_by_lang['.dart'], self.parsers['.dart']))
+            imports_map.update(dart_lang_module.pre_scan_dart(files_by_lang['.dart'], self.get_parser('.dart')))
         if '.pl' in files_by_lang:
             from .languages import perl as perl_lang_module
-            imports_map.update(perl_lang_module.pre_scan_perl(files_by_lang['.pl'], self.parsers['.pl']))
+            imports_map.update(perl_lang_module.pre_scan_perl(files_by_lang['.pl'], self.get_parser('.pl')))
         if '.pm' in files_by_lang:
             from .languages import perl as perl_lang_module
-            imports_map.update(perl_lang_module.pre_scan_perl(files_by_lang['.pm'], self.parsers['.pm']))
+            imports_map.update(perl_lang_module.pre_scan_perl(files_by_lang['.pm'], self.get_parser('.pm')))
         if '.ex' in files_by_lang:
             from .languages import elixir as elixir_lang_module
-            imports_map.update(elixir_lang_module.pre_scan_elixir(files_by_lang['.ex'], self.parsers['.ex']))
+            imports_map.update(elixir_lang_module.pre_scan_elixir(files_by_lang['.ex'], self.get_parser('.ex')))
         if '.exs' in files_by_lang:
             from .languages import elixir as elixir_lang_module
-            imports_map.update(elixir_lang_module.pre_scan_elixir(files_by_lang['.exs'], self.parsers['.exs']))
+            imports_map.update(elixir_lang_module.pre_scan_elixir(files_by_lang['.exs'], self.get_parser('.exs')))
 
         return imports_map
 
@@ -996,7 +1012,7 @@ class GraphBuilder:
 
     def parse_file(self, repo_path: Path, path: Path, is_dependency: bool = False) -> Dict:
         """Parses a file with the appropriate language parser and extracts code elements."""
-        parser = self.parsers.get(path.suffix)
+        parser = self.get_parser(path.suffix)
         if not parser:
             warning_logger(f"No parser found for file extension {path.suffix}. Skipping {path}")
             return {"path": str(path), "error": f"No parser for {path.suffix}"}
@@ -1123,9 +1139,9 @@ class GraphBuilder:
 
                 # Step 5: Tree-sitter supplement — add source text, complexity, imports and bases
                 file_path = Path(abs_path_str)
-                if file_path.exists() and file_path.suffix in self.parsers:
+                ts_parser = self.get_parser(file_path.suffix)
+                if file_path.exists() and ts_parser:
                     try:
-                        ts_parser = self.parsers[file_path.suffix]
                         ts_data = ts_parser.parse(file_path, is_dependency, index_source=True)
                         if "error" not in ts_data:
                             # 1. Functions: complexity, source, decorators