fix: support ANSI language with multiple themes (#1259)

Author: dan-lee

packages/core/src/highlight/code-to-tokens.ts

@@ -42,6 +42,7 @@ export function codeToTokens(
       primitive,
       code,
       options,
+      codeToTokensBase,
     )
 
     grammarState = getLastGrammarStateFromMap(themeTokens)

packages/primitive/src/highlight/code-to-tokens-themes.ts

@@ -14,14 +14,15 @@ export function codeToTokensWithThemes(
   primitive: ShikiPrimitive,
   code: string,
   options: CodeToTokensWithThemesOptions,
+  codeToTokensBaseFn: typeof codeToTokensBase = codeToTokensBase,
 ): ThemedTokenWithVariants[][] {
   const themes = Object
     .entries(options.themes)
     .filter(i => i[1])
     .map(i => ({ color: i[0], theme: i[1]! }))
 
   const themedTokens = themes.map((t) => {
-    const tokens = codeToTokensBase(primitive, code, {
+    const tokens = codeToTokensBaseFn(primitive, code, {
       ...options,
       theme: t.theme,
     })

packages/shiki/test/ansi.test.ts

@@ -44,3 +44,13 @@ Done in 15.7s`, { theme: 'dark-plus', lang: 'ansi' })
 
   await expect(out).toMatchFileSnapshot('./out/ansi-dark-plus.html')
 })
+
+// https://github.com/shikijs/shiki/issues/1257
+it('renders ansi with multiple themes', async () => {
+  const out = await codeToHtml('\x1B[32mhello\x1B[0m', {
+    lang: 'ansi',
+    themes: { light: 'github-light', dark: 'github-dark' },
+  })
+
+  await expect(out).toMatchFileSnapshot('./out/ansi-multi-themes.html')
+})