prev-cli 0.1.6 → 0.1.8

package/dist/cli.js

@@ -387,7 +387,10 @@ async function createViteConfig(options) {
         "@prev/theme": path4.join(srcRoot2, "theme"),
         react: path4.join(cliNodeModules, "react"),
         "react-dom": path4.join(cliNodeModules, "react-dom"),
-        "react-router-dom": path4.join(cliNodeModules, "react-router-dom")
+        "react-router-dom": path4.join(cliNodeModules, "react-router-dom"),
+        mermaid: path4.join(cliNodeModules, "mermaid"),
+        dayjs: path4.join(cliNodeModules, "dayjs"),
+        "@terrastruct/d2": path4.join(cliNodeModules, "@terrastruct/d2")
       }
     },
     optimizeDeps: {
@@ -398,7 +401,10 @@ async function createViteConfig(options) {
         "react-dom/client",
         "react-router-dom",
         "react/jsx-runtime",
-        "react/jsx-dev-runtime"
+        "react/jsx-dev-runtime",
+        "mermaid",
+        "dayjs",
+        "@terrastruct/d2"
       ],
       exclude: [
         "clsx",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prev-cli",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "Transform MDX directories into beautiful documentation websites",
   "type": "module",
   "license": "MIT",

package/src/theme/Layout.tsx

@@ -37,31 +37,47 @@ async function renderD2Diagrams() {
   const codeBlocks = document.querySelectorAll('code.language-d2, code.hljs.language-d2')
   if (codeBlocks.length === 0) return
 
-  const { D2 } = await import('@terrastruct/d2')
-  const d2 = new D2()
+  try {
+    const d2Module = await import('@terrastruct/d2')
+    const { D2 } = d2Module
+    const d2 = new D2()
 
-  for (const block of codeBlocks) {
-    const pre = block.parentElement as HTMLElement
-    if (!pre || pre.dataset.rendered) continue
-    pre.dataset.rendered = 'true'
+    for (const block of codeBlocks) {
+      const pre = block.parentElement as HTMLElement
+      if (!pre || pre.dataset.rendered) continue
+      pre.dataset.rendered = 'true'
 
-    const code = block.textContent || ''
-    const container = document.createElement('div')
-    container.className = 'd2-diagram'
+      const code = block.textContent || ''
+      const container = document.createElement('div')
+      container.className = 'd2-diagram'
 
-    try {
-      const result = await d2.compile(code)
-      const svg = await d2.render(result.diagram)
-      container.innerHTML = svg
-      // Hide original instead of replacing (avoids React DOM conflicts)
-      pre.style.display = 'none'
-      pre.insertAdjacentElement('afterend', container)
-    } catch (e) {
-      console.error('D2 render error:', e)
+      try {
+        const result = await d2.compile(code)
+        const svg = await d2.render(result.diagram, result.renderOptions)
+        container.innerHTML = svg
+        // Hide original instead of replacing (avoids React DOM conflicts)
+        pre.style.display = 'none'
+        pre.insertAdjacentElement('afterend', container)
+      } catch (e) {
+        console.error('D2 render error:', e)
+      }
     }
+  } catch (e) {
+    console.error('D2 library load error:', e)
   }
 }
 
+// Clean up rendered diagrams (for navigation)
+function cleanupDiagrams() {
+  // Remove all rendered diagram containers
+  document.querySelectorAll('.mermaid-diagram, .d2-diagram').forEach(el => el.remove())
+  // Reset rendered state on pre elements and restore visibility
+  document.querySelectorAll('pre[data-rendered="true"]').forEach(el => {
+    delete (el as HTMLElement).dataset.rendered;
+    (el as HTMLElement).style.display = ''
+  })
+}
+
 // Render all diagrams
 async function renderDiagrams() {
   await Promise.all([
@@ -74,9 +90,14 @@ export function Layout() {
   const location = useLocation()
 
   useEffect(() => {
+    // Clean up old diagrams before rendering new ones
+    cleanupDiagrams()
     // Render diagrams after content loads
     const timer = setTimeout(renderDiagrams, 100)
-    return () => clearTimeout(timer)
+    return () => {
+      clearTimeout(timer)
+      cleanupDiagrams()
+    }
   }, [location.pathname])
 
   return (