@bleedingdev/modern-js-runtime 3.9.0-ultramodern.2 → 3.9.0-ultramodern.21

package/dist/cjs/core/server/stream/createReadableStream.js

@@ -30,150 +30,247 @@ __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
     createReadableStreamFromElement: ()=>createReadableStreamFromElement
 });
-const node_namespaceObject = require("@modern-js/runtime-extensions/node");
-const runtime_utils_node_namespaceObject = require("@modern-js/runtime-utils/node");
+const context_namespaceObject = require("@modern-js/runtime/context");
+const node_namespaceObject = require("@modern-js/runtime-utils/node");
 const constants_namespaceObject = require("@modern-js/utils/universal/constants");
 const external_stream_namespaceObject = require("stream");
 const external_common_js_namespaceObject = require("../../../common.js");
 const external_constants_js_namespaceObject = require("../../constants.js");
-const index_js_namespaceObject = require("../../context/index.js");
-const monitors_index_js_namespaceObject = require("../../context/monitors/index.js");
+const index_js_namespaceObject = require("../../context/monitors/index.js");
+const wrapper_js_namespaceObject = require("../../react/wrapper.js");
 const external_helmet_js_namespaceObject = require("../helmet.js");
+const external_shared_js_namespaceObject = require("../shared.js");
 const external_deferredScript_js_namespaceObject = require("./deferredScript.js");
-const external_shared_js_namespaceObject = require("./shared.js");
+const external_shared_js_namespaceObject_1 = require("./shared.js");
 const external_template_js_namespaceObject = require("./template.js");
-const defaultExtender = {
-    modifyRootElement: (rootElement)=>rootElement,
-    getStyleTags: ()=>'',
-    processStream: (stream)=>stream
-};
 const createReadableStreamFromElement = async (request, rootElement, options)=>{
     const { renderToPipeableStream } = await import("react-dom/server");
-    const { runtimeContext, htmlTemplate, config, ssrConfig, entryName, moduleFederationCssAssets } = options;
-    let shellChunkStatus = external_shared_js_namespaceObject.ShellChunkStatus.START;
-    let renderLevel = external_constants_js_namespaceObject.RenderLevel.SERVER_RENDER;
+    const { runtimeContext, htmlTemplate, config, ssrConfig, entryName, resource } = options;
+    const hooks = (0, context_namespaceObject.getGlobalInternalRuntimeContext)().hooks;
+    const isRsc = true === (0, context_namespaceObject.getGlobalEnableRsc)();
+    const extenders = hooks.extendStreamSSR.call({
+        runtimeContext,
+        request,
+        resource,
+        config,
+        platform: 'node',
+        mode: 'stream',
+        monitors: (0, index_js_namespaceObject.getMonitors)(),
+        isRsc,
+        terminalMarker: external_common_js_namespaceObject.ESCAPED_SHELL_STREAM_END_MARK
+    }) || [];
+    const lifecycle = (0, external_shared_js_namespaceObject.createSSRRenderLifecycle)(extenders);
+    const reportError = (0, external_shared_js_namespaceObject.createSSRStreamErrorReporter)(options.onError);
     const forceStream2String = Boolean(process.env.MODERN_JS_STREAM_TO_STRING);
-    const { onReady } = (0, external_shared_js_namespaceObject.resolveStreamingMode)(request, forceStream2String);
-    const internalRuntimeContext = (0, index_js_namespaceObject.getGlobalInternalRuntimeContext)();
-    const hooks = internalRuntimeContext.hooks;
-    const extenders = hooks.extendStreamSSR.call() || [];
-    if (0 === extenders.length) extenders.push(defaultExtender);
-    extenders.forEach((extender)=>{
-        if (extender.init) extender.init({
-            rootElement,
-            forceStream2String
-        });
-    });
-    let processedRootElement = rootElement;
-    extenders.forEach((extender)=>{
-        if (extender.modifyRootElement) processedRootElement = extender.modifyRootElement(processedRootElement);
-    });
-    const chunkVec = [];
+    const { onReady } = (0, external_shared_js_namespaceObject_1.resolveStreamingMode)(request, forceStream2String);
+    let renderLevel = external_constants_js_namespaceObject.RenderLevel.SERVER_RENDER;
     let hasStartedPipe = false;
-    node_namespaceObject.beginHeadRender(runtimeContext);
-    const reportError = node_namespaceObject.createOnceErrorReporter(options.onError);
-    return new Promise((resolve)=>{
-        const { pipe: reactStreamingPipe } = renderToPipeableStream(processedRootElement, {
-            nonce: config.nonce,
-            identifierPrefix: constants_namespaceObject.SSR_HYDRATION_ID_PREFIX,
-            [onReady] () {
-                if (hasStartedPipe) return;
-                hasStartedPipe = true;
-                let styledComponentsStyleTags = '';
-                extenders.forEach((extender)=>{
-                    if (extender.getStyleTags) styledComponentsStyleTags += extender.getStyleTags();
+    let failed = false;
+    let reactStream;
+    const ownedStreams = new Set();
+    const streamCompletions = [];
+    let teardown;
+    return new Promise((resolve, reject)=>{
+        const destroyOwnedStreams = (reason)=>{
+            if (teardown) return teardown;
+            const error = void 0 === reason ? void 0 : reason instanceof Error ? reason : new Error(String(reason));
+            for (const stream of ownedStreams){
+                const destroyable = stream;
+                destroyable.destroy?.(error);
+            }
+            teardown = Promise.all(streamCompletions).then((errors)=>{
+                const cleanupError = errors.find((value)=>void 0 !== value && value !== error && 'ERR_STREAM_PREMATURE_CLOSE' !== value.code);
+                if (cleanupError) throw cleanupError;
+            });
+            return teardown;
+        };
+        const abortReact = (reason)=>reactStream?.abort(reason);
+        const fail = (error)=>{
+            if (failed) return;
+            failed = true;
+            request.signal.removeEventListener('abort', onStartupAbort);
+            if (request.signal.aborted) lifecycle.finish({
+                status: 'cancelled',
+                reason: request.signal.reason
+            });
+            else {
+                lifecycle.finish({
+                    status: 'error',
+                    error
                 });
-                options[onReady]?.();
-                (0, external_template_js_namespaceObject.getTemplates)(htmlTemplate, {
-                    request,
-                    ssrConfig,
-                    renderLevel,
-                    runtimeContext,
-                    config,
-                    entryName,
-                    moduleFederationCssAssets,
-                    styledComponentsStyleTags
-                }).then(({ shellAfter, shellBefore })=>{
-                    const pendingScripts = [];
-                    const body = new external_stream_namespaceObject.Transform({
-                        transform (chunk, _encoding, callback) {
-                            try {
-                                if (shellChunkStatus !== external_shared_js_namespaceObject.ShellChunkStatus.FINISH) {
-                                    chunkVec.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-                                    const concatedChunk = Buffer.concat(chunkVec).toString('utf-8');
-                                    const markerIndex = concatedChunk.indexOf(external_common_js_namespaceObject.ESCAPED_SHELL_STREAM_END_MARK);
-                                    if (-1 !== markerIndex) {
-                                        const beforeMark = concatedChunk.slice(0, markerIndex);
-                                        const afterMark = concatedChunk.slice(markerIndex + external_common_js_namespaceObject.ESCAPED_SHELL_STREAM_END_MARK.length);
-                                        node_namespaceObject.publishHeadRender(runtimeContext);
-                                        const completedShellBefore = (0, external_helmet_js_namespaceObject.createReplaceHelemt)((0, external_helmet_js_namespaceObject.getHelmetData)(runtimeContext))(shellBefore);
-                                        shellChunkStatus = external_shared_js_namespaceObject.ShellChunkStatus.FINISH;
-                                        this.push(`${completedShellBefore}${beforeMark}${shellAfter}`);
-                                        if (afterMark) this.push(afterMark);
-                                        if (pendingScripts.length > 0) for (const s of pendingScripts)this.push(s);
-                                    }
-                                } else this.push(chunk);
-                                callback();
-                            } catch (e) {
-                                e instanceof Error ? callback(e) : callback(new Error('Received unknown error when streaming'));
-                            }
+                reportError(error);
+            }
+            abortReact(error);
+            destroyOwnedStreams(error).then(()=>reject(error), ()=>reject(error));
+        };
+        const own = (stream)=>{
+            if (!ownedStreams.has(stream)) {
+                ownedStreams.add(stream);
+                streamCompletions.push(new Promise((done)=>{
+                    (0, external_stream_namespaceObject.finished)(stream, (error)=>done(error ?? void 0));
+                }));
+                stream.on('error', fail);
+            }
+            return stream;
+        };
+        const onStartupAbort = ()=>fail(request.signal.reason);
+        const deliver = (source)=>{
+            const stream = (0, external_shared_js_namespaceObject.observeSSRStream)(source, {
+                lifecycle,
+                signal: request.signal,
+                async onError (error) {
+                    failed = true;
+                    abortReact(error);
+                    await destroyOwnedStreams(error);
+                    reportError(error);
+                },
+                onCancel (reason) {
+                    failed = true;
+                    const cleanup = destroyOwnedStreams();
+                    abortReact(reason);
+                    return cleanup;
+                }
+            });
+            request.signal.removeEventListener('abort', onStartupAbort);
+            resolve(stream);
+        };
+        const templateOptions = ()=>({
+                request,
+                ssrConfig,
+                renderLevel,
+                runtimeContext,
+                config,
+                entryName,
+                lifecycle
+            });
+        const startOutput = async ()=>{
+            if (hasStartedPipe || failed) return;
+            hasStartedPipe = true;
+            const styledComponentsStyleTags = extenders.map((extender)=>extender.getStyleTags?.() ?? '').join('');
+            options[onReady]?.();
+            const { shellBefore, shellAfter } = await (0, external_template_js_namespaceObject.getTemplates)(htmlTemplate, {
+                ...templateOptions(),
+                styledComponentsStyleTags
+            });
+            if (failed) return;
+            const chunks = [];
+            let bufferedLength = 0;
+            const marker = Buffer.from(external_common_js_namespaceObject.ESCAPED_SHELL_STREAM_END_MARK);
+            let markerTail = Buffer.alloc(0);
+            const pendingScripts = [];
+            let shellChunkStatus = external_shared_js_namespaceObject_1.ShellChunkStatus.START;
+            const emitShell = (destination, buffered, markerIndex)=>{
+                const beforeMark = lifecycle.completedBody(buffered.subarray(0, markerIndex).toString('utf8'), 'shell');
+                const completedShellBefore = (0, external_helmet_js_namespaceObject.createReplaceHelemt)((0, external_helmet_js_namespaceObject.getHelmetData)(extenders))(shellBefore);
+                shellChunkStatus = external_shared_js_namespaceObject_1.ShellChunkStatus.FINISH;
+                chunks.length = 0;
+                bufferedLength = 0;
+                destination.push(`${completedShellBefore}${beforeMark}${shellAfter}`);
+                const afterMark = buffered.subarray(markerIndex + marker.length);
+                if (afterMark.length > 0) destination.push(afterMark);
+                for (const script of pendingScripts)destination.push(script);
+                pendingScripts.length = 0;
+            };
+            const body = new external_stream_namespaceObject.Transform({
+                transform (chunk, _encoding, callback) {
+                    try {
+                        if (shellChunkStatus === external_shared_js_namespaceObject_1.ShellChunkStatus.FINISH) this.push(chunk);
+                        else {
+                            const bytes = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk);
+                            const window = Buffer.concat([
+                                markerTail,
+                                bytes
+                            ]);
+                            const windowIndex = window.indexOf(marker);
+                            const windowStart = bufferedLength - markerTail.length;
+                            chunks.push(bytes);
+                            bufferedLength += bytes.length;
+                            if (-1 === windowIndex) markerTail = window.subarray(Math.max(0, window.length - marker.length + 1));
+                            else emitShell(this, Buffer.concat(chunks, bufferedLength), windowStart + windowIndex);
                         }
-                    });
-                    const passThrough = new external_stream_namespaceObject.PassThrough();
-                    const stream = external_stream_namespaceObject.Readable.toWeb(body);
-                    resolve(stream);
-                    let processedStream = passThrough;
-                    extenders.forEach((extender)=>{
-                        if (extender.processStream) processedStream = extender.processStream(processedStream);
-                    });
-                    node_namespaceObject.pipeNodeHeadStream({
-                        source: processedStream,
-                        destination: body,
-                        context: runtimeContext,
-                        terminalMarker: external_common_js_namespaceObject.ESCAPED_SHELL_STREAM_END_MARK,
-                        onError: reportError
-                    });
-                    reactStreamingPipe(passThrough);
+                        callback();
+                    } catch (error) {
+                        callback(error instanceof Error ? error : new Error(String(error)));
+                    }
+                },
+                flush (callback) {
                     try {
-                        const storageContext = runtime_utils_node_namespaceObject.storage.useContext?.();
-                        const activeDeferreds = storageContext?.activeDeferreds;
-                        const entries = activeDeferreds instanceof Map ? Array.from(activeDeferreds.entries()) : [];
-                        if (entries.length > 0) {
-                            const enqueueScript = (s)=>{
-                                if (shellChunkStatus === external_shared_js_namespaceObject.ShellChunkStatus.FINISH) body.write(s);
-                                else pendingScripts.push(s);
-                            };
-                            (0, external_deferredScript_js_namespaceObject.enqueueFromEntries)(entries, config.nonce, (s)=>enqueueScript(s));
+                        if (shellChunkStatus !== external_shared_js_namespaceObject_1.ShellChunkStatus.FINISH) {
+                            const buffered = Buffer.concat(chunks);
+                            emitShell(this, buffered, buffered.length);
                         }
-                    } catch (err) {
-                        const monitors = (0, monitors_index_js_namespaceObject.getMonitors)();
-                        monitors.error("cannot inject router data script", err);
+                        callback();
+                    } catch (error) {
+                        callback(error instanceof Error ? error : new Error(String(error)));
                     }
+                }
+            });
+            own(body);
+            const passThrough = new external_stream_namespaceObject.PassThrough();
+            let processedStream = own(passThrough);
+            for (const extender of (0, external_shared_js_namespaceObject.orderSSRStreamTransforms)(extenders))if (extender.processStream) processedStream = own(extender.processStream(processedStream));
+            (0, external_stream_namespaceObject.pipeline)(processedStream, body, (error)=>{
+                if (null != error && !failed) fail(error);
+            });
+            deliver(external_stream_namespaceObject.Readable.toWeb(body));
+            reactStream.pipe(passThrough);
+            try {
+                const activeDeferreds = node_namespaceObject.storage.useContext?.()?.activeDeferreds;
+                const entries = activeDeferreds instanceof Map ? Array.from(activeDeferreds.entries()) : [];
+                (0, external_deferredScript_js_namespaceObject.enqueueFromEntries)(entries, config.nonce, (script)=>{
+                    if (failed || body.destroyed || body.writableEnded) return;
+                    if (shellChunkStatus === external_shared_js_namespaceObject_1.ShellChunkStatus.FINISH) body.write(script);
+                    else pendingScripts.push(script);
                 });
-            },
-            onShellError (error) {
-                node_namespaceObject.abortHeadRender(runtimeContext);
-                renderLevel = external_constants_js_namespaceObject.RenderLevel.CLIENT_RENDER;
-                (0, external_template_js_namespaceObject.getTemplates)(htmlTemplate, {
-                    request,
-                    ssrConfig,
-                    renderLevel,
-                    runtimeContext,
-                    entryName,
-                    config,
-                    moduleFederationCssAssets
-                }).then(({ shellAfter, shellBefore })=>{
-                    const fallbackHtml = `${shellBefore}${shellAfter}`;
-                    const readableStream = (0, external_shared_js_namespaceObject.getReadableStreamFromString)(fallbackHtml);
-                    resolve(readableStream);
-                    options?.onShellError?.(error);
-                });
-            },
-            onError (error) {
-                renderLevel = external_constants_js_namespaceObject.RenderLevel.CLIENT_RENDER;
-                reportError(error);
+            } catch (error) {
+                (0, index_js_namespaceObject.getMonitors)().error("cannot inject router data script", error);
             }
+        };
+        const renderFallback = async (error)=>{
+            if (failed || hasStartedPipe) return;
+            lifecycle.finish({
+                status: 'fallback',
+                error
+            });
+            renderLevel = external_constants_js_namespaceObject.RenderLevel.CLIENT_RENDER;
+            const { shellBefore, shellAfter } = await (0, external_template_js_namespaceObject.getTemplates)(htmlTemplate, {
+                ...templateOptions(),
+                helmetData: (0, external_helmet_js_namespaceObject.getHelmetData)(extenders)
+            });
+            if (failed) return;
+            options.onShellError?.(error);
+            deliver((0, external_shared_js_namespaceObject_1.getReadableStreamFromString)(`${shellBefore}${shellAfter}`));
+        };
+        request.signal.addEventListener('abort', onStartupAbort, {
+            once: true
         });
+        try {
+            request.signal.throwIfAborted();
+            for (const extender of extenders)extender.init?.({
+                rootElement,
+                forceStream2String
+            });
+            let processedRootElement = isRsc && false === runtimeContext.isBrowser ? (0, wrapper_js_namespaceObject.wrapRuntimeComponentResolver)(rootElement, hooks) : rootElement;
+            for (const extender of extenders)processedRootElement = extender.modifyRootElement?.(processedRootElement) ?? processedRootElement;
+            lifecycle.beforeReact();
+            reactStream = renderToPipeableStream(processedRootElement, {
+                nonce: config.nonce,
+                identifierPrefix: constants_namespaceObject.SSR_HYDRATION_ID_PREFIX,
+                progressiveChunkSize: external_shared_js_namespaceObject_1.SHELL_PROGRESSIVE_CHUNK_SIZE,
+                [onReady] () {
+                    startOutput().catch(fail);
+                },
+                onShellError (error) {
+                    renderFallback(error).catch(fail);
+                },
+                onError (error) {
+                    renderLevel = external_constants_js_namespaceObject.RenderLevel.CLIENT_RENDER;
+                    if (!request.signal.aborted && !failed) reportError(error);
+                }
+            });
+        } catch (error) {
+            fail(error);
+        }
     });
 };
 exports.createReadableStreamFromElement = __webpack_exports__.createReadableStreamFromElement;