Function generate_html

code_review_graph/visualization.py:360–410

Generate a self-contained interactive HTML visualization. Args: store: The GraphStore to read graph data from. output_path: Path for the output HTML file. mode: Rendering mode — ``"auto"``, ``"full"``, ``"community"``, or ``"file"``. ``"auto"`` switches to

(
    store: GraphStore,
    output_path: str | Path,
    mode: str = "auto",
    max_full_nodes: int = 3000,
)

def generate_html(
    store: GraphStore,
    output_path: str | Path,
    mode: str = "auto",
    max_full_nodes: int = 3000,
) -> Path:
    """Generate a self-contained interactive HTML visualization.

    Args:
        store: The GraphStore to read graph data from.
        output_path: Path for the output HTML file.
        mode: Rendering mode — ``"auto"``, ``"full"``, ``"community"``,
              or ``"file"``.  ``"auto"`` switches to ``"community"`` when
              the node count exceeds *max_full_nodes*.
        max_full_nodes: Threshold for auto-switching to community mode.

    Writes the HTML file to *output_path* and returns the resolved Path.
    """
    output_path = Path(output_path)
    stats = store.get_stats()
    if stats.total_nodes > 50000:
        logger.warning(
            "Graph has %d nodes — visualization may be slow. "
            "Consider filtering by file pattern.", stats.total_nodes,
        )
    data = export_graph_data(store)

    # Determine effective mode
    effective_mode = mode
    if effective_mode == "auto":
        effective_mode = (
            "community" if stats.total_nodes > max_full_nodes else "full"
        )

    if effective_mode == "community":
        # Keep full data available for drill-down; aggregate for top-level
        agg = _aggregate_community(data)
        # Escape </script> inside JSON to prevent premature tag closure
        data_json = json.dumps(agg, default=str).replace("</", "<\\/")
        html = _AGGREGATED_HTML_TEMPLATE.replace("__GRAPH_DATA__", data_json)
    elif effective_mode == "file":
        agg = _aggregate_file(data)
        data_json = json.dumps(agg, default=str).replace("</", "<\\/")
        html = _AGGREGATED_HTML_TEMPLATE.replace("__GRAPH_DATA__", data_json)
    else:
        # full mode — original behavior
        data_json = json.dumps(data, default=str).replace("</", "<\\/")
        html = _HTML_TEMPLATE.replace("__GRAPH_DATA__", data_json)

    output_path.write_text(html, encoding="utf-8")
    return output_path


# ---------------------------------------------------------------------------