@bleedingdev/modern-js-runtime 3.9.0-ultramodern.1 → 3.9.0-ultramodern.10

package/dist/esm/core/context/runtime.mjs

@@ -1,6 +1,16 @@
-import { createContext } from "react";
+import { createContext, useContext } from "react";
 const ROUTE_MANIFEST = '_MODERNJS_ROUTE_MANIFEST';
-const InternalRuntimeContext = createContext({});
+function getRuntimeReactContext(kind, create) {
+    const key = Symbol.for(`@modern-js/runtime:react-context:v1:${kind}`);
+    const contexts = globalThis;
+    contexts[key] ??= create();
+    return contexts[key];
+}
+const InternalRuntimeContext = getRuntimeReactContext('internal', ()=>createContext({}));
+const RuntimeContext = getRuntimeReactContext('public', ()=>createContext({}));
+const RuntimeComponentResolverContext = getRuntimeReactContext('component-resolver', ()=>createContext(void 0));
+const ReactRuntimeContext = RuntimeContext;
+const useRuntimeContext = ()=>useContext(RuntimeContext);
 const getInitialContext = (isBrowser = true, routeManifest)=>{
     const requestContext = {
         request: {
@@ -24,5 +34,4 @@ const getInitialContext = (isBrowser = true, routeManifest)=>{
         context: requestContext
     };
 };
-export { ReactRuntimeContext, RuntimeContext, useRuntimeContext } from "./public.mjs";
-export { InternalRuntimeContext, getInitialContext };
+export { InternalRuntimeContext, ReactRuntimeContext, RuntimeComponentResolverContext, RuntimeContext, getInitialContext, useRuntimeContext };

package/dist/esm/core/react/wrapper.mjs

@@ -1,45 +1,14 @@
 import { jsx } from "react/jsx-runtime";
-import { HelmetProvider } from "react-helmet-async";
-import { InternalRuntimeContext, RuntimeContext, getGlobalEnableRsc, getInitialContext } from "../context/index.mjs";
-import { stripRuntimeContextExtensions } from "../context/extensions.mjs";
-import { ensureHelmetContext } from "../context/helmetContext.mjs";
-function createRscSafeRequestContext(ssrContext) {
-    const requestContext = getInitialContext().requestContext;
-    Object.defineProperty(requestContext.response, 'setHeader', {
-        enumerable: false
+import { InternalRuntimeContext, RuntimeContext, getGlobalEnableRsc, getGlobalInternalRuntimeContext } from "../context/index.mjs";
+import { RuntimeComponentResolverContext } from "../context/runtime.mjs";
+function wrapRuntimeComponentResolver(root, hooks = getGlobalInternalRuntimeContext()?.hooks) {
+    return /*#__PURE__*/ jsx(RuntimeComponentResolverContext.Provider, {
+        value: hooks?.resolveComponent?.call,
+        children: root
     });
-    Object.defineProperty(requestContext.response, 'status', {
-        enumerable: false
-    });
-    if (void 0 === ssrContext) return requestContext;
-    const { request, response } = ssrContext;
-    return {
-        request: {
-            url: request.url,
-            userAgent: request.userAgent,
-            cookie: request.cookie,
-            pathname: request.pathname,
-            query: {
-                ...request.query
-            },
-            params: {
-                ...request.params
-            },
-            headers: {
-                ...request.headers
-            },
-            host: request.host,
-            referer: request.referer
-        },
-        response: Object.assign(requestContext.response, {
-            locals: response.locals ?? {}
-        })
-    };
 }
 function wrapRuntimeContextProvider(App, contextValue) {
     const { isBrowser, initialData, routes, context, routeManifest, routerContext, unstable_getBlockNavState, ssrContext, _internalContext, _internalRouterBaseName, ...rest } = contextValue;
-    const internalContextValue = contextValue;
-    const helmetContext = ensureHelmetContext(internalContextValue);
     const runtimeContextValue = {
         isBrowser,
         initialData,
@@ -47,37 +16,22 @@ function wrapRuntimeContextProvider(App, contextValue) {
         context,
         ...rest
     };
-    stripRuntimeContextExtensions(runtimeContextValue);
-    if (true === getGlobalEnableRsc() && false === isBrowser) {
-        const rscSafeRequestContext = createRscSafeRequestContext(ssrContext);
-        const rscInternalContextValue = {
-            ...internalContextValue,
-            context: rscSafeRequestContext,
-            requestContext: rscSafeRequestContext
-        };
-        delete rscInternalContextValue.ssrContext;
-        const rscRuntimeContextValue = {
-            ...runtimeContextValue,
-            context: rscSafeRequestContext,
-            requestContext: rscSafeRequestContext
-        };
-        return /*#__PURE__*/ jsx(InternalRuntimeContext.Provider, {
-            value: rscInternalContextValue,
-            children: /*#__PURE__*/ jsx(RuntimeContext.Provider, {
-                value: rscRuntimeContextValue,
-                children: App
-            })
-        });
-    }
-    return /*#__PURE__*/ jsx(InternalRuntimeContext.Provider, {
-        value: internalContextValue,
+    const isRsc = true === getGlobalEnableRsc();
+    const projection = {
+        internalContext: contextValue,
+        publicContext: runtimeContextValue
+    };
+    const values = getGlobalInternalRuntimeContext()?.hooks.transformRuntimeContext?.call(projection, {
+        context: contextValue,
+        isRsc
+    }) ?? projection;
+    const root = /*#__PURE__*/ jsx(InternalRuntimeContext.Provider, {
+        value: values.internalContext,
         children: /*#__PURE__*/ jsx(RuntimeContext.Provider, {
-            value: runtimeContextValue,
-            children: /*#__PURE__*/ jsx(HelmetProvider, {
-                context: helmetContext,
-                children: App
-            })
+            value: values.publicContext,
+            children: App
         })
     });
+    return isRsc && false === isBrowser ? root : wrapRuntimeComponentResolver(root);
 }
-export { wrapRuntimeContextProvider };
+export { wrapRuntimeComponentResolver, wrapRuntimeContextProvider };

package/dist/esm/core/server/helmet.mjs

@@ -1,4 +1,3 @@
-import { getHelmetContext } from "../context/helmetContext.mjs";
 import { safeReplace } from "./utils.mjs";
 const EOL = '\n';
 const RE_HTML_ATTR = /<html[^>]*>/;
@@ -6,8 +5,10 @@ const RE_BODY_ATTR = /<body[^>]*>/;
 const RE_LAST_IN_HEAD = /<\/head>/;
 const RE_TITLE = /<title[^>]*>([\s\S\n\r]*?)<\/title>/;
 const TEST_TITLE_CONTENT = /(?<=<title[^>]*>)([\s\S\n\r]*?)([.|\S])([\s\S\n\r]*?)(?=<\/title>)/;
-function getHelmetData(runtimeContext) {
-    return getHelmetContext(runtimeContext)?.helmet ?? void 0;
+function getHelmetData(providers) {
+    let result;
+    for (const provider of providers)result = provider.getHeadData?.() ?? result;
+    return result;
 }
 function createReplaceHelemt(helmetData) {
     return helmetData ? (template)=>helmetReplace(template, helmetData) : (tempalte)=>tempalte;
@@ -19,7 +20,7 @@ function helmetReplace(content, helmetData) {
     const htmlAttributes = helmetData.htmlAttributes.toString();
     if (htmlAttributes) result = safeReplace(result, RE_HTML_ATTR, `<html ${htmlAttributes}>`);
     const base = helmetData.base.toString();
-    const priority = helmetData.priority.toString();
+    const priority = helmetData.priority?.toString() ?? '';
     const link = helmetData.link.toString();
     const meta = helmetData.meta.toString();
     const noscript = helmetData.noscript.toString();

package/dist/esm/core/server/requestResponse.mjs

@@ -1,11 +1,11 @@
-import { getRouterServerSnapshot } from "../../router/runtime/lifecycle.mjs";
+import { getRouterServerSnapshot } from "@modern-js/runtime-extensions/router-state";
 import { handleRSCRedirect } from "../../router/runtime/redirect.mjs";
 import { SSRErrors } from "./tracer.mjs";
 const isRedirectStatus = (status)=>301 === status || 302 === status || 303 === status || 307 === status || 308 === status;
 const isNullBodyStatus = (status)=>204 === status || 205 === status || 304 === status;
 const getRedirectLocation = (headers)=>{
     const location = headers.get('Location');
-    return location && URL.canParse(location, 'http://localhost') ? location : void 0;
+    return null !== location && '' !== location && URL.canParse(location, 'http://localhost') ? location : void 0;
 };
 const processRedirect = (headers, status, ctx)=>{
     headers.delete('content-length');
@@ -27,7 +27,7 @@ const createLoaderRedirectResponse = (beforeRenderResult, redirectCtx)=>{
     if (void 0 === beforeRenderResult || !isRedirectStatus(beforeRenderResult.status)) return;
     if (beforeRenderResult.headers.has('X-Modernjs-Redirect')) return beforeRenderResult;
     const redirectUrl = getRedirectLocation(beforeRenderResult.headers);
-    if (!redirectUrl) return;
+    if (void 0 === redirectUrl) return;
     return processRedirect(new Headers(beforeRenderResult.headers), beforeRenderResult.status, redirectCtx);
 };
 const finalizeRenderResponse = async (response, responseProxy, redirectCtx, routerCleanup)=>{

package/dist/esm/core/server/routerCleanup.mjs

@@ -1,4 +1,4 @@
-import { getRouterRuntimeState } from "../../router/runtime/lifecycle.mjs";
+import { getRouterRuntimeState } from "@modern-js/runtime-extensions/router-state";
 const ROUTER_CLEANUP_ERROR = 'An error occurs during router runtime cleanup';
 async function runWithRouterCleanupOnError(routerCleanup, callback) {
     try {

package/dist/esm/core/server/shared.mjs

@@ -1,4 +1,156 @@
+import { attributesToString, hasStylesheetLink, safeReplace } from "./utils.mjs";
 function buildHtml(template, callbacks) {
     return callbacks.reduce((promise, buildHtmlCb)=>promise.then((template)=>buildHtmlCb(template)), Promise.resolve(template));
 }
-export { buildHtml };
+function createSSRRenderLifecycle(observers) {
+    let terminal;
+    return {
+        transformAssets (groups, info) {
+            let assets = groups.flatMap((group)=>group.assets);
+            for (const observer of observers)assets = observer.transformAssets?.(assets, {
+                ...info,
+                groups
+            }) ?? assets;
+            return assets;
+        },
+        transformTemplateChunk (chunk) {
+            for (const observer of observers)chunk = observer.transformTemplateChunk?.(chunk, {
+                attributesToString: attributesToString,
+                hasStylesheetLink: hasStylesheetLink
+            }) ?? chunk;
+            return chunk;
+        },
+        getRouterData () {
+            let data;
+            for (const observer of observers)data = observer.getRouterData?.() ?? data;
+            return data;
+        },
+        get terminal () {
+            return terminal;
+        },
+        beforeReact () {
+            for (const observer of observers)observer.beforeReact?.();
+        },
+        completedBody (html, phase) {
+            for (const observer of observers)html = observer.completedBody?.(html, {
+                phase
+            }) ?? html;
+            return html;
+        },
+        finish (result) {
+            if (void 0 !== terminal) return;
+            terminal = result;
+            for (const observer of observers)try {
+                observer.onTerminal?.(result);
+            } catch  {}
+        }
+    };
+}
+function replaceSSRTemplateChunk(chunk, lifecycle, options = {}) {
+    const result = lifecycle?.transformTemplateChunk(chunk) ?? chunk;
+    if (true === options.preserveEmpty && '' === result.content) return result.template;
+    return safeReplace(result.template, result.placeholder, result.content);
+}
+function orderSSRStreamTransforms(extenders) {
+    return [
+        ...extenders.filter((extender)=>'body' !== extender.streamPhase),
+        ...extenders.filter((extender)=>'body' === extender.streamPhase)
+    ];
+}
+function createSSRStreamErrorReporter(onError) {
+    let reported = false;
+    return (error)=>{
+        if (reported) return;
+        reported = true;
+        try {
+            onError(error);
+        } catch  {}
+    };
+}
+function observeSSRStream(source, options) {
+    const reader = source.getReader();
+    let controller;
+    let ended = false;
+    let cancellation;
+    const cleanup = ()=>{
+        options.signal.removeEventListener('abort', onAbort);
+        reader.releaseLock();
+    };
+    const cancel = (reason)=>{
+        if (void 0 !== cancellation) return cancellation;
+        if (ended) return Promise.resolve();
+        ended = true;
+        options.lifecycle.finish({
+            status: 'cancelled',
+            reason
+        });
+        cancellation = (async ()=>{
+            try {
+                const observer = Promise.withResolvers();
+                try {
+                    observer.resolve(options.onCancel?.(reason));
+                } catch (error) {
+                    observer.reject(error);
+                }
+                const results = await Promise.allSettled([
+                    observer.promise,
+                    reader.cancel(reason)
+                ]);
+                const rejected = results.find((result)=>'rejected' === result.status);
+                if (rejected?.status === 'rejected') throw rejected.reason;
+            } finally{
+                cleanup();
+            }
+        })();
+        return cancellation;
+    };
+    const onAbort = ()=>{
+        if (ended) return;
+        const reason = options.signal.reason;
+        cancel(reason).then(()=>controller.error(reason), (error)=>{
+            controller.error(reason);
+            return options.onError(error);
+        });
+    };
+    return new ReadableStream({
+        start (value) {
+            controller = value;
+            options.signal.addEventListener('abort', onAbort, {
+                once: true
+            });
+            if (options.signal.aborted) onAbort();
+        },
+        async pull (value) {
+            if (ended) return void await cancellation;
+            try {
+                const result = await reader.read();
+                if (ended) return void await cancellation;
+                if (result.done) {
+                    await options.onComplete?.();
+                    if (ended) return void await cancellation;
+                    ended = true;
+                    options.lifecycle.finish({
+                        status: 'complete'
+                    });
+                    cleanup();
+                    value.close();
+                } else value.enqueue(result.value);
+            } catch (error) {
+                if (ended) return void await cancellation;
+                ended = true;
+                options.lifecycle.finish({
+                    status: 'error',
+                    error
+                });
+                try {
+                    await options.onError(error);
+                } finally{
+                    cleanup();
+                    value.error(error);
+                }
+            }
+        },
+        cancel
+    });
+}
+export { buildHtml, createSSRRenderLifecycle, createSSRStreamErrorReporter, observeSSRStream, orderSSRStreamTransforms, replaceSSRTemplateChunk };

package/dist/esm/core/server/stream/afterTemplate.mjs

@@ -1,12 +1,10 @@
 import { serializeJson } from "@modern-js/runtime-utils/node";
-import { getRouterHydrationScripts } from "../../../router/runtime/lifecycle.mjs";
 import { SSR_DATA_JSON_ID } from "../../constants.mjs";
-import { SSR_DATA_PLACEHOLDER } from "../constants.mjs";
-import { createRouteHydrationScriptTags, replaceChunkJsPlaceholder } from "../scriptOrder.mjs";
-import { buildHtml } from "../shared.mjs";
+import { CHUNK_JS_PLACEHOLDER, SSR_DATA_PLACEHOLDER } from "../constants.mjs";
+import { buildHtml, replaceSSRTemplateChunk } from "../shared.mjs";
 import { attributesToString } from "../utils.mjs";
 function buildShellAfterTemplate(afterAppTemplate, options) {
-    const { request, config, ssrConfig, runtimeContext, renderLevel, entryName } = options;
+    const { request, config, ssrConfig, runtimeContext, renderLevel, entryName, lifecycle } = options;
     const callbacks = [
         (template)=>injectJs(template, entryName, config.nonce),
         createReplaceSSRData({
@@ -16,21 +14,28 @@ function buildShellAfterTemplate(afterAppTemplate, options) {
             useJsonScript: config.useJsonScript,
             runtimeContext,
             renderLevel,
-            entryName
+            lifecycle
         })
     ];
     async function injectJs(template, entryName, nonce) {
-        const jsChunkStr = createRouteHydrationScriptTags(runtimeContext, entryName, {
-            nonce,
-            template
+        const assets = runtimeContext.routeManifest?.routeAssets?.[`async-${entryName}`]?.assets ?? [];
+        const jsChunkStr = assets.filter((asset)=>asset.endsWith('.js')).map((asset)=>`<script${attributesToString({
+                src: asset,
+                nonce
+            })}></script>`).join(' ');
+        return replaceSSRTemplateChunk({
+            name: "scripts",
+            template,
+            placeholder: CHUNK_JS_PLACEHOLDER,
+            content: jsChunkStr
+        }, lifecycle, {
+            preserveEmpty: true
         });
-        if (!jsChunkStr) return template;
-        return replaceChunkJsPlaceholder(template, jsChunkStr, entryName);
     }
     return buildHtml(afterAppTemplate, callbacks);
 }
 function createReplaceSSRData(options) {
-    const { runtimeContext, nonce, renderLevel, useJsonScript, ssrConfig, entryName } = options;
+    const { runtimeContext, nonce, renderLevel, useJsonScript, ssrConfig, lifecycle } = options;
     const { request, reporter } = runtimeContext.ssrContext;
     const headers = 'object' == typeof ssrConfig && ssrConfig.unsafeHeaders ? Object.fromEntries(Object.entries(request.headers).filter(([key, _])=>ssrConfig.unsafeHeaders?.map((header)=>header.toLowerCase())?.includes(key.toLowerCase()))) : void 0;
     const ssrData = {
@@ -59,11 +64,11 @@ function createReplaceSSRData(options) {
     });
     const serializeSSRData = serializeJson(ssrData);
     const ssrDataScript = useJsonScript ? `<script type="application/json" id="${SSR_DATA_JSON_ID}">${serializeSSRData}</script>` : `<script${attrsStr}>window._SSR_DATA = ${serializeSSRData}</script>`;
-    const hydrationScripts = getRouterHydrationScripts(runtimeContext);
-    const ssrScripts = hydrationScripts.length ? `${ssrDataScript}\n${hydrationScripts.join('\n')}` : ssrDataScript;
-    return (template)=>{
-        if (!template.includes(SSR_DATA_PLACEHOLDER)) return template;
-        return replaceChunkJsPlaceholder(template, ssrScripts, entryName, SSR_DATA_PLACEHOLDER);
-    };
+    return (template)=>replaceSSRTemplateChunk({
+            name: 'data',
+            template,
+            placeholder: SSR_DATA_PLACEHOLDER,
+            content: ssrDataScript
+        }, lifecycle);
 }
 export { buildShellAfterTemplate };

package/dist/esm/core/server/stream/beforeTemplate.mjs

@@ -1,10 +1,8 @@
 import { matchRoutes } from "@modern-js/runtime-utils/router";
-import { getRouterMatchedRouteIds } from "../../../router/runtime/lifecycle.mjs";
 import { CHUNK_CSS_PLACEHOLDER } from "../constants.mjs";
-import { createFederatedCssLinks } from "../federatedCss.mjs";
-import { createReplaceHelemt, getHelmetData } from "../helmet.mjs";
-import { buildHtml } from "../shared.mjs";
-import { checkIsNode, hasStylesheetLink, safeReplace } from "../utils.mjs";
+import { createReplaceHelemt } from "../helmet.mjs";
+import { buildHtml, replaceSSRTemplateChunk } from "../shared.mjs";
+import { checkIsNode, hasStylesheetLink } from "../utils.mjs";
 const readAsset = async (chunk)=>{
     const fs = await import("fs/promises");
     const path = await import("path");
@@ -17,8 +15,7 @@ const checkIsInline = (chunk, enableInline)=>{
     return Boolean(enableInline);
 };
 async function buildShellBeforeTemplate(beforeAppTemplate, options) {
-    const { config, runtimeContext, styledComponentsStyleTags, entryName, moduleFederationCssAssets } = options;
-    const helmetData = getHelmetData(runtimeContext);
+    const { config, runtimeContext, styledComponentsStyleTags, entryName, lifecycle, helmetData } = options;
     const callbacks = [
         createReplaceHelemt(helmetData),
         (template)=>injectCss(template, entryName, styledComponentsStyleTags)
@@ -27,36 +24,48 @@ async function buildShellBeforeTemplate(beforeAppTemplate, options) {
     async function injectCss(template, entryName, styledComponentsStyleTags) {
         let css = await getCssChunks();
         if (styledComponentsStyleTags) css += styledComponentsStyleTags;
-        css += createFederatedCssLinks(moduleFederationCssAssets, {
+        return replaceSSRTemplateChunk({
+            name: 'styles',
             template,
-            existingAssets: css.match(/href="([^"]+)"/g)?.map((item)=>item.replace(/^href="/, '').replace(/"$/, ''))
-        });
-        return safeReplace(template, CHUNK_CSS_PLACEHOLDER, css);
+            placeholder: CHUNK_CSS_PLACEHOLDER,
+            content: css,
+            emittedAssets: css.match(/href="([^"]+)"/g)?.map((item)=>item.replace(/^href="/, '').replace(/"$/, ''))
+        }, lifecycle);
         async function getCssChunks() {
             const { routeManifest, routerContext, routes } = runtimeContext;
             const routeAssets = routeManifest?.routeAssets;
             if (!routeAssets) return '';
             let matchedRouteManifests = [];
-            const matchedRouteIds = getRouterMatchedRouteIds(runtimeContext);
-            if (matchedRouteIds?.length) matchedRouteManifests = matchedRouteIds.map((routeId)=>routeAssets[routeId]).filter(Boolean);
-            else if (routerContext && routes) {
+            if (routerContext && routes) {
                 const matches = matchRoutes(routes, routerContext.location, routerContext.basename);
-                matchedRouteManifests = matches?.map((match, index)=>{
-                    if (!index) return;
+                matchedRouteManifests = matches?.flatMap((match, index)=>{
                     const routeId = match.route.id;
-                    if (routeId) return routeAssets[routeId];
-                }).filter((routeManifest)=>void 0 !== routeManifest) ?? [];
+                    const manifest = 0 !== index && routeId ? routeAssets[routeId] : void 0;
+                    return void 0 === manifest ? [] : [
+                        manifest
+                    ];
+                }) ?? [];
             }
-            const asyncEntry = routeAssets[`async-${entryName}`];
-            if (asyncEntry) matchedRouteManifests.push(asyncEntry);
-            const cssChunks = matchedRouteManifests.reduce((chunks, routeManifest)=>{
-                const { referenceCssAssets = [] } = routeManifest;
-                const _cssChunks = referenceCssAssets.filter((asset)=>asset?.endsWith('.css') && !hasStylesheetLink(template, asset));
-                return [
-                    ...chunks,
-                    ..._cssChunks
-                ];
-            }, []);
+            const createAsset = (url)=>({
+                    url
+                });
+            const groups = [
+                {
+                    name: 'route',
+                    assets: matchedRouteManifests.flatMap((manifest)=>manifest.referenceCssAssets ?? []).map(createAsset)
+                },
+                {
+                    name: 'async-entry',
+                    assets: (routeAssets[`async-${entryName}`]?.referenceCssAssets ?? []).map(createAsset)
+                }
+            ];
+            const assets = lifecycle?.transformAssets(groups, {
+                kind: 'style',
+                source: 'template',
+                template,
+                createAsset
+            }) ?? groups.flatMap((group)=>group.assets);
+            const cssChunks = assets.map((asset)=>asset.url).filter((asset)=>asset.endsWith('.css') && !hasStylesheetLink(template, asset));
             const { inlineStyles } = config;
             const styles = await Promise.all(cssChunks.map(async (chunk)=>{
                 const link = `<link href="${chunk}" rel="stylesheet" />`;

package/dist/esm/core/server/stream/beforeTemplate.worker.mjs

@@ -1,17 +1,14 @@
-import { getRouterMatchedRouteIds } from "../../../router/runtime/lifecycle.mjs";
 import { CHUNK_CSS_PLACEHOLDER } from "../constants.mjs";
-import { createFederatedCssLinks } from "../federatedCss.mjs";
-import { createReplaceHelemt, getHelmetData } from "../helmet.mjs";
-import { buildHtml } from "../shared.mjs";
-import { hasStylesheetLink, safeReplace } from "../utils.mjs";
+import { createReplaceHelemt } from "../helmet.mjs";
+import { buildHtml, replaceSSRTemplateChunk } from "../shared.mjs";
+import { hasStylesheetLink } from "../utils.mjs";
 const checkIsInline = (chunk, enableInline)=>{
     if ('production' !== process.env.NODE_ENV) return false;
     if (enableInline instanceof RegExp) return enableInline.test(chunk);
     return Boolean(enableInline);
 };
 async function buildShellBeforeTemplate(beforeAppTemplate, options) {
-    const { config, runtimeContext, styledComponentsStyleTags, entryName, moduleFederationCssAssets } = options;
-    const helmetData = getHelmetData(runtimeContext);
+    const { config, runtimeContext, styledComponentsStyleTags, entryName, lifecycle, helmetData } = options;
     const callbacks = [
         createReplaceHelemt(helmetData),
         (template)=>injectCss(template, entryName, styledComponentsStyleTags)
@@ -20,28 +17,38 @@ async function buildShellBeforeTemplate(beforeAppTemplate, options) {
     async function injectCss(template, entryName, styledComponentsStyleTags) {
         let css = await getCssChunks();
         if (styledComponentsStyleTags) css += styledComponentsStyleTags;
-        css += createFederatedCssLinks(moduleFederationCssAssets, {
+        return replaceSSRTemplateChunk({
+            name: 'styles',
             template,
-            existingAssets: css.match(/href="([^"]+)"/g)?.map((item)=>item.replace(/^href="/, '').replace(/"$/, ''))
-        });
-        return safeReplace(template, CHUNK_CSS_PLACEHOLDER, css);
+            placeholder: CHUNK_CSS_PLACEHOLDER,
+            content: css,
+            emittedAssets: css.match(/href="([^"]+)"/g)?.map((item)=>item.replace(/^href="/, '').replace(/"$/, ''))
+        }, lifecycle);
         async function getCssChunks() {
             const { routeManifest } = runtimeContext;
             const routeAssets = routeManifest?.routeAssets;
             if (!routeAssets) return '';
-            let matchedRouteManifests = [];
-            const matchedRouteIds = getRouterMatchedRouteIds(runtimeContext);
-            if (matchedRouteIds?.length) matchedRouteManifests = matchedRouteIds.map((routeId)=>routeAssets[routeId]).filter(Boolean);
-            const asyncEntry = routeAssets[`async-${entryName}`];
-            if (asyncEntry) matchedRouteManifests.push(asyncEntry);
-            const cssChunks = matchedRouteManifests.reduce((chunks, routeManifest)=>{
-                const { referenceCssAssets = [] } = routeManifest;
-                const _cssChunks = referenceCssAssets.filter((asset)=>asset?.endsWith('.css') && !hasStylesheetLink(template, asset));
-                return [
-                    ...chunks,
-                    ..._cssChunks
-                ];
-            }, []);
+            const matchedRouteManifests = [];
+            const createAsset = (url)=>({
+                    url
+                });
+            const groups = [
+                {
+                    name: 'route',
+                    assets: matchedRouteManifests.flatMap((manifest)=>manifest.referenceCssAssets ?? []).map(createAsset)
+                },
+                {
+                    name: 'async-entry',
+                    assets: (routeAssets[`async-${entryName}`]?.referenceCssAssets ?? []).map(createAsset)
+                }
+            ];
+            const assets = lifecycle?.transformAssets(groups, {
+                kind: 'style',
+                source: 'template',
+                template,
+                createAsset
+            }) ?? groups.flatMap((group)=>group.assets);
+            const cssChunks = assets.map((asset)=>asset.url).filter((asset)=>asset.endsWith('.css') && !hasStylesheetLink(template, asset));
             const { inlineStyles } = config;
             const styles = cssChunks.map((chunk)=>{
                 const link = `<link href="${chunk}" rel="stylesheet" />`;