From 469bdcef2fb938d3d884826cf59329663e6aae96 Mon Sep 17 00:00:00 2001
From: steph <steph@huynhicode.dev>
Date: Sun, 18 Dec 2022 20:15:29 -0800
Subject: [PATCH] add support for vitepress blocks

---
 packages/mermaid/src/docs.mts | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/packages/mermaid/src/docs.mts b/packages/mermaid/src/docs.mts
index 313d1f2de..a4ec0d718 100644
--- a/packages/mermaid/src/docs.mts
+++ b/packages/mermaid/src/docs.mts
@@ -146,9 +146,22 @@ const readSyncedUTF8file = (filename: string): string => {
   return readFileSync(filename, 'utf8');
 };
 
-const transformToBlockQuote = (content: string, type: string) => {
-  const title = type === 'warning' ? 'Warning' : 'Note';
-  return `> **${title}** \n> ${content.replace(/\n/g, '\n> ')}`;
+const blockIcons: Record<string, string> = {
+  tip: '💡 ',
+  danger: '‼️ ',
+};
+
+const capitalize = (word: string) => word[0].toUpperCase() + word.slice(1);
+
+const transformToBlockQuote = (content: string, type: string, customTitle?: string | null) => {
+  if (vitepress) {
+    const vitepressType = type === 'note' ? 'info' : type;
+    return `::: ${vitepressType} ${customTitle || ''}\n${content}\n:::`;
+  } else {
+    const icon = blockIcons[type] || '';
+    const title = `${icon}${customTitle || capitalize(type)}`;
+    return `> **${title}** \n> ${content.replace(/\n/g, '\n> ')}`;
+  }
 };
 
 const injectPlaceholders = (text: string): string =>
@@ -194,8 +207,8 @@ const transformMarkdown = (file: string) => {
     }
 
     // Transform codeblocks into block quotes.
-    if (['note', 'tip', 'warning'].includes(c.lang)) {
-      return [remark.parse(transformToBlockQuote(c.value, c.lang))];
+    if (['note', 'tip', 'warning', 'danger'].includes(c.lang)) {
+      return [remark.parse(transformToBlockQuote(c.value, c.lang, c.meta))];
     }
 
     return [c];