fix(core): fix the incorrect parameter when using a single theme (#723)

shikijs · Jul 25, 2024 · 5e2a69e · 5e2a69e
1 parent 4111b03
commit 5e2a69e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/packages/core/src/code-to-tokens.ts b/packages/core/src/code-to-tokens.ts
@@ -60,7 +60,7 @@ export function codeToTokens(
     rootStyle = defaultColor ? undefined : [fg, bg].join(';')
   }
   else if ('theme' in options) {
-    const colorReplacements = resolveColorReplacements(options.theme, options.colorReplacements)
+    const colorReplacements = resolveColorReplacements(options.theme, options)
 
     tokens = codeToTokensBase(
       internal,

diff --git a/packages/shiki/test/color-replacement.test.ts b/packages/shiki/test/color-replacement.test.ts
@@ -70,6 +70,20 @@ it('flat colorReplacements', async () => {
     `)
 })
 
+it('single theme colorReplacements', async () => {
+  const result = await codeToHtml('console.log("hi")', {
+    lang: 'js',
+    theme: 'vitesse-light',
+    colorReplacements: {
+      '#393a34': 'var(---replaced-1)',
+      '#b07d48': 'var(---replaced-2)',
+    },
+  })
+
+  expect(result).toContain('var(---replaced-1)')
+  expect(result).toContain('var(---replaced-2)')
+})
+
 it('scoped colorReplacements', async () => {
   const customLightTheme: ThemeRegistrationResolved = {
     name: 'custom-light',