Merge pull request #52 from johanhammar/mermaid-support-frontmatter

fix: support mermaid diagrams with frontmatter directives
johanneswuerbach · Jun 28, 2024 · 21d5b7f · 21d5b7f
2 parents 5f63012 + 7b0a8b4
commit 21d5b7f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/Mermaid/hooks.test.ts b/src/Mermaid/hooks.test.ts
@@ -57,6 +57,29 @@ A-->B`;
 
       expect(isMermaidCode(mermaidCode)).toBe(true);
     })
+
+    it("mermaid code exists with frontmatter", () => {
+      const mermaidCode = `---
+title: class diagram with frontmatter
+---
+classDiagram
+    direction LR`;
+
+      const result = isMermaidCode(mermaidCode);
+
+      expect(result).toBe(true);
+    });
+
+    it("mermaid code exists with comment", () => {
+      const mermaidCode = `%% mermaid code with comment
+graph LR
+A-->B`;
+
+      const result = isMermaidCode(mermaidCode);
+
+      expect(result).toBe(true);
+    });
+
   });
 
   describe("returns false when", () => {

diff --git a/src/Mermaid/hooks.ts b/src/Mermaid/hooks.ts
@@ -15,7 +15,7 @@
  */
 
 const mermaidStart =
-  /^(\s*)(graph|flowchart|sequenceDiagram|classDiagram|stateDiagram|erDiagram|journey|gantt|pie|requirementDiagram|gitGraph|C4Context|C4Container|C4Component|C4Dynamic|C4Deployment|timeline)/;
+  /^(\s*)(graph|flowchart|sequenceDiagram|classDiagram|stateDiagram|erDiagram|journey|gantt|pie|requirementDiagram|gitGraph|C4Context|C4Container|C4Component|C4Dynamic|C4Deployment|timeline)/gm;
 
 export const isMermaidCode = (code: string): boolean => {
   if (code.startsWith('%%{init')) {