|
| 1 | +import { createRequire } from 'node:module' |
1 | 2 | import type { BundledLanguage, BundledTheme, HighlighterGeneric } from 'shiki'
|
2 |
| -import { createHighlighter } from 'shiki' |
3 |
| -import { bundledLanguageNames } from '../../shiki.js' |
| 3 | +import { createHighlighter, isSpecialLang } from 'shiki' |
| 4 | +import { createSyncFn } from 'synckit' |
| 5 | +import type { ShikiResolveLang } from '../../resolveLang.js' |
4 | 6 | import type { ShikiHighlightOptions } from '../../types.js'
|
| 7 | +import { resolveLanguage } from '../../utils.js' |
| 8 | + |
| 9 | +const require = createRequire(import.meta.url) |
| 10 | + |
| 11 | +const resolveLangSync = createSyncFn<ShikiResolveLang>( |
| 12 | + require.resolve('@vuepress/plugin-shiki/resolveLang'), |
| 13 | +) |
| 14 | + |
| 15 | +export type ShikiLoadLang = (lang: string) => boolean |
5 | 16 |
|
6 | 17 | export const createShikiHighlighter = async ({
|
7 |
| - langs = bundledLanguageNames, |
| 18 | + langs = [], |
8 | 19 | langAlias = {},
|
9 | 20 | defaultLang,
|
10 | 21 | shikiSetup,
|
11 | 22 | ...options
|
12 |
| -}: ShikiHighlightOptions = {}): Promise< |
13 |
| - HighlighterGeneric<BundledLanguage, BundledTheme> |
14 |
| -> => { |
15 |
| - const shikiHighlighter = await createHighlighter({ |
16 |
| - langs, |
| 23 | +}: ShikiHighlightOptions = {}): Promise<{ |
| 24 | + highlighter: HighlighterGeneric<BundledLanguage, BundledTheme> |
| 25 | + loadLang: ShikiLoadLang |
| 26 | +}> => { |
| 27 | + const highlighter = await createHighlighter({ |
| 28 | + langs: [...langs, ...Object.values(langAlias)], |
17 | 29 | langAlias,
|
18 | 30 | themes:
|
19 | 31 | 'themes' in options
|
20 | 32 | ? Object.values(options.themes)
|
21 | 33 | : [options.theme ?? 'nord'],
|
22 | 34 | })
|
23 | 35 |
|
24 |
| - await shikiSetup?.(shikiHighlighter) |
| 36 | + const loadLang = (lang: string): boolean => { |
| 37 | + if (isSpecialLang(lang)) return true |
| 38 | + |
| 39 | + const loadedLangs = highlighter.getLoadedLanguages() |
| 40 | + |
| 41 | + if (!loadedLangs.includes(lang)) { |
| 42 | + const resolvedLang = resolveLangSync(lang) |
| 43 | + |
| 44 | + if (!resolvedLang.length) return false |
| 45 | + |
| 46 | + highlighter.loadLanguageSync(resolvedLang) |
| 47 | + } |
| 48 | + |
| 49 | + return true |
| 50 | + } |
| 51 | + |
| 52 | + // patch for twoslash - https://github.com/vuejs/vitepress/issues/4334 |
| 53 | + const rawGetLanguage = highlighter.getLanguage |
| 54 | + |
| 55 | + highlighter.getLanguage = (name) => { |
| 56 | + const lang = typeof name === 'string' ? name : name.name |
| 57 | + |
| 58 | + loadLang(resolveLanguage(lang)) |
| 59 | + |
| 60 | + return rawGetLanguage.call(highlighter, name) |
| 61 | + } |
| 62 | + |
| 63 | + await shikiSetup?.(highlighter) |
25 | 64 |
|
26 |
| - return shikiHighlighter |
| 65 | + return { highlighter, loadLang } |
27 | 66 | }
|
0 commit comments