From d0942d4897f923e9f632ecdc113776c75b46b88b Mon Sep 17 00:00:00 2001
From: omkarht <omkar@mermaidchart.com>
Date: Thu, 1 Aug 2024 16:34:10 +0530
Subject: [PATCH] feat: added half Rounded Rectangle Shape

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

diff --git a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
index 1598f2f5e2..da0ee40e4d 100644
--- a/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
+++ b/packages/mermaid/src/rendering-util/rendering-elements/nodes.js
@@ -27,6 +27,7 @@ import { trapezoid } from './shapes/trapezoid.js';
 import { inv_trapezoid } from './shapes/invertedTrapezoid.js';
 import { labelRect } from './shapes/labelRect.js';
 import { triangle } from './shapes/triangle.js';
+import { halfRoundedRectangle } from './shapes/halfRoundedRectangle.js';
 
 const shapes = {
   state,
@@ -57,6 +58,7 @@ const shapes = {
   shadedProcess,
   anchor,
   triangle,
+  halfRoundedRectangle,
 };
 
 const nodeElems = new Map();
diff --git a/packages/mermaid/src/rendering-util/rendering-elements/shapes/halfRoundedRectangle.ts b/packages/mermaid/src/rendering-util/rendering-elements/shapes/halfRoundedRectangle.ts
new file mode 100644
index 0000000000..5d694a87c6
--- /dev/null
+++ b/packages/mermaid/src/rendering-util/rendering-elements/shapes/halfRoundedRectangle.ts
@@ -0,0 +1,79 @@
+import { log } from '$root/logger.js';
+import { labelHelper, updateNodeBounds, getNodeClasses, createPathFromPoints } 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 createHalfRoundedRectShapePathD(
+  x: number,
+  y: number,
+  totalWidth: number,
+  totalHeight: number,
+  radius: number
+) {
+  const rw = totalWidth - radius;
+  const rh = totalHeight;
+
+  const points = [
+    { x: x + rw, y },
+    { x, y },
+    { x, y: y + rh },
+    { x: x + rw, y: y + rh },
+  ];
+
+  const rectPath = createPathFromPoints(points);
+  const arcPath = `M ${rw},0 A ${rh / 2} ${rh / 2} 0 0 1 ${rw} ${rh}`;
+  const finalPath = `${rectPath} ${arcPath}`.replace('Z', '');
+
+  return finalPath;
+}
+
+export const halfRoundedRectangle = async (parent: SVGAElement, node: Node) => {
+  const { labelStyles, nodeStyles } = styles2String(node);
+  node.labelStyle = labelStyles;
+  const { shapeSvg, bbox, label } = await labelHelper(parent, node, getNodeClasses(node));
+  const w = (bbox.width + node.padding) * 1.2;
+  const h = bbox.height + node.padding;
+  const radius = h / 2;
+  const { cssStyles } = node;
+
+  // @ts-ignore - rough is not typed
+  const rc = rough.svg(shapeSvg);
+  const options = userNodeOverrides(node, {});
+
+  if (node.look !== 'handdrawn') {
+    options.roughness = 0;
+    options.fillStyle = 'solid';
+  }
+
+  const pathData = createHalfRoundedRectShapePathD(0, 0, w, h, radius);
+  const shapeNode = rc.path(pathData, options);
+  const polygon = shapeSvg.insert(() => shapeNode, ':first-child');
+  polygon.attr('class', 'basic label-container');
+
+  if (cssStyles) {
+    polygon.attr('style', cssStyles);
+  }
+
+  if (nodeStyles) {
+    polygon.attr('style', nodeStyles);
+  }
+
+  polygon.attr('transform', `translate(${-w / 2}, ${-h / 2})`);
+
+  updateNodeBounds(node, polygon);
+
+  label.attr('transform', `translate(${-bbox.width / 2}, ${h / 2 - bbox.height})`);
+
+  node.intersect = function (point) {
+    log.info('Pill intersect', node, { radius, point });
+    const pos = intersect.rect(node, point);
+    return pos;
+  };
+
+  return shapeSvg;
+};