fix: inline code highlighter

wevm · Sep 27, 2024 · 6ebc071 · 6ebc071
1 parent 13db87c
commit 6ebc071
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/.changeset/pretty-rice-heal.md b/.changeset/pretty-rice-heal.md
@@ -0,0 +1,5 @@
+---
+"vocs": patch
+---
+
+Fixed inline code highlighter.
diff --git a/src/vite/plugins/rehype/inline-shiki.ts b/src/vite/plugins/rehype/inline-shiki.ts
@@ -1,7 +1,13 @@
 import type { RehypeShikiCoreOptions } from '@shikijs/rehype/core'
 import type { Root } from 'hast'
-import type { BuiltinLanguage, BuiltinTheme, Highlighter } from 'shiki'
-import { bundledLanguages, createHighlighter } from 'shiki'
+import type {
+  BuiltinLanguage,
+  BuiltinTheme,
+  BundledLanguage,
+  BundledTheme,
+  HighlighterGeneric,
+} from 'shiki'
+import { bundledLanguages, getSingletonHighlighter } from 'shiki'
 import type { LanguageInput } from 'shiki/core'
 import type { Plugin } from 'unified'
 import { visit } from 'unist-util-visit'
@@ -17,7 +23,7 @@ export type RehypeInlineShikiOptions = RehypeShikiCoreOptions & {
   langs?: Array<LanguageInput | BuiltinLanguage>
 }
 
-let promise: Promise<Highlighter>
+let promise: Promise<HighlighterGeneric<BundledLanguage, BundledTheme>>
 
 export const rehypeInlineShiki: Plugin<[RehypeInlineShikiOptions], Root> = function (
   options = {} as any,
@@ -29,7 +35,7 @@ export const rehypeInlineShiki: Plugin<[RehypeInlineShikiOptions], Root> = funct
 
   return async function (tree) {
     if (!promise)
-      promise = createHighlighter({
+      promise = getSingletonHighlighter({
         themes: themeNames,
         langs,
       })