codegraphcontext 0.5.5__tar.gz → 0.5.6__tar.gz

{codegraphcontext-0.5.5/src/codegraphcontext.egg-info → codegraphcontext-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.5.5
+Version: 0.5.6
 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

{codegraphcontext-0.5.5 → codegraphcontext-0.5.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.5.5"
+version = "0.5.6"
 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.5.5 → codegraphcontext-0.5.6}/src/codegraphcontext/tools/advanced_language_query_tool.py

@@ -82,10 +82,12 @@ class Advanced_language_query:
 
         if language not in self.TOOLKITS:
             raise ValueError(f"Unsupported language: {language}")
-        self.toolkit = self.TOOLKITS[language]()
-
-        # Getting the language query
-        cypher_query = self.toolkit.get_cypher_query(label)
+        try:
+            self.toolkit = self.TOOLKITS[language]()
+            # Getting the language query
+            cypher_query = self.toolkit.get_cypher_query(label)
+        except NotImplementedError as exc:
+            return {"error": str(exc), "language": language, "query": query}
         try:
             debug_log(f"Executing Cypher query: {cypher_query}")
             with self.db_manager.get_driver().session() as session:

{codegraphcontext-0.5.5 → codegraphcontext-0.5.6}/src/codegraphcontext/tools/query_tool_languages/java_toolkit.py

@@ -12,15 +12,23 @@ from typing import Any, Dict, List, Optional
 
 
 class JavaToolkit:
-    """Spring-aware Java analysis methods backed by CodeFinder / raw Cypher."""
+    """Spring-aware Java analysis methods backed by CodeFinder / raw Cypher.
 
-    def __init__(self, code_finder: Any):
-        """
-        Args:
-            code_finder: A ``CodeFinder`` instance (or compatible duck-type).
-        """
+    Note: The generic ``advanced_language_query`` path calls this toolkit
+    with no arguments (consistent with all other language toolkits). The
+    Spring-specific MCP tools (``find_java_spring_endpoints`` /
+    ``find_java_spring_beans``) pass a ``code_finder`` via ``_init_with``.
+    """
+
+    def __init__(self) -> None:
+        self._cf: Any = None
+        self._driver: Any = None
+
+    def _init_with(self, code_finder: Any) -> "JavaToolkit":
+        """Initialise with a live CodeFinder for Spring-specific queries."""
         self._cf = code_finder
         self._driver = code_finder.driver
+        return self
 
     # ── Internal helpers ──────────────────────────────────────────────────────
 
@@ -249,5 +257,6 @@ class JavaToolkit:
     # Legacy shim — keeps existing callers working
     def get_cypher_query(self, query: str) -> str:
         raise NotImplementedError(
+            "AdvancedLanguageQuery is not implemented yet for Java. "
             "Use find_spring_endpoints(), find_beans_by_stereotype(), etc. instead."
         )

{codegraphcontext-0.5.5 → codegraphcontext-0.5.6}/src/codegraphcontext/utils/visualize_graph.py

@@ -12,6 +12,7 @@ import json
 import os
 import tempfile
 import webbrowser
+from html import escape as html_escape
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
@@ -55,6 +56,24 @@ def edge_color(rel_type: str) -> str:
     return _EDGE_COLORS.get(rel_type, _EDGE_COLORS["default"])
 
 
+def _json_for_script(data: Any) -> str:
+    """Serialise ``data`` for inlining inside a ``<script>`` element.
+
+    ``json.dumps`` escapes quotes and backslashes but leaves ``<``, ``>`` and
+    ``&`` alone, so a node name or file path containing ``</script>`` closes
+    the element early and the rest of the document is parsed as markup. Those
+    three characters only ever appear inside string literals, so replacing
+    them with their ``\\uXXXX`` form keeps the payload valid JSON and valid
+    JavaScript while making it inert as markup.
+    """
+    return (
+        json.dumps(data, indent=2)
+        .replace("<", "\\u003c")
+        .replace(">", "\\u003e")
+        .replace("&", "\\u0026")
+    )
+
+
 def build_graph_data(
     nodes: List[Dict[str, Any]],
     edges: List[Dict[str, Any]],
@@ -125,14 +144,15 @@ def render_html(
     str
         Complete HTML document as a string.
     """
-    graph_json = json.dumps(graph_data, indent=2)
+    graph_json = _json_for_script(graph_data)
+    safe_title = html_escape(title)
 
     html = f"""<!DOCTYPE html>
 <html lang="en">
 <head>
   <meta charset="UTF-8" />
   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-  <title>{title}</title>
+  <title>{safe_title}</title>
   <style>
     * {{ box-sizing: border-box; margin: 0; padding: 0; }}
     body {{
@@ -198,7 +218,7 @@ def render_html(
 </head>
 <body>
   <div id="header">
-    <h1>🏗 {title}</h1>
+    <h1>🏗 {safe_title}</h1>
     <span id="stats"></span>
   </div>
   <div id="canvas-container">

{codegraphcontext-0.5.5 → codegraphcontext-0.5.6}/src/codegraphcontext/viz/server.py

@@ -96,13 +96,16 @@ def _get_eid(element):
     if isinstance(element, (int, str)):
         return str(element)
     
-    # If element is a dict (like Neo4j returned items or KuzuDB node/rel dicts)
+    # If element is a dict (like Neo4j returned items or KuzuDB/Ladybug node/rel
+    # dicts). KùzuDB spells its internal id field '_id'; LadybugDB spells the
+    # same field '_ID' (uppercase) — confirmed against a live query on each
+    # embedded backend.
     if isinstance(element, dict):
         # KuzuDB _src / _dst are directly {'offset': X, 'table': Y}
         if 'offset' in element and 'table' in element:
             return f"{element.get('table')}_{element.get('offset')}"
-            
-        for key in ['_id', 'id', 'element_id']:
+
+        for key in ['_id', '_ID', 'id', 'element_id']:
             if key in element:
                 val = element[key]
                 if val is not None:
@@ -116,7 +119,7 @@ def _get_eid(element):
     for attr in ['element_id', 'id', '_id']:
         if hasattr(element, attr):
             val = getattr(element, attr)
-            if val is not None: 
+            if val is not None:
                 # KuzuDB objects if any
                 if isinstance(val, dict):
                     return f"{val.get('table', 0)}_{val.get('offset', 0)}"
@@ -124,6 +127,18 @@ def _get_eid(element):
     return str(id(element))
 
 
+def _meta(d: dict, name: str):
+    """Read a KùzuDB/LadybugDB internal dict field regardless of its casing.
+
+    KùzuDB spells these fields lowercase (_label/_src/_dst); LadybugDB spells
+    the same fields uppercase (_LABEL/_SRC/_DST) — confirmed against a live
+    query on each embedded backend. Keying on one casing only silently drops
+    every row on the other backend.
+    """
+    lower = f"_{name}"
+    return d[lower] if lower in d else d.get(f"_{name.upper()}")
+
+
 @app.get("/api/graph")
 async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str] = None):
     if not db_manager:
@@ -186,9 +201,10 @@ async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str]
                                 # Extract labels
                                 labels = []
                                 if isinstance(node, dict):
-                                    # KuzuDB node label is under '_label'
-                                    if '_label' in node:
-                                        labels = [node['_label']]
+                                    # KuzuDB node label is under '_label' (LadybugDB: '_LABEL')
+                                    meta_label = _meta(node, 'label')
+                                    if meta_label is not None:
+                                        labels = [meta_label]
                                     elif 'label' in node:
                                         labels = [node['label']]
                                 else:
@@ -247,13 +263,17 @@ async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str]
                         # FalkorDB / KuzuDB may return rels as dicts OR objects
                         if isinstance(rel, dict):
                             rid = get_eid(rel)
-                            # KuzuDB uses _src and _dst, FalkorDB uses src_node/dest_node
-                            src = rel.get('_src', rel.get('src_node'))
-                            dst = rel.get('_dst', rel.get('dest_node'))
-                            
+                            # KuzuDB uses _src/_dst (LadybugDB: _SRC/_DST), FalkorDB uses src_node/dest_node
+                            src = _meta(rel, 'src')
+                            if src is None:
+                                src = rel.get('src_node')
+                            dst = _meta(rel, 'dst')
+                            if dst is None:
+                                dst = rel.get('dest_node')
+
                             source = get_eid(src) if src is not None else None
                             target = get_eid(dst) if dst is not None else None
-                            rel_type = str(rel.get('_label', rel.get('relation', rel.get('type', 'RELATED')))).upper()
+                            rel_type = str(_meta(rel, 'label') or rel.get('relation', rel.get('type', 'RELATED'))).upper()
                         else:
                             rid = get_eid(rel)
                             start_node = None
@@ -420,8 +440,9 @@ def parse_node(node, nodes_dict):
         
     labels = []
     if isinstance(node, dict):
-        if '_label' in node:
-            labels = [node['_label']]
+        meta_label = _meta(node, 'label')
+        if meta_label is not None:
+            labels = [meta_label]
         elif 'label' in node:
             labels = [node['label']]
     else:
@@ -463,11 +484,15 @@ def parse_node(node, nodes_dict):
 def parse_rel(rel, edges):
     rid = _get_eid(rel)
     if isinstance(rel, dict):
-        src = rel.get('_src', rel.get('src_node'))
-        dst = rel.get('_dst', rel.get('dest_node'))
+        src = _meta(rel, 'src')
+        if src is None:
+            src = rel.get('src_node')
+        dst = _meta(rel, 'dst')
+        if dst is None:
+            dst = rel.get('dest_node')
         source = _get_eid(src) if src is not None else None
         target = _get_eid(dst) if dst is not None else None
-        rel_type = str(rel.get('_label', rel.get('relation', rel.get('type', 'RELATED')))).upper()
+        rel_type = str(_meta(rel, 'label') or rel.get('relation', rel.get('type', 'RELATED'))).upper()
     else:
         start_node = None
         end_node = None
@@ -510,9 +535,16 @@ def parse_element(val, nodes_dict, edges):
         
     type_name = type(val).__name__
     
-    if type_name in ('Node', 'KuzuNode') or (isinstance(val, dict) and ('_label' in val or 'labels' in val) and not any(k in val for k in ('_src', '_dst', 'src_node', 'dest_node'))):
+    is_dict_rel = isinstance(val, dict) and (
+        any(k in val for k in ('_src', '_dst', '_SRC', '_DST', 'src_node', 'dest_node'))
+    )
+    is_dict_node = isinstance(val, dict) and (
+        '_label' in val or '_LABEL' in val or 'labels' in val
+    ) and not is_dict_rel
+
+    if type_name in ('Node', 'KuzuNode') or is_dict_node:
         parse_node(val, nodes_dict)
-    elif type_name in ('Relationship', 'KuzuRelationship') or (isinstance(val, dict) and any(k in val for k in ('_src', '_dst', 'src_node', 'dest_node'))):
+    elif type_name in ('Relationship', 'KuzuRelationship') or is_dict_rel:
         parse_rel(val, edges)
     elif isinstance(val, (list, tuple)):
         for item in val:

{codegraphcontext-0.5.5 → codegraphcontext-0.5.6/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.5.5
+Version: 0.5.6
 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