From 4a8c24a633476a654fefc92db5aaf813cf451d68 Mon Sep 17 00:00:00 2001
From: Ashish Jain <ashishjain0512@gmail.com>
Date: Mon, 17 Jun 2024 15:19:03 +0200
Subject: [PATCH] Add support for custom cssStyle and compiledStyles for custom
 classDefs

---
 packages/mermaid/src/diagrams/flowchart/flowDb.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/mermaid/src/diagrams/flowchart/flowDb.ts b/packages/mermaid/src/diagrams/flowchart/flowDb.ts
index c65c83677..66c852a14 100644
--- a/packages/mermaid/src/diagrams/flowchart/flowDb.ts
+++ b/packages/mermaid/src/diagrams/flowchart/flowDb.ts
@@ -824,8 +824,14 @@ const addNodeFromVertex = (
 function getCompiledStyles(classDefs: string[]) {
   let compiledStyles: string[] = [];
   for (const customClass of classDefs) {
-    if (classes.get(customClass)) {
-      compiledStyles = [...compiledStyles, ...(classes.get(customClass)?.styles ?? [])];
+    const cssClass = classes.get(customClass);
+    if (cssClass) {
+      if (cssClass.styles) {
+        compiledStyles = [...compiledStyles, ...(cssClass.styles ?? [])];
+      }
+      if (cssClass.textStyles) {
+        compiledStyles = [...compiledStyles, ...(cssClass.textStyles ?? [])];
+      }
     }
   }
   return compiledStyles;