mermaid-text 0.47.0

//! Hand-rolled parser for Mermaid `graph`/`flowchart` syntax.
//!
//! The parser works statement-by-statement. A "statement" is one logical
//! declaration separated by a newline or semicolon. Each statement is
//! classified as either:
//!
//! - A **node definition**: `A[Label]`, `A{Label}`, `A((Label))`, `A(Label)`, or bare `A`
//! - An **edge chain**: `A --> B --> C`, potentially with inline labels
//! - A **subgraph block**: `subgraph ID [Label]` … `end`
//! - A **header line**: `graph LR` / `flowchart TD` (handled before entering this module)
//! - A blank / comment line — silently ignored
//!
//! Edge style (`-->`, `---`, `-.->`, `==>`, `<-->`, `--o`, `--x`) is parsed
//! and stored on each [`Edge`] for the renderer to use.

use crate::{
    Error,
    parser::common::{
        apply_pending_classes, extract_class_modifier, parse_class_def_directive,
        parse_class_directive, parse_link_style_directive, parse_style_directive,
    },
    types::{
        ClickTarget, Direction, Edge, EdgeEndpoint, EdgeStyle, Graph, Node, NodeShape, Subgraph,
    },
};

// ---------------------------------------------------------------------------
// Public entry point
// ---------------------------------------------------------------------------

/// Parse a Mermaid `graph`/`flowchart` source string into a [`Graph`].
///
/// The function expects the *full* input including the header line
/// (`graph LR`, `flowchart TD`, etc.). Both newlines and semicolons are
/// treated as statement separators, so `graph LR; A-->B` is valid.
///
/// # Arguments
///
/// * `input` — the complete Mermaid source string
///
/// # Returns
///
/// A [`Graph`] containing all parsed nodes, edges, and subgraphs.
///
/// # Errors
///
/// Returns [`crate::Error::ParseError`] if the header statement is missing or
/// the direction keyword is unrecognised.
///
/// # Examples
///
/// ```
/// use mermaid_text::parser::parse;
/// use mermaid_text::{Direction, NodeShape};
///
/// let graph = parse("graph LR; A[Start] --> B[End]").unwrap();
/// assert_eq!(graph.direction, Direction::LeftToRight);
/// assert_eq!(graph.node("A").unwrap().label, "Start");
/// assert_eq!(graph.node("B").unwrap().shape, NodeShape::Rectangle);
/// assert_eq!(graph.edges.len(), 1);
/// ```
pub fn parse(input: &str) -> Result<Graph, Error> {
    // Normalise: replace newlines with semicolons, then split on ';'.
    // This means both `graph LR; A-->B` and multi-line input are handled
    // identically — the first non-blank, non-comment statement is the header.
    let normalised = input.replace('\n', ";").replace('\r', "");

    let statements: Vec<&str> = normalised
        .split(';')
        .map(str::trim)
        .filter(|s| !s.is_empty() && !s.starts_with("%%"))
        .collect();

    let mut iter = statements.iter().copied();

    // ---- Find and parse the header statement ----------------------------
    let direction = parse_header_stmt(&mut iter)?;
    let mut graph = Graph::new(direction);

    // ---- Parse each remaining statement ---------------------------------
    // We collect remaining statements into a Vec so we can do a stateful
    // multi-statement parse (subgraph blocks span multiple statements).
    //
    // `pending_classes` collects `(target_id, class_name)` from `class …`
    // directives and from inline `:::className` shorthands. Resolution
    // happens at end-of-parse so a `class A foo` statement can appear
    // before its `classDef foo …` definition (a real Mermaid pattern).
    let remaining: Vec<&str> = iter.collect();
    let mut pending_classes: Vec<(String, String)> = Vec::new();
    parse_statements(&remaining, &mut graph, &mut None, &mut pending_classes);
    apply_pending_classes(&mut graph, &pending_classes);

    Ok(graph)
}

// ---------------------------------------------------------------------------
// Header parsing
// ---------------------------------------------------------------------------

/// Consume the first statement from `stmts` and parse it as a
/// `graph`/`flowchart` header, returning the [`Direction`].
///
/// The direction is the first whitespace-delimited token after the keyword.
fn parse_header_stmt<'a>(stmts: &mut impl Iterator<Item = &'a str>) -> Result<Direction, Error> {
    let stmt = stmts
        .next()
        .ok_or_else(|| Error::ParseError("no 'graph'/'flowchart' header found".to_string()))?;

    // e.g. "graph LR" or "flowchart TD"
    let mut parts = stmt.splitn(3, |c: char| c.is_whitespace());
    let keyword = parts.next().unwrap_or("").to_lowercase();

    if keyword != "graph" && keyword != "flowchart" {
        return Err(Error::ParseError(format!(
            "expected 'graph' or 'flowchart', got '{keyword}'"
        )));
    }

    // The direction is the next whitespace-separated token (just the first
    // word — we ignore any trailing content on the header line since we
    // already split on semicolons above).
    let dir_str = parts
        .next()
        .map(str::trim)
        .filter(|s| !s.is_empty())
        .unwrap_or("TD"); // default to top-down if omitted

    Direction::parse(dir_str)
        .ok_or_else(|| Error::ParseError(format!("unknown direction '{dir_str}'")))
}

// ---------------------------------------------------------------------------
// Statement parsing
// ---------------------------------------------------------------------------

/// Parse a slice of statements into `graph`.
///
/// `current_subgraph_id` is `Some(id)` when we are inside a subgraph block
/// (used to register node membership). This function is called recursively
/// for nested subgraphs — the inner call consumes statements up through `end`
/// and the outer call continues from there.
///
/// Returns the index of the statement **after** the `end` that closed the
/// innermost subgraph block, or `stmts.len()` if there was no `end`.
fn parse_statements(
    stmts: &[&str],
    graph: &mut Graph,
    current_subgraph_id: &mut Option<String>,
    pending_classes: &mut Vec<(String, String)>,
) -> usize {
    let mut i = 0;
    while i < stmts.len() {
        let stmt = stmts[i];
        let first_word = stmt.split_whitespace().next().unwrap_or("");

        match first_word {
            "subgraph" => {
                // Parse the subgraph header: `subgraph ID` or `subgraph ID Label`
                let (sg_id, sg_label) = parse_subgraph_header(stmt);

                // Register this subgraph in the parent (or at the top level).
                if let Some(ref parent_id) = current_subgraph_id.clone() {
                    // Nested: link child into parent.
                    if let Some(parent) = graph.subgraphs.iter_mut().find(|s| &s.id == parent_id) {
                        parent.subgraph_ids.push(sg_id.clone());
                    }
                }
                graph.subgraphs.push(Subgraph::new(sg_id.clone(), sg_label));

                // Recurse into the subgraph body. The recursive call consumes
                // statements until it hits the matching `end` and returns the
                // index of the statement after `end`.
                let mut inner_sg = Some(sg_id);
                i += 1;
                let consumed = parse_statements(&stmts[i..], graph, &mut inner_sg, pending_classes);
                i += consumed;
            }
            "end" => {
                // Close the current subgraph block. Tell the caller we consumed
                // through (and including) this `end`.
                return i + 1;
            }
            "direction" => {
                // `direction TB` inside a subgraph: store it on the model.
                if let Some(ref sg_id) = current_subgraph_id.clone() {
                    let dir_word = stmt.split_whitespace().nth(1).unwrap_or("");
                    if let Some(dir) = Direction::parse(dir_word)
                        && let Some(sg) = graph.subgraphs.iter_mut().find(|s| s.id == *sg_id)
                    {
                        sg.direction = Some(dir);
                    }
                }
                i += 1;
            }
            "style" => {
                parse_style_directive(stmt, graph);
                i += 1;
            }
            "linkStyle" => {
                parse_link_style_directive(stmt, graph);
                i += 1;
            }
            "classDef" => {
                parse_class_def_directive(stmt, graph);
                i += 1;
            }
            "class" => {
                parse_class_directive(stmt, pending_classes);
                i += 1;
            }
            "click" => {
                parse_click_directive(stmt, graph);
                i += 1;
            }
            // Accessibility directives — no visual representation.
            "accTitle" | "accDescr" => {
                i += 1;
            }
            _ => {
                // Regular node definition or edge chain.
                parse_statement(stmt, graph, current_subgraph_id, pending_classes);
                i += 1;
            }
        }
    }
    // Consumed all statements without seeing `end` (top-level or unclosed block).
    stmts.len()
}

/// Parse a single statement (already trimmed, no leading/trailing whitespace).
///
/// A statement is either a standalone node definition or an edge chain that
/// may include inline node definitions.
///
/// Any nodes referenced in edges are auto-created if they have not been
/// explicitly defined yet. If `current_subgraph_id` is `Some`, newly seen
/// node IDs are recorded as members of that subgraph.
fn parse_statement(
    stmt: &str,
    graph: &mut Graph,
    current_subgraph_id: &mut Option<String>,
    pending_classes: &mut Vec<(String, String)>,
) {
    // Try to parse as an edge chain first (contains an arrow token).
    if looks_like_edge_chain(stmt) {
        parse_edge_chain(stmt, graph, current_subgraph_id, pending_classes);
    } else {
        // Pure node definition: A[label]:::cls, A{label}, A((label)), A(label), A
        let _ = materialize_node_token(stmt, graph, current_subgraph_id, pending_classes);
    }
}

fn materialize_node_token(
    token: &str,
    graph: &mut Graph,
    current_subgraph_id: &mut Option<String>,
    pending_classes: &mut Vec<(String, String)>,
) -> Option<String> {
    // Strip the optional `:::cls1:::cls2` shorthand BEFORE the shape
    // parser sees the token — shape parsing doesn't know about
    // class modifiers.
    let (clean_tok, classes) = extract_class_modifier(token);
    let node = parse_node_definition(&clean_tok)
        .unwrap_or_else(|| Node::new(clean_tok.clone(), clean_tok.clone(), NodeShape::Rectangle));
    let node_id = node.id.clone();
    for class_name in classes {
        pending_classes.push((node_id.clone(), class_name));
    }
    graph.upsert_node(node);
    register_node_in_subgraph(graph, &node_id, current_subgraph_id);
    Some(node_id)
}

/// Register `node_id` as a direct member of the current subgraph (if any).
///
/// Only registers if the node is not already a member (avoids duplicates from
/// multiple references to the same node within one subgraph body).
fn register_node_in_subgraph(
    graph: &mut Graph,
    node_id: &str,
    current_subgraph_id: &Option<String>,
) {
    if let Some(sg_id) = current_subgraph_id
        && let Some(sg) = graph.subgraphs.iter_mut().find(|s| s.id == *sg_id)
        && !sg.node_ids.contains(&node_id.to_string())
    {
        sg.node_ids.push(node_id.to_string());
    }
}

/// Parse the `subgraph` header statement and extract `(id, label)`.
///
/// Mermaid supports these forms:
/// - `subgraph ID`           — label defaults to ID
/// - `subgraph ID[Label]`    — label in square brackets (no space before `[`)
/// - `subgraph ID [Label]`   — same with a space
/// - `subgraph "Label"`      — quoted label used as both id and label
fn parse_subgraph_header(stmt: &str) -> (String, String) {
    // Strip the "subgraph" keyword.
    let rest = stmt.trim_start_matches("subgraph").trim();

    if rest.is_empty() {
        // Bare `subgraph` with no identifier — use a placeholder.
        return ("__sg__".to_string(), "".to_string());
    }

    // Check for bracket-style label: `ID[Label]` or `ID [Label]`.
    if let Some(bracket_pos) = rest.find('[') {
        let id = rest[..bracket_pos].trim().to_string();
        let rest_after = &rest[bracket_pos + 1..];
        let label = if let Some(close) = rest_after.find(']') {
            rest_after[..close].trim().to_string()
        } else {
            rest_after.trim().to_string()
        };
        let id = if id.is_empty() { label.clone() } else { id };
        return (id, label);
    }

    // No bracket: the entire rest is the ID, and label == ID.
    let id = rest.to_string();
    (id.clone(), id)
}

/// Return `true` if the statement appears to contain at least one edge arrow.
fn looks_like_edge_chain(s: &str) -> bool {
    tokenise_chain(s).len() >= 3
}

// ---------------------------------------------------------------------------
// Edge chain parsing
// ---------------------------------------------------------------------------

/// Parse an edge chain statement and push nodes + edges into `graph`.
///
/// The chain is tokenised by splitting on edge markers while preserving
/// edge-label content between `|...|` delimiters.
fn parse_edge_chain(
    stmt: &str,
    graph: &mut Graph,
    current_subgraph_id: &mut Option<String>,
    pending_classes: &mut Vec<(String, String)>,
) {
    // We build a list of (node_token, edge_label_or_none) pairs.
    // Strategy: walk char-by-char, extracting alternating node/edge segments.

    let tokens = tokenise_chain(stmt);
    if tokens.is_empty() {
        return;
    }

    // tokens = [node_tok, edge_tok, node_tok, edge_tok, node_tok, ...]
    // Odd indices are node tokens, even indices are edge (arrow+label) tokens.
    // Actually our tokeniser returns: node, arrow, node, arrow, node
    // i.e. length is always odd and ≥ 1.

    // Collect (node_token, Option<edge_label_before_next_node>) pairs.
    // We iterate pairs of (node_tok, Option<arrow_tok>).
    let mut i = 0;
    let mut prev_ids: Vec<String> = Vec::new();

    // Pending edge metadata carried forward between node tokens.
    let mut pending_edge_label: Option<String> = None;
    let mut pending_edge_style = EdgeStyle::Solid;
    let mut pending_edge_start = EdgeEndpoint::None;
    let mut pending_edge_end = EdgeEndpoint::Arrow;

    while i < tokens.len() {
        let tok = tokens[i].trim();

        if i % 2 == 0 {
            // Node token
            if tok.is_empty() {
                i += 1;
                continue;
            }
            let current_ids = split_grouped_node_token(tok)
                .into_iter()
                .filter_map(|part| {
                    materialize_node_token(&part, graph, current_subgraph_id, pending_classes)
                })
                .collect::<Vec<_>>();

            if !prev_ids.is_empty() {
                for from in &prev_ids {
                    for to in &current_ids {
                        let edge = Edge::new_styled(
                            from.clone(),
                            to.clone(),
                            pending_edge_label.clone(),
                            pending_edge_style,
                            pending_edge_start,
                            pending_edge_end,
                        );
                        graph.edges.push(edge);
                    }
                }
                pending_edge_label = None;
                pending_edge_style = EdgeStyle::Solid;
                pending_edge_start = EdgeEndpoint::None;
                pending_edge_end = EdgeEndpoint::Arrow;
            }
            prev_ids = current_ids;
        } else {
            // Arrow token — extract style and optional label.
            let (style, start, end) = classify_arrow(tok);
            pending_edge_style = style;
            pending_edge_start = start;
            pending_edge_end = end;
            pending_edge_label = extract_arrow_label(tok);
        }

        i += 1;
    }
}

/// Split a chain statement into alternating node/arrow tokens.
///
/// Returns a `Vec<String>` where even indices are node tokens and odd indices
/// are arrow tokens (including any `|label|` portion).
fn tokenise_chain(stmt: &str) -> Vec<String> {
    let mut tokens: Vec<String> = Vec::new();
    let chars: Vec<char> = stmt.chars().collect();
    let len = chars.len();
    let mut i = 0;
    let mut current = String::new();
    let mut depth = DelimiterDepth::default();
    let mut in_quotes = false;

    while i < len {
        let ch = chars[i];

        let is_potential_arrow_start = !depth.is_nested()
            && !in_quotes
            && (ch == '-' || ch == '=' || ch == '<')
            && !current.trim().is_empty();

        if is_potential_arrow_start {
            let remaining: String = chars[i..].iter().collect();
            if let Some((arrow_tok, consumed)) = try_consume_arrow(&remaining) {
                tokens.push(current.trim().to_string());
                current = String::new();
                tokens.push(arrow_tok);
                i += consumed;
                continue;
            }
        }

        current.push(ch);
        depth.observe(ch, in_quotes);
        if ch == '"' {
            in_quotes = !in_quotes;
        }
        i += 1;
    }

    // Push the last node token
    let last = current.trim().to_string();
    if !last.is_empty() {
        tokens.push(last);
    }

    tokens
}

#[derive(Debug, Default, Clone, Copy)]
struct DelimiterDepth {
    square: usize,
    round: usize,
    curly: usize,
}

impl DelimiterDepth {
    fn is_nested(self) -> bool {
        self.square > 0 || self.round > 0 || self.curly > 0
    }

    fn observe(&mut self, ch: char, in_quotes: bool) {
        if in_quotes && ch != '"' {
            return;
        }
        match ch {
            '[' => self.square += 1,
            ']' => self.square = self.square.saturating_sub(1),
            '(' => self.round += 1,
            ')' => self.round = self.round.saturating_sub(1),
            '{' => self.curly += 1,
            '}' => self.curly = self.curly.saturating_sub(1),
            _ => {}
        }
    }
}

fn split_grouped_node_token(token: &str) -> Vec<String> {
    let mut parts = Vec::new();
    let mut current = String::new();
    let mut depth = DelimiterDepth::default();
    let mut in_quotes = false;

    for ch in token.chars() {
        if ch == '&' && !depth.is_nested() && !in_quotes {
            let part = current.trim();
            if part.is_empty() {
                return vec![token.trim().to_string()];
            }
            parts.push(part.to_string());
            current.clear();
            continue;
        }
        current.push(ch);
        depth.observe(ch, in_quotes);
        if ch == '"' {
            in_quotes = !in_quotes;
        }
    }

    let tail = current.trim();
    if tail.is_empty() {
        return vec![token.trim().to_string()];
    }
    parts.push(tail.to_string());
    parts
}

fn try_consume_arrow(s: &str) -> Option<(String, usize)> {
    if let Some(rest) = s.strip_prefix("<-->") {
        let (label_part, extra) = try_consume_pipe_label(rest);
        return Some((format!("<-->{label_part}"), 4 + extra));
    }
    if let Some(arrow) = try_consume_labeled_dash_arrow(s) {
        let len = arrow.chars().count();
        return Some((arrow, len));
    }
    if let Some((tok, consumed)) = try_consume_inline_quoted_arrow(s) {
        return Some((tok, consumed));
    }
    if let Some((tok, consumed)) = try_consume_inline_compact_arrow(s) {
        return Some((tok, consumed));
    }
    if s.starts_with("-.-") {
        let base = if s.starts_with("-.->") { 4 } else { 3 };
        let (label_part, extra) = try_consume_pipe_label(&s[base..]);
        return Some((format!("{}{label_part}", &s[..base]), base + extra));
    }
    if s.starts_with("==") {
        let mut len = 0;
        for ch in s.chars() {
            if ch == '=' {
                len += 1;
            } else {
                break;
            }
        }
        let has_arrow = s[len..].starts_with('>');
        if has_arrow {
            len += 1;
        }
        let (label_part, extra) = try_consume_pipe_label(&s[len..]);
        return Some((format!("{}{label_part}", &s[..len]), len + extra));
    }
    if s.starts_with("--o") {
        return Some(("--o".to_string(), 3));
    }
    if s.starts_with("--x") {
        return Some(("--x".to_string(), 3));
    }
    if let Some(rest) = s.strip_prefix("-->") {
        let (label_part, extra) = try_consume_pipe_label(rest);
        return Some((format!("-->{label_part}"), 3 + extra));
    }
    if let Some(rest) = s.strip_prefix("---") {
        let (label_part, extra) = try_consume_pipe_label(rest);
        return Some((format!("---{label_part}"), 3 + extra));
    }
    if s.starts_with("--") {
        return Some((s[..2].to_string(), 2));
    }
    None
}

fn try_consume_inline_compact_arrow(s: &str) -> Option<(String, usize)> {
    if let Some(rest) = s.strip_prefix("-.")
        && !rest.starts_with("->")
        && let Some(end) = rest.find(".->")
    {
        let label = rest[..end].trim();
        if !label.is_empty() {
            return Some((format!("-.->|{label}|"), 2 + end + 3));
        }
    }
    if let Some(rest) = s.strip_prefix("==")
        && !rest.starts_with('>')
        && let Some(end) = rest.find("==>")
    {
        let label = rest[..end].trim();
        if !label.is_empty() {
            return Some((format!("==>|{label}|"), 2 + end + 3));
        }
    }
    None
}

/// Classify an arrow token into `(style, start_endpoint, end_endpoint)`.
///
/// The classification mirrors the Mermaid specification:
/// - `<-->` → bidirectional solid
/// - `==>` → thick with arrow
/// - `-.->` / `-..->` → dotted with arrow
/// - `-->` → solid with arrow (default)
/// - `---` → solid, no arrow
/// - `--o` → solid, circle endpoint
/// - `--x` → solid, cross endpoint
fn classify_arrow(arrow: &str) -> (EdgeStyle, EdgeEndpoint, EdgeEndpoint) {
    // Strip any |label| portion before classifying.
    let base = if let Some(pipe) = arrow.find('|') {
        &arrow[..pipe]
    } else {
        arrow
    }
    .trim();

    // Bidirectional: <-->
    if base.starts_with('<') && base.ends_with('>') {
        return (EdgeStyle::Solid, EdgeEndpoint::Arrow, EdgeEndpoint::Arrow);
    }
    // Circle endpoint: --o
    if base.ends_with('o') && base.starts_with('-') {
        return (EdgeStyle::Solid, EdgeEndpoint::None, EdgeEndpoint::Circle);
    }
    // Cross endpoint: --x
    if base.ends_with('x') && base.starts_with('-') {
        return (EdgeStyle::Solid, EdgeEndpoint::None, EdgeEndpoint::Cross);
    }
    // Thick with arrow: ==>
    if base.starts_with('=') {
        let has_arrow = base.ends_with('>');
        let end = if has_arrow {
            EdgeEndpoint::Arrow
        } else {
            EdgeEndpoint::None
        };
        return (EdgeStyle::Thick, EdgeEndpoint::None, end);
    }
    // Dotted: -.- or -.->
    if base.contains(".-") || base.contains("-.") {
        let has_arrow = base.ends_with('>');
        let end = if has_arrow {
            EdgeEndpoint::Arrow
        } else {
            EdgeEndpoint::None
        };
        return (EdgeStyle::Dotted, EdgeEndpoint::None, end);
    }
    // Solid no-arrow: ---  or "-- label --" (no trailing >)
    if base.starts_with('-') && !base.ends_with('>') && !base.ends_with('o') && !base.ends_with('x')
    {
        return (EdgeStyle::Solid, EdgeEndpoint::None, EdgeEndpoint::None);
    }
    // Default: solid arrow -->
    (EdgeStyle::Solid, EdgeEndpoint::None, EdgeEndpoint::Arrow)
}

/// Try to parse inline-quoted-label arrow forms, returning `(normalised_token, chars_consumed)`.
///
/// Recognised patterns and their normalised output (pipe-label form):
///
/// | Source syntax        | Normalised token |
/// |----------------------|------------------|
/// | `-. "label" .->`     | `-.->|"label"|`  |
/// | `== "label" ==>`     | `==>|"label"|`   |
///
/// Normalising to pipe-label keeps `classify_arrow` and `extract_arrow_label`
/// unchanged — all three inline-quoted styles (solid handled by
/// `try_consume_labeled_dash_arrow`, dashed and thick handled here) share the
/// same downstream normalization path via `extract_arrow_label`'s pipe-label branch.
///
/// Returns `None` when `s` does not start with a recognised inline-quoted prefix.
fn try_consume_inline_quoted_arrow(s: &str) -> Option<(String, usize)> {
    // All characters in the arrow tokens and ASCII quotes are single-byte UTF-8,
    // so byte length == char count for consumption purposes. The label text
    // itself may contain multi-byte chars, but we don't need to count it
    // separately — we measure the whole consumed slice as `s.len() - tail.len()`.

    // Dashed: `-. "label" .->`  →  normalised as `-.->|"label"|`
    // `strip_prefix` enforces the opening `-. "` in one step (avoids the
    // clippy::manual_strip lint that fires on `starts_with` + index-slice).
    if let Some(after_open) = s.strip_prefix("-. \"")
        && let Some(close_q) = after_open.find('"')
    {
        // after_open[..close_q] is the label text; after_open[close_q+1..] is
        // everything after the closing quote.
        let label = &after_open[..close_q];
        let tail_start = after_open[close_q + 1..].trim_start_matches(' ');
        if let Some(tail) = tail_start.strip_prefix(".->") {
            let pipe_tok = format!("-.->|\"{label}\"|");
            let consumed = s.len() - tail.len();
            return Some((pipe_tok, consumed));
        }
    }

    // Thick: `== "label" ==>`  →  normalised as `==>|"label"|`
    if let Some(after_open) = s.strip_prefix("== \"")
        && let Some(close_q) = after_open.find('"')
    {
        let label = &after_open[..close_q];
        let tail_start = after_open[close_q + 1..].trim_start_matches(' ');
        if let Some(tail) = tail_start.strip_prefix("==>") {
            let pipe_tok = format!("==>|\"{label}\"|");
            let consumed = s.len() - tail.len();
            return Some((pipe_tok, consumed));
        }
    }

    None
}

/// Try to parse `-- label -->` form. Returns the full token string if matched.
fn try_consume_labeled_dash_arrow(s: &str) -> Option<String> {
    // Must start with "-- " (dash dash space)
    if !s.starts_with("-- ") {
        return None;
    }
    // Find closing "-->"
    let rest = &s[3..];
    rest.find("-->").map(|end| {
        let full_len = 3 + end + 3; // "-- " + label + "-->"
        s[..full_len].to_string()
    })
}

/// Try to consume a `|label|` suffix. Returns `(consumed_string, char_count)`.
fn try_consume_pipe_label(s: &str) -> (String, usize) {
    if let Some(inner) = s.strip_prefix('|')
        && let Some(end) = inner.find('|')
    {
        let portion = &s[..end + 2]; // includes both pipes
        return (portion.to_string(), end + 2);
    }
    (String::new(), 0)
}

/// Extract a label string from an arrow token, if present.
///
/// Handles `-->|label|`, `-- label -->`, etc.
fn extract_arrow_label(arrow: &str) -> Option<String> {
    // Pipe-style: -->|label| or -.->|label|
    let raw = if let Some(start) = arrow.find('|')
        && let Some(end) = arrow[start + 1..].find('|')
    {
        Some(arrow[start + 1..start + 1 + end].trim())
    } else if arrow.starts_with("-- ")
        && let Some(end) = arrow.rfind("-->")
    {
        // Dash-style: -- label -->
        Some(arrow[3..end].trim())
    } else {
        None
    };
    raw.and_then(|s| {
        // Strip the optional surrounding quotes Mermaid allows so labels
        // with commas / spaces survive, then run through `normalize_label`
        // so HTML `<br>` tags become real newlines (matching node labels)
        // and overly-long lines get soft-wrapped.
        let unquoted = s.trim_matches('"');
        let normalised = normalize_label(unquoted);
        if normalised.is_empty() {
            None
        } else {
            Some(normalised)
        }
    })
}

// ---------------------------------------------------------------------------
// Node definition parsing
// ---------------------------------------------------------------------------

/// Parse a single node-definition token such as `A[Label]`, `B{text}`,
/// `C((name))`, `D(rounded)`, `E([Stadium])`, `F[[Sub]]`, etc., or bare `E`.
///
/// Shape patterns are matched **most-specific-first** to handle multi-char
/// delimiters like `(((`, `((`, `{{`, `[[`, `([`, `[(` before single chars.
///
/// Returns `None` if the token is empty or unparseable.
pub(crate) fn parse_node_definition(token: &str) -> Option<Node> {
    let token = token.trim();
    if token.is_empty() {
        return None;
    }

    // Asymmetric `>label]` — starts with `>`, no bracket for ID split.
    if token.starts_with('>') && token.ends_with(']') {
        // Mermaid: `A>label]` where `A` is the id extracted from the caller
        // context. Since we see the whole token here (id prepended), we must
        // find where the `>` starts.
        if let Some(pos) = token.find('>') {
            let id = token[..pos].trim().to_string();
            if !id.is_empty() {
                let inner = token[pos + 1..token.len() - 1].trim().to_string();
                let label = normalize_label(&inner);
                return Some(Node::new(id, label, NodeShape::Asymmetric));
            }
        }
    }

    // Find the first bracket/brace/paren/angle character to split id from shape.
    let shape_start = token.find(['[', '{', '(', '>']);

    let (id, label, shape) = if let Some(pos) = shape_start {
        let id = token[..pos].trim().to_string();
        let rest = &token[pos..];

        // --- Most specific first ---

        // Triple paren: A(((text))) → DoubleCircle
        if rest.starts_with("(((") && rest.ends_with(")))") {
            let inner = rest[3..rest.len() - 3].trim().to_string();
            (id, inner, NodeShape::DoubleCircle)
        }
        // Stadium: A([text])
        else if rest.starts_with("([") && rest.ends_with("])") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Stadium)
        }
        // Cylinder: A[(text)]
        else if rest.starts_with("[(") && rest.ends_with(")]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Cylinder)
        }
        // Subroutine: A[[text]]
        else if rest.starts_with("[[") && rest.ends_with("]]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Subroutine)
        }
        // Parallelogram (lean-right): A[/text/]
        else if rest.starts_with("[/") && rest.ends_with("/]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Parallelogram)
        }
        // Trapezoid (wider top): A[/text\]
        else if rest.starts_with("[/") && rest.ends_with("\\]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Trapezoid)
        }
        // Parallelogram (lean-left / backslash): A[\text\]
        else if rest.starts_with("[\\") && rest.ends_with("\\]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::ParallelogramBackslash)
        }
        // Inverted trapezoid (wider bottom): A[\text/]
        else if rest.starts_with("[\\") && rest.ends_with("/]") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::TrapezoidInverted)
        }
        // Hexagon: A{{text}}
        else if rest.starts_with("{{") && rest.ends_with("}}") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Hexagon)
        }
        // Double paren: A((text)) → Circle
        else if rest.starts_with("((") && rest.ends_with("))") {
            let inner = rest[2..rest.len() - 2].trim().to_string();
            (id, inner, NodeShape::Circle)
        }
        // Diamond: A{text}
        else if rest.starts_with('{') && rest.ends_with('}') {
            let inner = rest[1..rest.len() - 1].trim().to_string();
            (id, inner, NodeShape::Diamond)
        }
        // Rectangle: A[text]
        else if rest.starts_with('[') && rest.ends_with(']') {
            let inner = rest[1..rest.len() - 1].trim().to_string();
            (id, inner, NodeShape::Rectangle)
        }
        // Rounded: A(text)
        else if rest.starts_with('(') && rest.ends_with(')') {
            let inner = rest[1..rest.len() - 1].trim().to_string();
            (id, inner, NodeShape::Rounded)
        }
        // Asymmetric: A>text]
        else if rest.starts_with('>') && rest.ends_with(']') {
            let inner = rest[1..rest.len() - 1].trim().to_string();
            (id, inner, NodeShape::Asymmetric)
        } else {
            // Unrecognised bracket pattern — treat entire token as bare ID.
            let id = token.to_string();
            (id.clone(), id, NodeShape::Rectangle)
        }
    } else {
        // Bare ID
        (token.to_string(), token.to_string(), NodeShape::Rectangle)
    };

    if id.is_empty() {
        return None;
    }

    let label = normalize_label(&label);
    Some(Node::new(id, label, shape))
}

/// Soft-wrap threshold for a single label line. Lines longer than this get
/// wrapped at the nearest comma or space before the threshold, producing
/// additional line breaks. Lines without any break point remain intact so
/// a long identifier (`a_very_long_ident_without_separators`) isn't mangled.
const LABEL_WRAP_THRESHOLD: usize = 40;

/// Normalise a label for multi-row rendering.
///
/// Two transformations are applied, in order:
///
/// 1. HTML line-break tags (`<br/>`, `<br>`, `<br />`, case-insensitive on the
///    tag name) are replaced with `\n`. Mermaid uses these as explicit line
///    breaks inside node labels and we honor them.
/// 2. Any resulting line wider than [`LABEL_WRAP_THRESHOLD`] terminal cells is
///    soft-wrapped at the last comma or space at or before the threshold.
///    Words without any wrap-friendly break stay on a single line.
///
/// The renderer interprets `\n` as a line-break and draws each segment on its
/// own row inside the node box, widening the box vertically instead of
/// horizontally.
fn normalize_label(s: &str) -> String {
    // Step 1: replace HTML <br> variants with `\n`. Lower-case first; the
    // upper-case variants are the only other common spellings on the wild.
    let with_breaks = s
        .replace("<br/>", "\n")
        .replace("<br>", "\n")
        .replace("<br />", "\n")
        .replace("<BR/>", "\n")
        .replace("<BR>", "\n")
        .replace("<BR />", "\n");

    // Step 2: soft-wrap each resulting line.
    let mut out = String::with_capacity(with_breaks.len());
    let mut first = true;
    for line in with_breaks.lines() {
        if !first {
            out.push('\n');
        }
        first = false;
        soft_wrap_into(line, &mut out);
    }
    out
}

/// Append `line` to `out`, inserting `\n` breaks at word boundaries so that
/// no resulting row exceeds [`LABEL_WRAP_THRESHOLD`] columns.
///
/// The break character (comma or space) stays on the head side of the split —
/// a trailing space gets trimmed, a trailing comma is preserved so the user's
/// list formatting is kept.
fn soft_wrap_into(line: &str, out: &mut String) {
    use unicode_width::{UnicodeWidthChar, UnicodeWidthStr};

    if UnicodeWidthStr::width(line) <= LABEL_WRAP_THRESHOLD {
        out.push_str(line);
        return;
    }

    // Walk chars, tracking cumulative width and the byte index of the last
    // break-friendly char (comma or space) seen within the budget.
    let mut cum_w = 0usize;
    let mut last_break: Option<usize> = None;
    for (i, ch) in line.char_indices() {
        cum_w += UnicodeWidthChar::width(ch).unwrap_or(0);
        if cum_w > LABEL_WRAP_THRESHOLD {
            break;
        }
        if ch == ',' || ch == ' ' {
            last_break = Some(i);
        }
    }

    let Some(break_at) = last_break else {
        // No break point within the budget — emit the line as-is rather than
        // mangling a single long word.
        out.push_str(line);
        return;
    };

    // `split_at(break_at + 1)`: `break_at` is the byte index of the break
    // character; `+ 1` includes the break char (all break chars are ASCII,
    // so their UTF-8 length is 1) in the head.
    let (head, tail) = line.split_at(break_at + 1);
    let head = head.trim_end();
    let tail = tail.trim_start();
    out.push_str(head);
    out.push('\n');
    soft_wrap_into(tail, out);
}

// ---------------------------------------------------------------------------
// Style directive parsing
// ---------------------------------------------------------------------------

// ---------------------------------------------------------------------------
// Click directive parsing
// ---------------------------------------------------------------------------

/// Parse a `click` directive and record it on `graph.click_targets`.
///
/// Recognised syntaxes:
///
/// | Source                                     | Result                             |
/// |--------------------------------------------|------------------------------------|
/// | `click NodeId "url"`                       | URL, no tooltip                    |
/// | `click NodeId "url" "tooltip"`             | URL + tooltip                      |
/// | `click NodeId href "url"`                  | URL (alternate `href` keyword)     |
/// | `click NodeId href "url" "tooltip"`        | URL + tooltip                      |
/// | `click NodeId callbackFn`                  | silently ignored (JS callback)     |
/// | `click NodeId callbackFn "tooltip"`        | silently ignored                   |
///
/// The JS-callback forms have no renderable equivalent in a text terminal, so
/// they are silently dropped — matching Mermaid's own silent-fail behaviour
/// when a callback is undefined in the browser.
///
/// # Arguments
///
/// * `stmt`  — the raw statement string (already trimmed, `"click …"`)
/// * `graph` — the [`Graph`] to insert the result into
pub(crate) fn parse_click_directive(stmt: &str, graph: &mut Graph) {
    // Strip the "click " keyword prefix. The statement is guaranteed to start
    // with "click" because the parser dispatched on that first word.
    let rest = stmt.strip_prefix("click").unwrap_or("").trim();
    if rest.is_empty() {
        return;
    }

    // Extract the node ID — it's the first whitespace-delimited token.
    let mut parts = rest.splitn(2, |c: char| c.is_whitespace());
    let node_id = parts.next().unwrap_or("").trim();
    if node_id.is_empty() {
        return;
    }
    let after_id = parts.next().unwrap_or("").trim();

    // Check for the optional `href` keyword (alternate Mermaid syntax).
    // `click NodeId href "url" ["tooltip"]`
    let url_part = if let Some(stripped) = after_id.strip_prefix("href") {
        stripped.trim()
    } else {
        after_id
    };

    // The URL must start with a double-quote. If it doesn't, this is a
    // JS-callback form — silently ignore.
    let Some(url_rest) = url_part.strip_prefix('"') else {
        return;
    };

    // Find the closing double-quote for the URL.
    let Some(url_end) = url_rest.find('"') else {
        return;
    };
    let url = url_rest[..url_end].to_string();

    // Everything after the closing URL quote may contain an optional tooltip.
    let after_url = url_rest[url_end + 1..].trim();
    let tooltip = parse_quoted_string(after_url);

    graph
        .click_targets
        .insert(node_id.to_string(), ClickTarget { url, tooltip });
}

/// Extract a double-quoted string from the start of `s`, if present.
///
/// Returns `Some(inner)` when `s` starts with `"…"`, `None` otherwise.
fn parse_quoted_string(s: &str) -> Option<String> {
    let inner = s.strip_prefix('"')?;
    let end = inner.find('"')?;
    Some(inner[..end].to_string())
}

// ---------------------------------------------------------------------------
// Tests
// ---------------------------------------------------------------------------

#[cfg(test)]
mod tests {
    use super::*;
    use crate::types::{EdgeEndpoint, EdgeStyle, NodeShape, Rgb};

    #[test]
    fn parse_simple_lr() {
        let g = parse("graph LR\nA-->B-->C").unwrap();
        assert_eq!(g.direction, Direction::LeftToRight);
        assert!(g.has_node("A"));
        assert!(g.has_node("B"));
        assert!(g.has_node("C"));
        assert_eq!(g.edges.len(), 2);
    }

    #[test]
    fn parse_semicolons() {
        let g = parse("graph LR; A-->B; B-->C").unwrap();
        assert_eq!(g.edges.len(), 2);
    }

    #[test]
    fn parse_labeled_nodes() {
        let g = parse("graph LR\nA[Start] --> B[End]").unwrap();
        assert_eq!(g.node("A").unwrap().label, "Start");
        assert_eq!(g.node("B").unwrap().label, "End");
    }

    #[test]
    fn parse_diamond_node() {
        let g = parse("graph LR\nA{Decision}").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Diamond);
        assert_eq!(g.node("A").unwrap().label, "Decision");
    }

    #[test]
    fn parse_circle_node() {
        let g = parse("graph LR\nA((Circle))").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Circle);
    }

    #[test]
    fn circle_syntax_does_not_leak_parens_into_label() {
        // Regression test for Bug 1: the `((` / `))` delimiters in Mermaid's
        // circle syntax are structural — they must NOT appear in the rendered
        // label.  Previously the parser extracted "( Circle )" (with the inner
        // paren pair included), causing the box to display as "│( Circle )│".
        let g = parse("graph LR\nB((Circle))").unwrap();
        let node = g.node("B").unwrap();
        assert_eq!(node.shape, NodeShape::Circle);
        // The label must be "Circle", not "( Circle )" or "(Circle)".
        assert_eq!(node.label, "Circle");
        assert!(
            !node.label.contains('('),
            "label must not contain '(': {:?}",
            node.label
        );
        assert!(
            !node.label.contains(')'),
            "label must not contain ')': {:?}",
            node.label
        );
    }

    #[test]
    fn parse_rounded_node() {
        let g = parse("graph LR\nA(Rounded)").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Rounded);
    }

    #[test]
    fn parse_edge_label_pipe() {
        let g = parse("graph LR\nA -->|yes| B").unwrap();
        assert_eq!(g.edges[0].label.as_deref(), Some("yes"));
    }

    #[test]
    fn parse_edge_label_dash() {
        let g = parse("graph LR\nA -- hello --> B").unwrap();
        assert_eq!(g.edges[0].label.as_deref(), Some("hello"));
    }

    #[test]
    fn parse_flowchart_keyword() {
        let g = parse("flowchart TD\nA-->B").unwrap();
        assert_eq!(g.direction, Direction::TopToBottom);
    }

    #[test]
    fn bad_direction_returns_error() {
        assert!(parse("graph XY\nA-->B").is_err());
    }

    #[test]
    fn no_header_returns_error() {
        assert!(parse("A-->B").is_err());
    }

    #[test]
    fn parse_subgraph_basic() {
        let src = "graph LR\nsubgraph Supervisor\nF[Factory] --> W[Worker]\nend";
        let g = parse(src).unwrap();
        assert!(g.has_node("F"), "missing F");
        assert!(g.has_node("W"), "missing W");
        assert_eq!(g.subgraphs.len(), 1);
        assert_eq!(g.subgraphs[0].id, "Supervisor");
        assert_eq!(g.subgraphs[0].label, "Supervisor");
        // Both nodes should be members of the Supervisor subgraph.
        assert!(g.subgraphs[0].node_ids.contains(&"F".to_string()));
        assert!(g.subgraphs[0].node_ids.contains(&"W".to_string()));
    }

    #[test]
    fn parse_subgraph_with_direction() {
        let src = "graph LR\nsubgraph S\ndirection TB\nA-->B\nend";
        let g = parse(src).unwrap();
        assert_eq!(g.subgraphs[0].direction, Some(Direction::TopToBottom));
    }

    #[test]
    fn parse_nested_subgraphs() {
        let src = "graph TD\nsubgraph Outer\nsubgraph Inner\nA[A]\nend\nB[B]\nend";
        let g = parse(src).unwrap();
        // Both subgraphs should be registered.
        assert!(g.find_subgraph("Outer").is_some());
        assert!(g.find_subgraph("Inner").is_some());
        // Inner should be a child of Outer.
        let outer = g.find_subgraph("Outer").unwrap();
        assert!(outer.subgraph_ids.contains(&"Inner".to_string()));
        // A is in Inner, B is in Outer.
        let inner = g.find_subgraph("Inner").unwrap();
        assert!(inner.node_ids.contains(&"A".to_string()));
        assert!(outer.node_ids.contains(&"B".to_string()));
    }

    #[test]
    fn parse_subgraph_edge_crossing_boundary() {
        let src = "graph LR\nsubgraph S\nF[Factory] --> W[Worker]\nend\nW --> HB[Heartbeat]";
        let g = parse(src).unwrap();
        assert!(g.has_node("F"));
        assert!(g.has_node("W"));
        assert!(g.has_node("HB"));
        // W → HB edge should exist (crosses boundary).
        assert!(g.edges.iter().any(|e| e.from == "W" && e.to == "HB"));
        // HB should NOT be in subgraph S.
        let s = g.find_subgraph("S").unwrap();
        assert!(!s.node_ids.contains(&"HB".to_string()));
    }

    #[test]
    fn node_to_subgraph_map() {
        let src = "graph LR\nsubgraph S\nA-->B\nend\nC-->D";
        let g = parse(src).unwrap();
        let map = g.node_to_subgraph();
        assert_eq!(map.get("A").map(String::as_str), Some("S"));
        assert_eq!(map.get("B").map(String::as_str), Some("S"));
        assert!(!map.contains_key("C"));
        assert!(!map.contains_key("D"));
    }

    // ---- New node shape parser tests ------------------------------------

    #[test]
    fn parse_stadium_node() {
        let g = parse("graph LR\nA([Stadium])").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Stadium);
        assert_eq!(g.node("A").unwrap().label, "Stadium");
    }

    #[test]
    fn parse_subroutine_node() {
        let g = parse("graph LR\nA[[Sub]]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Subroutine);
        assert_eq!(g.node("A").unwrap().label, "Sub");
    }

    #[test]
    fn parse_cylinder_node() {
        let g = parse("graph LR\nA[(DB)]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Cylinder);
        assert_eq!(g.node("A").unwrap().label, "DB");
    }

    #[test]
    fn parse_hexagon_node() {
        let g = parse("graph LR\nA{{Hex}}").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Hexagon);
        assert_eq!(g.node("A").unwrap().label, "Hex");
    }

    #[test]
    fn parse_asymmetric_node() {
        let g = parse("graph LR\nA>Flag]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Asymmetric);
        assert_eq!(g.node("A").unwrap().label, "Flag");
    }

    #[test]
    fn parse_parallelogram_node() {
        let g = parse("graph LR\nA[/Lean/]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Parallelogram);
        assert_eq!(g.node("A").unwrap().label, "Lean");
    }

    #[test]
    fn parse_trapezoid_node() {
        let g = parse("graph LR\nA[/Trap\\]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Trapezoid);
        assert_eq!(g.node("A").unwrap().label, "Trap");
    }

    #[test]
    fn parse_parallelogram_backslash_node() {
        let g = parse("graph LR\nA[\\LeanLeft\\]").unwrap();
        assert_eq!(
            g.node("A").unwrap().shape,
            NodeShape::ParallelogramBackslash
        );
        assert_eq!(g.node("A").unwrap().label, "LeanLeft");
    }

    #[test]
    fn parse_trapezoid_inverted_node() {
        let g = parse("graph LR\nA[\\InvTrap/]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::TrapezoidInverted);
        assert_eq!(g.node("A").unwrap().label, "InvTrap");
    }

    #[test]
    fn parse_double_circle_node() {
        let g = parse("graph LR\nA(((Dbl)))").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::DoubleCircle);
        assert_eq!(g.node("A").unwrap().label, "Dbl");
    }

    // Disambiguation: (( before ((( — triple paren wins.
    #[test]
    fn triple_paren_beats_double_paren() {
        let g = parse("graph LR\nA(((X)))").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::DoubleCircle);
    }

    // Disambiguation: [[ before [ — double bracket wins.
    #[test]
    fn double_bracket_beats_single_bracket() {
        let g = parse("graph LR\nA[[Y]]").unwrap();
        assert_eq!(g.node("A").unwrap().shape, NodeShape::Subroutine);
    }

    // ---- Edge style parser tests ----------------------------------------

    #[test]
    fn parse_dotted_edge_style() {
        let g = parse("graph LR\nA-.->B").unwrap();
        assert_eq!(g.edges[0].style, EdgeStyle::Dotted);
        assert_eq!(g.edges[0].end, EdgeEndpoint::Arrow);
    }

    #[test]
    fn parse_thick_edge_style() {
        let g = parse("graph LR\nA==>B").unwrap();
        assert_eq!(g.edges[0].style, EdgeStyle::Thick);
        assert_eq!(g.edges[0].end, EdgeEndpoint::Arrow);
    }

    #[test]
    fn parse_plain_line_no_arrow() {
        let g = parse("graph LR\nA---B").unwrap();
        assert_eq!(g.edges[0].style, EdgeStyle::Solid);
        assert_eq!(g.edges[0].end, EdgeEndpoint::None);
        assert_eq!(g.edges[0].start, EdgeEndpoint::None);
    }

    #[test]
    fn parse_bidirectional_edge() {
        let g = parse("graph LR\nA<-->B").unwrap();
        assert_eq!(g.edges[0].style, EdgeStyle::Solid);
        assert_eq!(g.edges[0].start, EdgeEndpoint::Arrow);
        assert_eq!(g.edges[0].end, EdgeEndpoint::Arrow);
    }

    #[test]
    fn parse_circle_endpoint() {
        let g = parse("graph LR\nA--oB").unwrap();
        assert_eq!(g.edges[0].end, EdgeEndpoint::Circle);
    }

    #[test]
    fn parse_cross_endpoint() {
        let g = parse("graph LR\nA--xB").unwrap();
        assert_eq!(g.edges[0].end, EdgeEndpoint::Cross);
    }

    #[test]
    fn parse_style_directive_records_colors() {
        let src = "graph LR\nA-->B\nstyle A fill:#336,stroke:#fff,color:#fff";
        let g = parse(src).unwrap();
        let style = g.node_styles.get("A").copied().unwrap();
        assert_eq!(style.fill, Some(Rgb(0x33, 0x33, 0x66)));
        assert_eq!(style.stroke, Some(Rgb(0xff, 0xff, 0xff)));
        assert_eq!(style.color, Some(Rgb(0xff, 0xff, 0xff)));
    }

    #[test]
    fn parse_style_directive_ignores_unknown_keys_and_bad_hex() {
        let src = "graph LR\nA\nstyle A fill:#zzz,foo:bar,stroke:#000";
        let g = parse(src).unwrap();
        let style = g.node_styles.get("A").copied().unwrap();
        assert_eq!(style.fill, None);
        assert_eq!(style.stroke, Some(Rgb(0, 0, 0)));
    }

    #[test]
    fn parse_link_style_directive_per_index() {
        let src =
            "graph LR\nA-->B\nA-->C\nlinkStyle 0 stroke:#f00\nlinkStyle 1 stroke:#0f0,color:#fff";
        let g = parse(src).unwrap();
        let e0 = g.edge_styles.get(&0).copied().unwrap();
        assert_eq!(e0.stroke, Some(Rgb(0xff, 0, 0)));
        assert!(e0.color.is_none());
        let e1 = g.edge_styles.get(&1).copied().unwrap();
        assert_eq!(e1.stroke, Some(0).map(|_| Rgb(0, 0xff, 0)));
        assert_eq!(e1.color, Some(Rgb(0xff, 0xff, 0xff)));
    }

    #[test]
    fn parse_link_style_default_applies_to_all() {
        let src = "graph LR\nA-->B\nA-->C\nlinkStyle default stroke:#abc";
        let g = parse(src).unwrap();
        assert_eq!(
            g.edge_styles.get(&0).and_then(|e| e.stroke),
            Some(Rgb(0xaa, 0xbb, 0xcc))
        );
        assert_eq!(
            g.edge_styles.get(&1).and_then(|e| e.stroke),
            Some(Rgb(0xaa, 0xbb, 0xcc))
        );
    }

    // ---- classDef / class / ::: ---------------------------------------

    #[test]
    fn class_def_directive_records_palette() {
        let src = "graph LR\nA-->B\nclassDef cache fill:#234,stroke:#9cf,color:#fff";
        let g = parse(src).unwrap();
        let style = g.class_defs.get("cache").copied().unwrap();
        assert_eq!(style.fill, Some(Rgb(0x22, 0x33, 0x44)));
        assert_eq!(style.stroke, Some(Rgb(0x99, 0xcc, 0xff)));
        assert_eq!(style.color, Some(Rgb(0xff, 0xff, 0xff)));
    }

    #[test]
    fn class_directive_applies_palette_to_each_id() {
        let src = "graph LR\nA-->B-->C\nclassDef cache fill:#234\nclass A,B cache";
        let g = parse(src).unwrap();
        assert_eq!(
            g.node_styles.get("A").and_then(|s| s.fill),
            Some(Rgb(0x22, 0x33, 0x44))
        );
        assert_eq!(
            g.node_styles.get("B").and_then(|s| s.fill),
            Some(Rgb(0x22, 0x33, 0x44))
        );
        assert!(!g.node_styles.contains_key("C"), "C wasn't in `class` list");
    }

    #[test]
    fn triple_colon_shorthand_inline_on_node() {
        let src = "graph LR\nA[Start]:::cache --> B[End]\nclassDef cache fill:#234";
        let g = parse(src).unwrap();
        assert_eq!(
            g.node_styles.get("A").and_then(|s| s.fill),
            Some(Rgb(0x22, 0x33, 0x44))
        );
        // Label and shape parsing unaffected by the modifier suffix.
        assert_eq!(g.node("A").unwrap().label, "Start");
    }

    #[test]
    fn triple_colon_chained_classes_stack() {
        let src = "graph LR\nA:::base:::overlay --> B
classDef base fill:#111,stroke:#222
classDef overlay stroke:#999,color:#fff";
        let g = parse(src).unwrap();
        let s = g.node_styles.get("A").copied().unwrap();
        assert_eq!(s.fill, Some(Rgb(0x11, 0x11, 0x11))); // from base
        // overlay applied later (right of base in source) wins on stroke
        assert_eq!(s.stroke, Some(Rgb(0x99, 0x99, 0x99)));
        assert_eq!(s.color, Some(Rgb(0xff, 0xff, 0xff))); // from overlay
    }

    #[test]
    fn forward_reference_to_class_def_resolves_at_end_of_parse() {
        // `class A foo` appears BEFORE `classDef foo …` — must still
        // resolve.
        let src = "graph LR\nA-->B\nclass A foo\nclassDef foo fill:#abc";
        let g = parse(src).unwrap();
        assert_eq!(
            g.node_styles.get("A").and_then(|s| s.fill),
            Some(Rgb(0xaa, 0xbb, 0xcc))
        );
    }

    #[test]
    fn unknown_class_name_silently_dropped() {
        // Reference to an undefined class — match Mermaid's
        // best-effort semantics: no error, no application.
        let src = "graph LR\nA-->B\nclass A undefined";
        let g = parse(src).unwrap();
        assert!(!g.node_styles.contains_key("A"));
    }

    // ---- Inline-quoted edge label tests (B1/B2 regression) ----------------

    /// B1: dashed edge with inline-quoted label.
    ///
    /// `A -. "label" .-> B` must produce a single dotted edge labelled "label",
    /// not a ghost node containing the literal text `-. "label"`.
    #[test]
    fn inline_quoted_label_dashed_edge() {
        let g = parse("graph LR\nA -. \"my label\" .-> B").unwrap();
        // Must be exactly one edge — not zero (label consumed as node) and not two.
        assert_eq!(g.edges.len(), 1, "expected 1 edge, got {}", g.edges.len());
        let e = &g.edges[0];
        assert_eq!(e.from, "A");
        assert_eq!(e.to, "B");
        assert_eq!(e.style, EdgeStyle::Dotted);
        assert_eq!(e.end, EdgeEndpoint::Arrow);
        assert_eq!(
            e.label.as_deref(),
            Some("my label"),
            "label should be extracted without surrounding quotes"
        );
    }

    /// B2: thick edge with inline-quoted label.
    ///
    /// `A == "label" ==> B` must produce a single thick edge labelled "label".
    #[test]
    fn inline_quoted_label_thick_edge() {
        let g = parse("graph LR\nA == \"my label\" ==> B").unwrap();
        assert_eq!(g.edges.len(), 1, "expected 1 edge, got {}", g.edges.len());
        let e = &g.edges[0];
        assert_eq!(e.from, "A");
        assert_eq!(e.to, "B");
        assert_eq!(e.style, EdgeStyle::Thick);
        assert_eq!(e.end, EdgeEndpoint::Arrow);
        assert_eq!(e.label.as_deref(), Some("my label"));
    }

    /// Solid inline-quoted form (`-- "label" -->`) — this worked before the
    /// fix via `try_consume_labeled_dash_arrow`. Verify it still works and
    /// that quotes are stripped from the label.
    #[test]
    fn inline_quoted_label_solid_edge() {
        let g = parse("graph LR\nA -- \"my label\" --> B").unwrap();
        assert_eq!(g.edges.len(), 1, "expected 1 edge, got {}", g.edges.len());
        let e = &g.edges[0];
        assert_eq!(e.from, "A");
        assert_eq!(e.to, "B");
        assert_eq!(e.style, EdgeStyle::Solid);
        assert_eq!(e.end, EdgeEndpoint::Arrow);
        assert_eq!(e.label.as_deref(), Some("my label"));
    }

    /// Pipe-label forms must continue to work identically after the fix.
    /// Regression guard: B1/B2 fix must not break the pre-existing pipe path.
    #[test]
    fn pipe_label_dashed_and_thick_unaffected() {
        let g = parse("graph LR\nA -.->|pipe dashed| B\nA ==>|pipe thick| C").unwrap();
        assert_eq!(g.edges.len(), 2);
        let dashed = g.edges.iter().find(|e| e.to == "B").unwrap();
        assert_eq!(dashed.style, EdgeStyle::Dotted);
        assert_eq!(dashed.label.as_deref(), Some("pipe dashed"));
        let thick = g.edges.iter().find(|e| e.to == "C").unwrap();
        assert_eq!(thick.style, EdgeStyle::Thick);
        assert_eq!(thick.label.as_deref(), Some("pipe thick"));
    }

    /// Inline-quoted labels with spaces in the label text.
    #[test]
    fn inline_quoted_label_with_spaces() {
        let g = parse("graph LR\nA -. \"sends event to\" .-> B").unwrap();
        assert_eq!(g.edges.len(), 1);
        assert_eq!(g.edges[0].label.as_deref(), Some("sends event to"));
    }

    /// Inline-quoted label in a chain: `A -. "x" .-> B ==> C`.
    /// Both edges must parse correctly within the same chain statement.
    #[test]
    fn inline_quoted_label_in_chain() {
        let g = parse("graph LR\nA -. \"step one\" .-> B ==> C").unwrap();
        assert_eq!(g.edges.len(), 2, "expected 2 edges in chain");
        let ab = g
            .edges
            .iter()
            .find(|e| e.from == "A" && e.to == "B")
            .unwrap();
        assert_eq!(ab.style, EdgeStyle::Dotted);
        assert_eq!(ab.label.as_deref(), Some("step one"));
        let bc = g
            .edges
            .iter()
            .find(|e| e.from == "B" && e.to == "C")
            .unwrap();
        assert_eq!(bc.style, EdgeStyle::Thick);
        assert!(bc.label.is_none());
    }

    #[test]
    fn style_directive_overrides_class_for_same_id() {
        // Per-id `style` lands AFTER class application (style is
        // processed inline; class is resolved at end-of-parse — but
        // the resolver merges class as overlay over base, so the
        // earlier `style` survives unless the class also sets that
        // attribute).
        let src = "graph LR
A-->B
style A fill:#aaa
classDef cls fill:#bbb,stroke:#ccc
class A cls";
        let g = parse(src).unwrap();
        let s = g.node_styles.get("A").copied().unwrap();
        // class fill overlays the style fill (overlay wins per merge_node_style).
        assert_eq!(s.fill, Some(Rgb(0xbb, 0xbb, 0xbb)));
        // class also supplies stroke that wasn't in the style — stacks.
        assert_eq!(s.stroke, Some(Rgb(0xcc, 0xcc, 0xcc)));
    }

    // ---- click directive parser tests ----------------------------------------

    #[test]
    fn click_basic_url() {
        let src = "graph LR\nA-->B\nclick A \"https://example.com\"";
        let g = parse(src).unwrap();
        let ct = g.click_targets.get("A").expect("click target for A");
        assert_eq!(ct.url, "https://example.com");
        assert!(ct.tooltip.is_none());
        // B has no click directive.
        assert!(!g.click_targets.contains_key("B"));
    }

    #[test]
    fn click_with_tooltip() {
        let src = "graph LR\nA-->B\nclick A \"https://example.com\" \"Go to example\"";
        let g = parse(src).unwrap();
        let ct = g.click_targets.get("A").expect("click target for A");
        assert_eq!(ct.url, "https://example.com");
        assert_eq!(ct.tooltip.as_deref(), Some("Go to example"));
    }

    #[test]
    fn click_href_keyword_form() {
        // `click NodeId href "url"` is an alternate Mermaid syntax.
        let src = "graph LR\nA-->B\nclick A href \"https://example.com\"";
        let g = parse(src).unwrap();
        let ct = g.click_targets.get("A").expect("click target for A");
        assert_eq!(ct.url, "https://example.com");
        assert!(ct.tooltip.is_none());
    }

    #[test]
    fn click_href_keyword_with_tooltip() {
        let src = "graph LR\nA-->B\nclick A href \"https://example.com\" \"tooltip text\"";
        let g = parse(src).unwrap();
        let ct = g.click_targets.get("A").expect("click target");
        assert_eq!(ct.url, "https://example.com");
        assert_eq!(ct.tooltip.as_deref(), Some("tooltip text"));
    }

    #[test]
    fn click_js_callback_silently_ignored() {
        // JS callback forms have no URL quote — should produce zero click targets.
        let src = "graph LR\nA-->B\nclick A myCallback\nclick B myCallback \"tooltip\"";
        let g = parse(src).unwrap();
        assert!(
            g.click_targets.is_empty(),
            "JS callbacks must be silently ignored"
        );
    }

    #[test]
    fn click_does_not_affect_nodes_without_directive() {
        let src = "graph LR\nA-->B-->C\nclick B \"https://b.example\"";
        let g = parse(src).unwrap();
        assert!(!g.click_targets.contains_key("A"));
        assert!(g.click_targets.contains_key("B"));
        assert!(!g.click_targets.contains_key("C"));
    }

    #[test]
    fn click_multiple_nodes() {
        let src = "graph LR\nA-->B\nclick A \"https://a.example\"\nclick B \"https://b.example\"";
        let g = parse(src).unwrap();
        assert_eq!(
            g.click_targets.get("A").map(|c| c.url.as_str()),
            Some("https://a.example")
        );
        assert_eq!(
            g.click_targets.get("B").map(|c| c.url.as_str()),
            Some("https://b.example")
        );
    }

    #[test]
    fn ampersand_fanout_expands_to_multiple_edges() {
        let src = "flowchart LR\n    P1a & P1b --> K1";
        let g = parse(src).expect("parse must succeed");
        // Trap-check: K1 node must exist.
        assert!(g.nodes.iter().any(|n| n.id == "K1"), "K1 missing");
        // Acceptance: TWO edges exist (P1a → K1 and P1b → K1).
        let edges_to_k1 = g.edges.iter().filter(|e| e.to == "K1").count();
        assert_eq!(
            edges_to_k1, 2,
            "expected 2 edges into K1 from `P1a & P1b --> K1`; found {edges_to_k1}"
        );
        // Both nodes P1a and P1b must be parsed as separate nodes.
        assert!(g.nodes.iter().any(|n| n.id == "P1a"), "P1a missing");
        assert!(g.nodes.iter().any(|n| n.id == "P1b"), "P1b missing");
    }

    #[test]
    fn ampersand_fanout_cross_products_both_sides() {
        let src = "flowchart LR\n    A & B --> C & D";
        let g = parse(src).expect("parse must succeed");
        let edges: Vec<(&str, &str)> = g
            .edges
            .iter()
            .map(|e| (e.from.as_str(), e.to.as_str()))
            .collect();
        assert_eq!(edges, vec![("A", "C"), ("A", "D"), ("B", "C"), ("B", "D")]);
    }

    #[test]
    fn ampersand_fanout_expands_each_chain_step() {
        let src = "flowchart LR\n    A --> B & C --> D";
        let g = parse(src).expect("parse must succeed");
        let edges: Vec<(&str, &str)> = g
            .edges
            .iter()
            .map(|e| (e.from.as_str(), e.to.as_str()))
            .collect();
        assert_eq!(edges, vec![("A", "B"), ("A", "C"), ("B", "D"), ("C", "D")]);
    }

    #[test]
    fn ampersand_fanout_keeps_node_shapes_classes_and_labels() {
        let src = "flowchart LR
    A[one]:::left & B(two):::right --> C:::sink
    classDef left fill:#123
    classDef right fill:#456
    classDef sink fill:#789";
        let g = parse(src).expect("parse must succeed");
        assert_eq!(g.node("A").unwrap().label, "one");
        assert_eq!(g.node("B").unwrap().shape, NodeShape::Rounded);
        assert_eq!(
            g.node_styles.get("A").and_then(|s| s.fill),
            Some(Rgb(0x11, 0x22, 0x33))
        );
        assert_eq!(
            g.node_styles.get("B").and_then(|s| s.fill),
            Some(Rgb(0x44, 0x55, 0x66))
        );
        assert_eq!(
            g.node_styles.get("C").and_then(|s| s.fill),
            Some(Rgb(0x77, 0x88, 0x99))
        );
    }

    #[test]
    fn ampersand_inside_node_label_does_not_split_group() {
        let src = "flowchart LR\n    A[Research & Development] --> B";
        let g = parse(src).expect("parse must succeed");
        assert_eq!(g.edges.len(), 1);
        assert_eq!(g.node("A").unwrap().label, "Research & Development");
        assert_eq!(g.node("B").unwrap().label, "B");
    }

    #[test]
    fn inline_dotted_label_parses_as_edge_label() {
        let src = "flowchart TD\n    A[start] -.cancels.-> B[Sleep]";
        let g = parse(src).expect("parse must succeed");
        // Trap-check: BOTH nodes must exist as separate entities.
        assert!(g.nodes.iter().any(|n| n.id == "A"), "A missing");
        assert!(g.nodes.iter().any(|n| n.id == "B"), "B missing");
        // Acceptance: an edge from A to B with label "cancels".
        let edge = g
            .edges
            .iter()
            .find(|e| e.from == "A" && e.to == "B")
            .expect("edge A → B not parsed");
        assert_eq!(
            edge.label.as_deref(),
            Some("cancels"),
            "edge label should be 'cancels', got {:?}",
            edge.label
        );
    }

    #[test]
    fn inline_thick_label_parses_as_edge_label() {
        let src = "flowchart TD\n    A ==promotes==> B";
        let g = parse(src).expect("parse must succeed");
        let edge = g
            .edges
            .iter()
            .find(|e| e.from == "A" && e.to == "B")
            .expect("edge A → B not parsed");
        assert_eq!(edge.style, EdgeStyle::Thick);
        assert_eq!(edge.label.as_deref(), Some("promotes"));
    }

    #[test]
    fn inline_compact_labels_preserve_spacing_and_shapes() {
        let src = r#"flowchart TD
    Stop(["createOutboxRunner.stop()"]) -. cancels retry .-> Sleep
    Sleep == wakes worker ==> Wake"#;
        let g = parse(src).expect("parse must succeed");
        let dotted = g
            .edges
            .iter()
            .find(|e| e.from == "Stop" && e.to == "Sleep")
            .expect("dotted edge not parsed");
        assert_eq!(dotted.style, EdgeStyle::Dotted);
        assert_eq!(dotted.label.as_deref(), Some("cancels retry"));
        let thick = g
            .edges
            .iter()
            .find(|e| e.from == "Sleep" && e.to == "Wake")
            .expect("thick edge not parsed");
        assert_eq!(thick.style, EdgeStyle::Thick);
        assert_eq!(thick.label.as_deref(), Some("wakes worker"));
    }
}