From 191e4217fc3f968777cc17e734d494b4c56273ff Mon Sep 17 00:00:00 2001 From: yari-dewalt Date: Wed, 28 Aug 2024 10:54:09 -0700 Subject: [PATCH] Adjust classNode text when label is set --- packages/mermaid/src/diagrams/class/classDb.ts | 2 ++ 1 file changed, 2 insertions(+) diff --git a/packages/mermaid/src/diagrams/class/classDb.ts b/packages/mermaid/src/diagrams/class/classDb.ts index 06046f882..32d007634 100644 --- a/packages/mermaid/src/diagrams/class/classDb.ts +++ b/packages/mermaid/src/diagrams/class/classDb.ts @@ -58,6 +58,8 @@ export const setClassLabel = function (_id: string, label: string) { const { className } = splitClassNameAndType(id); classes.get(className)!.label = label; + classes.get(className)!.text = + `${label}${classes.get(className)!.type ? `<${classes.get(className)!.type}>` : ''}`; }; /**