merman-render 0.5.0

use crate::config::config_f64;
use crate::json::from_value_ref;
use crate::model::{
    Bounds, C4BoundaryLayout, C4DiagramLayout, C4ImageLayout, C4RelLayout, C4ShapeLayout,
    C4TextBlockLayout, LayoutPoint,
};
use crate::text::{TextMeasurer, TextStyle, WrapMode};
use crate::{Error, Result};
use merman_core::diagrams::c4::C4DiagramRenderModel;
use serde_json::Value;
use std::collections::HashMap;

type C4Model = C4DiagramRenderModel;

fn config_bool(cfg: &Value, path: &[&str]) -> Option<bool> {
    let mut cur = cfg;
    for key in path {
        cur = cur.get(*key)?;
    }
    cur.as_bool()
}

fn config_string(cfg: &Value, path: &[&str]) -> Option<String> {
    let mut cur = cfg;
    for key in path {
        cur = cur.get(*key)?;
    }
    cur.as_str().map(|s| s.to_string())
}

#[derive(Debug, Clone)]
struct C4Conf {
    diagram_margin_x: f64,
    diagram_margin_y: f64,
    c4_shape_margin: f64,
    c4_shape_padding: f64,
    width: f64,
    height: f64,
    wrap: bool,
    next_line_padding_x: f64,
    boundary_font_family: Option<String>,
    boundary_font_size: f64,
    boundary_font_weight: Option<String>,
    message_font_family: Option<String>,
    message_font_size: f64,
    message_font_weight: Option<String>,
}

impl C4Conf {
    fn from_effective_config(effective_config: &Value) -> Self {
        // Mermaid's C4 renderer (`packages/mermaid/src/diagrams/c4/c4Renderer.js`) calls
        // `setConf(diagObj.db.getConfig())`, where `getConfig()` yields the diagram config object
        // (i.e. `config.c4`), not the global config root. As a result, top-level `fontFamily`,
        // `fontSize`, and `fontWeight` do not override C4-specific font defaults.
        let message_font_family = config_string(effective_config, &["c4", "messageFontFamily"]);
        let message_font_size =
            config_f64(effective_config, &["c4", "messageFontSize"]).unwrap_or(12.0);
        let message_font_weight = config_string(effective_config, &["c4", "messageFontWeight"]);

        let boundary_font_family = config_string(effective_config, &["c4", "boundaryFontFamily"]);
        let boundary_font_size =
            config_f64(effective_config, &["c4", "boundaryFontSize"]).unwrap_or(14.0);
        let boundary_font_weight = config_string(effective_config, &["c4", "boundaryFontWeight"]);

        Self {
            diagram_margin_x: config_f64(effective_config, &["c4", "diagramMarginX"])
                .unwrap_or(50.0),
            diagram_margin_y: config_f64(effective_config, &["c4", "diagramMarginY"])
                .unwrap_or(10.0),
            c4_shape_margin: config_f64(effective_config, &["c4", "c4ShapeMargin"]).unwrap_or(50.0),
            c4_shape_padding: config_f64(effective_config, &["c4", "c4ShapePadding"])
                .unwrap_or(20.0),
            width: config_f64(effective_config, &["c4", "width"]).unwrap_or(216.0),
            height: config_f64(effective_config, &["c4", "height"]).unwrap_or(60.0),
            wrap: config_bool(effective_config, &["c4", "wrap"]).unwrap_or(true),
            next_line_padding_x: config_f64(effective_config, &["c4", "nextLinePaddingX"])
                .unwrap_or(0.0),
            boundary_font_family,
            boundary_font_size,
            boundary_font_weight,
            message_font_family,
            message_font_size,
            message_font_weight,
        }
    }

    fn boundary_font(&self) -> TextStyle {
        TextStyle {
            font_family: self.boundary_font_family.clone(),
            font_size: self.boundary_font_size,
            font_weight: self.boundary_font_weight.clone(),
        }
    }

    fn message_font(&self) -> TextStyle {
        TextStyle {
            font_family: self.message_font_family.clone(),
            font_size: self.message_font_size,
            font_weight: self.message_font_weight.clone(),
        }
    }

    fn c4_shape_font(&self, effective_config: &Value, type_c4_shape: &str) -> TextStyle {
        let key_family = format!("{type_c4_shape}FontFamily");
        let key_size = format!("{type_c4_shape}FontSize");
        let key_weight = format!("{type_c4_shape}FontWeight");

        let font_family = config_string(effective_config, &["c4", &key_family]);
        let font_size = config_f64(effective_config, &["c4", &key_size]).unwrap_or(14.0);
        let font_weight = config_string(effective_config, &["c4", &key_weight]);

        TextStyle {
            font_family,
            font_size,
            font_weight,
        }
    }
}

#[derive(Debug, Clone, Copy)]
struct TextMeasure {
    width: f64,
    height: f64,
    line_count: usize,
}

fn js_round_pos(v: f64) -> f64 {
    if !(v.is_finite() && v >= 0.0) {
        0.0
    } else {
        (v + 0.5).floor()
    }
}

fn c4_svg_bbox_line_height_px(style: &TextStyle) -> f64 {
    // C4 in Mermaid@11.12.2 uses `calculateTextDimensions(...).height`, which is measured via
    // SVG `getBBox()` and rounded with `Math.round`. Upstream fixtures show stable, integer
    // per-line heights for the default C4 fonts:
    // - 12px -> 14px
    // - 14px -> 16px
    // - 16px -> 17px
    //
    // These do not match our generic deterministic SVG line-height approximation (`1.1em`),
    // so C4 owns the small rule directly instead of keeping it in generated parity data.
    let fs = js_round_pos(style.font_size.max(1.0)) as i64;
    match fs {
        12 => 14.0,
        14 => 16.0,
        16 => 17.0,
        _ => js_round_pos(style.font_size.max(1.0) * 1.1),
    }
}

fn measure_c4_text(
    measurer: &dyn TextMeasurer,
    text: &str,
    style: &TextStyle,
    wrap: bool,
    text_limit_width: f64,
) -> TextMeasure {
    // Mermaid's `calculateTextWidth/Height` (used by C4) draws SVG `<text>` nodes, calls
    // `getBBox()`, and then applies `Math.round(...)` per line. To keep C4 layout + viewport
    // parity with upstream SVG baselines, we mirror that integer rounding behavior here.
    if wrap {
        let m = measurer.measure_wrapped(text, style, Some(text_limit_width), WrapMode::SvgLike);
        return TextMeasure {
            width: text_limit_width,
            height: c4_svg_bbox_line_height_px(style) * m.line_count.max(1) as f64,
            line_count: m.line_count,
        };
    }

    let mut width: f64 = 0.0;
    let lines = crate::text::DeterministicTextMeasurer::normalized_text_lines(text);
    for line in &lines {
        let m = measurer.measure(line, style);
        width = width.max(js_round_pos(m.width));
    }
    let height = c4_svg_bbox_line_height_px(style) * lines.len().max(1) as f64;
    TextMeasure {
        width,
        height,
        line_count: lines.len().max(1),
    }
}

#[derive(Debug, Clone, Default)]
struct BoundsData {
    startx: Option<f64>,
    stopx: Option<f64>,
    starty: Option<f64>,
    stopy: Option<f64>,
    width_limit: f64,
}

#[derive(Debug, Clone, Default)]
struct BoundsNext {
    startx: f64,
    stopx: f64,
    starty: f64,
    stopy: f64,
    cnt: usize,
}

#[derive(Debug, Clone, Default)]
struct BoundsState {
    data: BoundsData,
    next: BoundsNext,
}

impl BoundsState {
    fn set_data(&mut self, startx: f64, stopx: f64, starty: f64, stopy: f64) {
        self.next.startx = startx;
        self.data.startx = Some(startx);
        self.next.stopx = stopx;
        self.data.stopx = Some(stopx);
        self.next.starty = starty;
        self.data.starty = Some(starty);
        self.next.stopy = stopy;
        self.data.stopy = Some(stopy);
    }

    fn bump_last_margin(&mut self, margin: f64) {
        if let Some(v) = self.data.stopx.as_mut() {
            *v += margin;
        }
        if let Some(v) = self.data.stopy.as_mut() {
            *v += margin;
        }
    }

    fn update_val_opt(target: &mut Option<f64>, val: f64, fun: fn(f64, f64) -> f64) {
        match target {
            None => *target = Some(val),
            Some(existing) => *existing = fun(val, *existing),
        }
    }

    fn update_val(target: &mut f64, val: f64, fun: fn(f64, f64) -> f64) {
        *target = fun(val, *target);
    }

    fn insert_rect(&mut self, rect: &mut Rect, c4_shape_in_row: usize, conf: &C4Conf) {
        self.next.cnt += 1;

        let startx = if self.next.startx == self.next.stopx {
            self.next.stopx + rect.margin
        } else {
            self.next.stopx + rect.margin * 2.0
        };
        let mut stopx = startx + rect.size.width;
        let starty = self.next.starty + rect.margin * 2.0;
        let mut stopy = starty + rect.size.height;

        if startx >= self.data.width_limit
            || stopx >= self.data.width_limit
            || self.next.cnt > c4_shape_in_row
        {
            let startx2 = self.next.startx + rect.margin + conf.next_line_padding_x;
            let starty2 = self.next.stopy + rect.margin * 2.0;

            stopx = startx2 + rect.size.width;
            stopy = starty2 + rect.size.height;

            self.next.stopx = stopx;
            self.next.starty = self.next.stopy;
            self.next.stopy = stopy;
            self.next.cnt = 1;

            rect.origin.x = startx2;
            rect.origin.y = starty2;
        } else {
            rect.origin.x = startx;
            rect.origin.y = starty;
        }

        Self::update_val_opt(&mut self.data.startx, rect.origin.x, f64::min);
        Self::update_val_opt(&mut self.data.starty, rect.origin.y, f64::min);
        Self::update_val_opt(&mut self.data.stopx, stopx, f64::max);
        Self::update_val_opt(&mut self.data.stopy, stopy, f64::max);

        Self::update_val(&mut self.next.startx, rect.origin.x, f64::min);
        Self::update_val(&mut self.next.starty, rect.origin.y, f64::min);
        Self::update_val(&mut self.next.stopx, stopx, f64::max);
        Self::update_val(&mut self.next.stopy, stopy, f64::max);
    }
}

#[derive(Debug, Clone)]
struct Rect {
    origin: merman_core::geom::Point,
    size: merman_core::geom::Size,
    margin: f64,
}

struct C4LayoutContext<'a> {
    model: &'a C4Model,
    effective_config: &'a Value,
    conf: &'a C4Conf,
    c4_shape_in_row: usize,
    c4_boundary_in_row: usize,
    measurer: &'a dyn TextMeasurer,
    boundary_children: &'a HashMap<String, Vec<usize>>,
    shape_children: &'a HashMap<String, Vec<usize>>,
}

struct C4LayoutState {
    boundaries: HashMap<String, C4BoundaryLayout>,
    shapes: HashMap<String, C4ShapeLayout>,
    global_max_x: f64,
    global_max_y: f64,
}

fn has_sprite(v: &Option<Value>) -> bool {
    v.as_ref().is_some_and(|v| match v {
        Value::Null => false,
        Value::Bool(b) => *b,
        Value::Number(_) => true,
        Value::String(s) => !s.trim().is_empty(),
        Value::Array(a) => !a.is_empty(),
        Value::Object(o) => !o.is_empty(),
    })
}

fn intersect_point(from: &Rect, end_point: LayoutPoint) -> LayoutPoint {
    let x1 = from.origin.x;
    let y1 = from.origin.y;
    let x2 = end_point.x;
    let y2 = end_point.y;

    let from_center_x = x1 + from.size.width / 2.0;
    let from_center_y = y1 + from.size.height / 2.0;

    let dx = (x1 - x2).abs();
    let dy = (y1 - y2).abs();
    let tan_dyx = dy / dx;
    let from_dyx = from.size.height / from.size.width;

    let mut return_point: Option<LayoutPoint> = None;

    if y1 == y2 && x1 < x2 {
        return_point = Some(LayoutPoint {
            x: x1 + from.size.width,
            y: from_center_y,
        });
    } else if y1 == y2 && x1 > x2 {
        return_point = Some(LayoutPoint {
            x: x1,
            y: from_center_y,
        });
    } else if x1 == x2 && y1 < y2 {
        return_point = Some(LayoutPoint {
            x: from_center_x,
            y: y1 + from.size.height,
        });
    } else if x1 == x2 && y1 > y2 {
        return_point = Some(LayoutPoint {
            x: from_center_x,
            y: y1,
        });
    }

    if x1 > x2 && y1 < y2 {
        if from_dyx >= tan_dyx {
            return_point = Some(LayoutPoint {
                x: x1,
                y: from_center_y + (tan_dyx * from.size.width) / 2.0,
            });
        } else {
            return_point = Some(LayoutPoint {
                x: from_center_x - ((dx / dy) * from.size.height) / 2.0,
                y: y1 + from.size.height,
            });
        }
    } else if x1 < x2 && y1 < y2 {
        if from_dyx >= tan_dyx {
            return_point = Some(LayoutPoint {
                x: x1 + from.size.width,
                y: from_center_y + (tan_dyx * from.size.width) / 2.0,
            });
        } else {
            return_point = Some(LayoutPoint {
                x: from_center_x + ((dx / dy) * from.size.height) / 2.0,
                y: y1 + from.size.height,
            });
        }
    } else if x1 < x2 && y1 > y2 {
        if from_dyx >= tan_dyx {
            return_point = Some(LayoutPoint {
                x: x1 + from.size.width,
                y: from_center_y - (tan_dyx * from.size.width) / 2.0,
            });
        } else {
            return_point = Some(LayoutPoint {
                x: from_center_x + ((from.size.height / 2.0) * dx) / dy,
                y: y1,
            });
        }
    } else if x1 > x2 && y1 > y2 {
        if from_dyx >= tan_dyx {
            return_point = Some(LayoutPoint {
                x: x1,
                y: from_center_y - (from.size.width / 2.0) * tan_dyx,
            });
        } else {
            return_point = Some(LayoutPoint {
                x: from_center_x - ((from.size.height / 2.0) * dx) / dy,
                y: y1,
            });
        }
    }

    return_point.unwrap_or(LayoutPoint {
        x: from_center_x,
        y: from_center_y,
    })
}

fn intersect_points(from: &Rect, to: &Rect) -> (LayoutPoint, LayoutPoint) {
    let end_intersect_point = LayoutPoint {
        x: to.origin.x + to.size.width / 2.0,
        y: to.origin.y + to.size.height / 2.0,
    };
    let start_point = intersect_point(from, end_intersect_point);

    let end_intersect_point = LayoutPoint {
        x: from.origin.x + from.size.width / 2.0,
        y: from.origin.y + from.size.height / 2.0,
    };
    let end_point = intersect_point(to, end_intersect_point);

    (start_point, end_point)
}

fn layout_c4_shape_array(
    current_bounds: &mut BoundsState,
    shape_indices: &[usize],
    ctx: &C4LayoutContext<'_>,
    state: &mut C4LayoutState,
) {
    for idx in shape_indices {
        let shape = &ctx.model.shapes[*idx];
        let mut y = ctx.conf.c4_shape_padding;

        let type_c4_shape = shape.type_c4_shape.as_str().to_string();
        let mut type_conf = ctx.conf.c4_shape_font(ctx.effective_config, &type_c4_shape);
        type_conf.font_size -= 2.0;

        let type_text = format!("«{}»", type_c4_shape);
        let type_metrics = ctx.measurer.measure(&type_text, &type_conf);
        let type_block = C4TextBlockLayout {
            text: type_text,
            y,
            width: type_metrics.width,
            height: type_conf.font_size + 2.0,
            line_count: 1,
        };
        y = y + type_block.height - 4.0;

        let mut image = C4ImageLayout {
            width: 0.0,
            height: 0.0,
            y: 0.0,
        };
        if matches!(type_c4_shape.as_str(), "person" | "external_person") {
            image.width = 48.0;
            image.height = 48.0;
            image.y = y;
            y = image.y + image.height;
        }
        if has_sprite(&shape.sprite) {
            image.width = 48.0;
            image.height = 48.0;
            image.y = y;
            y = image.y + image.height;
        }

        let text_wrap = shape.wrap && ctx.conf.wrap;
        let text_limit_width = ctx.conf.width - ctx.conf.c4_shape_padding * 2.0;

        let mut label_conf = ctx.conf.c4_shape_font(ctx.effective_config, &type_c4_shape);
        label_conf.font_size += 2.0;
        label_conf.font_weight = Some("bold".to_string());

        let label_text = shape.label.as_str().to_string();
        let label_m = measure_c4_text(
            ctx.measurer,
            &label_text,
            &label_conf,
            text_wrap,
            text_limit_width,
        );
        let label = C4TextBlockLayout {
            text: label_text,
            y: y + 8.0,
            width: label_m.width,
            height: label_m.height,
            line_count: label_m.line_count,
        };
        y = label.y + label.height;

        let mut ty_block: Option<C4TextBlockLayout> = None;
        let mut techn_block: Option<C4TextBlockLayout> = None;

        if let Some(ty) = shape.ty.as_ref().filter(|t| !t.as_str().is_empty()) {
            let type_text = format!("[{}]", ty.as_str());
            let type_conf = ctx.conf.c4_shape_font(ctx.effective_config, &type_c4_shape);
            let m = measure_c4_text(
                ctx.measurer,
                &type_text,
                &type_conf,
                text_wrap,
                text_limit_width,
            );
            let block = C4TextBlockLayout {
                text: type_text,
                y: y + 5.0,
                width: m.width,
                height: m.height,
                line_count: m.line_count,
            };
            y = block.y + block.height;
            ty_block = Some(block);
        } else if let Some(techn) = shape.techn.as_ref().filter(|t| !t.as_str().is_empty()) {
            let techn_text = format!("[{}]", techn.as_str());
            // Mermaid@11.12.2 C4 renderer quirk: `techn` text is measured with
            // `c4ShapeFont(conf, c4Shape.techn.text)`, where `c4Shape.techn.text` already contains
            // the bracketed string (e.g. `[Rust]`). That key does not exist in the config object,
            // so the downstream `calculateTextDimensions` falls back to its defaults
            // (`fontSize=12`, `fontFamily='Arial'`).
            //
            // Upstream SVG baselines encode this behavior into shape heights and ultimately the
            // root viewBox. Mirror it here for parity.
            let techn_conf = TextStyle {
                font_family: Some("Arial".to_string()),
                font_size: 12.0,
                font_weight: None,
            };
            let m = measure_c4_text(
                ctx.measurer,
                &techn_text,
                &techn_conf,
                text_wrap,
                text_limit_width,
            );
            let block = C4TextBlockLayout {
                text: techn_text,
                y: y + 5.0,
                width: m.width,
                height: m.height,
                line_count: m.line_count,
            };
            y = block.y + block.height;
            techn_block = Some(block);
        }

        let mut rect_height = y;
        let mut rect_width = label.width;

        let mut descr_block: Option<C4TextBlockLayout> = None;
        if let Some(descr) = shape.descr.as_ref().filter(|t| !t.as_str().is_empty()) {
            let descr_text = descr.as_str().to_string();
            let descr_conf = ctx.conf.c4_shape_font(ctx.effective_config, &type_c4_shape);
            let m = measure_c4_text(
                ctx.measurer,
                &descr_text,
                &descr_conf,
                text_wrap,
                text_limit_width,
            );
            let block = C4TextBlockLayout {
                text: descr_text,
                y: y + 20.0,
                width: m.width,
                height: m.height,
                line_count: m.line_count,
            };
            y = block.y + block.height;
            rect_width = rect_width.max(block.width);
            rect_height = y - block.line_count as f64 * 5.0;
            descr_block = Some(block);
        }

        rect_width += ctx.conf.c4_shape_padding;

        let width = ctx.conf.width.max(rect_width);
        let height = ctx.conf.height.max(rect_height);
        let margin = ctx.conf.c4_shape_margin;

        let mut rect = Rect {
            origin: merman_core::geom::point(0.0, 0.0),
            size: merman_core::geom::Size::new(width, height),
            margin,
        };
        current_bounds.insert_rect(&mut rect, ctx.c4_shape_in_row, ctx.conf);

        state.shapes.insert(
            shape.alias.clone(),
            C4ShapeLayout {
                alias: shape.alias.clone(),
                parent_boundary: shape.parent_boundary.clone(),
                type_c4_shape: type_c4_shape.clone(),
                x: rect.origin.x,
                y: rect.origin.y,
                width: rect.size.width,
                height: rect.size.height,
                margin: rect.margin,
                image,
                type_block,
                label,
                ty: ty_block,
                techn: techn_block,
                descr: descr_block,
            },
        );
    }

    current_bounds.bump_last_margin(ctx.conf.c4_shape_margin);
}

fn layout_inside_boundary(
    parent_bounds: &mut BoundsState,
    boundary_indices: &[usize],
    ctx: &C4LayoutContext<'_>,
    state: &mut C4LayoutState,
) -> Result<()> {
    let mut current_bounds = BoundsState::default();

    let denom = ctx.c4_boundary_in_row.min(boundary_indices.len().max(1));
    let width_limit = parent_bounds.data.width_limit / denom as f64;
    current_bounds.data.width_limit = width_limit;

    for (i, idx) in boundary_indices.iter().enumerate() {
        let boundary = &ctx.model.boundaries[*idx];
        let mut y = 0.0;

        let mut image = C4ImageLayout {
            width: 0.0,
            height: 0.0,
            y: 0.0,
        };
        if has_sprite(&boundary.sprite) {
            image.width = 48.0;
            image.height = 48.0;
            image.y = y;
            y = image.y + image.height;
        }

        let text_wrap = boundary.wrap.unwrap_or(ctx.model.wrap) && ctx.conf.wrap;
        let mut label_conf = ctx.conf.boundary_font();
        label_conf.font_size += 2.0;
        label_conf.font_weight = Some("bold".to_string());

        let label_text = boundary.label.as_str().to_string();
        let label_m = measure_c4_text(
            ctx.measurer,
            &label_text,
            &label_conf,
            text_wrap,
            width_limit,
        );
        let label = C4TextBlockLayout {
            text: label_text,
            y: y + 8.0,
            width: label_m.width,
            height: label_m.height,
            line_count: label_m.line_count,
        };
        y = label.y + label.height;

        let mut ty_block: Option<C4TextBlockLayout> = None;
        if let Some(ty) = boundary.ty.as_ref().filter(|t| !t.as_str().is_empty()) {
            let ty_text = format!("[{}]", ty.as_str());
            let ty_conf = ctx.conf.boundary_font();
            let m = measure_c4_text(ctx.measurer, &ty_text, &ty_conf, text_wrap, width_limit);
            let block = C4TextBlockLayout {
                text: ty_text,
                y: y + 5.0,
                width: m.width,
                height: m.height,
                line_count: m.line_count,
            };
            y = block.y + block.height;
            ty_block = Some(block);
        }

        let mut descr_block: Option<C4TextBlockLayout> = None;
        if let Some(descr) = boundary.descr.as_ref().filter(|t| !t.as_str().is_empty()) {
            let descr_text = descr.as_str().to_string();
            let mut descr_conf = ctx.conf.boundary_font();
            descr_conf.font_size -= 2.0;
            let m = measure_c4_text(
                ctx.measurer,
                &descr_text,
                &descr_conf,
                text_wrap,
                width_limit,
            );
            let block = C4TextBlockLayout {
                text: descr_text,
                y: y + 20.0,
                width: m.width,
                height: m.height,
                line_count: m.line_count,
            };
            y = block.y + block.height;
            descr_block = Some(block);
        }

        let parent_startx = parent_bounds
            .data
            .startx
            .ok_or_else(|| Error::InvalidModel {
                message: "c4: parent bounds missing startx".to_string(),
            })?;
        let parent_stopy = parent_bounds
            .data
            .stopy
            .ok_or_else(|| Error::InvalidModel {
                message: "c4: parent bounds missing stopy".to_string(),
            })?;

        if i == 0 || i % ctx.c4_boundary_in_row == 0 {
            let x = parent_startx + ctx.conf.diagram_margin_x;
            let y0 = parent_stopy + ctx.conf.diagram_margin_y + y;
            current_bounds.set_data(x, x, y0, y0);
        } else {
            let startx = current_bounds.data.startx.unwrap_or(parent_startx);
            let stopx = current_bounds.data.stopx.unwrap_or(startx);
            let x = if stopx != startx {
                stopx + ctx.conf.diagram_margin_x
            } else {
                startx
            };
            let y0 = current_bounds.data.starty.unwrap_or(parent_stopy);
            current_bounds.set_data(x, x, y0, y0);
        }

        if let Some(shape_indices) = ctx.shape_children.get(&boundary.alias) {
            if !shape_indices.is_empty() {
                layout_c4_shape_array(&mut current_bounds, shape_indices, ctx, state);
            }
        }

        if let Some(next_boundaries) = ctx.boundary_children.get(&boundary.alias) {
            if !next_boundaries.is_empty() {
                layout_inside_boundary(&mut current_bounds, next_boundaries, ctx, state)?;
            }
        }

        let startx = current_bounds.data.startx.unwrap_or(0.0);
        let stopx = current_bounds.data.stopx.unwrap_or(startx);
        let starty = current_bounds.data.starty.unwrap_or(0.0);
        let stopy = current_bounds.data.stopy.unwrap_or(starty);

        state.boundaries.insert(
            boundary.alias.clone(),
            C4BoundaryLayout {
                alias: boundary.alias.clone(),
                parent_boundary: boundary.parent_boundary.clone(),
                x: startx,
                y: starty,
                width: stopx - startx,
                height: stopy - starty,
                image,
                label,
                ty: ty_block,
                descr: descr_block,
            },
        );

        let stopx_with_margin = stopx + ctx.conf.c4_shape_margin;
        let stopy_with_margin = stopy + ctx.conf.c4_shape_margin;
        parent_bounds.data.stopx = Some(
            parent_bounds
                .data
                .stopx
                .unwrap_or(stopx_with_margin)
                .max(stopx_with_margin),
        );
        parent_bounds.data.stopy = Some(
            parent_bounds
                .data
                .stopy
                .unwrap_or(stopy_with_margin)
                .max(stopy_with_margin),
        );

        state.global_max_x = state
            .global_max_x
            .max(parent_bounds.data.stopx.unwrap_or(state.global_max_x));
        state.global_max_y = state
            .global_max_y
            .max(parent_bounds.data.stopy.unwrap_or(state.global_max_y));
    }

    Ok(())
}

pub(crate) fn layout_c4_diagram_typed(
    model: &C4DiagramRenderModel,
    effective_config: &Value,
    measurer: &dyn TextMeasurer,
    viewport_width: f64,
    viewport_height: f64,
) -> Result<C4DiagramLayout> {
    let conf = C4Conf::from_effective_config(effective_config);

    let c4_shape_in_row = (model.layout.c4_shape_in_row.max(1)) as usize;
    let c4_boundary_in_row = (model.layout.c4_boundary_in_row.max(1)) as usize;

    let mut boundary_children: HashMap<String, Vec<usize>> = HashMap::new();
    for (i, b) in model.boundaries.iter().enumerate() {
        boundary_children
            .entry(b.parent_boundary.clone())
            .or_default()
            .push(i);
    }
    let mut shape_children: HashMap<String, Vec<usize>> = HashMap::new();
    for (i, s) in model.shapes.iter().enumerate() {
        shape_children
            .entry(s.parent_boundary.clone())
            .or_default()
            .push(i);
    }

    let mut state = C4LayoutState {
        boundaries: HashMap::new(),
        shapes: HashMap::new(),
        global_max_x: conf.diagram_margin_x,
        global_max_y: conf.diagram_margin_y,
    };

    let mut screen_bounds = BoundsState::default();
    screen_bounds.set_data(
        conf.diagram_margin_x,
        conf.diagram_margin_x,
        conf.diagram_margin_y,
        conf.diagram_margin_y,
    );
    screen_bounds.data.width_limit = viewport_width;

    let root_boundaries = boundary_children.get("").cloned().unwrap_or_default();
    if root_boundaries.is_empty() {
        return Err(Error::InvalidModel {
            message: "c4: expected at least the implicit global boundary".to_string(),
        });
    }

    let ctx = C4LayoutContext {
        model,
        effective_config,
        conf: &conf,
        c4_shape_in_row,
        c4_boundary_in_row,
        measurer,
        boundary_children: &boundary_children,
        shape_children: &shape_children,
    };

    layout_inside_boundary(&mut screen_bounds, &root_boundaries, &ctx, &mut state)?;

    screen_bounds.data.stopx = Some(state.global_max_x);
    screen_bounds.data.stopy = Some(state.global_max_y);

    let box_startx = screen_bounds.data.startx.unwrap_or(0.0);
    let box_starty = screen_bounds.data.starty.unwrap_or(0.0);
    let box_stopx = screen_bounds.data.stopx.unwrap_or(conf.diagram_margin_x);
    let box_stopy = screen_bounds.data.stopy.unwrap_or(conf.diagram_margin_y);

    let width = (box_stopx - box_startx) + 2.0 * conf.diagram_margin_x;
    let height = (box_stopy - box_starty) + 2.0 * conf.diagram_margin_y;

    let bounds = Some(Bounds {
        min_x: box_startx,
        min_y: box_starty,
        max_x: box_stopx,
        max_y: box_stopy,
    });

    let mut shape_rects: HashMap<&str, Rect> = HashMap::new();
    for s in model.shapes.iter() {
        let Some(l) = state.shapes.get(&s.alias) else {
            continue;
        };
        shape_rects.insert(
            s.alias.as_str(),
            Rect {
                origin: merman_core::geom::point(l.x, l.y),
                size: merman_core::geom::Size::new(l.width, l.height),
                margin: l.margin,
            },
        );
    }

    let rel_font = conf.message_font();
    let mut rels_out: Vec<C4RelLayout> = Vec::new();
    for (i, rel) in model.rels.iter().enumerate() {
        let mut label_text = rel.label.as_str().to_string();
        if model.c4_type == "C4Dynamic" {
            label_text = format!("{}: {}", i + 1, label_text);
        }

        let rel_text_wrap = rel.wrap && conf.wrap;

        let label_limit = measurer.measure(&label_text, &rel_font).width;
        let label_m = measure_c4_text(measurer, &label_text, &rel_font, rel_text_wrap, label_limit);
        let label = C4TextBlockLayout {
            text: label_text,
            y: 0.0,
            width: label_m.width,
            height: label_m.height,
            line_count: label_m.line_count,
        };

        let techn = rel
            .techn
            .as_ref()
            .filter(|t| !t.as_str().is_empty())
            .map(|t| {
                let text = t.as_str().to_string();
                let limit = measurer.measure(&text, &rel_font).width;
                let m = measure_c4_text(measurer, &text, &rel_font, rel_text_wrap, limit);
                C4TextBlockLayout {
                    text,
                    y: 0.0,
                    width: m.width,
                    height: m.height,
                    line_count: m.line_count,
                }
            });

        let descr = rel
            .descr
            .as_ref()
            .filter(|t| !t.as_str().is_empty())
            .map(|t| {
                let text = t.as_str().to_string();
                let limit = measurer.measure(&text, &rel_font).width;
                let m = measure_c4_text(measurer, &text, &rel_font, rel_text_wrap, limit);
                C4TextBlockLayout {
                    text,
                    y: 0.0,
                    width: m.width,
                    height: m.height,
                    line_count: m.line_count,
                }
            });

        let from = shape_rects
            .get(rel.from_alias.as_str())
            .ok_or_else(|| Error::InvalidModel {
                message: format!(
                    "c4: relationship references missing from shape {}",
                    rel.from_alias
                ),
            })?;
        let to = shape_rects
            .get(rel.to_alias.as_str())
            .ok_or_else(|| Error::InvalidModel {
                message: format!(
                    "c4: relationship references missing to shape {}",
                    rel.to_alias
                ),
            })?;

        let (start_point, end_point) = intersect_points(from, to);

        rels_out.push(C4RelLayout {
            from: rel.from_alias.clone(),
            to: rel.to_alias.clone(),
            rel_type: rel.rel_type.clone(),
            start_point,
            end_point,
            offset_x: rel.offset_x,
            offset_y: rel.offset_y,
            label,
            techn,
            descr,
        });
    }

    let mut boundaries_out = Vec::with_capacity(model.boundaries.len());
    for b in &model.boundaries {
        let Some(l) = state.boundaries.get(&b.alias) else {
            return Err(Error::InvalidModel {
                message: format!("c4: missing boundary layout for {}", b.alias),
            });
        };
        boundaries_out.push(l.clone());
    }

    let mut shapes_out = Vec::with_capacity(model.shapes.len());
    for s in &model.shapes {
        let Some(l) = state.shapes.get(&s.alias) else {
            return Err(Error::InvalidModel {
                message: format!("c4: missing shape layout for {}", s.alias),
            });
        };
        shapes_out.push(l.clone());
    }

    Ok(C4DiagramLayout {
        bounds,
        width,
        height,
        viewport_width,
        viewport_height,
        c4_type: model.c4_type.clone(),
        title: model.title.clone(),
        boundaries: boundaries_out,
        shapes: shapes_out,
        rels: rels_out,
    })
}

pub(crate) fn layout_c4_diagram(
    model: &Value,
    effective_config: &Value,
    measurer: &dyn TextMeasurer,
    viewport_width: f64,
    viewport_height: f64,
) -> Result<C4DiagramLayout> {
    let model: C4DiagramRenderModel = from_value_ref(model)?;
    layout_c4_diagram_typed(
        &model,
        effective_config,
        measurer,
        viewport_width,
        viewport_height,
    )
}

#[cfg(test)]
mod tests {
    use super::{TextStyle, c4_svg_bbox_line_height_px};

    #[test]
    fn c4_svg_bbox_line_height_uses_owner_rules() {
        fn style(font_size: f64) -> TextStyle {
            TextStyle {
                font_size,
                ..Default::default()
            }
        }

        assert_eq!(c4_svg_bbox_line_height_px(&style(12.0)), 14.0);

        assert_eq!(c4_svg_bbox_line_height_px(&style(14.0)), 16.0);

        assert_eq!(c4_svg_bbox_line_height_px(&style(16.0)), 17.0);

        assert_eq!(c4_svg_bbox_line_height_px(&style(15.0)), 17.0);
    }
}