Function get_graph

src/codegraphcontext/viz/server.py:143–349

(repo_path: Optional[str] = None, cypher_query: Optional[str] = None)

@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")
    
    get_eid = _get_eid

    try:
        nodes_dict = {}
        edges = []

        print(f"DEBUG: Starting get_graph with repo_path={repo_path}", flush=True)

        # This endpoint only ever reads. Open the session in READ access mode so
        # the database enforces read-only (Neo4j default_access_mode / FalkorDB
        # GRAPH.RO_QUERY) in addition to the regex guard on any user query.
        with db_manager.get_driver().session(**_read_session_kwargs(db_manager)) as session:
            if cypher_query:
                if not _is_read_only_cypher(cypher_query):
                    raise HTTPException(
                        status_code=400,
                        detail=(
                            "This endpoint only supports read-only Cypher queries. "
                            "Prohibited keywords like CREATE, MERGE, DELETE, SET, REMOVE, "
                            "DROP, or CALL apoc are not allowed."
                        ),
                    )
                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)
                # Get all nodes within the repository scope
                query = """
                MATCH (r:Repository {path: $repo_path})
                OPTIONAL MATCH (r)-[:CONTAINS*0..]->(n)
                WITH DISTINCT r, COLLECT(DISTINCT n) as repo_nodes
                UNWIND repo_nodes as node
                OPTIONAL MATCH (node)-[rel]->(target)
                WITH r, node, rel, target, repo_nodes
                WHERE target IN repo_nodes OR target = r
                RETURN node as n, rel, target as 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 50000"
                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: