From b807e330c8ee2509b2fd0f5413512795009c8bf8 Mon Sep 17 00:00:00 2001
From: omkarht <omkar@mermaidchart.com>
Date: Thu, 8 Aug 2024 18:25:15 +0530
Subject: [PATCH] added new Wave Rectangle shape

---
 .../rendering-elements/nodes.js               |  2 +
 .../shapes/waveRectangle.ts                   | 66 +++++++++++++++++++
 2 files changed, 68 insertions(+)
 create mode 100644 packages/mermaid/src/rendering-util/rendering-elements/shapes/waveRectangle.ts

diff --git a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
index 60f1ee7580..fae98e8e65 100644
--- a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
+++ b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
@@ -33,6 +33,7 @@ import { slopedRect } from './shapes/slopedRect.js';
 import { bowTieRect } from './shapes/bowTieRect.js';
 import { dividedRect } from './shapes/dividedRect.js';
 import { crossedCircle } from './shapes/crossedCircle.js';
+import { waveRectangle } from './shapes/waveRectangle.js';
 
 const shapes = {
   state,
@@ -69,6 +70,7 @@ const shapes = {
   bowTieRect,
   dividedRect,
   crossedCircle,
+  waveRectangle,
 };
 
 const nodeElems = new Map();
diff --git a/packages/mermaid/src/rendering-util/rendering-elements/shapes/waveRectangle.ts b/packages/mermaid/src/rendering-util/rendering-elements/shapes/waveRectangle.ts
new file mode 100644
index 0000000000..19130f214b
--- /dev/null
+++ b/packages/mermaid/src/rendering-util/rendering-elements/shapes/waveRectangle.ts
@@ -0,0 +1,66 @@
+import { labelHelper, updateNodeBounds, getNodeClasses } from './util.js';
+import intersect from '../intersect/index.js';
+import type { Node } from '$root/rendering-util/types.d.ts';
+import {
+  styles2String,
+  userNodeOverrides,
+} from '$root/rendering-util/rendering-elements/shapes/handdrawnStyles.js';
+import rough from 'roughjs';
+
+function createWaveRectanglePathD(x: number, y: number, width: number, height: number) {
+  const halfWidth = width / 2;
+  const halfHeight = height / 2;
+
+  const pathData = `M ${x} ${y}
+    Q ${x + halfWidth / 2} ${y + halfHeight}, ${x + halfWidth} ${y}
+    Q ${x + (3 * halfWidth) / 2} ${y - halfHeight}, ${x + width} ${y}
+    L ${x + width} ${y + height}
+    Q ${x + (3 * halfWidth) / 2} ${y + halfHeight}, ${x + halfWidth} ${y + height}
+    Q ${x + halfWidth / 2} ${y + 3 * halfHeight}, ${x} ${y + height}
+    L ${x} ${y}
+    Z`;
+  return pathData;
+}
+
+export const waveRectangle = async (parent: SVGAElement, node: Node) => {
+  const { labelStyles, nodeStyles } = styles2String(node);
+  node.labelStyle = labelStyles;
+  const { shapeSvg, bbox } = await labelHelper(parent, node, getNodeClasses(node));
+  const w = bbox.width + node.padding;
+  const h = bbox.height + node.padding + 20;
+
+  let shape: d3.Selection<SVGPathElement | SVGGElement, unknown, null, undefined>;
+  const { cssStyles } = node;
+
+  if (node.look === 'handdrawn') {
+    // @ts-ignore - rough is not typed
+    const rc = rough.svg(shapeSvg);
+    const pathData = createWaveRectanglePathD(0, 0, w, h);
+    const shapeNode = rc.path(pathData, userNodeOverrides(node, {}));
+
+    shape = shapeSvg.insert(() => shapeNode, ':first-child');
+    shape.attr('class', 'basic label-container');
+    if (cssStyles) {
+      shape.attr('style', cssStyles);
+    }
+  } else {
+    const pathData = createWaveRectanglePathD(0, 0, w, h);
+    shape = shapeSvg
+      .insert('path', ':first-child')
+      .attr('d', pathData)
+      .attr('class', 'basic label-container')
+      .attr('style', cssStyles)
+      .attr('style', nodeStyles);
+  }
+
+  shape.attr('transform', `translate(${-w / 2}, ${-h / 2})`);
+
+  updateNodeBounds(node, shape);
+
+  node.intersect = function (point) {
+    const pos = intersect.polygon(node, point);
+    return pos;
+  };
+
+  return shapeSvg;
+};