one
Version:
One is a new React Framework that makes Vite serve both native and web.
114 lines (112 loc) • 4.19 kB
JavaScript
import { VIRTUAL_SSR_CSS_ENTRY, VIRTUAL_SSR_CSS_HREF } from "../../constants.mjs";
function SSRCSSPlugin(pluginOpts) {
let server;
return {
name: "one-plugin-ssr-css",
apply: "serve",
configureServer(server_) {
server = server_, server.middlewares.use(async (req, res, next) => {
if (req.url?.includes(VIRTUAL_SSR_CSS_HREF)) {
invalidateModule(server, "\0" + VIRTUAL_SSR_CSS_ENTRY + "?direct");
const code = await collectStyle(server, pluginOpts.entries);
res.setHeader("Content-Type", "text/css"), res.setHeader("Cache-Control", "no-store"), res.setHeader("Vary", "*"), res.write(code), res.end();
return;
}
next();
});
},
// virtual module
// (use `startsWith` since Vite adds `?direct` for raw css request)
resolveId(source, _importer, _options) {
return source.startsWith(VIRTUAL_SSR_CSS_ENTRY) ? "\0" + source : void 0;
},
async load(id, _options) {
if (id.startsWith("\0" + VIRTUAL_SSR_CSS_ENTRY)) {
const url = new URL(id.slice(1), "https://test.local");
let code = await collectStyle(server, pluginOpts.entries);
return url.searchParams.has("direct") || (code = `export default ${JSON.stringify(code)}`), code;
}
},
// also expose via transformIndexHtml
transformIndexHtml: {
handler: async () => [{
tag: "link",
injectTo: "head",
attrs: {
rel: "stylesheet",
href: VIRTUAL_SSR_CSS_HREF,
"data-ssr-css": !0
}
}, {
tag: "script",
injectTo: "head",
attrs: {
type: "module"
},
children: (/* js */
`
import { createHotContext } from "/@vite/client";
const hot = createHotContext("/__clear_ssr_css");
hot.on("vite:afterUpdate", () => {
document
.querySelectorAll("[data-ssr-css]")
.forEach(node => node.remove());
});
`)
}]
}
};
}
function invalidateModule(server, id) {
const mod = server.moduleGraph.getModuleById(id);
mod && server.moduleGraph.invalidateModule(mod);
}
async function collectStyle(server, entries) {
const {
transform
} = await import("lightningcss"),
urls = await collectStyleUrls(server, entries);
let out = (await Promise.all(urls.map(async url => {
const code = (await server.transformRequest(url + "?direct"))?.code || "",
prefix = `/* [collectStyle] ${url} */`;
try {
const buffer = Buffer.from(code),
codeOut = new Uint8Array(buffer.buffer, buffer.byteOffset, buffer.byteLength),
processed = transform({
filename: "code.css",
code: codeOut,
...server.config.css.lightningcss
}).code.toString();
return [prefix, processed];
} catch (err) {
return console.error(` [one] Error post-processing CSS, leaving un-processed: ${err}`), [prefix, code];
}
}))).flat().filter(Boolean).join(`
`);
try {
const buffer = Buffer.from(out),
codeOut = new Uint8Array(buffer.buffer, buffer.byteOffset, buffer.byteLength);
out = transform({
filename: "code.css",
code: codeOut,
...server.config.css.lightningcss
}).code.toString();
} catch {
console.error(" [one] Error post-processing merged CSS, leaving un-processed");
}
return out;
}
async function collectStyleUrls(server, entries) {
const visited = /* @__PURE__ */new Set();
async function traverse(url) {
const [, id] = await server.moduleGraph.resolveUrl(url);
if (visited.has(id)) return;
visited.add(id);
const mod = server.moduleGraph.getModuleById(id);
mod && (await Promise.all([...mod.importedModules].map(childMod => traverse(childMod.url))));
}
return await Promise.all(entries.map(e => server.transformRequest(e))), await Promise.all(entries.map(url => traverse(url))), [...visited].filter(url => url.match(CSS_LANGS_RE));
}
const CSS_LANGS_RE = /\.(css|less|sass|scss|styl|stylus|pcss|postcss|sss)(?:$|\?)/;
export { SSRCSSPlugin, collectStyle };
//# sourceMappingURL=SSRCSSPlugin.mjs.map