@zombie-mermaid/svg-renderer 4.2.2 → 5.0.0

package/src/er/layout.ts

@@ -1,216 +0,0 @@
-/**
- * ER diagram layout engine (ELK.js).
- *
- * Each entity box has:
- *   1. Header (entity name)
- *   2. Attribute rows (type, name, keys)
- */
-
-import type { ElkNode, ElkExtendedEdge } from 'elkjs'
-import type {
-  ErDiagram,
-  ErEntity,
-  PositionedErDiagram,
-  PositionedErEntity,
-  PositionedErRelationship,
-} from '@zombie-mermaid/mermaid-parser'
-import type { ErRenderOptions } from '@zombie-mermaid/core'
-import {
-  estimateTextWidth,
-  estimateMonoTextWidth,
-  FONT_WEIGHTS,
-  resolveFontSizes,
-} from '../styles.ts'
-import { elkLayoutSync } from '../elk-instance.ts'
-import { extractEdgePoints } from '../layout-engine/elk-adapter-utils.ts'
-import {
-  ELK_DIRECTION_FALLBACK,
-  baseElkLayoutOptions,
-  buildElkEdge,
-  buildElkLeafNode,
-  directionToElk,
-} from '../layout-engine/elk-graph-builder.ts'
-
-/** Layout constants for ER diagrams */
-const ER = {
-  padding: 40,
-  boxPadX: 14,
-  headerHeight: 34,
-  rowHeight: 22,
-  minWidth: 140,
-  attrFontSize: 11,
-  attrFontWeight: 400,
-  nodeSpacing: 70,
-  layerSpacing: 90,
-} as const
-
-type EntitySizeMap = Map<string, { width: number; height: number }>
-
-/** Build ELK graph and size map from an ER diagram. */
-/**
- * Width of an attribute row sized for the worst-case glyph advance.
- *
- * The mono face advances 0.6em per glyph (6.6px at the 11px attribute size),
- * but browsers that don't position glyphs at subpixel offsets (headless
- * Chromium on Linux) round each advance to a whole pixel, so a long row draws
- * wider than `estimateMonoTextWidth` says and the type runs into the
- * right-aligned name (#1262, same cause as class members in #1238).
- */
-function attrTextWidth(text: string): number {
-  return text.length * Math.ceil(estimateMonoTextWidth('M', ER.attrFontSize))
-}
-
-function buildErElkGraph(
-  diagram: ErDiagram,
-  options: ErRenderOptions,
-): { elkGraph: ElkNode; entitySizes: EntitySizeMap } {
-  const entitySizes: EntitySizeMap = new Map()
-  const fontSizes = resolveFontSizes(options.fontSizes)
-
-  for (const entity of diagram.entities) {
-    const headerTextW = estimateTextWidth(
-      entity.label,
-      fontSizes.nodeLabel,
-      FONT_WEIGHTS.nodeLabel,
-    )
-    let maxAttrW = 0
-    for (const attr of entity.attributes) {
-      const attrText = `${attr.type}  ${attr.name}${attr.keys.length > 0 ? '  ' + attr.keys.join(',') : ''}`
-      const w = attrTextWidth(attrText)
-      if (w > maxAttrW) maxAttrW = w
-    }
-    const width = Math.max(
-      ER.minWidth,
-      headerTextW + ER.boxPadX * 2,
-      maxAttrW + ER.boxPadX * 2,
-    )
-    const height =
-      ER.headerHeight + Math.max(entity.attributes.length, 1) * ER.rowHeight
-    entitySizes.set(entity.id, { width, height })
-  }
-
-  // Iterate entitySizes directly (populated above, in diagram.entities order)
-  // rather than looking each entity back up by id — sidesteps needing an
-  // assertion or invariant check for a lookup that can't actually miss.
-  const children: ElkNode[] = []
-  for (const [id, size] of entitySizes) {
-    children.push(buildElkLeafNode(id, size))
-  }
-
-  // ER edge labels carry no per-label layout options — placement is set
-  // once on the root graph below (`elk.edgeLabels.placement: CENTER`).
-  const labelStyle = { fontSize: fontSizes.edgeLabel }
-
-  const edges: ElkExtendedEdge[] = []
-  for (const [i, rel] of diagram.relationships.entries()) {
-    edges.push(
-      buildElkEdge({
-        id: `e${i}`,
-        source: rel.entity1,
-        target: rel.entity2,
-        label: rel.label,
-        labelStyle,
-      }),
-    )
-  }
-
-  const elkGraph: ElkNode = {
-    id: 'root',
-    layoutOptions: {
-      ...baseElkLayoutOptions({
-        // A source with no `direction` statement lays out top-to-bottom,
-        // as official Mermaid does (#1250). See ELK_DIRECTION_FALLBACK.
-        direction: directionToElk(diagram.direction, ELK_DIRECTION_FALLBACK.er),
-        nodeSpacing: ER.nodeSpacing,
-        layerSpacing: ER.layerSpacing,
-        padding: ER.padding,
-      }),
-      'elk.edgeLabels.placement': 'CENTER',
-      // Keep declaration order as the crossing-minimization tie-break and
-      // center children under their parents, as official Mermaid does.
-      'elk.layered.considerModelOrder.strategy': 'NODES_AND_EDGES',
-      'elk.layered.nodePlacement.bk.fixedAlignment': 'BALANCED',
-    },
-    children,
-    edges,
-  }
-
-  return { elkGraph, entitySizes }
-}
-
-/** Extract positioned entities and relationships from ELK result. */
-function extractErLayout(
-  result: ElkNode,
-  diagram: ErDiagram,
-  entitySizes: EntitySizeMap,
-): PositionedErDiagram {
-  const entityLookup = new Map<string, ErEntity>()
-  for (const entity of diagram.entities) entityLookup.set(entity.id, entity)
-
-  const positionedEntities: PositionedErEntity[] = []
-  for (const child of result.children ?? []) {
-    const entity = entityLookup.get(child.id)
-    if (entity) {
-      const size = entitySizes.get(entity.id)
-      if (!size) {
-        // Unreachable — entitySizes is populated for every diagram.entities
-        // entry, and entityLookup/entity.id come from that same list.
-        /* v8 ignore next */
-        throw new Error(`Missing computed size for entity "${entity.id}"`)
-      }
-      positionedEntities.push({
-        id: entity.id,
-        label: entity.label,
-        attributes: entity.attributes,
-        x: child.x ?? 0,
-        y: child.y ?? 0,
-        width: child.width ?? size.width,
-        height: child.height ?? size.height,
-        headerHeight: ER.headerHeight,
-        rowHeight: ER.rowHeight,
-      })
-    }
-  }
-
-  const relationships: PositionedErRelationship[] = []
-  for (const [i, elkEdge] of (result.edges ?? []).entries()) {
-    // diagram.relationships[i] is guaranteed by construction: buildErElkGraph
-    // creates exactly one ELK edge per relationship, in the same order.
-    const rel = diagram.relationships[i]!
-
-    const points = extractEdgePoints(elkEdge)
-
-    relationships.push({
-      entity1: rel.entity1,
-      entity2: rel.entity2,
-      cardinality1: rel.cardinality1,
-      cardinality2: rel.cardinality2,
-      label: rel.label,
-      identifying: rel.identifying,
-      points,
-    })
-  }
-
-  return {
-    width: result.width ?? 600,
-    height: result.height ?? 400,
-    entities: positionedEntities,
-    relationships,
-  }
-}
-
-/**
- * Lay out a parsed ER diagram using ELK.js (synchronous).
- */
-export function layoutErDiagramSync(
-  diagram: ErDiagram,
-  options: ErRenderOptions = {},
-): PositionedErDiagram {
-  if (diagram.entities.length === 0) {
-    return { width: 0, height: 0, entities: [], relationships: [] }
-  }
-
-  const { elkGraph, entitySizes } = buildErElkGraph(diagram, options)
-  const result = elkLayoutSync(elkGraph, options.layoutCache)
-  return extractErLayout(result, diagram, entitySizes)
-}