feat: expose getLastGrammarState shorthand

packages/core/src/bundle-factory.ts

@@ -1,5 +1,5 @@
 import type { Root } from 'hast'
-import type { BundledHighlighterOptions, CodeToHastOptions, CodeToTokensBaseOptions, CodeToTokensOptions, CodeToTokensWithThemesOptions, HighlighterCoreOptions, HighlighterGeneric, LanguageInput, RequireKeys, SpecialLanguage, SpecialTheme, ThemeInput, ThemedToken, ThemedTokenWithVariants, TokensResult } from './types'
+import type { BundledHighlighterOptions, CodeToHastOptions, CodeToTokensBaseOptions, CodeToTokensOptions, CodeToTokensWithThemesOptions, GrammarState, HighlighterCoreOptions, HighlighterGeneric, LanguageInput, RequireKeys, SpecialLanguage, SpecialTheme, ThemeInput, ThemedToken, ThemedTokenWithVariants, TokensResult } from './types'
 import { isSpecialLang, isSpecialTheme } from './utils'
 import { createHighlighterCore } from './highlighter'
 import { ShikiError } from './error'
@@ -115,9 +115,15 @@ export interface ShorthandsBundle<L extends string, T extends string> {
   codeToTokensWithThemes: (code: string, options: RequireKeys<CodeToTokensWithThemesOptions<L, T>, 'themes' | 'lang'>) => Promise<ThemedTokenWithVariants[][]>
 
   /**
-   * Get internal singleton highlighter.
+   * Get the singleton highlighter.
    */
   getSingletonHighlighter: (options?: Partial<BundledHighlighterOptions<L, T>>) => Promise<HighlighterGeneric<L, T>>
+
+  /**
+   * Shorthand for `getLastGrammarState` with auto-loaded theme and language.
+   * A singleton highlighter it maintained internally.
+   */
+  getLastGrammarState: (code: string, options: CodeToTokensBaseOptions<L, T>) => Promise<GrammarState>
 }
 
 export function makeSingletonHighlighter<L extends string, T extends string>(createHighlighter: CreateHighlighterFactory<L, T>) {
@@ -196,5 +202,13 @@ export function createSingletonShorthands<L extends string, T extends string >(
       })
       return shiki.codeToTokensWithThemes(code, options)
     },
+
+    async getLastGrammarState(code, options) {
+      const shiki = await getSingletonHighlighter({
+        langs: [options.lang as L],
+        themes: [options.theme as T],
+      })
+      return shiki.getLastGrammarState(code, options)
+    },
   }
 }

packages/shiki/src/bundle-full.ts

@@ -40,6 +40,7 @@ export const {
   codeToTokensBase,
   codeToTokensWithThemes,
   getSingletonHighlighter,
+  getLastGrammarState,
 } = /* @__PURE__ */ createSingletonShorthands<
   BundledLanguage,
   BundledTheme

packages/shiki/src/bundle-web.ts

@@ -40,6 +40,7 @@ export const {
   codeToTokens,
   codeToTokensWithThemes,
   getSingletonHighlighter,
+  getLastGrammarState,
 } = /* @__PURE__ */ createSingletonShorthands<
   BundledLanguage,
   BundledTheme