Function emit_dot

atomic-cli/src/commands/query/mod.rs:493–530

Generate a DOT (Graphviz) representation of a subgraph.

(nodes: &[KgNode], edges: &[KgEdge])

/// Generate a DOT (Graphviz) representation of a subgraph.
fn emit_dot(nodes: &[KgNode], edges: &[KgEdge]) -> String {
    let mut out = String::new();
    out.push_str("digraph query {\n");
    out.push_str("  rankdir=LR;\n");
    out.push_str(
        "  node [shape=box, style=\"filled,rounded\", fontname=\"Helvetica\", fontsize=10];\n",
    );
    out.push_str("  edge [fontname=\"Helvetica\", fontsize=8, color=\"#888888\"];\n");
    out.push('\n');

    // Nodes
    for node in nodes {
        let (fillcolor, shape) = node_style(&node.kind);
        let label = node_label(node);
        out.push_str(&format!(
            "  \"{}\" [label=\"{}\", fillcolor=\"{}\", shape={}];\n",
            dot_escape_id(&node.id),
            dot_escape_label(&label),
            fillcolor,
            shape,
        ));
    }

    out.push('\n');

    // Edges
    for edge in edges {
        out.push_str(&format!(
            "  \"{}\" -> \"{}\" [label=\"{}\"];\n",
            dot_escape_id(&edge.from_id),
            dot_escape_id(&edge.to_id),
            dot_escape_label(&edge.kind),
        ));
    }

    out.push_str("}\n");
    out
}

/// Returns `(fillcolor, shape)` for a given node kind.
fn node_style(kind: &str) -> (&'static str, &'static str) {