feat: allow passing grammarStates to codeToTokensWithThemes

shikijs · Oct 6, 2024 · 6c3a2d3 · 6c3a2d3
1 parent ea4b8dd
commit 6c3a2d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/packages/core/src/highlight/code-to-tokens-themes.ts b/packages/core/src/highlight/code-to-tokens-themes.ts
@@ -22,6 +22,7 @@ export function codeToTokensWithThemes(
   const tokens = syncThemesTokenization(
     ...themes.map(t => codeToTokensBase(internal, code, {
       ...options,
+      grammarState: options.grammarStates?.[t.color],
       theme: t.theme,
     })),
   )

diff --git a/packages/types/src/tokens.ts b/packages/types/src/tokens.ts
@@ -41,6 +41,9 @@ export interface CodeToTokensWithThemesOptions<Languages = string, Themes = stri
    * ```
    */
   themes: Partial<Record<string, Themes | ThemeRegistrationAny | SpecialTheme>>
+
+  // TODO:
+  grammarStates?: Partial<Record<string, GrammarState>>
 }
 
 export interface ThemedTokenScopeExplanation {