run blocking

GitbookIO · Apr 13, 2024 · b7b0490 · b7b0490
1 parent b56490d
commit b7b0490
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/src/components/DocumentView/CodeBlock/highlight.ts b/src/components/DocumentView/CodeBlock/highlight.ts
@@ -46,10 +46,13 @@ export async function highlight(block: DocumentBlockCode): Promise<HighlightLine
         return plainHighlighting(block);
     }
 
-    lineCount += block.nodes.length;
-    blockCount++;
+    const overLimit = await renderer.runBlocking(async () => {
+            lineCount += block.nodes.length;
+            blockCount++;
+        return lineCount > LINE_LIMIT;
+    })
 
-    if (lineCount > LINE_LIMIT) {
+    if (overLimit) {
         return plainHighlighting(block);
     }
 
@@ -60,14 +63,12 @@ export async function highlight(block: DocumentBlockCode): Promise<HighlightLine
         return a.start - b.start;
     });
 
-    const lines = await renderer.runBlocking(async () => {
-        const highlighter = await loadHighlighter();
-        await loadHighlighterLanguage(highlighter, langName);
+    const highlighter = await loadHighlighter();
+    await loadHighlighterLanguage(highlighter, langName);
 
-        return highlighter.codeToTokensBase(code, {
-            lang: langName,
-            tokenizeMaxLineLength: 120,
-        });
+    const lines = highlighter.codeToTokensBase(code, {
+        lang: langName,
+        tokenizeMaxLineLength: 120,
     });
 
     console.log(