feat: allow html blocks inside code groups (#2719)

src/client/app/composables/codeGroups.ts

@@ -1,6 +1,17 @@
-import { inBrowser } from 'vitepress'
+import { inBrowser, onContentUpdated } from 'vitepress'
 
 export function useCodeGroups() {
+  if (import.meta.env.DEV) {
+    onContentUpdated(() => {
+      document.querySelectorAll('.vp-code-group > .blocks').forEach((el) => {
+        Array.from(el.children).forEach((child) => {
+          child.classList.remove('active')
+        })
+        el.children[0].classList.add('active')
+      })
+    })
+  }
+
   if (inBrowser) {
     window.addEventListener('click', (e) => {
       const el = e.target as HTMLInputElement
@@ -8,18 +19,25 @@ export function useCodeGroups() {
       if (el.matches('.vp-code-group input')) {
         // input <- .tabs <- .vp-code-group
         const group = el.parentElement?.parentElement
-        const i = Array.from(group?.querySelectorAll('input') || []).indexOf(el)
+        if (!group) return
 
-        const current = group?.querySelector('div[class*="language-"].active')
-        const next = group?.querySelectorAll(
-          'div[class*="language-"]:not(.language-id)'
-        )?.[i]
+        const i = Array.from(group.querySelectorAll('input')).indexOf(el)
+        if (i < 0) return
+
+        const blocks = group.querySelector('.blocks')
+        if (!blocks) return
+
+        const current = Array.from(blocks.children).find((child) =>
+          child.classList.contains('active')
+        )
+        if (!current) return
+
+        const next = blocks.children[i]
+        if (!next || current === next) return
 
-        if (current && next && current !== next) {
         current.classList.remove('active')
         next.classList.add('active')
       }
-      }
     })
   }
 }

src/client/app/index.ts

@@ -59,7 +59,7 @@ const VitePressApp = defineComponent({
     useCodeGroups()
 
     if (Theme.setup) Theme.setup()
-    return () => h(Theme.Layout)
+    return () => h(Theme.Layout!)
   }
 })
 

src/client/app/theme.ts

@@ -9,7 +9,7 @@ export interface EnhanceAppContext {
 }
 
 export interface Theme {
-  Layout: Component
+  Layout?: Component
   enhanceApp?: (ctx: EnhanceAppContext) => Awaitable<void>
   extends?: Theme
 

src/client/theme-default/styles/components/vp-code-group.css

@@ -66,13 +66,19 @@
   background-color: var(--vp-code-tab-active-bar-color);
 }
 
-.vp-code-group div[class*='language-'] {
+.vp-code-group div[class*='language-'],
+.vp-block {
   display: none;
   margin-top: 0 !important;
   border-top-left-radius: 0 !important;
   border-top-right-radius: 0 !important;
 }
 
-.vp-code-group div[class*='language-'].active {
+.vp-code-group div[class*='language-'].active,
+.vp-block.active {
   display: block;
 }
+
+.vp-block {
+  padding: 20px 24px;
+}

src/client/theme-default/styles/components/vp-doc.css

@@ -278,7 +278,8 @@
   color: var(--vp-c-brand-dark);
 }
 
-.vp-doc div[class*='language-'] {
+.vp-doc div[class*='language-'],
+.vp-block {
   position: relative;
   margin: 16px -24px;
   background-color: var(--vp-code-block-bg);
@@ -287,7 +288,8 @@
 }
 
 @media (min-width: 640px) {
-  .vp-doc div[class*='language-'] {
+  .vp-doc div[class*='language-'],
+  .vp-block {
     border-radius: 8px;
     margin: 16px 0;
   }

src/node/markdown/plugins/containers.ts

@@ -74,13 +74,22 @@ function createCodeGroup(options: Options): ContainerArgs {
             );
             ++i
           ) {
-            if (tokens[i].type === 'fence' && tokens[i].tag === 'code') {
-              const title = extractTitle(tokens[i].info)
+            const isHtml = tokens[i].type === 'html_block'
+
+            if (
+              (tokens[i].type === 'fence' && tokens[i].tag === 'code') ||
+              isHtml
+            ) {
+              const title = extractTitle(
+                isHtml ? tokens[i].content : tokens[i].info,
+                isHtml
+              )
+
+              if (title) {
                 const id = nanoid(7)
                 tabs += `<input type="radio" name="group-${name}" id="tab-${id}" ${checked}><label for="tab-${id}">${title}</label>`
 
-              if (checked) {
-                tokens[i].info += ' active'
+                if (checked && !isHtml) tokens[i].info += ' active'
                 checked = ''
               }
             }

src/node/markdown/plugins/preWrapper.ts

@@ -9,14 +9,18 @@ export function preWrapperPlugin(md: MarkdownIt, options: Options) {
   md.renderer.rules.fence = (...args) => {
     const [tokens, idx] = args
     const token = tokens[idx]
+
     // remove title from info
     token.info = token.info.replace(/\[.*\]/, '')
 
+    const active = / active( |$)/.test(token.info) ? ' active' : ''
+    token.info = token.info.replace(/ active$/, '').replace(/ active /, ' ')
+
     const lang = extractLang(token.info)
     const rawCode = fence(...args)
-    return `<div class="language-${lang}${getAdaptiveThemeMarker(options)}${
-      / active( |$)/.test(token.info) ? ' active' : ''
-    }"><button title="Copy Code" class="copy"></button><span class="lang">${lang}</span>${rawCode}</div>`
+    return `<div class="language-${lang}${getAdaptiveThemeMarker(
+      options
+    )}${active}"><button title="Copy Code" class="copy"></button><span class="lang">${lang}</span>${rawCode}</div>`
   }
 }
 
@@ -24,7 +28,12 @@ export function getAdaptiveThemeMarker(options: Options) {
   return options.hasSingleTheme ? '' : ' vp-adaptive-theme'
 }
 
-export function extractTitle(info: string) {
+export function extractTitle(info: string, html = false) {
+  if (html) {
+    return (
+      info.replace(/<!--[^]*?-->/g, '').match(/data-title="(.*?)"/)?.[1] || ''
+    )
+  }
   return info.match(/\[(.*)\]/)?.[1] || extractLang(info) || 'txt'
 }