@mui/internal-docs-infra 0.2.0-alpha.1 → 0.2.0

package/esm/CodeHighlighter/CodeHighlighter.js

@@ -14,11 +14,11 @@ import { getFileNameFromUrl } from "../pipeline/loaderUtils/getFileNameFromUrl.j
 import { codeToFallbackProps } from "./codeToFallbackProps.js";
 import * as Errors from "./errors.js";
 import { jsx as _jsx } from "react/jsx-runtime";
-var DEFAULT_HIGHLIGHT_AT = 'stream';
 var DEBUG = false; // Set to true for debugging purposes
 
 function createClientProps(props) {
-  var highlightAt = props.highlightAt === 'stream' ? 'init' : props.highlightAt;
+  var highlightAfter = props.highlightAfter === 'stream' ? 'init' : props.highlightAfter;
+  var enhanceAfter = props.enhanceAfter === 'stream' ? 'init' : props.enhanceAfter;
   var contentProps = _extends({
     code: props.code || props.precompute,
     components: props.components,
@@ -37,7 +37,8 @@ function createClientProps(props) {
     fileName: props.fileName,
     initialVariant: props.initialVariant,
     defaultVariant: props.defaultVariant,
-    highlightAt: highlightAt || 'init',
+    highlightAfter: highlightAfter || 'idle',
+    enhanceAfter: enhanceAfter || 'idle',
     skipFallback: props.skipFallback,
     controlled: props.controlled,
     name: props.name,
@@ -169,12 +170,19 @@ function _CodeSourceLoader() {
                 _iterator.f();
               }
             }
+            var output = 'hastGzip';
+            if (props.deferParsing === 'json') {
+              output = 'hastJson';
+            } else if (props.deferParsing === 'none') {
+              output = 'hast';
+            }
             return loadVariant(variantUrl, variantName, variantCode, {
               sourceParser: props.sourceParser,
               loadSource: props.loadSource,
               loadVariantMeta: props.loadVariantMeta,
               sourceTransformers: props.sourceTransformers,
-              globalsCode: resolvedGlobalsCode
+              globalsCode: resolvedGlobalsCode,
+              output: output
             }).then(function (variant) {
               return {
                 name: variantName,
@@ -247,6 +255,11 @@ function _CodeHighlighterSuspense() {
     return _regenerator().w(function (_context3) {
       while (1) switch (_context3.n) {
         case 0:
+          _context3.n = 1;
+          return new Promise(function (resolve) {
+            setTimeout(resolve, 0);
+          });
+        case 1:
           return _context3.a(2, props.children);
       }
     }, _callee3);
@@ -280,40 +293,46 @@ function renderWithInitialSource(props) {
     components: components
   }, fallbackProps), props.contentProps);
   var fallback = /*#__PURE__*/_jsx(ContentLoading, _extends({}, contentProps));
-  if (props.forceClient) {
-    return renderCodeHighlighter(_extends(_extends({}, props), {}, {
-      fallback: fallback
-    }));
+  if (props.highlightAfter === 'stream' && !props.forceClient) {
+    return /*#__PURE__*/_jsx(React.Suspense, {
+      fallback: fallback,
+      children: /*#__PURE__*/_jsx(CodeHighlighterSuspense, {
+        children: renderCodeHighlighter(_extends(_extends({}, props), {}, {
+          fallback: fallback,
+          skipFallback: props.enhanceAfter === 'stream'
+        }))
+      })
+    });
   }
-  return /*#__PURE__*/_jsx(React.Suspense, {
-    fallback: fallback,
-    children: /*#__PURE__*/_jsx(CodeHighlighterSuspense, {
-      children: renderCodeHighlighter(_extends(_extends({}, props), {}, {
-        fallback: fallback,
-        skipFallback: true
-      }))
-    })
-  });
+  return renderCodeHighlighter(_extends(_extends({}, props), {}, {
+    fallback: fallback
+  }));
 }
 function CodeInitialSourceLoader(_x3) {
   return _CodeInitialSourceLoader.apply(this, arguments);
 }
 function _CodeInitialSourceLoader() {
   _CodeInitialSourceLoader = _asyncToGenerator(/*#__PURE__*/_regenerator().m(function _callee4(props) {
-    var url, initialVariant, highlightAt, fallbackUsesExtraFiles, fallbackUsesAllVariants, sourceParser, loadSource, loadVariantMeta, loadCodeMeta, fileName, variants, globalsCode, ContentLoading, _yield$loadFallbackCo, code, initialFilename, initialSource, initialExtraFiles, processedGlobalsCode;
+    var url, initialVariant, highlightAfter, fallbackUsesExtraFiles, fallbackUsesAllVariants, sourceParser, loadSource, loadVariantMeta, loadCodeMeta, fileName, variants, globalsCode, ContentLoading, output, _yield$loadFallbackCo, code, initialFilename, initialSource, initialExtraFiles, processedGlobalsCode;
     return _regenerator().w(function (_context4) {
       while (1) switch (_context4.n) {
         case 0:
-          url = props.url, initialVariant = props.initialVariant, highlightAt = props.highlightAt, fallbackUsesExtraFiles = props.fallbackUsesExtraFiles, fallbackUsesAllVariants = props.fallbackUsesAllVariants, sourceParser = props.sourceParser, loadSource = props.loadSource, loadVariantMeta = props.loadVariantMeta, loadCodeMeta = props.loadCodeMeta, fileName = props.fileName, variants = props.variants, globalsCode = props.globalsCode, ContentLoading = props.ContentLoading;
+          url = props.url, initialVariant = props.initialVariant, highlightAfter = props.highlightAfter, fallbackUsesExtraFiles = props.fallbackUsesExtraFiles, fallbackUsesAllVariants = props.fallbackUsesAllVariants, sourceParser = props.sourceParser, loadSource = props.loadSource, loadVariantMeta = props.loadVariantMeta, loadCodeMeta = props.loadCodeMeta, fileName = props.fileName, variants = props.variants, globalsCode = props.globalsCode, ContentLoading = props.ContentLoading;
           if (url) {
             _context4.n = 1;
             break;
           }
           throw new Errors.ErrorCodeHighlighterServerMissingUrl();
         case 1:
+          output = 'hastGzip';
+          if (props.deferParsing === 'json') {
+            output = 'hastJson';
+          } else if (props.deferParsing === 'none') {
+            output = 'hast';
+          }
           _context4.n = 2;
           return loadFallbackCode(url, initialVariant, props.code, {
-            shouldHighlight: highlightAt === 'init',
+            shouldHighlight: highlightAfter === 'init',
             fallbackUsesExtraFiles: fallbackUsesExtraFiles,
             fallbackUsesAllVariants: fallbackUsesAllVariants,
             sourceParser: sourceParser,
@@ -322,7 +341,8 @@ function _CodeInitialSourceLoader() {
             loadCodeMeta: loadCodeMeta,
             initialFilename: fileName,
             variants: variants,
-            globalsCode: globalsCode
+            globalsCode: globalsCode,
+            output: output
           });
         case 2:
           _yield$loadFallbackCo = _context4.v;
@@ -345,7 +365,7 @@ function _CodeInitialSourceLoader() {
   return _CodeInitialSourceLoader.apply(this, arguments);
 }
 export function CodeHighlighter(props) {
-  var _code, _props$components2;
+  var _code, _props$precompute, _props$components2;
   // Validate mutually exclusive props
   if (props.children && (props.code || props.precompute)) {
     throw new Errors.ErrorCodeHighlighterServerInvalidProps();
@@ -381,35 +401,25 @@ export function CodeHighlighter(props) {
   }
   var ContentLoading = props.ContentLoading;
   if (!ContentLoading) {
-    if (props.highlightAt === 'stream') {
-      // if the user explicitly sets highlightAt to 'stream', we need a ContentLoading component
+    if (props.highlightAfter === 'stream') {
+      // if the user explicitly sets highlightAfter to 'stream', we need a ContentLoading component
       throw new Errors.ErrorCodeHighlighterServerMissingContentLoading();
     }
     return renderCodeHighlighter(_extends(_extends({}, props), {}, {
       code: code
     }));
   }
-
-  // Check if any loader functions are available
-  var hasAnyLoaderFunction = !!(props.loadCodeMeta || props.loadVariantMeta || props.loadSource || props.sourceParser || props.sourceTransformers);
-
-  // If no loader functions are available, skip async loading and go directly to client
-  if (!hasAnyLoaderFunction) {
-    return renderCodeHighlighter(_extends(_extends({}, props), {}, {
-      code: code
-    }));
-  }
   var initialKey = props.initialVariant || props.variant || props.defaultVariant || variants[0];
-  var initial = (_code = code) == null ? void 0 : _code[initialKey];
+  var initial = ((_code = code) == null ? void 0 : _code[initialKey]) || ((_props$precompute = props.precompute) == null ? void 0 : _props$precompute[initialKey]);
   if (!initial && !((_props$components2 = props.components) != null && _props$components2[initialKey])) {
     throw new Errors.ErrorCodeHighlighterServerMissingVariant(initialKey);
   }
 
   // TODO: use initial.filesOrder to determing which source to use
-  var highlightAt = props.highlightAt || DEFAULT_HIGHLIGHT_AT;
+
   var _maybeInitialData = maybeInitialData(variants, initialKey, code || props.precompute, undefined,
     // TODO: use initial.filesOrder if provided?
-    highlightAt === 'init', props.fallbackUsesExtraFiles, props.fallbackUsesAllVariants),
+    props.highlightAfter === 'init', props.fallbackUsesExtraFiles, props.fallbackUsesAllVariants),
     initialData = _maybeInitialData.initialData,
     reason = _maybeInitialData.reason;
   if (!initialData) {
@@ -417,8 +427,13 @@ export function CodeHighlighter(props) {
       // eslint-disable-next-line no-console
       console.log('Initial data not found:', reason);
     }
-    if (props.forceClient) {
-      if (highlightAt === 'init') {
+
+    // Check if any loader functions are available
+    var hasAnyLoaderFunction = !!(props.loadCodeMeta || props.loadVariantMeta || props.loadSource || props.sourceParser || props.sourceTransformers);
+
+    // If no loader functions are available, skip async loading and go directly to client
+    if (!hasAnyLoaderFunction || props.forceClient) {
+      if (props.highlightAfter === 'init') {
         throw new Errors.ErrorCodeHighlighterServerInvalidClientMode();
       }
       return renderCodeHighlighter(_extends(_extends({}, props), {}, {

package/esm/CodeHighlighter/CodeHighlighterClient.js

@@ -26,7 +26,7 @@ function useInitialData(_ref) {
     setCode = _ref.setCode,
     fileName = _ref.fileName,
     url = _ref.url,
-    highlightAt = _ref.highlightAt,
+    highlightAfter = _ref.highlightAfter,
     fallbackUsesExtraFiles = _ref.fallbackUsesExtraFiles,
     fallbackUsesAllVariants = _ref.fallbackUsesAllVariants,
     isControlled = _ref.isControlled,
@@ -39,8 +39,8 @@ function useInitialData(_ref) {
     loadSource = _useCodeContext.loadSource,
     loadFallbackCode = _useCodeContext.loadFallbackCode;
   var _React$useMemo = React.useMemo(function () {
-      return maybeInitialData(variants, variantName, code, fileName, highlightAt === 'init', fallbackUsesExtraFiles, fallbackUsesAllVariants);
-    }, [variants, variantName, code, fileName, highlightAt, fallbackUsesExtraFiles, fallbackUsesAllVariants]),
+      return maybeInitialData(variants, variantName, code, fileName, highlightAfter === 'init', fallbackUsesExtraFiles, fallbackUsesAllVariants);
+    }, [variants, variantName, code, fileName, highlightAfter, fallbackUsesExtraFiles, fallbackUsesAllVariants]),
     initialData = _React$useMemo.initialData,
     reason = _React$useMemo.reason;
   var needsFallback = !initialData && !isControlled;
@@ -74,7 +74,7 @@ function useInitialData(_ref) {
             }
             _context.n = 1;
             return loadFallbackCode(url, variantName, code, {
-              shouldHighlight: highlightAt === 'init',
+              shouldHighlight: highlightAfter === 'init',
               fallbackUsesExtraFiles: fallbackUsesExtraFiles,
               fallbackUsesAllVariants: fallbackUsesAllVariants,
               sourceParser: sourceParser,
@@ -105,7 +105,7 @@ function useInitialData(_ref) {
         }
       }, _callee);
     }))();
-  }, [initialData, reason, needsFallback, variantName, code, setCode, highlightAt, url, sourceParser, loadSource, loadVariantMeta, loadCodeMeta, fallbackUsesExtraFiles, fallbackUsesAllVariants, fileName, variants, globalsCode, setProcessedGlobalsCode, loadFallbackCode]);
+  }, [initialData, reason, needsFallback, variantName, code, setCode, highlightAfter, url, sourceParser, loadSource, loadVariantMeta, loadCodeMeta, fallbackUsesExtraFiles, fallbackUsesAllVariants, fileName, variants, globalsCode, setProcessedGlobalsCode, loadFallbackCode]);
 }
 function useAllVariants(_ref3) {
   var readyForContent = _ref3.readyForContent,
@@ -302,36 +302,33 @@ function useAllVariants(_ref3) {
     readyForContent: readyForContent
   };
 }
+function yieldToMain() {
+  var _scheduler;
+  if ((_scheduler = globalThis.scheduler) != null && _scheduler["yield"]) {
+    return globalThis.scheduler["yield"]();
+  }
+
+  // Fall back to yielding with setTimeout.
+  return new Promise(function (resolve) {
+    setTimeout(resolve, 0);
+  });
+}
 function useCodeParsing(_ref6) {
   var code = _ref6.code,
     readyForContent = _ref6.readyForContent,
-    highlightAt = _ref6.highlightAt,
+    highlightAfter = _ref6.highlightAfter,
+    isHydrated = _ref6.isHydrated,
     forceClient = _ref6.forceClient,
     url = _ref6.url;
   var _useCodeContext3 = useCodeContext(),
     parseSource = _useCodeContext3.parseSource,
     parseCode = _useCodeContext3.parseCode;
-
-  // Use useSyncExternalStore to detect hydration
-  var subscribe = React.useCallback(function () {
-    return function () {};
-  }, []);
-  var getSnapshot = React.useCallback(function () {
-    return true;
-  }, []);
-  var getServerSnapshot = React.useCallback(function () {
-    return false;
-  }, []);
-  var useIsHydrated = function useIsHydrated() {
-    return React.useSyncExternalStore(subscribe, getSnapshot, getServerSnapshot);
-  };
-  var isHydrated = useIsHydrated();
-  var _React$useState = React.useState(highlightAt === 'init' || highlightAt === 'hydration' && isHydrated),
+  var _React$useState = React.useState(highlightAfter === 'init' || highlightAfter === 'hydration' && isHydrated),
     _React$useState2 = _slicedToArray(_React$useState, 2),
     isHighlightAllowed = _React$useState2[0],
     setIsHighlightAllowed = _React$useState2[1];
   React.useEffect(function () {
-    if (highlightAt === 'idle') {
+    if (highlightAfter === 'idle') {
       var idleRequest = window.requestIdleCallback(function () {
         setIsHighlightAllowed(true);
       });
@@ -340,16 +337,20 @@ function useCodeParsing(_ref6) {
       };
     }
     return undefined;
-  }, [highlightAt]);
+  }, [highlightAfter]);
 
   // Update highlight allowed state when hydration completes
   React.useEffect(function () {
-    if (highlightAt === 'hydration' && isHydrated) {
-      setIsHighlightAllowed(true);
+    if (highlightAfter === 'hydration' && isHydrated) {
+      // we should ensure that each code highlighter is enhanced as a separate task
+      // this should run from top to bottom
+      yieldToMain().then(function () {
+        return setIsHighlightAllowed(true);
+      });
     }
-  }, [highlightAt, isHydrated]);
+  }, [highlightAfter, isHydrated]);
 
-  // Determine if we should highlight based on the highlightAt setting
+  // Determine if we should highlight based on the highlightAfter setting
   var shouldHighlight = React.useMemo(function () {
     if (!readyForContent) {
       return false;
@@ -851,7 +852,8 @@ export function CodeHighlighterClient(props) {
   }
   var fileName = (controlled == null || (_controlled$selection2 = controlled.selection) == null ? void 0 : _controlled$selection2.fileName) || props.fileName || initialFilename;
   var url = props.url,
-    highlightAt = props.highlightAt,
+    highlightAfter = props.highlightAfter,
+    enhanceAfter = props.enhanceAfter,
     fallbackUsesExtraFiles = props.fallbackUsesExtraFiles,
     fallbackUsesAllVariants = props.fallbackUsesAllVariants;
   useInitialData({
@@ -861,13 +863,54 @@ export function CodeHighlighterClient(props) {
     setCode: setCode,
     fileName: fileName,
     url: url,
-    highlightAt: highlightAt,
+    highlightAfter: highlightAfter,
     fallbackUsesExtraFiles: fallbackUsesExtraFiles,
     fallbackUsesAllVariants: fallbackUsesAllVariants,
     isControlled: isControlled,
     globalsCode: props.globalsCode,
     setProcessedGlobalsCode: setProcessedGlobalsCode
   });
+
+  // Use useSyncExternalStore to detect hydration
+  var subscribe = React.useCallback(function () {
+    return function () {};
+  }, []);
+  var getSnapshot = React.useCallback(function () {
+    return true;
+  }, []);
+  var getServerSnapshot = React.useCallback(function () {
+    return false;
+  }, []);
+  var useIsHydrated = function useIsHydrated() {
+    return React.useSyncExternalStore(subscribe, getSnapshot, getServerSnapshot);
+  };
+  var isHydrated = useIsHydrated();
+  var _React$useState1 = React.useState(enhanceAfter === 'init' || enhanceAfter === 'hydration' && isHydrated),
+    _React$useState10 = _slicedToArray(_React$useState1, 2),
+    isEnhanceAllowed = _React$useState10[0],
+    setIsEnhanceAllowed = _React$useState10[1];
+  React.useEffect(function () {
+    if (enhanceAfter === 'idle') {
+      var idleRequest = window.requestIdleCallback(function () {
+        setIsEnhanceAllowed(true);
+      });
+      return function () {
+        return window.cancelIdleCallback(idleRequest);
+      };
+    }
+    return undefined;
+  }, [enhanceAfter]);
+
+  // Update enhance allowed state when hydration completes
+  React.useEffect(function () {
+    if (enhanceAfter === 'hydration' && isHydrated) {
+      // we should ensure that each code highlighter is enhanced as a separate task
+      // this should run from top to bottom
+      yieldToMain().then(function () {
+        return setIsEnhanceAllowed(true);
+      });
+    }
+  }, [enhanceAfter, isHydrated]);
   var readyForContent = React.useMemo(function () {
     if (!code) {
       return false;
@@ -877,7 +920,7 @@ export function CodeHighlighterClient(props) {
 
   // Separate check for activeCode to determine when to show fallback
   var activeCodeReady = React.useMemo(function () {
-    if (!activeCode) {
+    if (!activeCode || !isEnhanceAllowed) {
       return false;
     }
 
@@ -889,7 +932,7 @@ export function CodeHighlighterClient(props) {
     // For regular code, use the existing hasAllVariants function
     var regularCode = props.code || code;
     return regularCode ? hasAllVariants(variants, regularCode) : false;
-  }, [activeCode, controlled == null ? void 0 : controlled.code, variants, props.code, code]);
+  }, [activeCode, isEnhanceAllowed, controlled == null ? void 0 : controlled.code, variants, props.code, code]);
   useAllVariants({
     readyForContent: readyForContent,
     variants: variants,
@@ -927,7 +970,8 @@ export function CodeHighlighterClient(props) {
   var _useCodeParsing = useCodeParsing({
       code: codeWithGlobals,
       readyForContent: readyForContent || Boolean(props.code),
-      highlightAt: highlightAt,
+      highlightAfter: highlightAfter,
+      isHydrated: isHydrated,
       forceClient: props.forceClient,
       url: props.url
     }),

package/esm/CodeHighlighter/errors.js

@@ -190,7 +190,7 @@ export var ErrorCodeHighlighterServerMissingContentLoading = /*#__PURE__*/functi
 export var ErrorCodeHighlighterServerInvalidClientMode = /*#__PURE__*/function (_ErrorCodeHighlighter17) {
   function ErrorCodeHighlighterServerInvalidClientMode() {
     _classCallCheck(this, ErrorCodeHighlighterServerInvalidClientMode);
-    return _callSuper(this, ErrorCodeHighlighterServerInvalidClientMode, ["Invalid client mode - Client only mode with highlightAt: init requires precomputed and parsed source code"]);
+    return _callSuper(this, ErrorCodeHighlighterServerInvalidClientMode, ["Invalid client mode - Client only mode with highlightAfter: init requires precomputed and parsed source code"]);
   }
   _inherits(ErrorCodeHighlighterServerInvalidClientMode, _ErrorCodeHighlighter17);
   return _createClass(ErrorCodeHighlighterServerInvalidClientMode);

package/esm/CodeHighlighter/loadFallbackCode.js

@@ -99,6 +99,7 @@ function _loadFallbackCode() {
       initialFilename,
       variants,
       globalsCode,
+      output,
       initial,
       _allFileNames,
       fileSource,
@@ -142,7 +143,7 @@ function _loadFallbackCode() {
       while (1) switch (_context4.p = _context4.n) {
         case 0:
           options = _args4.length > 3 && _args4[3] !== undefined ? _args4[3] : {};
-          shouldHighlight = options.shouldHighlight, fallbackUsesExtraFiles = options.fallbackUsesExtraFiles, fallbackUsesAllVariants = options.fallbackUsesAllVariants, sourceParser = options.sourceParser, loadSource = options.loadSource, loadVariantMeta = options.loadVariantMeta, loadCodeMeta = options.loadCodeMeta, initialFilename = options.initialFilename, variants = options.variants, globalsCode = options.globalsCode;
+          shouldHighlight = options.shouldHighlight, fallbackUsesExtraFiles = options.fallbackUsesExtraFiles, fallbackUsesAllVariants = options.fallbackUsesAllVariants, sourceParser = options.sourceParser, loadSource = options.loadSource, loadVariantMeta = options.loadVariantMeta, loadCodeMeta = options.loadCodeMeta, initialFilename = options.initialFilename, variants = options.variants, globalsCode = options.globalsCode, output = options.output;
           loaded = _extends({}, loaded);
 
           // Step 1: Ensure we have the initial variant loaded
@@ -458,7 +459,9 @@ function _loadFallbackCode() {
             // Don't apply transforms for fallback
             disableParsing: !shouldHighlight,
             // Only parse if highlighting is needed
-            globalsCode: resolvedGlobalsCode // Pass resolved globalsCode
+            globalsCode: resolvedGlobalsCode,
+            // Pass resolved globalsCode
+            output: output
           });
         case 38:
           _yield$loadVariant = _context4.v;
@@ -565,6 +568,7 @@ function _loadFallbackCode() {
                       // sourceTransformers
                       disableTransforms: true,
                       disableParsing: !shouldHighlight,
+                      output: output,
                       globalsCode: globalsCodeObjects && globalsCodeObjects.length > 0 ? function () {
                         // Convert globalsCodeObjects to VariantCode | string for this specific variant
                         var variantGlobalsCode = [];

package/esm/CodeHighlighter/loadVariant.js

@@ -6,10 +6,25 @@ import _createForOfIteratorHelper from "@babel/runtime/helpers/esm/createForOfIt
 import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
 import _asyncToGenerator from "@babel/runtime/helpers/esm/asyncToGenerator";
 import * as path from 'path-module';
+import { compress, strToU8 } from 'fflate';
+import { encode } from 'uint8-to-base64';
 import { transformSource } from "./transformSource.js";
 import { transformParsedSource } from "./transformParsedSource.js";
 import { getFileNameFromUrl } from "../pipeline/loaderUtils/index.js";
 import { mergeExternals } from "../pipeline/loaderUtils/mergeExternals.js";
+function compressAsync(input) {
+  var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+  return new Promise(function (resolve, reject) {
+    compress(input, options, function (err, output) {
+      if (err) {
+        reject(err);
+      } else {
+        resolve(output);
+      }
+    });
+  });
+}
+
 /**
  * Check if a path is absolute (either filesystem absolute or URL)
  */
@@ -141,10 +156,12 @@ function _loadSingleFile() {
       finalTransforms,
       sourceString,
       parseSource,
+      hastGzip,
       _args = arguments,
       _t,
       _t2,
-      _t3;
+      _t3,
+      _t4;
     return _regenerator().w(function (_context) {
       while (1) switch (_context.p = _context.n) {
         case 0:
@@ -306,7 +323,7 @@ function _loadSingleFile() {
           finalTransforms = _context.v;
         case 22:
           if (!(typeof finalSource === 'string' && !disableParsing)) {
-            _context.n = 28;
+            _context.n = 31;
             break;
           }
           if (sourceParser) {
@@ -331,13 +348,38 @@ function _loadSingleFile() {
         case 25:
           finalTransforms = _context.v;
         case 26:
-          _context.n = 28;
-          break;
+          if (!(options.output === 'hastGzip' && process.env.NODE_ENV === 'production')) {
+            _context.n = 28;
+            break;
+          }
+          _t3 = encode;
+          _context.n = 27;
+          return compressAsync(strToU8(JSON.stringify(finalSource)), {
+            consume: true,
+            level: 9
+          });
         case 27:
-          _context.p = 27;
-          _t3 = _context.v;
-          throw new Error("Failed to parse source code (variant: ".concat(variantName, ", file: ").concat(fileName, ", url: ").concat(url, "): ").concat(_t3 instanceof Error ? _t3.message : ''));
+          hastGzip = _t3(_context.v);
+          finalSource = {
+            hastGzip: hastGzip
+          };
+          _context.n = 29;
+          break;
         case 28:
+          if (options.output === 'hastJson' || options.output === 'hastGzip') {
+            // in development, we skip compression but still convert to JSON
+            finalSource = {
+              hastJson: JSON.stringify(finalSource)
+            };
+          }
+        case 29:
+          _context.n = 31;
+          break;
+        case 30:
+          _context.p = 30;
+          _t4 = _context.v;
+          throw new Error("Failed to parse source code (variant: ".concat(variantName, ", file: ").concat(fileName, ", url: ").concat(url, "): ").concat(_t4 instanceof Error ? _t4.message : ''));
+        case 31:
           return _context.a(2, {
             source: finalSource,
             transforms: finalTransforms,
@@ -346,7 +388,7 @@ function _loadSingleFile() {
             externals: externalsFromSource
           });
       }
-    }, _callee, null, [[23, 27], [8, 13, 14, 15], [2, 18]]);
+    }, _callee, null, [[23, 30], [8, 13, 14, 15], [2, 18]]);
   }));
   return _loadSingleFile.apply(this, arguments);
 }
@@ -397,7 +439,7 @@ function _loadExtraFiles() {
       convertedKey,
       normalizedConvertedKey,
       _args4 = arguments,
-      _t5;
+      _t6;
     return _regenerator().w(function (_context4) {
       while (1) switch (_context4.p = _context4.n) {
         case 0:
@@ -417,7 +459,7 @@ function _loadExtraFiles() {
           allExternals = {}; // Start loading all extra files in parallel
           extraFilePromises = Object.entries(extraFiles).map(/*#__PURE__*/function () {
             var _ref2 = _asyncToGenerator(/*#__PURE__*/_regenerator().m(function _callee2(_ref) {
-              var _ref3, fileName, fileData, fileUrl, sourceData, transforms, fileResult, filesUsedFromFile, externalsFromFile, _t4;
+              var _ref3, fileName, fileData, fileUrl, sourceData, transforms, fileResult, filesUsedFromFile, externalsFromFile, _t5;
               return _regenerator().w(function (_context2) {
                 while (1) switch (_context2.p = _context2.n) {
                   case 0:
@@ -475,8 +517,8 @@ function _loadExtraFiles() {
                     });
                   case 6:
                     _context2.p = 6;
-                    _t4 = _context2.v;
-                    throw new Error("Failed to load extra file (variant: ".concat(variantName, ", file: ").concat(fileName, ", url: ").concat(baseUrl, "): ").concat(_t4 instanceof Error ? _t4.message : ''));
+                    _t5 = _context2.v;
+                    throw new Error("Failed to load extra file (variant: ".concat(variantName, ", file: ").concat(fileName, ", url: ").concat(baseUrl, "): ").concat(_t5 instanceof Error ? _t5.message : ''));
                   case 7:
                     return _context2.a(2);
                 }
@@ -565,8 +607,8 @@ function _loadExtraFiles() {
           break;
         case 7:
           _context4.p = 7;
-          _t5 = _context4.v;
-          _iterator2.e(_t5);
+          _t6 = _context4.v;
+          _iterator2.e(_t6);
         case 8:
           _context4.p = 8;
           _iterator2.f();
@@ -681,7 +723,7 @@ function _loadVariant() {
       _extraFilesResult,
       finalVariant,
       _args6 = arguments,
-      _t8;
+      _t9;
     return _regenerator().w(function (_context6) {
       while (1) switch (_context6.p = _context6.n) {
         case 0:
@@ -726,8 +768,8 @@ function _loadVariant() {
           break;
         case 5:
           _context6.p = 5;
-          _t8 = _context6.v;
-          throw new Error("Failed to load variant code (variant: ".concat(variantName, ", url: ").concat(variant, "): ").concat(JSON.stringify(_t8)));
+          _t9 = _context6.v;
+          throw new Error("Failed to load variant code (variant: ".concat(variantName, ", url: ").concat(variant, "): ").concat(JSON.stringify(_t9)));
         case 6:
           loadedFiles = new Set();
           if (url) {
@@ -833,7 +875,7 @@ function _loadVariant() {
           // Process all globals items in parallel
           globalsPromises = globalsCode.map(/*#__PURE__*/function () {
             var _ref4 = _asyncToGenerator(/*#__PURE__*/_regenerator().m(function _callee4(globalsItem) {
-              var globalsVariant, _getFileNameFromUrl2, globalsFileName, globalsResult, _t6, _t7;
+              var globalsVariant, _getFileNameFromUrl2, globalsFileName, globalsResult, _t7, _t8;
               return _regenerator().w(function (_context5) {
                 while (1) switch (_context5.p = _context5.n) {
                   case 0:
@@ -869,8 +911,8 @@ function _loadVariant() {
                     break;
                   case 4:
                     _context5.p = 4;
-                    _t6 = _context5.v;
-                    throw new Error("Failed to load globalsCode variant metadata (variant: ".concat(variantName, ", url: ").concat(globalsItem, "): ").concat(JSON.stringify(_t6)));
+                    _t7 = _context5.v;
+                    throw new Error("Failed to load globalsCode variant metadata (variant: ".concat(variantName, ", url: ").concat(globalsItem, "): ").concat(JSON.stringify(_t7)));
                   case 5:
                     _context5.n = 7;
                     break;
@@ -888,8 +930,8 @@ function _loadVariant() {
                     return _context5.a(2, globalsResult);
                   case 9:
                     _context5.p = 9;
-                    _t7 = _context5.v;
-                    throw new Error("Failed to load globalsCode (variant: ".concat(variantName, "): ").concat(_t7 instanceof Error ? _t7.message : JSON.stringify(_t7)));
+                    _t8 = _context5.v;
+                    throw new Error("Failed to load globalsCode (variant: ".concat(variantName, "): ").concat(_t8 instanceof Error ? _t8.message : JSON.stringify(_t8)));
                   case 10:
                     return _context5.a(2);
                 }

package/esm/CodeHighlighter/maybeInitialData.d.ts

@@ -67,7 +67,7 @@ import { Code, VariantExtraFiles, VariantSource } from "./types.js";
  *   initialKey,
  *   code || props.precompute,
  *   undefined,
- *   highlightAt === 'init',
+ *   highlightAfter === 'init',
  *   props.fallbackUsesExtraFiles,
  *   props.fallbackUsesAllVariants,
  * );
@@ -84,7 +84,7 @@ import { Code, VariantExtraFiles, VariantSource } from "./types.js";
  *     variantName,
  *     code,
  *     fileName,
- *     highlightAt === 'init',
+ *     highlightAfter === 'init',
  *     fallbackUsesExtraFiles,
  *     fallbackUsesAllVariants,
  *   ), [dependencies]);