updated imageSquare shape

mermaid-js · Sep 24, 2024 · 6a649d3 · 6a649d3
1 parent fd37294
commit 6a649d3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/packages/mermaid/src/diagrams/flowchart/flowDb.ts b/packages/mermaid/src/diagrams/flowchart/flowDb.ts
@@ -161,11 +161,9 @@ export const addVertex = function (
       if (!doc.label?.trim() && vertex.text === id) {
         vertex.text = '';
       }
-      if (doc?.constraint) {
-        vertex.constraint = doc.constraint;
-      } else {
-        vertex.constraint = 'off';
-      }
+    }
+    if (doc?.constraint) {
+      vertex.constraint = doc.constraint;
     }
     if (doc.w) {
       vertex.assetWidth = Number(doc.w);

diff --git a/packages/mermaid/src/rendering-util/rendering-elements/shapes/imageSquare.ts b/packages/mermaid/src/rendering-util/rendering-elements/shapes/imageSquare.ts
@@ -26,10 +26,18 @@ export const imageSquare = async (
   const defaultWidth = flowchart?.wrappingWidth;
   node.defaultWidth = flowchart?.wrappingWidth;
 
-  const imageWidth = Math.max(
+  const imageRawWidth = Math.max(
     node.label ? (defaultWidth ?? 0) : 0,
     node?.assetWidth ?? imageNaturalWidth
   );
+
+  const imageWidth =
+    node.constraint === 'on'
+      ? node?.assetHeight
+        ? node.assetHeight * node.imageAspectRatio
+        : imageRawWidth
+      : imageRawWidth;
+
   const imageHeight =
     node.constraint === 'on'
       ? imageWidth / node.imageAspectRatio