fd_core/

emitter.rs

//! Emitter: SceneGraph → FD text format.
//!
//! Produces minimal, token-efficient output that round-trips through the parser.

use crate::id::NodeId;
use crate::model::*;
use petgraph::graph::NodeIndex;
use std::fmt::Write;

/// Emit a `SceneGraph` as an FD text document.
#[must_use]
pub fn emit_document(graph: &SceneGraph) -> String {
    let mut out = String::with_capacity(1024);

    // Count sections to decide if separators add value
    let has_imports = !graph.imports.is_empty();
    let has_styles = !graph.styles.is_empty();
    let children = graph.children(graph.root);
    let has_constraints = graph.graph.node_indices().any(|idx| {
        graph.graph[idx]
            .constraints
            .iter()
            .any(|c| !matches!(c, Constraint::Position { .. }))
    });
    let has_edges = !graph.edges.is_empty();
    let section_count =
        has_imports as u8 + has_styles as u8 + has_constraints as u8 + has_edges as u8;
    let use_separators = section_count >= 2;

    // Emit imports
    for import in &graph.imports {
        let _ = writeln!(out, "import \"{}\" as {}", import.path, import.namespace);
    }
    if has_imports {
        out.push('\n');
    }

    // Emit style definitions
    if use_separators && has_styles {
        out.push_str("# ─── Themes ───\n\n");
    }
    let mut styles: Vec<_> = graph.styles.iter().collect();
    styles.sort_by_key(|(id, _)| id.as_str().to_string());
    for (name, style) in &styles {
        emit_style_block(&mut out, name, style, 0);
        out.push('\n');
    }

    // Emit root's children (node tree)
    if use_separators && !children.is_empty() {
        out.push_str("# ─── Layout ───\n\n");
    }
    for child_idx in &children {
        emit_node(&mut out, graph, *child_idx, 0);
        out.push('\n');
    }

    // Emit top-level constraints (skip Position — emitted inline as x:/y:)
    if use_separators && has_constraints {
        out.push_str("# ─── Constraints ───\n\n");
    }
    for idx in graph.graph.node_indices() {
        let node = &graph.graph[idx];
        for constraint in &node.constraints {
            if matches!(constraint, Constraint::Position { .. }) {
                continue; // emitted inline inside node block
            }
            emit_constraint(&mut out, &node.id, constraint);
        }
    }

    // Emit edges
    if use_separators && has_edges {
        if has_constraints {
            out.push('\n');
        }
        out.push_str("# ─── Flows ───\n\n");
    }
    for edge in &graph.edges {
        emit_edge(&mut out, edge, graph);
    }

    out
}

fn indent(out: &mut String, depth: usize) {
    for _ in 0..depth {
        out.push_str("  ");
    }
}

fn emit_style_block(out: &mut String, name: &NodeId, style: &Style, depth: usize) {
    indent(out, depth);
    writeln!(out, "theme {} {{", name.as_str()).unwrap();

    if let Some(ref fill) = style.fill {
        emit_paint_prop(out, "fill", fill, depth + 1);
    }
    if let Some(ref font) = style.font {
        emit_font_prop(out, font, depth + 1);
    }
    if let Some(radius) = style.corner_radius {
        indent(out, depth + 1);
        writeln!(out, "corner: {}", format_num(radius)).unwrap();
    }
    if let Some(opacity) = style.opacity {
        indent(out, depth + 1);
        writeln!(out, "opacity: {}", format_num(opacity)).unwrap();
    }
    if let Some(ref shadow) = style.shadow {
        indent(out, depth + 1);
        writeln!(
            out,
            "shadow: ({},{},{},{})",
            format_num(shadow.offset_x),
            format_num(shadow.offset_y),
            format_num(shadow.blur),
            shadow.color.to_hex()
        )
        .unwrap();
    }
    // Text alignment
    if style.text_align.is_some() || style.text_valign.is_some() {
        let h = match style.text_align {
            Some(TextAlign::Left) => "left",
            Some(TextAlign::Right) => "right",
            _ => "center",
        };
        let v = match style.text_valign {
            Some(TextVAlign::Top) => "top",
            Some(TextVAlign::Bottom) => "bottom",
            _ => "middle",
        };
        indent(out, depth + 1);
        writeln!(out, "align: {h} {v}").unwrap();
    }

    indent(out, depth);
    out.push_str("}\n");
}

fn emit_node(out: &mut String, graph: &SceneGraph, idx: NodeIndex, depth: usize) {
    let node = &graph.graph[idx];

    // Emit preserved `# comment` lines before the node declaration
    for comment in &node.comments {
        indent(out, depth);
        writeln!(out, "# {comment}").unwrap();
    }

    indent(out, depth);

    // Node kind keyword + optional @id + optional inline text
    match &node.kind {
        NodeKind::Root => return,
        NodeKind::Generic => write!(out, "@{}", node.id.as_str()).unwrap(),
        NodeKind::Group => write!(out, "group @{}", node.id.as_str()).unwrap(),
        NodeKind::Frame { .. } => write!(out, "frame @{}", node.id.as_str()).unwrap(),
        NodeKind::Rect { .. } => write!(out, "rect @{}", node.id.as_str()).unwrap(),
        NodeKind::Ellipse { .. } => write!(out, "ellipse @{}", node.id.as_str()).unwrap(),
        NodeKind::Path { .. } => write!(out, "path @{}", node.id.as_str()).unwrap(),
        NodeKind::Text { content } => {
            write!(out, "text @{} \"{}\"", node.id.as_str(), content).unwrap();
        }
    }

    out.push_str(" {\n");

    // Annotations (spec block)
    emit_annotations(out, &node.annotations, depth + 1);

    // Children — emitted right after spec so the structural skeleton
    // is visible first. Visual styling comes at the tail for clean folding.
    let children = graph.children(idx);
    for child_idx in &children {
        emit_node(out, graph, *child_idx, depth + 1);
    }

    // Group is purely organizational — no layout mode emission

    // Layout mode (for frames)
    if let NodeKind::Frame { layout, .. } = &node.kind {
        match layout {
            LayoutMode::Free => {}
            LayoutMode::Column { gap, pad } => {
                indent(out, depth + 1);
                writeln!(
                    out,
                    "layout: column gap={} pad={}",
                    format_num(*gap),
                    format_num(*pad)
                )
                .unwrap();
            }
            LayoutMode::Row { gap, pad } => {
                indent(out, depth + 1);
                writeln!(
                    out,
                    "layout: row gap={} pad={}",
                    format_num(*gap),
                    format_num(*pad)
                )
                .unwrap();
            }
            LayoutMode::Grid { cols, gap, pad } => {
                indent(out, depth + 1);
                writeln!(
                    out,
                    "layout: grid cols={cols} gap={} pad={}",
                    format_num(*gap),
                    format_num(*pad)
                )
                .unwrap();
            }
        }
    }

    // Dimensions
    match &node.kind {
        NodeKind::Rect { width, height } => {
            indent(out, depth + 1);
            writeln!(out, "w: {} h: {}", format_num(*width), format_num(*height)).unwrap();
        }
        NodeKind::Frame { width, height, .. } => {
            indent(out, depth + 1);
            writeln!(out, "w: {} h: {}", format_num(*width), format_num(*height)).unwrap();
        }
        NodeKind::Ellipse { rx, ry } => {
            indent(out, depth + 1);
            writeln!(out, "w: {} h: {}", format_num(*rx), format_num(*ry)).unwrap();
        }
        _ => {}
    }

    // Clip property (for frames only)
    if let NodeKind::Frame { clip: true, .. } = &node.kind {
        indent(out, depth + 1);
        writeln!(out, "clip: true").unwrap();
    }

    // Style references
    for style_ref in &node.use_styles {
        indent(out, depth + 1);
        writeln!(out, "use: {}", style_ref.as_str()).unwrap();
    }

    // Inline style properties
    if let Some(ref fill) = node.style.fill {
        emit_paint_prop(out, "fill", fill, depth + 1);
    }
    if let Some(ref stroke) = node.style.stroke {
        indent(out, depth + 1);
        match &stroke.paint {
            Paint::Solid(c) => {
                writeln!(out, "stroke: {} {}", c.to_hex(), format_num(stroke.width)).unwrap()
            }
            _ => writeln!(out, "stroke: #000 {}", format_num(stroke.width)).unwrap(),
        }
    }
    if let Some(radius) = node.style.corner_radius {
        indent(out, depth + 1);
        writeln!(out, "corner: {}", format_num(radius)).unwrap();
    }
    if let Some(ref font) = node.style.font {
        emit_font_prop(out, font, depth + 1);
    }
    if let Some(opacity) = node.style.opacity {
        indent(out, depth + 1);
        writeln!(out, "opacity: {}", format_num(opacity)).unwrap();
    }
    if let Some(ref shadow) = node.style.shadow {
        indent(out, depth + 1);
        writeln!(
            out,
            "shadow: ({},{},{},{})",
            format_num(shadow.offset_x),
            format_num(shadow.offset_y),
            format_num(shadow.blur),
            shadow.color.to_hex()
        )
        .unwrap();
    }

    // Text alignment
    if node.style.text_align.is_some() || node.style.text_valign.is_some() {
        let h = match node.style.text_align {
            Some(TextAlign::Left) => "left",
            Some(TextAlign::Right) => "right",
            _ => "center",
        };
        let v = match node.style.text_valign {
            Some(TextVAlign::Top) => "top",
            Some(TextVAlign::Bottom) => "bottom",
            _ => "middle",
        };
        indent(out, depth + 1);
        writeln!(out, "align: {h} {v}").unwrap();
    }

    // Inline position (x: / y:) — emitted here for token efficiency
    for constraint in &node.constraints {
        if let Constraint::Position { x, y } = constraint {
            if *x != 0.0 {
                indent(out, depth + 1);
                writeln!(out, "x: {}", format_num(*x)).unwrap();
            }
            if *y != 0.0 {
                indent(out, depth + 1);
                writeln!(out, "y: {}", format_num(*y)).unwrap();
            }
        }
    }

    // Animations (when blocks)
    for anim in &node.animations {
        emit_anim(out, anim, depth + 1);
    }

    indent(out, depth);
    out.push_str("}\n");
}

fn emit_annotations(out: &mut String, annotations: &[Annotation], depth: usize) {
    if annotations.is_empty() {
        return;
    }

    // Single description → inline shorthand: `spec "desc"`
    if annotations.len() == 1
        && let Annotation::Description(s) = &annotations[0]
    {
        indent(out, depth);
        writeln!(out, "spec \"{s}\"").unwrap();
        return;
    }

    // Multiple annotations → block form: `spec { ... }`
    indent(out, depth);
    out.push_str("spec {\n");
    for ann in annotations {
        indent(out, depth + 1);
        match ann {
            Annotation::Description(s) => writeln!(out, "\"{s}\"").unwrap(),
            Annotation::Accept(s) => writeln!(out, "accept: \"{s}\"").unwrap(),
            Annotation::Status(s) => writeln!(out, "status: {s}").unwrap(),
            Annotation::Priority(s) => writeln!(out, "priority: {s}").unwrap(),
            Annotation::Tag(s) => writeln!(out, "tag: {s}").unwrap(),
        }
    }
    indent(out, depth);
    out.push_str("}\n");
}

fn emit_paint_prop(out: &mut String, name: &str, paint: &Paint, depth: usize) {
    indent(out, depth);
    match paint {
        Paint::Solid(c) => {
            let hex = c.to_hex();
            let hint = color_hint(&hex);
            if hint.is_empty() {
                writeln!(out, "{name}: {hex}").unwrap();
            } else {
                writeln!(out, "{name}: {hex}  # {hint}").unwrap();
            }
        }
        Paint::LinearGradient { angle, stops } => {
            write!(out, "{name}: linear({}deg", format_num(*angle)).unwrap();
            for stop in stops {
                write!(out, ", {} {}", stop.color.to_hex(), format_num(stop.offset)).unwrap();
            }
            writeln!(out, ")").unwrap();
        }
        Paint::RadialGradient { stops } => {
            write!(out, "{name}: radial(").unwrap();
            for (i, stop) in stops.iter().enumerate() {
                if i > 0 {
                    write!(out, ", ").unwrap();
                }
                write!(out, "{} {}", stop.color.to_hex(), format_num(stop.offset)).unwrap();
            }
            writeln!(out, ")").unwrap();
        }
    }
}

fn emit_font_prop(out: &mut String, font: &FontSpec, depth: usize) {
    indent(out, depth);
    let weight_str = weight_number_to_name(font.weight);
    writeln!(
        out,
        "font: \"{}\" {} {}",
        font.family,
        weight_str,
        format_num(font.size)
    )
    .unwrap();
}

/// Map numeric font weight to human-readable name.
fn weight_number_to_name(weight: u16) -> &'static str {
    match weight {
        100 => "thin",
        200 => "extralight",
        300 => "light",
        400 => "regular",
        500 => "medium",
        600 => "semibold",
        700 => "bold",
        800 => "extrabold",
        900 => "black",
        _ => "400", // fallback
    }
}

/// Classify a hex color into a human-readable hue name.
fn color_hint(hex: &str) -> &'static str {
    let hex = hex.trim_start_matches('#');
    let bytes = hex.as_bytes();
    let Some((r, g, b)) = (match bytes.len() {
        3 | 4 => {
            let r = crate::model::hex_val(bytes[0]).unwrap_or(0) * 17;
            let g = crate::model::hex_val(bytes[1]).unwrap_or(0) * 17;
            let b = crate::model::hex_val(bytes[2]).unwrap_or(0) * 17;
            Some((r, g, b))
        }
        6 | 8 => {
            let r = (crate::model::hex_val(bytes[0]).unwrap_or(0) << 4)
                | crate::model::hex_val(bytes[1]).unwrap_or(0);
            let g = (crate::model::hex_val(bytes[2]).unwrap_or(0) << 4)
                | crate::model::hex_val(bytes[3]).unwrap_or(0);
            let b = (crate::model::hex_val(bytes[4]).unwrap_or(0) << 4)
                | crate::model::hex_val(bytes[5]).unwrap_or(0);
            Some((r, g, b))
        }
        _ => None,
    }) else {
        return "";
    };

    // Achromatic check
    let max = r.max(g).max(b);
    let min = r.min(g).min(b);
    let diff = max - min;
    if diff < 15 {
        return match max {
            0..=30 => "black",
            31..=200 => "gray",
            _ => "white",
        };
    }

    // Hue classification
    let rf = r as f32;
    let gf = g as f32;
    let bf = b as f32;
    let hue = if max == r {
        60.0 * (((gf - bf) / diff as f32) % 6.0)
    } else if max == g {
        60.0 * (((bf - rf) / diff as f32) + 2.0)
    } else {
        60.0 * (((rf - gf) / diff as f32) + 4.0)
    };
    let hue = if hue < 0.0 { hue + 360.0 } else { hue };

    match hue as u16 {
        0..=14 | 346..=360 => "red",
        15..=39 => "orange",
        40..=64 => "yellow",
        65..=79 => "lime",
        80..=159 => "green",
        160..=179 => "teal",
        180..=199 => "cyan",
        200..=259 => "blue",
        260..=279 => "purple",
        280..=319 => "pink",
        320..=345 => "rose",
        _ => "",
    }
}

fn emit_anim(out: &mut String, anim: &AnimKeyframe, depth: usize) {
    indent(out, depth);
    let trigger = match &anim.trigger {
        AnimTrigger::Hover => "hover",
        AnimTrigger::Press => "press",
        AnimTrigger::Enter => "enter",
        AnimTrigger::Custom(s) => s.as_str(),
    };
    writeln!(out, "when :{trigger} {{").unwrap();

    if let Some(ref fill) = anim.properties.fill {
        emit_paint_prop(out, "fill", fill, depth + 1);
    }
    if let Some(opacity) = anim.properties.opacity {
        indent(out, depth + 1);
        writeln!(out, "opacity: {}", format_num(opacity)).unwrap();
    }
    if let Some(scale) = anim.properties.scale {
        indent(out, depth + 1);
        writeln!(out, "scale: {}", format_num(scale)).unwrap();
    }
    if let Some(rotate) = anim.properties.rotate {
        indent(out, depth + 1);
        writeln!(out, "rotate: {}", format_num(rotate)).unwrap();
    }

    let ease_name = match &anim.easing {
        Easing::Linear => "linear",
        Easing::EaseIn => "ease_in",
        Easing::EaseOut => "ease_out",
        Easing::EaseInOut => "ease_in_out",
        Easing::Spring => "spring",
        Easing::CubicBezier(_, _, _, _) => "cubic",
    };
    indent(out, depth + 1);
    writeln!(out, "ease: {ease_name} {}ms", anim.duration_ms).unwrap();

    indent(out, depth);
    out.push_str("}\n");
}

fn emit_constraint(out: &mut String, node_id: &NodeId, constraint: &Constraint) {
    match constraint {
        Constraint::CenterIn(target) => {
            writeln!(
                out,
                "@{} -> center_in: {}",
                node_id.as_str(),
                target.as_str()
            )
            .unwrap();
        }
        Constraint::Offset { from, dx, dy } => {
            writeln!(
                out,
                "@{} -> offset: @{} {}, {}",
                node_id.as_str(),
                from.as_str(),
                format_num(*dx),
                format_num(*dy)
            )
            .unwrap();
        }
        Constraint::FillParent { pad } => {
            writeln!(
                out,
                "@{} -> fill_parent: {}",
                node_id.as_str(),
                format_num(*pad)
            )
            .unwrap();
        }
        Constraint::Position { .. } => {
            // Emitted inline as x: / y: inside node block — skip here
        }
    }
}

fn emit_edge(out: &mut String, edge: &Edge, graph: &SceneGraph) {
    writeln!(out, "edge @{} {{", edge.id.as_str()).unwrap();

    // Annotations
    emit_annotations(out, &edge.annotations, 1);

    // Nested text child
    if let Some(text_id) = edge.text_child
        && let Some(node) = graph.get_by_id(text_id)
        && let NodeKind::Text { content } = &node.kind
    {
        writeln!(out, "  text @{} \"{}\" {{}}", text_id.as_str(), content).unwrap();
    }

    // from / to
    match &edge.from {
        EdgeAnchor::Node(id) => writeln!(out, "  from: @{}", id.as_str()).unwrap(),
        EdgeAnchor::Point(x, y) => {
            writeln!(out, "  from: {} {}", format_num(*x), format_num(*y)).unwrap()
        }
    }
    match &edge.to {
        EdgeAnchor::Node(id) => writeln!(out, "  to: @{}", id.as_str()).unwrap(),
        EdgeAnchor::Point(x, y) => {
            writeln!(out, "  to: {} {}", format_num(*x), format_num(*y)).unwrap()
        }
    }

    // Style references
    for style_ref in &edge.use_styles {
        writeln!(out, "  use: {}", style_ref.as_str()).unwrap();
    }

    // Stroke
    if let Some(ref stroke) = edge.style.stroke {
        match &stroke.paint {
            Paint::Solid(c) => {
                writeln!(out, "  stroke: {} {}", c.to_hex(), format_num(stroke.width)).unwrap();
            }
            _ => {
                writeln!(out, "  stroke: #000 {}", format_num(stroke.width)).unwrap();
            }
        }
    }

    // Opacity
    if let Some(opacity) = edge.style.opacity {
        writeln!(out, "  opacity: {}", format_num(opacity)).unwrap();
    }

    // Arrow
    if edge.arrow != ArrowKind::None {
        let name = match edge.arrow {
            ArrowKind::None => "none",
            ArrowKind::Start => "start",
            ArrowKind::End => "end",
            ArrowKind::Both => "both",
        };
        writeln!(out, "  arrow: {name}").unwrap();
    }

    // Curve
    if edge.curve != CurveKind::Straight {
        let name = match edge.curve {
            CurveKind::Straight => "straight",
            CurveKind::Smooth => "smooth",
            CurveKind::Step => "step",
        };
        writeln!(out, "  curve: {name}").unwrap();
    }

    // Flow animation
    if let Some(ref flow) = edge.flow {
        let kind = match flow.kind {
            FlowKind::Pulse => "pulse",
            FlowKind::Dash => "dash",
        };
        writeln!(out, "  flow: {} {}ms", kind, flow.duration_ms).unwrap();
    }

    // Label offset (dragged position)
    if let Some((ox, oy)) = edge.label_offset {
        writeln!(out, "  label_offset: {} {}", format_num(ox), format_num(oy)).unwrap();
    }

    // Trigger animations
    for anim in &edge.animations {
        emit_anim(out, anim, 1);
    }

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

// ─── Read Modes (filtered emit for AI agents) ────────────────────────────

/// What an AI agent wants to read from the document.
///
/// Each mode selectively emits only the properties relevant to a specific
/// concern, saving 50-80% tokens while preserving structural understanding.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ReadMode {
    /// Full file — no filtering (identical to `emit_document`).
    Full,
    /// Node types, `@id`s, parent-child nesting only.
    Structure,
    /// Structure + dimensions (`w:`/`h:`) + `layout:` directives + constraints.
    Layout,
    /// Structure + themes/styles + `fill:`/`stroke:`/`font:`/`corner:`/`use:` refs.
    Design,
    /// Structure + `spec {}` blocks + annotations.
    Spec,
    /// Layout + Design + When combined — the full visual story.
    Visual,
    /// Structure + `when :trigger { ... }` animation blocks only.
    When,
    /// Structure + `edge @id { ... }` blocks.
    Edges,
}

/// Emit a `SceneGraph` filtered to show only the properties relevant to `mode`.
///
/// - `Full`: identical to `emit_document`.
/// - `Structure`: node kind + `@id` + children. No styles, dims, anims, specs.
/// - `Layout`: structure + `w:`/`h:` + `layout:` + constraints (`->`).
/// - `Design`: structure + themes + `fill:`/`stroke:`/`font:`/`corner:`/`use:`.
/// - `Spec`: structure + `spec {}` blocks.
/// - `Visual`: layout + design + when combined.
/// - `When`: structure + `when :trigger { ... }` blocks.
/// - `Edges`: structure + `edge @id { ... }` blocks.
#[must_use]
pub fn emit_filtered(graph: &SceneGraph, mode: ReadMode) -> String {
    if mode == ReadMode::Full {
        return emit_document(graph);
    }

    let mut out = String::with_capacity(1024);

    let children = graph.children(graph.root);
    let include_themes = matches!(mode, ReadMode::Design | ReadMode::Visual);
    let include_constraints = matches!(mode, ReadMode::Layout | ReadMode::Visual);
    let include_edges = matches!(mode, ReadMode::Edges | ReadMode::Visual);

    // Themes (Design and Visual modes)
    if include_themes && !graph.styles.is_empty() {
        let mut styles: Vec<_> = graph.styles.iter().collect();
        styles.sort_by_key(|(id, _)| id.as_str().to_string());
        for (name, style) in &styles {
            emit_style_block(&mut out, name, style, 0);
            out.push('\n');
        }
    }

    // Node tree (always emitted, but with per-mode filtering)
    for child_idx in &children {
        emit_node_filtered(&mut out, graph, *child_idx, 0, mode);
        out.push('\n');
    }

    // Constraints (Layout and Visual modes)
    if include_constraints {
        for idx in graph.graph.node_indices() {
            let node = &graph.graph[idx];
            for constraint in &node.constraints {
                if matches!(constraint, Constraint::Position { .. }) {
                    continue;
                }
                emit_constraint(&mut out, &node.id, constraint);
            }
        }
    }

    // Edges (Edges and Visual modes)
    if include_edges {
        for edge in &graph.edges {
            emit_edge(&mut out, edge, graph);
            out.push('\n');
        }
    }

    out
}

/// Emit a single node with mode-based property filtering.
fn emit_node_filtered(
    out: &mut String,
    graph: &SceneGraph,
    idx: NodeIndex,
    depth: usize,
    mode: ReadMode,
) {
    let node = &graph.graph[idx];

    if matches!(node.kind, NodeKind::Root) {
        return;
    }

    indent(out, depth);

    // Node kind + @id (always emitted)
    match &node.kind {
        NodeKind::Root => return,
        NodeKind::Generic => write!(out, "@{}", node.id.as_str()).unwrap(),
        NodeKind::Group => write!(out, "group @{}", node.id.as_str()).unwrap(),
        NodeKind::Frame { .. } => write!(out, "frame @{}", node.id.as_str()).unwrap(),
        NodeKind::Rect { .. } => write!(out, "rect @{}", node.id.as_str()).unwrap(),
        NodeKind::Ellipse { .. } => write!(out, "ellipse @{}", node.id.as_str()).unwrap(),
        NodeKind::Path { .. } => write!(out, "path @{}", node.id.as_str()).unwrap(),
        NodeKind::Text { content } => {
            write!(out, "text @{} \"{}\"", node.id.as_str(), content).unwrap();
        }
    }

    out.push_str(" {\n");

    // Spec annotations (Spec mode only)
    if mode == ReadMode::Spec {
        emit_annotations(out, &node.annotations, depth + 1);
    }

    // Children (always recurse)
    let children = graph.children(idx);
    for child_idx in &children {
        emit_node_filtered(out, graph, *child_idx, depth + 1, mode);
    }

    // Layout directives (Layout and Visual modes)
    if matches!(mode, ReadMode::Layout | ReadMode::Visual) {
        emit_layout_mode_filtered(out, &node.kind, depth + 1);
    }

    // Dimensions (Layout and Visual modes)
    if matches!(mode, ReadMode::Layout | ReadMode::Visual) {
        emit_dimensions_filtered(out, &node.kind, depth + 1);
    }

    // Style properties (Design and Visual modes)
    if matches!(mode, ReadMode::Design | ReadMode::Visual) {
        for style_ref in &node.use_styles {
            indent(out, depth + 1);
            writeln!(out, "use: {}", style_ref.as_str()).unwrap();
        }
        if let Some(ref fill) = node.style.fill {
            emit_paint_prop(out, "fill", fill, depth + 1);
        }
        if let Some(ref stroke) = node.style.stroke {
            indent(out, depth + 1);
            match &stroke.paint {
                Paint::Solid(c) => {
                    writeln!(out, "stroke: {} {}", c.to_hex(), format_num(stroke.width)).unwrap();
                }
                _ => writeln!(out, "stroke: #000 {}", format_num(stroke.width)).unwrap(),
            }
        }
        if let Some(radius) = node.style.corner_radius {
            indent(out, depth + 1);
            writeln!(out, "corner: {}", format_num(radius)).unwrap();
        }
        if let Some(ref font) = node.style.font {
            emit_font_prop(out, font, depth + 1);
        }
        if let Some(opacity) = node.style.opacity {
            indent(out, depth + 1);
            writeln!(out, "opacity: {}", format_num(opacity)).unwrap();
        }
    }

    // Inline position (Layout and Visual modes)
    if matches!(mode, ReadMode::Layout | ReadMode::Visual) {
        for constraint in &node.constraints {
            if let Constraint::Position { x, y } = constraint {
                if *x != 0.0 {
                    indent(out, depth + 1);
                    writeln!(out, "x: {}", format_num(*x)).unwrap();
                }
                if *y != 0.0 {
                    indent(out, depth + 1);
                    writeln!(out, "y: {}", format_num(*y)).unwrap();
                }
            }
        }
    }

    // Animations / when blocks (When and Visual modes)
    if matches!(mode, ReadMode::When | ReadMode::Visual) {
        for anim in &node.animations {
            emit_anim(out, anim, depth + 1);
        }
    }

    indent(out, depth);
    out.push_str("}\n");
}

/// Emit layout mode directive for groups and frames (filtered path).
fn emit_layout_mode_filtered(out: &mut String, kind: &NodeKind, depth: usize) {
    let layout = match kind {
        NodeKind::Frame { layout, .. } => layout,
        _ => return, // Group is always Free — no layout emission
    };
    match layout {
        LayoutMode::Free => {}
        LayoutMode::Column { gap, pad } => {
            indent(out, depth);
            writeln!(
                out,
                "layout: column gap={} pad={}",
                format_num(*gap),
                format_num(*pad)
            )
            .unwrap();
        }
        LayoutMode::Row { gap, pad } => {
            indent(out, depth);
            writeln!(
                out,
                "layout: row gap={} pad={}",
                format_num(*gap),
                format_num(*pad)
            )
            .unwrap();
        }
        LayoutMode::Grid { cols, gap, pad } => {
            indent(out, depth);
            writeln!(
                out,
                "layout: grid cols={cols} gap={} pad={}",
                format_num(*gap),
                format_num(*pad)
            )
            .unwrap();
        }
    }
}

/// Emit dimension properties (w/h) for sized nodes (filtered path).
fn emit_dimensions_filtered(out: &mut String, kind: &NodeKind, depth: usize) {
    match kind {
        NodeKind::Rect { width, height } | NodeKind::Frame { width, height, .. } => {
            indent(out, depth);
            writeln!(out, "w: {} h: {}", format_num(*width), format_num(*height)).unwrap();
        }
        NodeKind::Ellipse { rx, ry } => {
            indent(out, depth);
            writeln!(out, "w: {} h: {}", format_num(*rx), format_num(*ry)).unwrap();
        }
        _ => {}
    }
}

// ─── Spec Markdown Export ─────────────────────────────────────────────────

/// Emit a `SceneGraph` as a markdown spec document.
///
/// Extracts only `@id` names, `spec { ... }` annotations, hierarchy, and edges —
/// all visual properties (fill, stroke, dimensions, animations) are omitted.
/// Intended for PM-facing spec reports.
#[must_use]
pub fn emit_spec_markdown(graph: &SceneGraph, title: &str) -> String {
    let mut out = String::with_capacity(512);
    writeln!(out, "# Spec: {title}\n").unwrap();

    // Emit root's children
    let children = graph.children(graph.root);
    for child_idx in &children {
        emit_spec_node(&mut out, graph, *child_idx, 2);
    }

    // Emit edges as flow descriptions
    if !graph.edges.is_empty() {
        out.push_str("\n---\n\n## Flows\n\n");
        for edge in &graph.edges {
            let from_str = match &edge.from {
                EdgeAnchor::Node(id) => format!("@{}", id.as_str()),
                EdgeAnchor::Point(x, y) => format!("({}, {})", x, y),
            };
            let to_str = match &edge.to {
                EdgeAnchor::Node(id) => format!("@{}", id.as_str()),
                EdgeAnchor::Point(x, y) => format!("({}, {})", x, y),
            };
            write!(out, "- **{}** → **{}**", from_str, to_str).unwrap();
            if let Some(text_id) = edge.text_child
                && let Some(node) = graph.get_by_id(text_id)
                && let NodeKind::Text { content } = &node.kind
            {
                write!(out, " — {content}").unwrap();
            }
            out.push('\n');
            emit_spec_annotations(&mut out, &edge.annotations, "  ");
        }
    }

    out
}

fn emit_spec_node(out: &mut String, graph: &SceneGraph, idx: NodeIndex, heading_level: usize) {
    let node = &graph.graph[idx];

    // Skip nodes with no annotations and no annotated children
    let has_annotations = !node.annotations.is_empty();
    let children = graph.children(idx);
    let has_annotated_children = children
        .iter()
        .any(|c| has_annotations_recursive(graph, *c));

    if !has_annotations && !has_annotated_children {
        return;
    }

    // Heading: ## @node_id (kind)
    let hashes = "#".repeat(heading_level.min(6));
    let kind_label = match &node.kind {
        NodeKind::Root => return,
        NodeKind::Generic => "spec",
        NodeKind::Group => "group",
        NodeKind::Frame { .. } => "frame",
        NodeKind::Rect { .. } => "rect",
        NodeKind::Ellipse { .. } => "ellipse",
        NodeKind::Path { .. } => "path",
        NodeKind::Text { .. } => "text",
    };
    writeln!(out, "{hashes} @{} `{kind_label}`\n", node.id.as_str()).unwrap();

    // Annotation details
    emit_spec_annotations(out, &node.annotations, "");

    // Children (recurse with deeper heading level)
    for child_idx in &children {
        emit_spec_node(out, graph, *child_idx, heading_level + 1);
    }
}

fn has_annotations_recursive(graph: &SceneGraph, idx: NodeIndex) -> bool {
    let node = &graph.graph[idx];
    if !node.annotations.is_empty() {
        return true;
    }
    graph
        .children(idx)
        .iter()
        .any(|c| has_annotations_recursive(graph, *c))
}

fn emit_spec_annotations(out: &mut String, annotations: &[Annotation], prefix: &str) {
    for ann in annotations {
        match ann {
            Annotation::Description(s) => writeln!(out, "{prefix}> {s}").unwrap(),
            Annotation::Accept(s) => writeln!(out, "{prefix}- [ ] {s}").unwrap(),
            Annotation::Status(s) => writeln!(out, "{prefix}- **Status:** {s}").unwrap(),
            Annotation::Priority(s) => writeln!(out, "{prefix}- **Priority:** {s}").unwrap(),
            Annotation::Tag(s) => writeln!(out, "{prefix}- **Tag:** {s}").unwrap(),
        }
    }
    if !annotations.is_empty() {
        out.push('\n');
    }
}

/// Format a float without trailing zeros for compact output.
fn format_num(n: f32) -> String {
    if n == n.floor() {
        format!("{}", n as i32)
    } else {
        format!("{n:.2}")
            .trim_end_matches('0')
            .trim_end_matches('.')
            .to_string()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::parser::parse_document;

    #[test]
    fn roundtrip_simple() {
        let input = r#"
rect @box {
  w: 100
  h: 50
  fill: #FF0000
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Re-parse the emitted output
        let graph2 = parse_document(&output).expect("re-parse of emitted output failed");
        let node2 = graph2.get_by_id(NodeId::intern("box")).unwrap();

        match &node2.kind {
            NodeKind::Rect { width, height } => {
                assert_eq!(*width, 100.0);
                assert_eq!(*height, 50.0);
            }
            _ => panic!("expected Rect"),
        }
    }

    #[test]
    fn roundtrip_ellipse() {
        let input = r#"
ellipse @dot {
  w: 40 h: 40
  fill: #00FF00
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of ellipse failed");
        let node = graph2.get_by_id(NodeId::intern("dot")).unwrap();
        match &node.kind {
            NodeKind::Ellipse { rx, ry } => {
                assert_eq!(*rx, 40.0);
                assert_eq!(*ry, 40.0);
            }
            _ => panic!("expected Ellipse"),
        }
    }

    #[test]
    fn roundtrip_text_with_font() {
        let input = r#"
text @title "Hello" {
  font: "Inter" 700 32
  fill: #1A1A2E
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of text failed");
        let node = graph2.get_by_id(NodeId::intern("title")).unwrap();
        match &node.kind {
            NodeKind::Text { content } => assert_eq!(content, "Hello"),
            _ => panic!("expected Text"),
        }
        let font = node.style.font.as_ref().expect("font missing");
        assert_eq!(font.family, "Inter");
        assert_eq!(font.weight, 700);
        assert_eq!(font.size, 32.0);
    }

    #[test]
    fn roundtrip_nested_group() {
        let input = r#"
group @card {
  layout: column gap=16 pad=24

  text @heading "Title" {
    font: "Inter" 600 20
    fill: #333333
  }

  rect @body {
    w: 300 h: 200
    fill: #F5F5F5
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of nested group failed");
        let card_idx = graph2.index_of(NodeId::intern("card")).unwrap();
        assert_eq!(graph2.children(card_idx).len(), 2);
    }

    #[test]
    fn roundtrip_animation() {
        let input = r#"
rect @btn {
  w: 200 h: 48
  fill: #6C5CE7

  anim :hover {
    fill: #5A4BD1
    scale: 1.02
    ease: spring 300ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of animation failed");
        let btn = graph2.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(btn.animations.len(), 1);
        assert_eq!(btn.animations[0].trigger, AnimTrigger::Hover);
    }

    #[test]
    fn roundtrip_style_and_use() {
        let input = r#"
style accent {
  fill: #6C5CE7
  corner: 10
}

rect @btn {
  w: 200 h: 48
  use: accent
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of style+use failed");
        assert!(graph2.styles.contains_key(&NodeId::intern("accent")));
        let btn = graph2.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(btn.use_styles.len(), 1);
    }

    #[test]
    fn roundtrip_annotation_description() {
        let input = r#"
rect @box {
  spec "Primary container for content"
  w: 100 h: 50
  fill: #FF0000
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("box")).unwrap();
        assert_eq!(node.annotations.len(), 1);
        assert_eq!(
            node.annotations[0],
            Annotation::Description("Primary container for content".into())
        );

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of annotation failed");
        let node2 = graph2.get_by_id(NodeId::intern("box")).unwrap();
        assert_eq!(node2.annotations.len(), 1);
        assert_eq!(node2.annotations[0], node.annotations[0]);
    }

    #[test]
    fn roundtrip_annotation_accept() {
        let input = r#"
rect @login_btn {
  spec {
    accept: "disabled state when fields empty"
    accept: "loading spinner during auth"
  }
  w: 280 h: 48
  fill: #6C5CE7
}
"#;
        let graph = parse_document(input).unwrap();
        let btn = graph.get_by_id(NodeId::intern("login_btn")).unwrap();
        assert_eq!(btn.annotations.len(), 2);
        assert_eq!(
            btn.annotations[0],
            Annotation::Accept("disabled state when fields empty".into())
        );
        assert_eq!(
            btn.annotations[1],
            Annotation::Accept("loading spinner during auth".into())
        );

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of accept annotation failed");
        let btn2 = graph2.get_by_id(NodeId::intern("login_btn")).unwrap();
        assert_eq!(btn2.annotations, btn.annotations);
    }

    #[test]
    fn roundtrip_annotation_status_priority() {
        let input = r#"
rect @card {
  spec {
    status: doing
    priority: high
    tag: mvp
  }
  w: 300 h: 200
}
"#;
        let graph = parse_document(input).unwrap();
        let card = graph.get_by_id(NodeId::intern("card")).unwrap();
        assert_eq!(card.annotations.len(), 3);
        assert_eq!(card.annotations[0], Annotation::Status("doing".into()));
        assert_eq!(card.annotations[1], Annotation::Priority("high".into()));
        assert_eq!(card.annotations[2], Annotation::Tag("mvp".into()));

        let output = emit_document(&graph);
        let graph2 =
            parse_document(&output).expect("re-parse of status/priority/tag annotation failed");
        let card2 = graph2.get_by_id(NodeId::intern("card")).unwrap();
        assert_eq!(card2.annotations, card.annotations);
    }

    #[test]
    fn roundtrip_annotation_nested() {
        let input = r#"
group @form {
  layout: column gap=16 pad=32
  spec "User authentication entry point"

  rect @email {
    spec {
      accept: "validates email format"
    }
    w: 280 h: 44
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let form = graph.get_by_id(NodeId::intern("form")).unwrap();
        assert_eq!(form.annotations.len(), 1);
        let email = graph.get_by_id(NodeId::intern("email")).unwrap();
        assert_eq!(email.annotations.len(), 1);

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of nested annotation failed");
        let form2 = graph2.get_by_id(NodeId::intern("form")).unwrap();
        assert_eq!(form2.annotations, form.annotations);
        let email2 = graph2.get_by_id(NodeId::intern("email")).unwrap();
        assert_eq!(email2.annotations, email.annotations);
    }

    #[test]
    fn parse_annotation_freeform() {
        let input = r#"
rect @widget {
  spec {
    "Description line"
    accept: "criterion one"
    status: done
    priority: low
    tag: design
  }
  w: 100 h: 100
}
"#;
        let graph = parse_document(input).unwrap();
        let w = graph.get_by_id(NodeId::intern("widget")).unwrap();
        assert_eq!(w.annotations.len(), 5);
        assert_eq!(
            w.annotations[0],
            Annotation::Description("Description line".into())
        );
        assert_eq!(w.annotations[1], Annotation::Accept("criterion one".into()));
        assert_eq!(w.annotations[2], Annotation::Status("done".into()));
        assert_eq!(w.annotations[3], Annotation::Priority("low".into()));
        assert_eq!(w.annotations[4], Annotation::Tag("design".into()));
    }

    #[test]
    fn roundtrip_edge_basic() {
        let input = r#"
rect @box_a {
  w: 100 h: 50
}

rect @box_b {
  w: 100 h: 50
}

edge @a_to_b {
  from: @box_a
  to: @box_b
  label: "next step"
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.id.as_str(), "a_to_b");
        assert_eq!(edge.from, EdgeAnchor::Node(NodeId::intern("box_a")));
        assert_eq!(edge.to, EdgeAnchor::Node(NodeId::intern("box_b")));
        // label: "next step" auto-creates text child
        let text_id = edge.text_child.expect("text_child should be set");
        assert_eq!(text_id.as_str(), "_a_to_b_label");
        let text_node = graph.get_by_id(text_id).expect("text node should exist");
        if let NodeKind::Text { content } = &text_node.kind {
            assert_eq!(content, "next step");
        } else {
            panic!("expected Text node");
        }
        assert_eq!(edge.arrow, ArrowKind::End);

        // Re-parse roundtrip — emitter writes nested text block
        let output = emit_document(&graph);
        assert!(output.contains("text @_a_to_b_label \"next step\" {}"));

        let graph2 = parse_document(&output).expect("roundtrip failed");
        assert_eq!(graph2.edges.len(), 1);
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.from, EdgeAnchor::Node(NodeId::intern("box_a")));
        assert_eq!(edge2.to, EdgeAnchor::Node(NodeId::intern("box_b")));
        assert!(edge2.text_child.is_some());
        assert_eq!(edge2.arrow, ArrowKind::End);
    }

    #[test]
    fn roundtrip_edge_styled() {
        let input = r#"
rect @s1 { w: 50 h: 50 }
rect @s2 { w: 50 h: 50 }

edge @flow {
  from: @s1
  to: @s2
  stroke: #6C5CE7 2
  arrow: both
  curve: smooth
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.arrow, ArrowKind::Both);
        assert_eq!(edge.curve, CurveKind::Smooth);
        assert!(edge.style.stroke.is_some());

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("styled edge roundtrip failed");
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.arrow, ArrowKind::Both);
        assert_eq!(edge2.curve, CurveKind::Smooth);
    }

    #[test]
    fn roundtrip_edge_with_annotations() {
        let input = r#"
rect @login { w: 200 h: 100 }
rect @dashboard { w: 200 h: 100 }

edge @login_flow {
  spec {
    "Main authentication flow"
    accept: "must redirect within 2s"
  }
  from: @login
  to: @dashboard
  label: "on success"
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        let edge = &graph.edges[0];
        assert_eq!(edge.annotations.len(), 2);
        assert_eq!(
            edge.annotations[0],
            Annotation::Description("Main authentication flow".into())
        );
        assert_eq!(
            edge.annotations[1],
            Annotation::Accept("must redirect within 2s".into())
        );

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("annotated edge roundtrip failed");
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.annotations, edge.annotations);
    }

    #[test]
    fn roundtrip_generic_node() {
        let input = r#"
@login_btn {
  spec {
    "Primary CTA — triggers login API call"
    accept: "disabled when fields empty"
    status: doing
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("login_btn")).unwrap();
        assert!(matches!(node.kind, NodeKind::Generic));
        assert_eq!(node.annotations.len(), 3);

        let output = emit_document(&graph);
        assert!(output.contains("@login_btn {"));
        // Should NOT have a type prefix
        assert!(!output.contains("rect @login_btn"));
        assert!(!output.contains("group @login_btn"));

        let graph2 = parse_document(&output).expect("re-parse of generic node failed");
        let node2 = graph2.get_by_id(NodeId::intern("login_btn")).unwrap();
        assert!(matches!(node2.kind, NodeKind::Generic));
        assert_eq!(node2.annotations, node.annotations);
    }

    #[test]
    fn roundtrip_generic_nested() {
        let input = r#"
group @form {
  layout: column gap=16 pad=32

  @email_input {
    spec {
      "Email field"
      accept: "validates format on blur"
    }
  }

  @password_input {
    spec {
      "Password field"
      accept: "min 8 chars"
    }
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let form_idx = graph.index_of(NodeId::intern("form")).unwrap();
        assert_eq!(graph.children(form_idx).len(), 2);

        let email = graph.get_by_id(NodeId::intern("email_input")).unwrap();
        assert!(matches!(email.kind, NodeKind::Generic));
        assert_eq!(email.annotations.len(), 2);

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of nested generic failed");
        let email2 = graph2.get_by_id(NodeId::intern("email_input")).unwrap();
        assert!(matches!(email2.kind, NodeKind::Generic));
        assert_eq!(email2.annotations, email.annotations);
    }

    #[test]
    fn parse_generic_with_properties() {
        let input = r#"
@card {
  fill: #FFFFFF
  corner: 8
}
"#;
        let graph = parse_document(input).unwrap();
        let card = graph.get_by_id(NodeId::intern("card")).unwrap();
        assert!(matches!(card.kind, NodeKind::Generic));
        assert!(card.style.fill.is_some());
        assert_eq!(card.style.corner_radius, Some(8.0));
    }

    #[test]
    fn roundtrip_edge_with_trigger_anim() {
        let input = r#"
rect @a { w: 50 h: 50 }
rect @b { w: 50 h: 50 }

edge @hover_edge {
  from: @a
  to: @b
  stroke: #6C5CE7 2
  arrow: end

  anim :hover {
    opacity: 0.5
    ease: ease_out 200ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.animations.len(), 1);
        assert_eq!(edge.animations[0].trigger, AnimTrigger::Hover);
        assert_eq!(edge.animations[0].duration_ms, 200);

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("trigger anim roundtrip failed");
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.animations.len(), 1);
        assert_eq!(edge2.animations[0].trigger, AnimTrigger::Hover);
    }

    #[test]
    fn roundtrip_edge_with_flow() {
        let input = r#"
rect @src { w: 50 h: 50 }
rect @dst { w: 50 h: 50 }

edge @data {
  from: @src
  to: @dst
  arrow: end
  flow: pulse 800ms
}
"#;
        let graph = parse_document(input).unwrap();
        let edge = &graph.edges[0];
        assert!(edge.flow.is_some());
        let flow = edge.flow.unwrap();
        assert_eq!(flow.kind, FlowKind::Pulse);
        assert_eq!(flow.duration_ms, 800);

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("flow roundtrip failed");
        let edge2 = &graph2.edges[0];
        let flow2 = edge2.flow.unwrap();
        assert_eq!(flow2.kind, FlowKind::Pulse);
        assert_eq!(flow2.duration_ms, 800);
    }

    #[test]
    fn roundtrip_edge_dash_flow() {
        let input = r#"
rect @x { w: 50 h: 50 }
rect @y { w: 50 h: 50 }

edge @dashed {
  from: @x
  to: @y
  stroke: #EF4444 1
  flow: dash 400ms
  arrow: both
  curve: step
}
"#;
        let graph = parse_document(input).unwrap();
        let edge = &graph.edges[0];
        let flow = edge.flow.unwrap();
        assert_eq!(flow.kind, FlowKind::Dash);
        assert_eq!(flow.duration_ms, 400);
        assert_eq!(edge.arrow, ArrowKind::Both);
        assert_eq!(edge.curve, CurveKind::Step);

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("dash flow roundtrip failed");
        let edge2 = &graph2.edges[0];
        let flow2 = edge2.flow.unwrap();
        assert_eq!(flow2.kind, FlowKind::Dash);
        assert_eq!(flow2.duration_ms, 400);
    }

    #[test]
    fn roundtrip_edge_point_anchors() {
        let input = r#"
edge @standalone {
  from: 100 200
  to: 300 150
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.from, EdgeAnchor::Point(100.0, 200.0));
        assert_eq!(edge.to, EdgeAnchor::Point(300.0, 150.0));
        assert_eq!(edge.arrow, ArrowKind::End);

        let output = emit_document(&graph);
        assert!(output.contains("from: 100 200"));
        assert!(output.contains("to: 300 150"));

        let graph2 = parse_document(&output).expect("point anchor roundtrip failed");
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.from, EdgeAnchor::Point(100.0, 200.0));
        assert_eq!(edge2.to, EdgeAnchor::Point(300.0, 150.0));
    }

    #[test]
    fn roundtrip_edge_mixed_anchors() {
        let input = r#"
rect @start { w: 50 h: 50 }

edge @dangling {
  from: @start
  to: 400 300
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.from, EdgeAnchor::Node(NodeId::intern("start")));
        assert_eq!(edge.to, EdgeAnchor::Point(400.0, 300.0));

        let output = emit_document(&graph);
        assert!(output.contains("from: @start"));
        assert!(output.contains("to: 400 300"));

        let graph2 = parse_document(&output).expect("mixed anchor roundtrip failed");
        let edge2 = &graph2.edges[0];
        assert_eq!(edge2.from, EdgeAnchor::Node(NodeId::intern("start")));
        assert_eq!(edge2.to, EdgeAnchor::Point(400.0, 300.0));
    }

    #[test]
    fn parse_edge_omitted_anchors_default() {
        let input = r#"
edge @no_endpoints {
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.edges.len(), 1);
        let edge = &graph.edges[0];
        assert_eq!(edge.from, EdgeAnchor::Point(0.0, 0.0));
        assert_eq!(edge.to, EdgeAnchor::Point(0.0, 0.0));
    }

    #[test]
    fn test_spec_markdown_basic() {
        let input = r#"
rect @login_btn {
  spec {
    "Primary CTA for login"
    accept: "disabled when fields empty"
    status: doing
    priority: high
    tag: auth
  }
  w: 280 h: 48
  fill: #6C5CE7
}
"#;
        let graph = parse_document(input).unwrap();
        let md = emit_spec_markdown(&graph, "login.fd");

        assert!(md.starts_with("# Spec: login.fd\n"));
        assert!(md.contains("## @login_btn `rect`"));
        assert!(md.contains("> Primary CTA for login"));
        assert!(md.contains("- [ ] disabled when fields empty"));
        assert!(md.contains("- **Status:** doing"));
        assert!(md.contains("- **Priority:** high"));
        assert!(md.contains("- **Tag:** auth"));
        // Visual props must NOT appear
        assert!(!md.contains("280"));
        assert!(!md.contains("6C5CE7"));
    }

    #[test]
    fn test_spec_markdown_nested() {
        let input = r#"
group @form {
  layout: column gap=16 pad=32
  spec {
    "Shipping address form"
    accept: "autofill from saved addresses"
  }

  rect @email {
    spec {
      "Email input"
      accept: "validates email format"
    }
    w: 280 h: 44
  }

  rect @no_annotations {
    w: 100 h: 50
    fill: #CCC
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let md = emit_spec_markdown(&graph, "checkout.fd");

        assert!(md.contains("## @form `group`"));
        assert!(md.contains("### @email `rect`"));
        assert!(md.contains("> Shipping address form"));
        assert!(md.contains("- [ ] autofill from saved addresses"));
        assert!(md.contains("- [ ] validates email format"));
        // Node without annotations should be skipped
        assert!(!md.contains("no_annotations"));
    }

    #[test]
    fn test_spec_markdown_with_edges() {
        let input = r#"
rect @login { w: 200 h: 100 }
rect @dashboard {
  spec "Main dashboard"
  w: 200 h: 100
}

edge @auth_flow {
  spec {
    "Authentication flow"
    accept: "redirect within 2s"
  }
  from: @login
  to: @dashboard
  label: "on success"
  arrow: end
}
"#;
        let graph = parse_document(input).unwrap();
        let md = emit_spec_markdown(&graph, "flow.fd");

        assert!(md.contains("## Flows"));
        assert!(md.contains("**@login** → **@dashboard**"));
        assert!(md.contains("on success"));
        assert!(md.contains("> Authentication flow"));
        assert!(md.contains("- [ ] redirect within 2s"));
    }

    #[test]
    fn roundtrip_import_basic() {
        let input = "import \"components/buttons.fd\" as btn\nrect @hero { w: 200 h: 100 }\n";
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.imports.len(), 1);
        assert_eq!(graph.imports[0].path, "components/buttons.fd");
        assert_eq!(graph.imports[0].namespace, "btn");

        let output = emit_document(&graph);
        assert!(output.contains("import \"components/buttons.fd\" as btn"));

        let graph2 = parse_document(&output).expect("re-parse of import failed");
        assert_eq!(graph2.imports.len(), 1);
        assert_eq!(graph2.imports[0].path, "components/buttons.fd");
        assert_eq!(graph2.imports[0].namespace, "btn");
    }

    #[test]
    fn roundtrip_import_multiple() {
        let input = "import \"tokens.fd\" as tokens\nimport \"buttons.fd\" as btn\nrect @box { w: 50 h: 50 }\n";
        let graph = parse_document(input).unwrap();
        assert_eq!(graph.imports.len(), 2);
        assert_eq!(graph.imports[0].namespace, "tokens");
        assert_eq!(graph.imports[1].namespace, "btn");

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of multiple imports failed");
        assert_eq!(graph2.imports.len(), 2);
        assert_eq!(graph2.imports[0].namespace, "tokens");
        assert_eq!(graph2.imports[1].namespace, "btn");
    }

    #[test]
    fn parse_import_without_alias_errors() {
        let input = "import \"missing_alias.fd\"\nrect @box { w: 50 h: 50 }\n";
        // This should fail because "as namespace" is missing
        let result = parse_document(input);
        assert!(result.is_err());
    }

    #[test]
    fn roundtrip_comment_preserved() {
        // A `# comment` before a node should survive parse → emit → parse.
        let input = r#"
# This is a section header
rect @box {
  w: 100 h: 50
  fill: #FF0000
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        assert!(
            output.contains("# This is a section header"),
            "comment should appear in emitted output: {output}"
        );
        // Re-parse should also preserve it
        let graph2 = parse_document(&output).expect("re-parse of commented document failed");
        let node = graph2.get_by_id(NodeId::intern("box")).unwrap();
        assert_eq!(node.comments, vec!["This is a section header"]);
    }

    #[test]
    fn roundtrip_multiple_comments_preserved() {
        let input = r#"
# Header section
# Subheading
rect @panel {
  w: 300 h: 200
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse failed");
        let node = graph2.get_by_id(NodeId::intern("panel")).unwrap();
        assert_eq!(node.comments.len(), 2);
        assert_eq!(node.comments[0], "Header section");
        assert_eq!(node.comments[1], "Subheading");
    }

    #[test]
    fn roundtrip_inline_position() {
        let input = r#"
rect @placed {
  x: 100
  y: 200
  w: 50 h: 50
  fill: #FF0000
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("placed")).unwrap();

        // Should have a Position constraint from x:/y: parsing
        assert!(
            node.constraints
                .iter()
                .any(|c| matches!(c, Constraint::Position { .. })),
            "should have Position constraint"
        );

        // Emit and verify x:/y: appear inline (not as top-level arrow)
        let output = emit_document(&graph);
        assert!(output.contains("x: 100"), "should emit x: inline");
        assert!(output.contains("y: 200"), "should emit y: inline");
        assert!(
            !output.contains("-> absolute"),
            "should NOT emit old absolute arrow"
        );
        assert!(
            !output.contains("-> position"),
            "should NOT emit position arrow"
        );

        // Round-trip: re-parse emitted output
        let graph2 = parse_document(&output).expect("re-parse of inline position failed");
        let node2 = graph2.get_by_id(NodeId::intern("placed")).unwrap();
        let pos = node2
            .constraints
            .iter()
            .find_map(|c| match c {
                Constraint::Position { x, y } => Some((*x, *y)),
                _ => None,
            })
            .expect("Position constraint missing after roundtrip");
        assert_eq!(pos, (100.0, 200.0));
    }

    #[test]
    fn emit_children_before_styles() {
        let input = r#"
rect @box {
  w: 200 h: 100
  fill: #FF0000
  corner: 10
  text @label "Hello" {
    fill: #FFFFFF
    font: "Inter" 600 14
  }
  anim :hover {
    fill: #CC0000
    ease: ease_out 200ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Children should appear before inline appearance properties
        let child_pos = output.find("text @label").expect("child missing");
        let fill_pos = output.find("fill: #FF0000").expect("fill missing");
        let corner_pos = output.find("corner: 10").expect("corner missing");
        let anim_pos = output.find("when :hover").expect("when missing");

        assert!(
            child_pos < fill_pos,
            "children should appear before fill: child_pos={child_pos} fill_pos={fill_pos}"
        );
        assert!(
            child_pos < corner_pos,
            "children should appear before corner"
        );
        assert!(fill_pos < anim_pos, "fill should appear before animations");
    }

    #[test]
    fn emit_section_separators() {
        let input = r#"
style accent {
  fill: #6C5CE7
}

rect @a {
  w: 100 h: 50
}

rect @b {
  w: 100 h: 50
}

edge @flow {
  from: @a
  to: @b
  arrow: end
}

@a -> center_in: canvas
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        assert!(
            output.contains("# ─── Themes ───"),
            "should have Themes separator"
        );
        assert!(
            output.contains("# ─── Layout ───"),
            "should have Layout separator"
        );
        assert!(
            output.contains("# ─── Flows ───"),
            "should have Flows separator"
        );
    }

    #[test]
    fn roundtrip_children_before_styles() {
        let input = r#"
group @card {
  layout: column gap=12 pad=20
  text @title "Dashboard" {
    font: "Inter" 600 20
    fill: #111111
  }
  rect @body {
    w: 300 h: 200
    fill: #F5F5F5
  }
  fill: #FFFFFF
  corner: 8
  shadow: (0,2,8,#00000011)
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Re-parse the re-ordered output
        let graph2 = parse_document(&output).expect("re-parse of reordered output failed");
        let card_idx = graph2.index_of(NodeId::intern("card")).unwrap();
        assert_eq!(
            graph2.children(card_idx).len(),
            2,
            "card should still have 2 children after roundtrip"
        );

        // Verify children appear before appearance
        let child_pos = output.find("text @title").expect("child missing");
        let fill_pos = output.find("fill: #FFFFFF").expect("card fill missing");
        assert!(
            child_pos < fill_pos,
            "children should appear before parent fill"
        );
    }

    #[test]
    fn roundtrip_theme_keyword() {
        // Verify that `theme` keyword parses and emits correctly
        let input = r#"
theme accent {
  fill: #6C5CE7
  corner: 12
}

rect @btn {
  w: 120 h: 40
  use: accent
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Emitter should output `theme`, not `style`
        assert!(
            output.contains("theme accent"),
            "should emit `theme` keyword"
        );
        assert!(
            !output.contains("style accent"),
            "should NOT emit `style` keyword"
        );

        // Round-trip: re-parse emitted output
        let graph2 = parse_document(&output).expect("re-parse of theme output failed");
        assert!(
            graph2.styles.contains_key(&NodeId::intern("accent")),
            "theme definition should survive roundtrip"
        );
    }

    #[test]
    fn roundtrip_when_keyword() {
        // Verify that `when` keyword parses and emits correctly
        let input = r#"
rect @btn {
  w: 120 h: 40
  fill: #6C5CE7
  when :hover {
    fill: #5A4BD1
    ease: ease_out 200ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Emitter should output `when`, not `anim`
        assert!(output.contains("when :hover"), "should emit `when` keyword");
        assert!(
            !output.contains("anim :hover"),
            "should NOT emit `anim` keyword"
        );

        // Round-trip: re-parse emitted output
        let graph2 = parse_document(&output).expect("re-parse of when output failed");
        let node = graph2.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(
            node.animations.len(),
            1,
            "animation should survive roundtrip"
        );
        assert_eq!(
            node.animations[0].trigger,
            AnimTrigger::Hover,
            "trigger should be Hover"
        );
    }

    #[test]
    fn parse_old_style_keyword_compat() {
        // Old `style` keyword must still be accepted by the parser
        let input = r#"
style accent {
  fill: #6C5CE7
}

rect @btn {
  w: 120 h: 40
  use: accent
}
"#;
        let graph = parse_document(input).unwrap();
        assert!(
            graph.styles.contains_key(&NodeId::intern("accent")),
            "old `style` keyword should parse into a theme definition"
        );

        // Emitter should upgrade to `theme`
        let output = emit_document(&graph);
        assert!(
            output.contains("theme accent"),
            "emitter should upgrade `style` to `theme`"
        );
    }

    #[test]
    fn parse_old_anim_keyword_compat() {
        // Old `anim` keyword must still be accepted by the parser
        let input = r#"
rect @btn {
  w: 120 h: 40
  fill: #6C5CE7
  anim :press {
    scale: 0.95
    ease: spring 150ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(
            node.animations.len(),
            1,
            "old `anim` keyword should parse into animation"
        );
        assert_eq!(
            node.animations[0].trigger,
            AnimTrigger::Press,
            "trigger should be Press"
        );

        // Emitter should upgrade to `when`
        let output = emit_document(&graph);
        assert!(
            output.contains("when :press"),
            "emitter should upgrade `anim` to `when`"
        );
    }

    #[test]
    fn roundtrip_theme_import() {
        // Verify that import + theme references work together
        let input = r#"
import "tokens.fd" as tokens

theme card_base {
  fill: #FFFFFF
  corner: 16
}

rect @card {
  w: 300 h: 200
  use: card_base
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);

        // Both import and theme should appear in output
        assert!(
            output.contains("import \"tokens.fd\" as tokens"),
            "import should survive roundtrip"
        );
        assert!(
            output.contains("theme card_base"),
            "theme should survive roundtrip"
        );

        // Re-parse
        let graph2 = parse_document(&output).expect("re-parse failed");
        assert_eq!(graph2.imports.len(), 1, "import count should survive");
        assert!(
            graph2.styles.contains_key(&NodeId::intern("card_base")),
            "theme def should survive"
        );
    }

    // ─── Hardening: edge-case round-trip tests ─────────────────────────────

    #[test]
    fn roundtrip_empty_group() {
        let input = "group @empty {\n}\n";
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of empty group failed");
        let node = graph2.get_by_id(NodeId::intern("empty")).unwrap();
        assert!(matches!(node.kind, NodeKind::Group));
    }

    #[test]
    fn roundtrip_deeply_nested_groups() {
        let input = r#"
group @outer {
  group @middle {
    group @inner {
      rect @leaf {
        w: 40 h: 20
        fill: #FF0000
      }
    }
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of nested groups failed");
        let leaf = graph2.get_by_id(NodeId::intern("leaf")).unwrap();
        assert!(matches!(leaf.kind, NodeKind::Rect { .. }));
        // Verify 3-level nesting preserved
        let inner_idx = graph2.index_of(NodeId::intern("inner")).unwrap();
        assert_eq!(graph2.children(inner_idx).len(), 1);
        let middle_idx = graph2.index_of(NodeId::intern("middle")).unwrap();
        assert_eq!(graph2.children(middle_idx).len(), 1);
    }

    #[test]
    fn roundtrip_unicode_text() {
        let input = "text @emoji \"Hello 🎨 café 日本語\" {\n  fill: #333333\n}\n";
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        assert!(
            output.contains("Hello 🎨 café 日本語"),
            "unicode should survive emit"
        );
        let graph2 = parse_document(&output).expect("re-parse of unicode failed");
        let node = graph2.get_by_id(NodeId::intern("emoji")).unwrap();
        match &node.kind {
            NodeKind::Text { content } => {
                assert!(content.contains("🎨"));
                assert!(content.contains("café"));
                assert!(content.contains("日本語"));
            }
            _ => panic!("expected Text node"),
        }
    }

    #[test]
    fn roundtrip_spec_all_fields() {
        let input = r#"
rect @full_spec {
  spec {
    "Full specification node"
    accept: "all fields present"
    status: doing
    priority: high
    tag: mvp, auth
  }
  w: 100 h: 50
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("full_spec")).unwrap();
        assert_eq!(node.annotations.len(), 5, "should have 5 annotations");

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of full spec failed");
        let node2 = graph2.get_by_id(NodeId::intern("full_spec")).unwrap();
        assert_eq!(node2.annotations.len(), 5);
        assert_eq!(node2.annotations, node.annotations);
    }

    #[test]
    fn roundtrip_path_node() {
        let input = "path @sketch {\n}\n";
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of path failed");
        let node = graph2.get_by_id(NodeId::intern("sketch")).unwrap();
        assert!(matches!(node.kind, NodeKind::Path { .. }));
    }

    #[test]
    fn roundtrip_gradient_linear() {
        let input = r#"
rect @grad {
  w: 200 h: 100
  fill: linear(90deg, #FF0000 0, #0000FF 1)
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("grad")).unwrap();
        assert!(matches!(
            node.style.fill,
            Some(Paint::LinearGradient { .. })
        ));

        let output = emit_document(&graph);
        assert!(output.contains("linear("), "should emit linear gradient");
        let graph2 = parse_document(&output).expect("re-parse of linear gradient failed");
        let node2 = graph2.get_by_id(NodeId::intern("grad")).unwrap();
        assert!(matches!(
            node2.style.fill,
            Some(Paint::LinearGradient { .. })
        ));
    }

    #[test]
    fn roundtrip_gradient_radial() {
        let input = r#"
rect @radial_box {
  w: 100 h: 100
  fill: radial(#FFFFFF 0, #000000 1)
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("radial_box")).unwrap();
        assert!(matches!(
            node.style.fill,
            Some(Paint::RadialGradient { .. })
        ));

        let output = emit_document(&graph);
        assert!(output.contains("radial("), "should emit radial gradient");
        let graph2 = parse_document(&output).expect("re-parse of radial gradient failed");
        let node2 = graph2.get_by_id(NodeId::intern("radial_box")).unwrap();
        assert!(matches!(
            node2.style.fill,
            Some(Paint::RadialGradient { .. })
        ));
    }

    #[test]
    fn roundtrip_shadow_property() {
        let input = r#"
rect @shadowed {
  w: 200 h: 100
  shadow: (0,4,20,#000000)
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("shadowed")).unwrap();
        let shadow = node.style.shadow.as_ref().expect("shadow should exist");
        assert_eq!(shadow.blur, 20.0);

        let output = emit_document(&graph);
        assert!(output.contains("shadow:"), "should emit shadow");
        let graph2 = parse_document(&output).expect("re-parse of shadow failed");
        let node2 = graph2.get_by_id(NodeId::intern("shadowed")).unwrap();
        let shadow2 = node2.style.shadow.as_ref().expect("shadow should survive");
        assert_eq!(shadow2.offset_y, 4.0);
        assert_eq!(shadow2.blur, 20.0);
    }

    #[test]
    fn roundtrip_opacity() {
        let input = r#"
rect @faded {
  w: 100 h: 100
  fill: #6C5CE7
  opacity: 0.5
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("faded")).unwrap();
        assert_eq!(node.style.opacity, Some(0.5));

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of opacity failed");
        let node2 = graph2.get_by_id(NodeId::intern("faded")).unwrap();
        assert_eq!(node2.style.opacity, Some(0.5));
    }

    #[test]
    fn roundtrip_clip_frame() {
        let input = r#"
frame @clipped {
  w: 300 h: 200
  clip: true
  fill: #FFFFFF
  corner: 12
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        assert!(output.contains("clip: true"), "should emit clip");
        let graph2 = parse_document(&output).expect("re-parse of clip frame failed");
        let node = graph2.get_by_id(NodeId::intern("clipped")).unwrap();
        match &node.kind {
            NodeKind::Frame { clip, .. } => assert!(clip, "clip should be true"),
            _ => panic!("expected Frame node"),
        }
    }

    #[test]
    fn roundtrip_multiple_animations() {
        let input = r#"
rect @animated {
  w: 120 h: 40
  fill: #6C5CE7
  when :hover {
    fill: #5A4BD1
    scale: 1.05
    ease: ease_out 200ms
  }
  when :press {
    scale: 0.95
    ease: spring 150ms
  }
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("animated")).unwrap();
        assert_eq!(node.animations.len(), 2, "should have 2 animations");

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of multi-anim failed");
        let node2 = graph2.get_by_id(NodeId::intern("animated")).unwrap();
        assert_eq!(node2.animations.len(), 2);
        assert_eq!(node2.animations[0].trigger, AnimTrigger::Hover);
        assert_eq!(node2.animations[1].trigger, AnimTrigger::Press);
    }

    #[test]
    fn roundtrip_inline_spec_shorthand() {
        let input = r#"
rect @btn {
  spec "Primary action button"
  w: 180 h: 48
  fill: #6C5CE7
}
"#;
        let graph = parse_document(input).unwrap();
        let node = graph.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(node.annotations.len(), 1);
        assert!(matches!(
            &node.annotations[0],
            Annotation::Description(d) if d == "Primary action button"
        ));

        let output = emit_document(&graph);
        let graph2 = parse_document(&output).expect("re-parse of inline spec failed");
        let node2 = graph2.get_by_id(NodeId::intern("btn")).unwrap();
        assert_eq!(node2.annotations, node.annotations);
    }

    #[test]
    fn roundtrip_layout_modes() {
        let input = r#"
frame @col {
  w: 400 h: 300
  layout: column gap=16 pad=24
  rect @c1 { w: 100 h: 50 }
}

frame @rw {
  w: 400 h: 300
  layout: row gap=8 pad=12
  rect @r1 { w: 50 h: 50 }
}

frame @grd {
  w: 400 h: 300
  layout: grid cols=2 gap=10 pad=20
  rect @g1 { w: 80 h: 80 }
}
"#;
        let graph = parse_document(input).unwrap();
        let output = emit_document(&graph);
        assert!(output.contains("layout: column gap=16 pad=24"));
        assert!(output.contains("layout: row gap=8 pad=12"));
        assert!(output.contains("layout: grid cols=2 gap=10 pad=20"));

        let graph2 = parse_document(&output).expect("re-parse of layout modes failed");
        let col = graph2.get_by_id(NodeId::intern("col")).unwrap();
        assert!(matches!(col.kind, NodeKind::Frame { .. }));
        let rw = graph2.get_by_id(NodeId::intern("rw")).unwrap();
        assert!(matches!(rw.kind, NodeKind::Frame { .. }));
        let grd = graph2.get_by_id(NodeId::intern("grd")).unwrap();
        assert!(matches!(grd.kind, NodeKind::Frame { .. }));
    }

    // ─── emit_filtered tests ─────────────────────────────────────────────

    fn make_test_graph() -> SceneGraph {
        // A rich document with styles, layout, anims, specs, and edges
        let input = r#"
theme accent {
  fill: #6C5CE7
  font: "Inter" bold 16
}

frame @container {
  w: 600 h: 400
  layout: column gap=16 pad=24

  rect @card {
    w: 200 h: 100
    use: accent
    fill: #FFFFFF
    corner: 12
    spec {
      "Main card component"
      status: done
    }
    when :hover {
      fill: #F0EDFF
      scale: 1.05
      ease: ease_out 200ms
    }
  }

  text @label "Hello" {
    font: "Inter" regular 14
    fill: #333333
    x: 20
    y: 40
  }
}

edge @card_to_label {
  from: @card
  to: @label
  label: "displays"
}
"#;
        parse_document(input).unwrap()
    }

    #[test]
    fn emit_filtered_full_matches_emit_document() {
        let graph = make_test_graph();
        let full = emit_filtered(&graph, ReadMode::Full);
        let doc = emit_document(&graph);
        assert_eq!(full, doc, "Full mode should be identical to emit_document");
    }

    #[test]
    fn emit_filtered_structure() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Structure);
        // Should have node declarations
        assert!(out.contains("frame @container"), "should include frame");
        assert!(out.contains("rect @card"), "should include rect");
        assert!(out.contains("text @label"), "should include text");
        // Should NOT have styles, dimensions, specs, or anims
        assert!(!out.contains("fill:"), "no fill in structure mode");
        assert!(!out.contains("spec"), "no spec in structure mode");
        assert!(!out.contains("when"), "no when in structure mode");
        assert!(!out.contains("theme"), "no theme in structure mode");
        assert!(!out.contains("edge"), "no edges in structure mode");
    }

    #[test]
    fn emit_filtered_layout() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Layout);
        // Should have layout + dimensions
        assert!(out.contains("layout: column"), "should include layout");
        assert!(out.contains("w: 200 h: 100"), "should include dims");
        assert!(out.contains("x: 20"), "should include position");
        // Should NOT have styles or anims
        assert!(!out.contains("fill:"), "no fill in layout mode");
        assert!(!out.contains("theme"), "no theme in layout mode");
        assert!(!out.contains("when :hover"), "no when in layout mode");
    }

    #[test]
    fn emit_filtered_design() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Design);
        // Should have themes + styles
        assert!(out.contains("theme accent"), "should include theme");
        assert!(out.contains("use: accent"), "should include use ref");
        assert!(out.contains("fill:"), "should include fill");
        assert!(out.contains("corner: 12"), "should include corner");
        // Should NOT have layout or anims
        assert!(!out.contains("layout:"), "no layout in design mode");
        assert!(!out.contains("w: 200"), "no dims in design mode");
        assert!(!out.contains("when :hover"), "no when in design mode");
    }

    #[test]
    fn emit_filtered_spec() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Spec);
        // Should have spec blocks
        assert!(out.contains("spec"), "should include spec");
        assert!(out.contains("Main card component"), "should include desc");
        assert!(out.contains("status: done"), "should include status");
        // Should NOT have styles or anims
        assert!(!out.contains("fill:"), "no fill in spec mode");
        assert!(!out.contains("when"), "no when in spec mode");
    }

    #[test]
    fn emit_filtered_visual() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Visual);
        // Visual = Layout + Design + When
        assert!(out.contains("theme accent"), "should include theme");
        assert!(out.contains("layout: column"), "should include layout");
        assert!(out.contains("w: 200 h: 100"), "should include dims");
        assert!(out.contains("fill:"), "should include fill");
        assert!(out.contains("corner: 12"), "should include corner");
        assert!(out.contains("when :hover"), "should include when");
        assert!(out.contains("scale: 1.05"), "should include anim props");
        assert!(out.contains("edge @card_to_label"), "should include edges");
        // Should NOT have spec blocks
        assert!(
            !out.contains("Main card component"),
            "no spec desc in visual mode"
        );
    }

    #[test]
    fn emit_filtered_when() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::When);
        // Should have when blocks
        assert!(out.contains("when :hover"), "should include when");
        assert!(out.contains("scale: 1.05"), "should include anim props");
        // Should NOT have node-level styles, layout, or spec
        assert!(!out.contains("corner:"), "no corner in when mode");
        assert!(!out.contains("w: 200"), "no dims in when mode");
        assert!(!out.contains("theme"), "no theme in when mode");
        assert!(!out.contains("spec"), "no spec in when mode");
    }

    #[test]
    fn emit_filtered_edges() {
        let graph = make_test_graph();
        let out = emit_filtered(&graph, ReadMode::Edges);
        // Should have edges
        assert!(out.contains("edge @card_to_label"), "should include edge");
        assert!(out.contains("from: @card"), "should include from");
        assert!(out.contains("to: @label"), "should include to");
        assert!(
            out.contains("text @_card_to_label_label \"displays\" {}"),
            "should include text child"
        );
        // Should NOT have styles or anims
        assert!(!out.contains("fill:"), "no fill in edges mode");
        assert!(!out.contains("when"), "no when in edges mode");
    }
}