Update packages/mermaid/src/diagrams/class/classDb.ts

Co-authored-by: Sidharth Vinod <[email protected]>
mermaid-js · Dec 5, 2023 · 02b2d71 · 02b2d71
1 parent e25515a
commit 02b2d71
Showing 1 changed file with 10 additions and 14 deletions.
diff --git a/packages/mermaid/src/diagrams/class/classDb.ts b/packages/mermaid/src/diagrams/class/classDb.ts
@@ -457,20 +457,16 @@ export const addClassesToNamespace = function (id: string, classNames: string[])
   }
 };
 
-export const setCssStyle = function (id: string, style: string[]) {
-  if (style !== undefined && style !== null) {
-    const thisClass = classes[id];
-    if (thisClass !== undefined) {
-      style.forEach(function (s) {
-        if (s.includes(',')) {
-          const styles = s.split(',');
-          styles.forEach(function (newStyle) {
-            thisClass.styles.push(newStyle);
-          });
-        } else {
-          thisClass.styles.push(s);
-        }
-      });
+export const setCssStyle = function (id: string, styles: string[]) {
+  const thisClass = classes[id];
+  if (!styles || !thisClass) {
+    return;
+  }
+  for (const s of styles) {
+    if (s.includes(',')) {
+      thisClass.styles = thisClass.styles.concat(s.split(','));
+    } else {
+      thisClass.styles.push(s);
     }
   }
 };