From 7f721028f5ecba645549df5475d9e05a6f250401 Mon Sep 17 00:00:00 2001 From: sapphi-red Date: Fri, 15 Apr 2022 09:26:52 +0900 Subject: [PATCH] fix(css): preserve dynamic import css code --- .../__tests__/css-codesplit.spec.ts | 18 +++++++++++++++++- packages/playground/css-codesplit/async.css | 3 +++ packages/playground/css-codesplit/index.html | 9 +++++++++ packages/playground/css-codesplit/inline.css | 3 +++ packages/playground/css-codesplit/main.js | 15 ++++++++++++--- .../playground/css-codesplit/mod.module.css | 3 +++ .../src/node/plugins/importAnalysisBuild.ts | 18 ++++++++++++------ 7 files changed, 59 insertions(+), 10 deletions(-) create mode 100644 packages/playground/css-codesplit/async.css create mode 100644 packages/playground/css-codesplit/inline.css create mode 100644 packages/playground/css-codesplit/mod.module.css diff --git a/packages/playground/css-codesplit/__tests__/css-codesplit.spec.ts b/packages/playground/css-codesplit/__tests__/css-codesplit.spec.ts index 95fe97a1b953ba..789adba23ae021 100644 --- a/packages/playground/css-codesplit/__tests__/css-codesplit.spec.ts +++ b/packages/playground/css-codesplit/__tests__/css-codesplit.spec.ts @@ -1,8 +1,23 @@ import { findAssetFile, getColor, isBuild, readManifest } from '../../testUtils' -test('should load both stylesheets', async () => { +test('should load all stylesheets', async () => { expect(await getColor('h1')).toBe('red') expect(await getColor('h2')).toBe('blue') + expect(await getColor('.dynamic')).toBe('green') +}) + +test('should load dynamic import with inline', async () => { + const css = await page.textContent('.dynamic-inline') + expect(css).toMatch('.inline') + + expect(await getColor('.inline')).not.toBe('yellow') +}) + +test('should load dynamic import with module', async () => { + const css = await page.textContent('.dynamic-module') + expect(css).toMatch('_mod_') + + expect(await getColor('.mod')).toBe('yellow') }) if (isBuild) { @@ -10,6 +25,7 @@ if (isBuild) { expect(findAssetFile(/style.*\.js$/)).toBe('') expect(findAssetFile('main.*.js$')).toMatch(`/* empty css`) expect(findAssetFile('other.*.js$')).toMatch(`/* empty css`) + expect(findAssetFile(/async.*\.js$/)).toBe('') }) test('should generate correct manifest', async () => { diff --git a/packages/playground/css-codesplit/async.css b/packages/playground/css-codesplit/async.css new file mode 100644 index 00000000000000..4902b2e7bee811 --- /dev/null +++ b/packages/playground/css-codesplit/async.css @@ -0,0 +1,3 @@ +.dynamic { + color: green; +} diff --git a/packages/playground/css-codesplit/index.html b/packages/playground/css-codesplit/index.html index 6b7b3bb2b4dc2d..63bdb59e11dc6b 100644 --- a/packages/playground/css-codesplit/index.html +++ b/packages/playground/css-codesplit/index.html @@ -1,2 +1,11 @@ +

This should be red

+

This should be blue

+ +

This should be green

+

This should not be yellow

+

+

This should be yellow

+

+
diff --git a/packages/playground/css-codesplit/inline.css b/packages/playground/css-codesplit/inline.css new file mode 100644 index 00000000000000..b2a2b5f1ead51f --- /dev/null +++ b/packages/playground/css-codesplit/inline.css @@ -0,0 +1,3 @@ +.inline { + color: yellow; +} diff --git a/packages/playground/css-codesplit/main.js b/packages/playground/css-codesplit/main.js index 8c80df2c181511..eb6e703f79e718 100644 --- a/packages/playground/css-codesplit/main.js +++ b/packages/playground/css-codesplit/main.js @@ -1,6 +1,15 @@ import './style.css' import './main.css' -document.getElementById( - 'app' -).innerHTML = `

This should be red

This should be blue

` +import('./async.css') + +import('./inline.css?inline').then((css) => { + document.querySelector('.dynamic-inline').textContent = css.default +}) + +import('./mod.module.css').then((css) => { + document.querySelector('.dynamic-module').textContent = JSON.stringify( + css.default + ) + document.querySelector('.mod').classList.add(css.default.mod) +}) diff --git a/packages/playground/css-codesplit/mod.module.css b/packages/playground/css-codesplit/mod.module.css new file mode 100644 index 00000000000000..7f84410485a32c --- /dev/null +++ b/packages/playground/css-codesplit/mod.module.css @@ -0,0 +1,3 @@ +.mod { + color: yellow; +} diff --git a/packages/vite/src/node/plugins/importAnalysisBuild.ts b/packages/vite/src/node/plugins/importAnalysisBuild.ts index 91ce663b9f8111..446a575ba9f998 100644 --- a/packages/vite/src/node/plugins/importAnalysisBuild.ts +++ b/packages/vite/src/node/plugins/importAnalysisBuild.ts @@ -177,11 +177,15 @@ export function buildImportAnalysisPlugin(config: ResolvedConfig): Plugin { continue } - if (dynamicIndex > -1 && insertPreload) { + const isDynamic = dynamicIndex > -1 + + if (isDynamic && insertPreload) { needPreloadHelper = true - const original = source.slice(expStart, expEnd) - const replacement = `${preloadMethod}(() => ${original},${isModernFlag}?"${preloadMarker}":void 0)` - str().overwrite(expStart, expEnd, replacement, { contentOnly: true }) + str().prependLeft(expStart, `${preloadMethod}(() => `) + str().appendRight( + expEnd, + `,${isModernFlag}?"${preloadMarker}":void 0)` + ) } // Differentiate CSS imports that use the default export from those that @@ -191,12 +195,14 @@ export function buildImportAnalysisPlugin(config: ResolvedConfig): Plugin { if ( specifier && isCSSRequest(specifier) && - source.slice(expStart, start).includes('from') && + // always inject ?used query when it is a dynamic import + // because there is no way to check whether the default export is used + (source.slice(expStart, start).includes('from') || isDynamic) && // edge case for package names ending with .css (e.g normalize.css) !(bareImportRE.test(specifier) && !specifier.includes('/')) ) { const url = specifier.replace(/\?|$/, (m) => `?used${m ? '&' : ''}`) - str().overwrite(start, end, dynamicIndex > -1 ? `'${url}'` : url, { + str().overwrite(start, end, isDynamic ? `'${url}'` : url, { contentOnly: true }) }