From fb44e769f2f3306a20614d9511a2677280d589c9 Mon Sep 17 00:00:00 2001
From: Knut Sveidqvist <knsv@sveido.com>
Date: Sun, 6 Oct 2024 18:34:24 +0200
Subject: [PATCH] 5432 WIP, rendering sections

---
 cypress/platform/knsv2.html                   |   6 +-
 .../mermaid/src/diagrams/kanban/kanbanDb.ts   |  27 +++-
 .../src/diagrams/kanban/kanbanRenderer.ts     | 116 ++++++------------
 .../mermaid/src/diagrams/kanban/styles.ts     |  10 ++
 .../rendering-elements/clusters.js            | 113 ++++++++---------
 .../rendering-elements/nodes.js               |   2 +
 .../rendering-elements/shapes/kanbanItem.ts   |  59 +++++++++
 7 files changed, 186 insertions(+), 147 deletions(-)
 create mode 100644 packages/mermaid/src/rendering-util/rendering-elements/shapes/kanbanItem.ts

diff --git a/cypress/platform/knsv2.html b/cypress/platform/knsv2.html
index 5b974fa26..8d3065a5e 100644
--- a/cypress/platform/knsv2.html
+++ b/cypress/platform/knsv2.html
@@ -90,9 +90,13 @@ kanban
     id3[Update DB function]
     id4[Create parsing tests]
     id5[define getData]
-    id6[Create renderer]
+    id6[Create renderer so that it works in all cases]
   id7[In progress]
     id8[Design grammar]
+  id9[Ready for deploy]
+  id10[Ready for test]
+  id11[Done]
+  id12[Can't reproduce]
     </pre>
     <script type="module">
       import mermaid from './mermaid.esm.mjs';
diff --git a/packages/mermaid/src/diagrams/kanban/kanbanDb.ts b/packages/mermaid/src/diagrams/kanban/kanbanDb.ts
index aaf196147..b68c0fdcc 100644
--- a/packages/mermaid/src/diagrams/kanban/kanbanDb.ts
+++ b/packages/mermaid/src/diagrams/kanban/kanbanDb.ts
@@ -50,7 +50,6 @@ const getSection = function (level: number) {
 };
 
 const getSections = function () {
-  console.log('sections', sections);
   return sections;
 };
 
@@ -58,13 +57,29 @@ const getData = function () {
   const edges = [] as Edge[];
   const nodes: Node[] = [];
 
+  const sections = getSections();
+  const conf = getConfig();
   // const id: string = sanitizeText(id, conf) || 'identifier' + cnt++;
 
-  // const node = {
-  //   id,
-  //   label: sanitizeText(descr, conf),
-  //   isGroup,
-  // } satisfies Node;
+  for (const section of sections) {
+    const node = {
+      id: section.nodeId,
+      label: sanitizeText(section.descr, conf),
+      isGroup: true,
+      shape: 'kanbanSection',
+    } satisfies Node;
+    nodes.push(node);
+    for (const item of section.children) {
+      const childNode = {
+        id: item.nodeId,
+        parentId: section.nodeId,
+        label: sanitizeText(item.descr, conf),
+        isGroup: false,
+        shape: 'kanbanItem',
+      } satisfies Node;
+      nodes.push(childNode);
+    }
+  }
 
   return { nodes, edges, other: {}, config: getConfig() };
 };
diff --git a/packages/mermaid/src/diagrams/kanban/kanbanRenderer.ts b/packages/mermaid/src/diagrams/kanban/kanbanRenderer.ts
index 3835694cf..bd6620086 100644
--- a/packages/mermaid/src/diagrams/kanban/kanbanRenderer.ts
+++ b/packages/mermaid/src/diagrams/kanban/kanbanRenderer.ts
@@ -1,4 +1,4 @@
-import cytoscape from 'cytoscape';
+import type cytoscape from 'cytoscape';
 // @ts-expect-error No types available
 import coseBilkent from 'cytoscape-cose-bilkent';
 import { select } from 'd3';
@@ -9,13 +9,10 @@ import { log } from '../../logger.js';
 import type { D3Element } from '../../types.js';
 import { selectSvgElement } from '../../rendering-util/selectSvgElement.js';
 import { setupGraphViewbox } from '../../setupGraphViewbox.js';
-import type { FilledKanbanNode, KanbanDB, KanbanNode } from './kanbanTypes.js';
-import { drawNode, positionNode } from './svgDraw.js';
+import type { KanbanDB, KanbanNode } from './kanbanTypes.js';
 import defaultConfig from '../../defaultConfig.js';
-
-// Inject the layout algorithm into cytoscape
-
-async function drawSection(section: FilledKanbanNode, svg: D3Element, conf: MermaidConfig) {}
+import { insertCluster, positionCluster } from '../../rendering-util/rendering-elements/clusters';
+import { insertNode, positionNode } from '../../rendering-util/rendering-elements/nodes';
 
 declare module 'cytoscape' {
   interface EdgeSingular {
@@ -85,72 +82,11 @@ function addNodes(mindmap: KanbanNode, cy: cytoscape.Core, conf: MermaidConfig,
   }
 }
 
-function layoutMindmap(node: KanbanNode, conf: MermaidConfig): Promise<cytoscape.Core> {
-  return new Promise((resolve) => {
-    // Add temporary render element
-    const renderEl = select('body').append('div').attr('id', 'cy').attr('style', 'display:none');
-    const cy = cytoscape({
-      container: document.getElementById('cy'), // container to render in
-      style: [
-        {
-          selector: 'edge',
-          style: {
-            'curve-style': 'bezier',
-          },
-        },
-      ],
-    });
-    // Remove element after layout
-    renderEl.remove();
-    addNodes(node, cy, conf, 0);
-
-    // Make cytoscape care about the dimensions of the nodes
-    cy.nodes().forEach(function (n) {
-      n.layoutDimensions = () => {
-        const data = n.data();
-        return { w: data.width, h: data.height };
-      };
-    });
-
-    cy.layout({
-      name: 'cose-bilkent',
-      // @ts-ignore Types for cose-bilkent are not correct?
-      quality: 'proof',
-      styleEnabled: false,
-      animate: false,
-    }).run();
-    cy.ready((e) => {
-      log.info('Ready', e);
-      resolve(cy);
-    });
-  });
-}
-
-function positionNodes(db: KanbanDB, cy: cytoscape.Core) {
-  cy.nodes().map((node, id) => {
-    const data = node.data();
-    data.x = node.position().x;
-    data.y = node.position().y;
-    positionNode(db, data);
-    const el = db.getElementById(data.nodeId);
-    log.info('Id:', id, 'Position: (', node.position().x, ', ', node.position().y, ')', data);
-    el.attr(
-      'transform',
-      `translate(${node.position().x - data.width / 2}, ${node.position().y - data.height / 2})`
-    );
-    el.attr('attr', `apa-${id})`);
-  });
-}
-
 export const draw: DrawDefinition = async (text, id, _version, diagObj) => {
   log.debug('Rendering mindmap diagram\n' + text);
 
   const db = diagObj.db as KanbanDB;
-  const sections = db.getSections();
-  // const sections = db.getData();
-  if (!sections) {
-    return;
-  }
+  const data4Layout = db.getData();
 
   const conf = getConfig();
   conf.htmlLabels = false;
@@ -160,19 +96,43 @@ export const draw: DrawDefinition = async (text, id, _version, diagObj) => {
   // Draw the graph and start with drawing the nodes without proper position
   // this gives us the size of the nodes and we can set the positions later
 
-  const edgesElem = svg.append('g');
-  edgesElem.attr('class', 'sections');
+  const sectionsElem = svg.append('g');
+  sectionsElem.attr('class', 'sections');
   const nodesElem = svg.append('g');
   nodesElem.attr('class', 'items');
-  await drawSections(db, nodesElem, sections as FilledKanbanNode, -1, conf);
+  const sections = data4Layout.nodes.filter((node) => node.isGroup);
+  let cnt = 0;
+  // TODO set padding
+  const padding = 10;
 
-  // Next step is to layout the mindmap, giving each node a position
+  for (const section of sections) {
+    let y = 0;
+    cnt = cnt + 1;
+    const WIDTH = 300;
+    section.x = WIDTH * cnt + ((cnt - 1) * padding) / 2;
+    section.width = WIDTH;
+    section.y = 0;
+    section.height = WIDTH;
+    section.rx = 5;
+    section.ry = 5;
 
-  const cy = await layoutMindmap(sections, conf);
-
-  // After this we can draw, first the edges and the then nodes with the correct position
-  drawEdges(edgesElem, cy);
-  positionNodes(db, cy);
+    // Todo, use theme variable THEME_COLOR_LIMIT instead of 10
+    section.cssClasses = section.cssClasses + ' section-' + cnt;
+    const cluster = await insertCluster(sectionsElem, section);
+    const sectionItems = data4Layout.nodes.filter((node) => node.parentId === section.id);
+    // positionCluster(section);
+    for (const item of sectionItems) {
+      item.x = section.x;
+      item.width = WIDTH - padding * 2;
+      // item.height = 100;
+      const nodeEl = await insertNode(nodesElem, item);
+      console.log('ITEM', item, 'bbox=', nodeEl.node().getBBox());
+      item.y = y;
+      item.height = 150;
+      await positionNode(item);
+      y = y + 1.5 * nodeEl.node().getBBox().height + padding / 2;
+    }
+  }
 
   // Setup the view box and size of the svg element
   setupGraphViewbox(
diff --git a/packages/mermaid/src/diagrams/kanban/styles.ts b/packages/mermaid/src/diagrams/kanban/styles.ts
index fffa6e4d9..090c4ed67 100644
--- a/packages/mermaid/src/diagrams/kanban/styles.ts
+++ b/packages/mermaid/src/diagrams/kanban/styles.ts
@@ -46,6 +46,16 @@ const genSections: DiagramStylesProvider = (options) => {
     .disabled text {
       fill: #efefef;
     }
+
+  .node rect,
+  .node circle,
+  .node ellipse,
+  .node polygon,
+  .node path {
+    fill: ${options.mainBkg};
+    stroke: ${options.nodeBorder};
+    stroke-width: 1px;
+  }
     `;
   }
   return sections;
diff --git a/packages/mermaid/src/rendering-util/rendering-elements/clusters.js b/packages/mermaid/src/rendering-util/rendering-elements/clusters.js
index 27206b268..8fa2bc896 100644
--- a/packages/mermaid/src/rendering-util/rendering-elements/clusters.js
+++ b/packages/mermaid/src/rendering-util/rendering-elements/clusters.js
@@ -281,30 +281,30 @@ const roundedWithTitle = async (parent, node) => {
   return { cluster: shapeSvg, labelBBox: bbox };
 };
 const kanbanSection = async (parent, node) => {
+  log.info('Creating subgraph rect for ', node.id, node);
   const siteConfig = getConfig();
-
   const { themeVariables, handDrawnSeed } = siteConfig;
-  const { altBackground, compositeBackground, compositeTitleBackground, nodeBorder } =
-    themeVariables;
+  const { clusterBkg, clusterBorder } = themeVariables;
+
+  const { labelStyles, nodeStyles, borderStyles, backgroundStyles } = styles2String(node);
 
   // Add outer g element
   const shapeSvg = parent
     .insert('g')
-    .attr('class', node.cssClasses)
+    .attr('class', 'cluster ' + node.cssClasses)
     .attr('id', node.id)
-    .attr('data-id', node.id)
     .attr('data-look', node.look);
 
-  // add the rect
-  const outerRectG = shapeSvg.insert('g', ':first-child');
+  const useHtmlLabels = evaluate(siteConfig.flowchart.htmlLabels);
 
   // Create the label and insert it after the rect
-  const label = shapeSvg.insert('g').attr('class', 'cluster-label');
-  let innerRect = shapeSvg.append('rect');
+  const labelEl = shapeSvg.insert('g').attr('class', 'cluster-label ');
 
-  const text = label
-    .node()
-    .appendChild(await createLabel(node.label, node.labelStyle, undefined, true));
+  const text = await createText(labelEl, node.label, {
+    style: node.labelStyle,
+    useHtmlLabels,
+    isNode: true,
+  });
 
   // Get the size of the label
   let bbox = text.getBBox();
@@ -317,83 +317,72 @@ const kanbanSection = async (parent, node) => {
     dv.attr('height', bbox.height);
   }
 
-  // Rounded With Title
-  const padding = 0 * node.padding;
-  const halfPadding = padding / 2;
-
-  const width =
-    (node.width <= bbox.width + node.padding ? bbox.width + node.padding : node.width) + padding;
+  const width = node.width <= bbox.width + node.padding ? bbox.width + node.padding : node.width;
   if (node.width <= bbox.width + node.padding) {
     node.diff = (width - node.width) / 2 - node.padding;
   } else {
     node.diff = -node.padding;
   }
 
-  const height = node.height + padding;
-  // const height = node.height + padding;
-  const innerHeight = node.height + padding - bbox.height - 6;
+  const height = node.height;
   const x = node.x - width / 2;
-  const y = node.y - height / 2;
-  node.width = width;
-  const innerY = node.y - node.height / 2 - halfPadding + bbox.height + 2;
+  const y = node.y;
 
-  // add the rect
+  log.trace('Data ', node, JSON.stringify(node));
   let rect;
   if (node.look === 'handDrawn') {
-    const isAlt = node.cssClasses.includes('statediagram-cluster-alt');
+    // @ts-ignore TODO: Fix rough typings
     const rc = rough.svg(shapeSvg);
-    const roughOuterNode =
-      node.rx || node.ry
-        ? rc.path(createRoundedRectPathD(x, y, width, height, 10), {
-            roughness: 0.7,
-            fill: compositeTitleBackground,
-            fillStyle: 'solid',
-            stroke: nodeBorder,
-            seed: handDrawnSeed,
-          })
-        : rc.rectangle(x, y, width, height, { seed: handDrawnSeed });
-
-    rect = shapeSvg.insert(() => roughOuterNode, ':first-child');
-    const roughInnerNode = rc.rectangle(x, innerY, width, innerHeight, {
-      fill: isAlt ? altBackground : compositeBackground,
-      fillStyle: isAlt ? 'hachure' : 'solid',
-      stroke: nodeBorder,
+    const options = userNodeOverrides(node, {
+      roughness: 0.7,
+      fill: clusterBkg,
+      // fill: 'red',
+      stroke: clusterBorder,
+      fillWeight: 3,
       seed: handDrawnSeed,
     });
-
-    rect = shapeSvg.insert(() => roughOuterNode, ':first-child');
-    innerRect = shapeSvg.insert(() => roughInnerNode);
+    const roughNode = rc.path(createRoundedRectPathD(x, y, width, height, node.rx), options);
+    rect = shapeSvg.insert(() => {
+      log.debug('Rough node insert CXC', roughNode);
+      return roughNode;
+    }, ':first-child');
+    // Should we affect the options instead of doing this?
+    rect.select('path:nth-child(2)').attr('style', borderStyles.join(';'));
+    rect.select('path').attr('style', backgroundStyles.join(';').replace('fill', 'stroke'));
   } else {
-    rect = outerRectG.insert('rect', ':first-child');
-    const outerRectClass = 'outer';
-
+    // add the rect
+    rect = shapeSvg.insert('rect', ':first-child');
     // center the rect around its coordinate
     rect
-      .attr('class', outerRectClass)
+      .attr('style', nodeStyles)
+      .attr('rx', node.rx)
+      .attr('ry', node.ry)
       .attr('x', x)
       .attr('y', y)
       .attr('width', width)
-      .attr('height', height)
-      .attr('data-look', node.look);
-    innerRect
-      .attr('class', 'inner')
-      .attr('x', x)
-      .attr('y', innerY)
-      .attr('width', width)
-      .attr('height', innerHeight);
+      .attr('height', height);
   }
-
-  label.attr(
+  const { subGraphTitleTopMargin } = getSubGraphTitleMargins(siteConfig);
+  labelEl.attr(
     'transform',
-    `translate(${node.x - bbox.width / 2}, ${y + 1 - (evaluate(siteConfig.flowchart.htmlLabels) ? 0 : 3)})`
+    // This puts the label on top of the box instead of inside it
+    `translate(${node.x - bbox.width / 2}, ${node.y - node.height / 2 + subGraphTitleTopMargin})`
   );
 
+  if (labelStyles) {
+    const span = labelEl.select('span');
+    if (span) {
+      span.attr('style', labelStyles);
+    }
+  }
+  // Center the label
+
   const rectBox = rect.node().getBBox();
-  node.height = rectBox.height;
   node.offsetX = 0;
+  node.width = rectBox.width;
+  node.height = rectBox.height;
   // Used by layout engine to position subgraph in parent
   node.offsetY = bbox.height - node.padding / 2;
-  node.labelBBox = bbox;
 
   node.intersect = function (point) {
     return intersectRect(node, point);
diff --git a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
index 55deb1b78..52b90bf73 100644
--- a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
+++ b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
@@ -55,6 +55,7 @@ import { iconCircle } from './shapes/iconCircle.js';
 import { icon } from './shapes/icon.js';
 import { imageSquare } from './shapes/imageSquare.js';
 import { iconRounded } from './shapes/iconRounded.js';
+import { kanbanItem } from './shapes/kanbanItem.js';
 
 //Use these names as the left side to render shapes.
 export const shapes = {
@@ -299,6 +300,7 @@ export const shapes = {
   icon,
   iconRounded,
   imageSquare,
+  kanbanItem,
 };
 
 const nodeElems = new Map();
diff --git a/packages/mermaid/src/rendering-util/rendering-elements/shapes/kanbanItem.ts b/packages/mermaid/src/rendering-util/rendering-elements/shapes/kanbanItem.ts
new file mode 100644
index 000000000..94995abdc
--- /dev/null
+++ b/packages/mermaid/src/rendering-util/rendering-elements/shapes/kanbanItem.ts
@@ -0,0 +1,59 @@
+import { labelHelper, updateNodeBounds, getNodeClasses } from './util.js';
+import intersect from '../intersect/index.js';
+import type { Node } from '../../types.js';
+import { createRoundedRectPathD } from './roundedRectPath.js';
+import { userNodeOverrides, styles2String } from './handDrawnShapeStyles.js';
+import rough from 'roughjs';
+
+export const kanbanItem = async (parent: SVGAElement, node: Node) => {
+  const { labelStyles, nodeStyles } = styles2String(node);
+  node.labelStyle = labelStyles;
+  // console.log('IPI labelStyles:', labelStyles);
+  const { shapeSvg, bbox } = await labelHelper(parent, node, getNodeClasses(node));
+  const labelPaddingX = 10;
+  const labelPaddingY = 10;
+  const totalWidth = Math.max(bbox.width + labelPaddingX * 2, node?.width || 0);
+  const totalHeight = Math.max(bbox.height + labelPaddingY * 2, node?.height || 0);
+  const x = -totalWidth / 2;
+  const y = -totalHeight / 2;
+
+  // log.info('IPI node = ', node);
+
+  let rect;
+  const { rx, ry } = node;
+  const { cssStyles } = node;
+
+  if (node.look === 'handDrawn') {
+    // @ts-ignore TODO: Fix rough typings
+    const rc = rough.svg(shapeSvg);
+    const options = userNodeOverrides(node, {});
+
+    const roughNode =
+      rx || ry
+        ? rc.path(createRoundedRectPathD(x, y, totalWidth, totalHeight, rx || 0), options)
+        : rc.rectangle(x, y, totalWidth, totalHeight, options);
+
+    rect = shapeSvg.insert(() => roughNode, ':first-child');
+    rect.attr('class', 'basic label-container').attr('style', cssStyles);
+  } else {
+    rect = shapeSvg.insert('rect', ':first-child');
+
+    rect
+      .attr('class', 'basic label-container __APA__')
+      .attr('style', nodeStyles)
+      .attr('rx', rx)
+      .attr('ry', ry)
+      .attr('x', x)
+      .attr('y', y)
+      .attr('width', totalWidth)
+      .attr('height', totalHeight);
+  }
+
+  updateNodeBounds(node, rect);
+
+  node.intersect = function (point) {
+    return intersect.rect(node, point);
+  };
+
+  return shapeSvg;
+};