feat: accept shiki theme option (#40)

ocavue · Jun 21, 2024 · fbfbfc9 · fbfbfc9
1 parent 3a38dc0
commit fbfbfc9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/shiki.ts b/src/shiki.ts
@@ -1,16 +1,20 @@
 import { Decoration } from 'prosemirror-view'
-import { type BundledLanguage, type Highlighter } from 'shiki'
+import type { BundledLanguage, BundledTheme, Highlighter } from 'shiki'
 
 import type { Parser } from './types'
 
 export type { Parser }
 
-export function createParser(highlighter: Highlighter): Parser {
+export function createParser(
+  highlighter: Highlighter,
+  options?: { theme?: BundledTheme },
+): Parser {
   return function parser({ content, language, pos }) {
     const decorations: Decoration[] = []
 
     const tokens = highlighter.codeToTokensBase(content, {
       lang: language as BundledLanguage,
+      theme: options?.theme,
     })
 
     let from = pos + 1