fix: cache based on theme (#187)

rehype-pretty · Mar 29, 2024 · 50e9a32 · 50e9a32
1 parent 0782823
commit 50e9a32
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -162,7 +162,7 @@ function apply(
     .flatMap((c) => c);
 }
 
-const globalHighlighterCache = new WeakMap<Options, Promise<Highlighter>>();
+const globalHighlighterCache = new Map<string, Promise<Highlighter>>();
 const hastParser = unified().use(rehypeParse, { fragment: true });
 
 export default function rehypePrettyCode(
@@ -184,7 +184,9 @@ export default function rehypePrettyCode(
     onVisitCaption,
   } = options;
 
-  let cachedHighlighter = globalHighlighterCache.get(options);
+  const key = JSON.stringify(theme);
+
+  let cachedHighlighter = globalHighlighterCache.get(key);
   if (!cachedHighlighter) {
     cachedHighlighter = getHighlighter({
       themes:
@@ -193,7 +195,7 @@ export default function rehypePrettyCode(
           : Object.values(theme),
       langs: ['plaintext'],
     });
-    globalHighlighterCache.set(options, cachedHighlighter);
+    globalHighlighterCache.set(key, cachedHighlighter);
   }
 
   const defaultCodeBlockLang =