michael/astro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix hoisted script scanning

Browse source
This commit is contained in:
unknown 2022-06-17 16:04:50 -04:00
parent b21c9576e8
commit e1712020d4
1 changed files with 38 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
packages/astro/src/core/build/vite-plugin-analyzer.ts
View file

@ -18,25 +18,41 @@ export function vitePluginAnalyzer(
function hoistedScriptScanner() {
const uniqueHoistedIds = new Map<string, string>();
return function(
this: PluginContext,
scripts: AstroPluginMetadata['astro']['scripts'],
from: string
) {
const hoistedScripts = new Set<string>();
for(let i = 0; i < scripts.length; i++) {
const hid = `${from.replace('/@fs', '')}?astro&type=script&index=${i}`;
hoistedScripts.add(hid);
}
const pageScripts = new Map<string, Set<string>>();
for(const pageId of getTopLevelPages(from, this)) {
const pageData = getPageDataByViteID(internals, pageId);
if(!pageData) continue;
const { component } = pageData;
const astroModuleId = prependForwardSlash(component);
return {
scan(
this: PluginContext,
scripts: AstroPluginMetadata['astro']['scripts'],
from: string
) {
const hoistedScripts = new Set<string>();
for(let i = 0; i < scripts.length; i++) {
const hid = `${from.replace('/@fs', '')}?astro&type=script&index=${i}`;
hoistedScripts.add(hid);
}
if (hoistedScripts.size) {
for(const pageId of getTopLevelPages(from, this)) {
for(const hid of hoistedScripts) {
if(pageScripts.has(pageId)) {
pageScripts.get(pageId)?.add(hid);
} else {
pageScripts.set(pageId, new Set([hid]));
}
}
}
}
},
finalize() {
for(const [pageId, hoistedScripts] of pageScripts) {
const pageData = getPageDataByViteID(internals, pageId);
if(!pageData) continue;
const { component } = pageData;
const astroModuleId = prependForwardSlash(component);
const uniqueHoistedId = JSON.stringify(Array.from(hoistedScripts).sort());
let moduleId: string;
@ -60,13 +76,13 @@ export function vitePluginAnalyzer(
}
}
}
}
};
}
return {
name: '@astro/rollup-plugin-astro-analyzer',
generateBundle() {
const scanHoisted = hoistedScriptScanner();
const hoistScanner = hoistedScriptScanner();
const ids = this.getModuleIds();
for(const id of ids) {
@ -80,7 +96,7 @@ export function vitePluginAnalyzer(
}
// Scan hoisted scripts
scanHoisted.call(this, astro.scripts, id);
hoistScanner.scan.call(this, astro.scripts, id);
if(astro.clientOnlyComponents.length) {
const clientOnlys: string[] = [];
@ -99,6 +115,9 @@ export function vitePluginAnalyzer(
}
}
}
// Finalize hoisting
hoistScanner.finalize();
}
};
}