From d71fe28a3541497c0648aea6ec4d710cf88d0be6 Mon Sep 17 00:00:00 2001
From: Sidharth Vinod <sidharthv96@gmail.com>
Date: Sat, 13 Jul 2024 15:11:53 +0530
Subject: [PATCH] fix: Handle spaces after newline

---
 .../src/rendering-util/handle-markdown-text.spec.ts  | 12 +++++++++++-
 .../src/rendering-util/handle-markdown-text.ts       |  4 ++--
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/packages/mermaid/src/rendering-util/handle-markdown-text.spec.ts b/packages/mermaid/src/rendering-util/handle-markdown-text.spec.ts
index a83179e5d..a96b69e61 100644
--- a/packages/mermaid/src/rendering-util/handle-markdown-text.spec.ts
+++ b/packages/mermaid/src/rendering-util/handle-markdown-text.spec.ts
@@ -295,5 +295,15 @@ test('markdownToHTML - no auto wrapping', () => {
   you do?`,
       { markdownAutoWrap: false }
     )
-  ).toMatchInlineSnapshot(`"<p>Hello,&nbsp;how&nbsp;do<br/>&nbsp;&nbsp;you&nbsp;do?</p>"`);
+  ).toMatchInlineSnapshot(`"<p>Hello,&nbsp;how&nbsp;do<br/>you&nbsp;do?</p>"`);
+});
+
+test('markdownToHTML - auto wrapping', () => {
+  expect(
+    markdownToHTML(
+      `Hello, how do
+  you do?`,
+      { markdownAutoWrap: true }
+    )
+  ).toMatchInlineSnapshot(`"<p>Hello, how do<br/>you do?</p>"`);
 });
diff --git a/packages/mermaid/src/rendering-util/handle-markdown-text.ts b/packages/mermaid/src/rendering-util/handle-markdown-text.ts
index ee49463df..3846e7f37 100644
--- a/packages/mermaid/src/rendering-util/handle-markdown-text.ts
+++ b/packages/mermaid/src/rendering-util/handle-markdown-text.ts
@@ -66,9 +66,9 @@ export function markdownToHTML(markdown: string, { markdownAutoWrap }: MermaidCo
   function output(node: Token): string {
     if (node.type === 'text') {
       if (markdownAutoWrap === false) {
-        return node.text.replace(/\n/g, '<br/>').replace(/ /g, '&nbsp;');
+        return node.text.replace(/\n */g, '<br/>').replace(/ /g, '&nbsp;');
       }
-      return node.text.replace(/\n/g, '<br/>');
+      return node.text.replace(/\n */g, '<br/>');
     } else if (node.type === 'strong') {
       return `<strong>${node.tokens?.map(output).join('')}</strong>`;
     } else if (node.type === 'em') {