@bleedingdev/modern-js-runtime 3.9.0-ultramodern.4 → 3.9.0-ultramodern.6

package/dist/esm-node/core/server/stream/createReadableStream.worker.mjs

@@ -1,122 +1,181 @@
 import "node:module";
 import { renderSSRStream } from "@modern-js/render/ssr";
 import { storage } from "@modern-js/runtime-utils/node";
+import react from "react";
 import { ESCAPED_SHELL_STREAM_END_MARK } from "../../../common.mjs";
 import { RenderLevel } from "../../constants.mjs";
+import { getGlobalEnableRsc, getGlobalInternalRuntimeContext } from "../../context/index.mjs";
+import { wrapRuntimeComponentResolver } from "../../react/wrapper.mjs";
 import { createReplaceHelemt, getHelmetData } from "../helmet.mjs";
+import { createSSRRenderLifecycle, createSSRStreamErrorReporter, observeSSRStream, orderSSRStreamTransforms } from "../shared.mjs";
 import { enqueueFromEntries } from "./deferredScript.mjs";
-import { ShellChunkStatus, encodeForWebStream, getReadableStreamFromString, resolveStreamingMode } from "./shared.mjs";
+import { ShellChunkStatus, getReadableStreamFromString, resolveStreamingMode } from "./shared.mjs";
 import { getTemplates } from "./template.mjs";
-import * as __rspack_external__modern_js_runtime_extensions_ce611631 from "@modern-js/runtime-extensions";
 const createReadableStreamFromElement = async (request, rootElement, options)=>{
-    let shellChunkStatus = ShellChunkStatus.START;
-    const chunkVec = [];
-    const { htmlTemplate, runtimeContext, config, ssrConfig, entryName, moduleFederationCssAssets, rscManifest, rscRoot } = options;
-    __rspack_external__modern_js_runtime_extensions_ce611631.beginHeadRender(runtimeContext);
-    const { shellBefore, shellAfter } = await getTemplates(htmlTemplate, {
-        renderLevel: RenderLevel.SERVER_RENDER,
+    const { htmlTemplate, runtimeContext, config, ssrConfig, entryName, resource, rscManifest, rscRoot } = options;
+    const hooks = getGlobalInternalRuntimeContext().hooks;
+    const isRsc = true === getGlobalEnableRsc();
+    const extenders = hooks.extendStreamSSR.call({
+        runtimeContext,
+        request,
+        resource,
+        config,
+        platform: 'web',
+        mode: 'stream',
+        isRsc,
+        terminalMarker: ESCAPED_SHELL_STREAM_END_MARK
+    }) || [];
+    const lifecycle = createSSRRenderLifecycle(extenders);
+    const reportOnce = createSSRStreamErrorReporter(options.onError);
+    const reportError = (error)=>{
+        if (!request.signal.aborted && lifecycle.terminal?.status !== 'cancelled') reportOnce(error);
+    };
+    const forceStream2String = Boolean("u" > typeof process && process.env?.MODERN_JS_STREAM_TO_STRING);
+    let pendingSource;
+    let bodyOpen = true;
+    let bodyPipe;
+    const templateOptions = {
         runtimeContext,
         ssrConfig,
         request,
         config,
         entryName,
-        moduleFederationCssAssets
-    });
+        lifecycle
+    };
     try {
+        request.signal.throwIfAborted();
+        for (const extender of extenders)extender.init?.({
+            rootElement,
+            forceStream2String
+        });
         const readableOriginal = await renderSSRStream(rootElement, {
             request,
+            signal: request.signal,
             nonce: config.nonce,
             rscManifest,
             rscRoot: rscRoot,
             routes: runtimeContext.routes,
-            onError (error) {
-                options.onError?.(error);
+            onError: reportError,
+            wrapHtmlRoot (root) {
+                let element = react.isValidElement(root) ? root : react.createElement(react.Fragment, null, root);
+                if (isRsc && false === runtimeContext.isBrowser) element = wrapRuntimeComponentResolver(element, hooks);
+                for (const extender of extenders)element = extender.modifyRootElement?.(element) ?? element;
+                lifecycle.beforeReact();
+                return element;
             }
         });
+        pendingSource = readableOriginal;
         options.onShellReady?.();
-        readableOriginal.allReady.then(()=>{
-            options?.onAllReady?.();
-        });
-        const forceStreamToString = Boolean("u" > typeof process && process.env?.MODERN_JS_STREAM_TO_STRING);
-        const { waitForAllReady } = resolveStreamingMode(request, forceStreamToString);
+        readableOriginal.allReady.then(()=>options.onAllReady?.()).catch(reportError);
+        const { waitForAllReady } = resolveStreamingMode(request, forceStream2String);
         if (waitForAllReady) await readableOriginal.allReady;
-        const reader = __rspack_external__modern_js_runtime_extensions_ce611631.createConservingWebShellStream(readableOriginal, runtimeContext, ESCAPED_SHELL_STREAM_END_MARK, options.onError).getReader();
-        const stream = new ReadableStream({
+        request.signal.throwIfAborted();
+        const styledComponentsStyleTags = extenders.map((extender)=>extender.getStyleTags?.() ?? '').join('');
+        const { shellBefore, shellAfter } = await getTemplates(htmlTemplate, {
+            ...templateOptions,
+            renderLevel: RenderLevel.SERVER_RENDER,
+            styledComponentsStyleTags
+        });
+        for (const extender of orderSSRStreamTransforms(extenders))if (void 0 !== extender.processReadableStream) pendingSource = extender.processReadableStream(pendingSource);
+        const decoder = new TextDecoder();
+        const encoder = new TextEncoder();
+        const pendingScripts = [];
+        let buffered = '';
+        let shellChunkStatus = ShellChunkStatus.START;
+        let bodyController;
+        const emit = (chunk, controller)=>{
+            if (shellChunkStatus === ShellChunkStatus.FINISH) {
+                if (chunk.length > 0) controller.enqueue(encoder.encode(chunk));
+                return;
+            }
+            buffered += chunk;
+            const markerIndex = buffered.indexOf(ESCAPED_SHELL_STREAM_END_MARK);
+            if (-1 === markerIndex) return;
+            const beforeMark = lifecycle.completedBody(buffered.slice(0, markerIndex), 'shell');
+            const afterMark = buffered.slice(markerIndex + ESCAPED_SHELL_STREAM_END_MARK.length);
+            const completedShellBefore = createReplaceHelemt(getHelmetData(extenders))(shellBefore);
+            shellChunkStatus = ShellChunkStatus.FINISH;
+            buffered = '';
+            controller.enqueue(encoder.encode(`${completedShellBefore}${beforeMark}${shellAfter}`));
+            if (afterMark.length > 0) controller.enqueue(encoder.encode(afterMark));
+            for (const script of pendingScripts)controller.enqueue(encoder.encode(script));
+            pendingScripts.length = 0;
+        };
+        const body = new TransformStream({
             start (controller) {
-                const pendingScripts = [];
-                let isClosed = false;
-                const safeEnqueue = (chunk)=>{
-                    if (isClosed) return;
-                    try {
-                        controller.enqueue(chunk);
-                    } catch  {
-                        isClosed = true;
-                    }
-                };
-                const closeController = ()=>{
-                    if (!isClosed) {
-                        isClosed = true;
-                        try {
-                            controller.close();
-                        } catch  {}
-                    }
-                };
-                const flushPendingScripts = ()=>{
-                    for (const s of pendingScripts)safeEnqueue(encodeForWebStream(s));
-                    pendingScripts.length = 0;
-                };
-                const enqueueScript = (script)=>{
-                    if (shellChunkStatus === ShellChunkStatus.FINISH) safeEnqueue(encodeForWebStream(script));
-                    else pendingScripts.push(script);
-                };
-                const storageContext = storage.useContext?.();
-                const activeDeferreds = storageContext?.activeDeferreds;
-                const entries = activeDeferreds instanceof Map ? Array.from(activeDeferreds.entries()) : [];
-                if (entries.length > 0) enqueueFromEntries(entries, config.nonce, enqueueScript);
-                async function push() {
-                    try {
-                        const { done, value } = await reader.read();
-                        if (done) return void closeController();
-                        if (isClosed) return;
-                        if (shellChunkStatus !== ShellChunkStatus.FINISH) {
-                            chunkVec.push(new TextDecoder().decode(value));
-                            const concatedChunk = chunkVec.join('');
-                            const markerIndex = concatedChunk.indexOf(ESCAPED_SHELL_STREAM_END_MARK);
-                            if (-1 !== markerIndex) {
-                                const beforeMark = concatedChunk.slice(0, markerIndex);
-                                const afterMark = concatedChunk.slice(markerIndex + ESCAPED_SHELL_STREAM_END_MARK.length);
-                                __rspack_external__modern_js_runtime_extensions_ce611631.publishHeadRender(runtimeContext);
-                                const completedShellBefore = createReplaceHelemt(getHelmetData(runtimeContext))(shellBefore);
-                                shellChunkStatus = ShellChunkStatus.FINISH;
-                                safeEnqueue(encodeForWebStream(`${completedShellBefore}${beforeMark}${shellAfter}`));
-                                if (afterMark.length > 0) safeEnqueue(encodeForWebStream(afterMark));
-                                flushPendingScripts();
-                            }
-                        } else safeEnqueue(value);
-                        if (!isClosed) push();
-                    } catch (error) {
-                        __rspack_external__modern_js_runtime_extensions_ce611631.abortHeadRender(runtimeContext);
-                        if (!isClosed) {
-                            isClosed = true;
-                            try {
-                                controller.error(error);
-                            } catch  {}
-                        }
-                    }
-                }
-                push();
+                bodyController = controller;
+            },
+            transform (chunk, controller) {
+                emit(decoder.decode(chunk, {
+                    stream: true
+                }), controller);
             },
-            cancel (reason) {
-                __rspack_external__modern_js_runtime_extensions_ce611631.abortHeadRender(runtimeContext);
-                reader.cancel(reason).catch(()=>{});
+            flush (controller) {
+                bodyOpen = false;
+                emit(decoder.decode(), controller);
+                if (shellChunkStatus !== ShellChunkStatus.FINISH) throw new Error('React SSR stream ended before the shell marker');
             }
         });
-        return stream;
-    } catch (e) {
-        __rspack_external__modern_js_runtime_extensions_ce611631.abortHeadRender(runtimeContext);
-        const fallbackHtml = `${shellBefore}${shellAfter}`;
-        const stream = getReadableStreamFromString(fallbackHtml);
-        return stream;
+        bodyPipe = pendingSource.pipeTo(body.writable);
+        bodyPipe.catch(()=>{});
+        pendingSource = body.readable;
+        const activeDeferreds = storage.useContext?.()?.activeDeferreds;
+        const entries = activeDeferreds instanceof Map ? Array.from(activeDeferreds.entries()) : [];
+        enqueueFromEntries(entries, config.nonce, (script)=>{
+            if (!bodyOpen) return;
+            if (shellChunkStatus === ShellChunkStatus.FINISH) try {
+                bodyController.enqueue(encoder.encode(script));
+            } catch  {
+                bodyOpen = false;
+            }
+            else pendingScripts.push(script);
+        });
+        return observeSSRStream(pendingSource, {
+            lifecycle,
+            async onError (error) {
+                bodyOpen = false;
+                await bodyPipe?.catch(()=>{});
+                reportError(error);
+            },
+            signal: request.signal,
+            async onCancel (reason) {
+                bodyOpen = false;
+                await bodyPipe?.catch((error)=>{
+                    if (error !== reason) throw error;
+                });
+            },
+            onComplete: ()=>bodyPipe
+        });
+    } catch (error) {
+        bodyOpen = false;
+        if (void 0 !== pendingSource) try {
+            await pendingSource.cancel(error);
+            await bodyPipe?.catch(()=>{});
+        } catch (cancelError) {
+            if (!request.signal.aborted) reportError(cancelError);
+        }
+        if (request.signal.aborted) {
+            lifecycle.finish({
+                status: 'cancelled',
+                reason: request.signal.reason
+            });
+            throw request.signal.reason;
+        }
+        lifecycle.finish({
+            status: 'fallback',
+            error
+        });
+        reportError(error);
+        const { shellBefore, shellAfter } = await getTemplates(htmlTemplate, {
+            ...templateOptions,
+            renderLevel: RenderLevel.CLIENT_RENDER,
+            helmetData: getHelmetData(extenders)
+        });
+        return observeSSRStream(getReadableStreamFromString(`${shellBefore}${shellAfter}`), {
+            lifecycle,
+            onError: reportError,
+            signal: request.signal
+        });
     }
 };
 export { createReadableStreamFromElement };

package/dist/esm-node/core/server/stream/deferredScript.mjs

@@ -21,13 +21,13 @@ function isDeferredDataLike(v) {
     return hasPending && hasData;
 }
 function isPromiseLike(value) {
-    return !!value && 'function' == typeof value.then;
+    return Boolean(value) && 'function' == typeof value.then;
 }
 function toErrorInfo(error) {
     if (shouldRedactServerError()) return {
         message: 'Unexpected Server Error'
     };
-    if (error && 'object' == typeof error) {
+    if (null !== error && 'object' == typeof error) {
         const maybeMsg = error.message;
         const maybeStack = error.stack;
         return {
@@ -42,7 +42,7 @@ function toErrorInfo(error) {
 function buildDeferredDataScript(nonce, args) {
     const payload = JSON.stringify(args);
     const escaped = escapeAttr(payload);
-    const nonceAttr = nonce ? ` nonce="${nonce}"` : '';
+    const nonceAttr = void 0 !== nonce && '' !== nonce ? ` nonce="${nonce}"` : '';
     return `<script async${nonceAttr} data-fn-name="r" data-script-src="modern-run-router-data-fn" data-fn-args='${escaped}' suppressHydrationWarning>${runRouterDataFnStr}</script>`;
 }
 function enqueueFromEntries(entries, nonce, emit) {

package/dist/esm-node/core/server/stream/shared.mjs

@@ -64,7 +64,7 @@ function createRenderStreaming(createReadableStreamPromise) {
         const end = time();
         const { runtimeContext, config, resource } = options;
         const { onError, onTiming } = options;
-        const { htmlTemplate, entryName, moduleFederationCssAssets } = resource;
+        const { htmlTemplate, entryName } = resource;
         const ssrConfig = getSSRConfigByEntry(entryName, config.ssr, config.ssrByEntries);
         const StreamServerRootWrapper = ({ children })=>/*#__PURE__*/ jsxs(Fragment, {
                 children: [
@@ -84,7 +84,7 @@ function createRenderStreaming(createReadableStreamPromise) {
             runtimeContext,
             ssrConfig,
             entryName,
-            moduleFederationCssAssets,
+            resource,
             rscClientManifest: options.rscClientManifest,
             rscSSRManifest: options.rscSSRManifest,
             rscServerManifest: options.rscServerManifest,

package/dist/esm-node/core/server/string/index.mjs

@@ -3,17 +3,15 @@ import { time } from "@modern-js/runtime-utils/time";
 import { SSR_HYDRATION_ID_PREFIX } from "@modern-js/utils/universal/constants";
 import server from "react-dom/server";
 import { RenderLevel } from "../../constants.mjs";
-import { getGlobalInternalRuntimeContext } from "../../context/index.mjs";
-import { wrapRuntimeContextProvider } from "../../react/wrapper.mjs";
-import { CHUNK_CSS_PLACEHOLDER, HTML_PLACEHOLDER, SSR_DATA_PLACEHOLDER } from "../constants.mjs";
+import { getGlobalEnableRsc, getGlobalInternalRuntimeContext } from "../../context/index.mjs";
+import { wrapRuntimeComponentResolver, wrapRuntimeContextProvider } from "../../react/wrapper.mjs";
+import { CHUNK_CSS_PLACEHOLDER, CHUNK_JS_PLACEHOLDER, HTML_PLACEHOLDER, SSR_DATA_PLACEHOLDER } from "../constants.mjs";
 import { createReplaceHelemt, getHelmetData } from "../helmet.mjs";
-import { replaceChunkJsPlaceholder } from "../scriptOrder.mjs";
-import { buildHtml } from "../shared.mjs";
+import { buildHtml, createSSRRenderLifecycle, replaceSSRTemplateChunk } from "../shared.mjs";
 import { SSRErrors, SSRTimings } from "../tracer.mjs";
 import { getSSRConfigByEntry, safeReplace } from "../utils.mjs";
 import { LoadableCollector } from "./loadable.mjs";
 import { SSRDataCollector } from "./ssrData.mjs";
-import * as __rspack_external__modern_js_runtime_extensions_ce611631 from "@modern-js/runtime-extensions";
 const renderString = async (request, serverRoot, options)=>{
     const { resource, runtimeContext, config, onError, onTiming } = options;
     const tracer = {
@@ -21,7 +19,7 @@ const renderString = async (request, serverRoot, options)=>{
         onTiming
     };
     const routerContext = runtimeContext.routerContext;
-    const { htmlTemplate, entryName, loadableStats, routeManifest, moduleFederationCssAssets } = resource;
+    const { htmlTemplate, entryName, loadableStats, routeManifest } = resource;
     const ssrConfig = getSSRConfigByEntry(entryName, config.ssr, config.ssrByEntries);
     const chunkSet = {
         renderLevel: RenderLevel.CLIENT_RENDER,
@@ -29,83 +27,121 @@ const renderString = async (request, serverRoot, options)=>{
         jsChunk: '',
         cssChunk: ''
     };
-    const collectors = [
-        new LoadableCollector({
-            stats: loadableStats,
-            nonce: config.nonce,
-            routeManifest,
-            runtimeContext,
-            template: htmlTemplate,
-            entryName,
-            moduleFederationCssAssets,
-            chunkSet,
-            config
-        }),
-        new SSRDataCollector({
-            runtimeContext,
-            request,
-            ssrConfig,
-            ssrContext: runtimeContext.ssrContext,
-            chunkSet,
-            routerContext,
-            nonce: config.nonce,
-            useJsonScript: config.useJsonScript
-        })
-    ];
+    const collectors = [];
+    const lifecycle = createSSRRenderLifecycle(collectors);
+    const loadableCollector = new LoadableCollector({
+        stats: loadableStats,
+        nonce: config.nonce,
+        routeManifest,
+        template: htmlTemplate,
+        entryName,
+        chunkSet,
+        config,
+        lifecycle
+    });
+    collectors.push(loadableCollector, new SSRDataCollector({
+        runtimeContext,
+        request,
+        ssrConfig,
+        ssrContext: runtimeContext.ssrContext,
+        chunkSet,
+        routerContext,
+        nonce: config.nonce,
+        useJsonScript: config.useJsonScript,
+        lifecycle
+    }));
     const internalRuntimeContext = getGlobalInternalRuntimeContext();
     const hooks = internalRuntimeContext.hooks;
     const extraCollectors = hooks.extendStringSSRCollectors.call({
-        chunkSet
+        chunkSet,
+        render: {
+            runtimeContext,
+            request,
+            resource,
+            config,
+            platform: 'node',
+            mode: 'string',
+            isRsc: true === getGlobalEnableRsc()
+        }
     });
     for (const c of extraCollectors)if (c) collectors.unshift(c);
-    const rootElement = wrapRuntimeContextProvider(serverRoot, Object.assign(runtimeContext, {
+    let rootElement = wrapRuntimeContextProvider(serverRoot, Object.assign(runtimeContext, {
         ssr: true
     }));
-    const html = await generateHtml(rootElement, htmlTemplate, chunkSet, collectors, runtimeContext.ssrContext?.htmlModifiers || [], runtimeContext, entryName, tracer);
+    if (true === getGlobalEnableRsc() && false === runtimeContext.isBrowser) rootElement = wrapRuntimeComponentResolver(rootElement, hooks);
+    const html = await generateHtml(rootElement, htmlTemplate, chunkSet, collectors, runtimeContext.ssrContext?.htmlModifiers || [], tracer, request.signal, lifecycle, loadableCollector.stylesheetInfo);
     return html;
 };
-async function generateHtml(App, htmlTemplate, chunkSet, collectors, htmlModifiers, runtimeContext, entryName, { onError, onTiming }) {
+async function generateHtml(App, htmlTemplate, chunkSet, collectors, htmlModifiers, { onError, onTiming }, signal, lifecycle, stylesheetInfo) {
     let html = '';
     let helmetData;
-    const finalApp = collectors.reduce((pre, creator)=>creator.collect?.(pre) || pre, App);
-    __rspack_external__modern_js_runtime_extensions_ce611631.beginHeadRender(runtimeContext);
     try {
-        const end = time();
-        html = server.renderToString(finalApp, {
-            identifierPrefix: SSR_HYDRATION_ID_PREFIX
+        signal.throwIfAborted();
+        try {
+            const finalApp = collectors.reduce((pre, creator)=>creator.collect?.(pre) || pre, App);
+            lifecycle.beforeReact();
+            const end = time();
+            html = server.renderToString(finalApp, {
+                identifierPrefix: SSR_HYDRATION_ID_PREFIX
+            });
+            html = lifecycle.completedBody(html, 'complete');
+            chunkSet.renderLevel = RenderLevel.SERVER_RENDER;
+            helmetData = getHelmetData(collectors);
+            const cost = end();
+            onTiming(SSRTimings.RENDER_HTML, cost);
+        } catch (error) {
+            lifecycle.finish({
+                status: 'fallback',
+                error
+            });
+            html = '';
+            chunkSet.renderLevel = RenderLevel.CLIENT_RENDER;
+            onError(error, SSRErrors.RENDER_HTML);
+        }
+        await Promise.all(collectors.map((component)=>component.effect()));
+        signal.throwIfAborted();
+        const { ssrScripts, cssChunk, jsChunk } = chunkSet;
+        const finalHtml = await buildHtml(htmlTemplate, [
+            createReplaceHtml(html),
+            (template)=>replaceSSRTemplateChunk({
+                    name: "scripts",
+                    template,
+                    placeholder: CHUNK_JS_PLACEHOLDER,
+                    content: jsChunk
+                }, lifecycle),
+            (template)=>replaceSSRTemplateChunk({
+                    name: 'styles',
+                    template,
+                    placeholder: CHUNK_CSS_PLACEHOLDER,
+                    content: cssChunk,
+                    ...stylesheetInfo
+                }, lifecycle),
+            (template)=>replaceSSRTemplateChunk({
+                    name: 'data',
+                    template,
+                    placeholder: SSR_DATA_PLACEHOLDER,
+                    content: ssrScripts
+                }, lifecycle),
+            createReplaceHelemt(helmetData),
+            ...htmlModifiers
+        ]);
+        signal.throwIfAborted();
+        lifecycle.finish({
+            status: 'complete'
+        });
+        return finalHtml;
+    } catch (error) {
+        lifecycle.finish(signal.aborted ? {
+            status: 'cancelled',
+            reason: signal.reason
+        } : {
+            status: 'error',
+            error
         });
-        html = __rspack_external__modern_js_runtime_extensions_ce611631.completeHeadRender(runtimeContext, html);
-        chunkSet.renderLevel = RenderLevel.SERVER_RENDER;
-        helmetData = getHelmetData(runtimeContext);
-        const cost = end();
-        onTiming(SSRTimings.RENDER_HTML, cost);
-    } catch (e) {
-        __rspack_external__modern_js_runtime_extensions_ce611631.abortHeadRender(runtimeContext);
-        chunkSet.renderLevel = RenderLevel.CLIENT_RENDER;
-        onError(e, SSRErrors.RENDER_HTML);
+        throw error;
     }
-    await Promise.all(collectors.map((component)=>component.effect()));
-    const { ssrScripts, cssChunk, jsChunk } = chunkSet;
-    const finalHtml = await buildHtml(htmlTemplate, [
-        createReplaceHtml(html),
-        createReplaceChunkJs(jsChunk, entryName),
-        createReplaceChunkCss(cssChunk),
-        createReplaceSSRDataScript(ssrScripts, entryName),
-        createReplaceHelemt(helmetData),
-        ...htmlModifiers
-    ]);
-    return finalHtml;
 }
 function createReplaceHtml(html) {
     return (template)=>safeReplace(template, HTML_PLACEHOLDER, html);
 }
-function createReplaceSSRDataScript(data, entryName) {
-    return (template)=>replaceChunkJsPlaceholder(template, data, entryName, SSR_DATA_PLACEHOLDER);
-}
-function createReplaceChunkJs(js, entryName) {
-    return (template)=>replaceChunkJsPlaceholder(template, js, entryName);
-}
-function createReplaceChunkCss(css) {
-    return (template)=>safeReplace(template, CHUNK_CSS_PLACEHOLDER, css);
-}
 export { renderString };

package/dist/esm-node/core/server/string/loadable.mjs

@@ -1,7 +1,5 @@
 import "node:module";
 import { ChunkExtractor } from "@loadable/server";
-import { createFederatedCssLinks } from "../federatedCss.mjs";
-import { getMatchedRouteChunks, orderHydrationScriptChunks } from "../scriptOrder.mjs";
 import { attributesToString, checkIsNode, hasStylesheetLink } from "../utils.mjs";
 import { fileURLToPath as __rspack_fileURLToPath } from "node:url";
 import { dirname as __rspack_dirname } from "node:path";
@@ -11,7 +9,7 @@ const extname = (uri)=>{
     return `.${uri?.split('.').pop()}` || '';
 };
 const generateChunks = (chunks, ext)=>chunks.filter((chunk)=>Boolean(chunk.url)).filter((chunk)=>extname(chunk.url).slice(1) === ext);
-const routeAssetToChunk = (asset)=>({
+const assetToChunk = (asset)=>({
         chunk: asset,
         filename: asset.replace(/^\//, ''),
         linkType: 'preload',
@@ -57,16 +55,32 @@ class LoadableCollector {
             ]));
         } catch (e) {}
         const collectedChunks = extractor ? extractor.getChunkAssets(extractor.chunks) : [];
-        const matchedRouteChunks = getMatchedRouteChunks(options.runtimeContext, options.routeManifest, routeAssetToChunk);
-        const orderedScriptChunks = orderHydrationScriptChunks({
-            asyncEntryChunks: asyncChunks,
-            collectedChunks,
-            matchedRouteChunks,
-            entryName
-        });
-        const chunks = [].concat(asyncChunks).concat(collectedChunks).concat(matchedRouteChunks);
-        const scriptChunks = generateChunks(orderedScriptChunks, 'js');
-        const styleChunks = generateChunks(chunks, 'css');
+        const groups = [
+            {
+                name: 'async-entry',
+                assets: asyncChunks
+            },
+            {
+                name: 'loadable',
+                assets: collectedChunks
+            }
+        ];
+        const nativeChunks = [
+            ...asyncChunks,
+            ...collectedChunks
+        ];
+        const prepare = (kind)=>options.lifecycle?.transformAssets(groups, {
+                kind,
+                source: 'loadable',
+                template: options.template,
+                createAsset: assetToChunk
+            }) ?? nativeChunks;
+        const scriptChunks = generateChunks([
+            ...prepare("script")
+        ], 'js');
+        const styleChunks = generateChunks([
+            ...prepare('style')
+        ], 'css');
         if (extractor) {
             this.emitLoadableScripts(extractor);
             await this.emitScriptAssets(scriptChunks);
@@ -105,7 +119,7 @@ class LoadableCollector {
         chunkSet.jsChunk += scripts.filter((script)=>Boolean(script)).join('');
     }
     async emitStyleAssets(chunks) {
-        const { template, chunkSet, config, moduleFederationCssAssets } = this.options;
+        const { template, chunkSet, config } = this.options;
         const { inlineStyles } = config;
         const atrributes = attributesToString(this.generateAttributes());
         const emittedAssetUrls = new Set();
@@ -120,11 +134,8 @@ class LoadableCollector {
             return link;
         }));
         chunkSet.cssChunk += css.filter((css)=>Boolean(css)).join('');
-        chunkSet.cssChunk += createFederatedCssLinks(moduleFederationCssAssets, {
-            template,
-            attributes: this.generateAttributes(),
-            existingAssets: emittedChunks.map((chunk)=>chunk.url)
-        });
+        this.stylesheetInfo.emittedAssets = emittedChunks.map((chunk)=>chunk.url);
+        this.stylesheetInfo.attributes = this.generateAttributes();
     }
     generateAttributes(extraAtr = {}) {
         const { config } = this.options;
@@ -137,6 +148,10 @@ class LoadableCollector {
         };
     }
     constructor(options){
+        this.stylesheetInfo = {
+            emittedAssets: [],
+            attributes: {}
+        };
         this.options = options;
     }
 }

package/dist/esm-node/core/server/string/ssrData.mjs

@@ -1,6 +1,5 @@
 import "node:module";
 import { serializeJson } from "@modern-js/runtime-utils/node";
-import { getRouterHydrationScripts, getRouterServerSnapshot } from "../../../router/runtime/lifecycle.mjs";
 import { ROUTER_DATA_JSON_ID, SSR_DATA_JSON_ID } from "../../constants.mjs";
 import { attributesToString, serializeErrors } from "../utils.mjs";
 function _check_private_redeclaration(obj, privateCollection) {
@@ -89,20 +88,15 @@ function getSSRData() {
     };
 }
 function getRouterData() {
-    const { routerContext, runtimeContext } = _class_private_field_get(this, _options);
-    const snapshotRouterData = getRouterServerSnapshot(runtimeContext)?.routerData;
-    if (snapshotRouterData) return {
-        loaderData: snapshotRouterData.loaderData,
-        errors: serializeErrors(snapshotRouterData.errors || null)
-    };
-    return routerContext ? {
-        loaderData: routerContext.loaderData,
-        errors: serializeErrors(routerContext.errors)
+    const { routerContext, lifecycle } = _class_private_field_get(this, _options);
+    const data = lifecycle?.getRouterData() ?? routerContext;
+    return data ? {
+        loaderData: data.loaderData,
+        errors: serializeErrors(data.errors || null)
     } : void 0;
 }
 function getSSRDataScripts(ssrData, routerData) {
     const { nonce, useJsonScript = false } = _class_private_field_get(this, _options);
-    const hydrationScripts = getRouterHydrationScripts(_class_private_field_get(this, _options).runtimeContext);
     const serializeSSRData = serializeJson(ssrData);
     const attrsStr = attributesToString({
         nonce
@@ -112,7 +106,6 @@ function getSSRDataScripts(ssrData, routerData) {
         const serializedRouterData = serializeJson(routerData);
         ssrDataScripts += useJsonScript ? `\n<script type="application/json" id="${ROUTER_DATA_JSON_ID}">${serializedRouterData}</script>` : `\n<script${attrsStr}>window._ROUTER_DATA = ${serializedRouterData}</script>`;
     }
-    if (hydrationScripts.length) ssrDataScripts += `\n${hydrationScripts.join('\n')}`;
     return ssrDataScripts;
 }
 export { SSRDataCollector };