bedazzlr 0.2.6 → 0.2.7

package/lib/index.js

@@ -1,8 +1,8 @@
-/* global document window console */
+/* global document window console requestAnimationFrame */
 
 import emptyStack from '@iter-tools/imm-stack';
 import classNames from 'classnames';
-import { streamParse } from 'bablr';
+import { buildModule } from 'bablr/enhanceable';
 
 import { CloseNodeTag, LiteralTag, OpenNodeTag, ReferenceTag } from '@bablr/agast-helpers/symbols';
 import { printSelfClosingNodeTag } from '@bablr/agast-helpers/print';
@@ -16,7 +16,7 @@ import {
   buildReferenceMatcher,
 } from '@bablr/helpers/builders';
 import { buildEmbeddedMatcher } from '@bablr/agast-vm-helpers/builders';
-import { hoistTrivia } from '@bablr/agast-helpers/stream';
+import { getStreamIterator, hoistTrivia, StreamIterable, wait } from '@bablr/agast-helpers/stream';
 
 let anchorStyle =
   'display: inline-block; position: relative; vertical-align: top; pointer-events: none;';
@@ -116,10 +116,34 @@ export const removeDocumentListeners = () => {
   }
 };
 
-export const highlightCode = (el, language, matcher = language.defaultMatcher, options) => {
-  let tags = hoistTrivia(
-    streamParse(language, matcher, el.innerText, null, { ...options, holdShiftedNodes: true }),
-  )[Symbol.iterator]();
+function* __chunkify(chunkSize, str) {
+  let i = -1;
+  for (const chr of str) {
+    if (++i === chunkSize) {
+      yield wait(new Promise((resolve) => requestAnimationFrame(resolve)));
+      i = -1;
+    }
+    yield chr;
+  }
+}
+
+const chunkify = (chunkSize, str) => {
+  return new StreamIterable(__chunkify(chunkSize, str));
+};
+
+let { streamParse } = buildModule();
+
+function* __highlightCode(el, language, matcher, options = {}) {
+  let { chunkSize, bablr: bablrOptions = {} } = options;
+  let input = chunkSize ? chunkify(chunkSize, el.innerText) : el.innerText;
+  let tags = getStreamIterator(
+    hoistTrivia(
+      streamParse(language, matcher, input, null, {
+        ...bablrOptions,
+        holdShiftedNodes: true,
+      }),
+    ),
+  );
 
   let open;
   let referenceTag = null;
@@ -131,7 +155,13 @@ export const highlightCode = (el, language, matcher = language.defaultMatcher, o
 
   let step = tags.next();
 
-  while (!step.done) {
+  while (true) {
+    if (step instanceof Promise) {
+      step = yield wait(step);
+    }
+
+    if (step.done) break;
+
     let tag = step.value;
 
     if (tag.type === OpenNodeTag) {
@@ -472,9 +502,13 @@ export const highlightCode = (el, language, matcher = language.defaultMatcher, o
       store.selectionState = 'none';
     }
   });
+}
+
+export const highlightCode = (el, language, matcher = language.defaultMatcher, options) => {
+  return new StreamIterable(__highlightCode(el, language, matcher, options));
 };
 
-export const highlightAll = (languages, options) => {
+function* __highlightAll(languages, options) {
   let codeBlocks = document.querySelectorAll('code');
 
   for (let block of codeBlocks) {
@@ -508,4 +542,8 @@ export const highlightAll = (languages, options) => {
       console.warn(e);
     }
   }
+}
+
+export const highlightAll = (languages, options) => {
+  return new StreamIterable(__highlightAll(languages, options));
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bedazzlr",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "Code block syntax highlighting using BABLR",
   "author": "Conrad Buck<conartist6@gmail.com>",
   "type": "module",
@@ -20,7 +20,7 @@
     "@bablr/boot": "0.11.3",
     "@bablr/helpers": "0.25.2",
     "@iter-tools/imm-stack": "1.2.0",
-    "bablr": "0.11.9",
+    "bablr": "0.11.10",
     "classnames": "2.5.1"
   },
   "devDependencies": {