fix(rehype): support special languages (#808)

shikijs · Oct 24, 2024 · b8471aa · b8471aa
1 parent f76a371
commit b8471aa
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/packages/rehype/src/core.ts b/packages/rehype/src/core.ts
@@ -6,6 +6,7 @@ import type { Root } from 'hast'
 import type { Transformer } from 'unified'
 import type { RehypeShikiHandler } from './handlers'
 import type { RehypeShikiCoreOptions } from './types'
+import { isSpecialLang } from 'shiki/core'
 import { visit } from 'unist-util-visit'
 import { InlineCodeHandlers, PreHandler } from './handlers'
 
@@ -92,7 +93,7 @@ function rehypeShikiFromHighlighter(
       if (!lang)
         return defaultLanguage
 
-      if (highlighter.getLoadedLanguages().includes(lang))
+      if (highlighter.getLoadedLanguages().includes(lang) || isSpecialLang(lang))
         return lang
 
       if (lazy) {

diff --git a/packages/rehype/test/fixtures/a.core.out.html b/packages/rehype/test/fixtures/a.core.out.html
diff --git a/packages/rehype/test/fixtures/a.md b/packages/rehype/test/fixtures/a.md
diff --git a/packages/rehype/test/fixtures/a.out.html b/packages/rehype/test/fixtures/a.out.html