-
-
Notifications
You must be signed in to change notification settings - Fork 9.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
148 additions
and
28 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
31 changes: 31 additions & 0 deletions
31
code/lib/builder-vite/src/plugins/externals-plugin.test.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,31 @@ | ||
| import { rewriteImport } from './externals-plugin'; | ||
|
|
||
| const packageName = '@storybook/package'; | ||
| const globals = { [packageName]: '_STORYBOOK_PACKAGE_' }; | ||
|
|
||
| const cases = [ | ||
| { | ||
| globals, | ||
| packageName, | ||
| input: `import { Rain, Jour as Day, Nuit as Night, Sun } from "${packageName}"`, | ||
| output: `const { Rain, Jour: Day, Nuit: Night, Sun } = ${globals[packageName]}`, | ||
| }, | ||
| { | ||
| globals, | ||
| packageName, | ||
| input: `import{Rain,Jour as Day,Nuit as Night,Sun}from '${packageName}'`, | ||
| output: `const {Rain,Jour: Day,Nuit: Night,Sun} = ${globals[packageName]}`, | ||
| }, | ||
| { | ||
| globals, | ||
| packageName, | ||
| input: `const { Afternoon } = await import('${packageName}')`, | ||
| output: `const { Afternoon } = ${globals[packageName]}`, | ||
| }, | ||
| ]; | ||
|
|
||
| test('rewriteImport', () => { | ||
| cases.forEach(({ input, output, globals: caseGlobals, packageName: casePackage }) => { | ||
| expect(rewriteImport<false>(input, caseGlobals, casePackage)).toStrictEqual(output); | ||
| }); | ||
| }); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,112 @@ | ||
| import { join } from 'node:path'; | ||
| import { init, parse } from 'es-module-lexer'; | ||
| import MagicString from 'magic-string'; | ||
| import { emptyDir, ensureDir, ensureFile, writeFile } from 'fs-extra'; | ||
| import { mergeAlias } from 'vite'; | ||
| import type { Alias, Plugin } from 'vite'; | ||
| import { globals } from '@storybook/preview/globals'; | ||
|
|
||
| type SingleGlobalName = keyof typeof globals; | ||
| type Globals = typeof globals & Record<string, string>; | ||
|
|
||
| const replacementMap = new Map([ | ||
| ['import ', 'const '], | ||
| ['import{', 'const {'], | ||
| [' as ', ': '], | ||
| [' from ', ' = '], | ||
| ['}from', '} ='], | ||
| ]); | ||
|
|
||
| /** | ||
| * This plugin swaps out imports of pre-bundled storybook preview modules for destructures from global | ||
| * variables that are added in runtime.mjs. | ||
| * | ||
| * For instance: | ||
| * | ||
| * ```js | ||
| * import { useMemo as useMemo2, useEffect as useEffect2 } from "@storybook/preview-api"; | ||
| * ``` | ||
| * | ||
| * becomes | ||
| * | ||
| * ```js | ||
| * const { useMemo: useMemo2, useEffect: useEffect2 } = __STORYBOOK_MODULE_PREVIEW_API__; | ||
| * ``` | ||
| * | ||
| * It is based on existing plugins like https://github.com/crcong/vite-plugin-externals | ||
| * and https://github.com/eight04/rollup-plugin-external-globals, but simplified to meet our simple needs. | ||
| */ | ||
| export async function externalsPlugin() { | ||
| await init; | ||
| return { | ||
| name: 'storybook:externals-plugin', | ||
| enforce: 'post', | ||
| // In dev (serve), we set up aliases to files that we write into node_modules/.cache. | ||
| async config(config, { command }) { | ||
| if (command !== 'serve') { | ||
| return undefined; | ||
| } | ||
| const newAlias = mergeAlias([], config.resolve?.alias) as Alias[]; | ||
|
|
||
| const cachePath = join(process.cwd(), 'node_modules', '.cache', 'vite-plugin-externals'); | ||
| await ensureDir(cachePath); | ||
| await emptyDir(cachePath); | ||
| await Promise.all( | ||
| (Object.keys(globals) as Array<keyof typeof globals>).map(async (externalKey) => { | ||
| const externalCachePath = join(cachePath, `${externalKey}.js`); | ||
| newAlias.push({ find: new RegExp(`^${externalKey}$`), replacement: externalCachePath }); | ||
| await ensureFile(externalCachePath); | ||
| await writeFile(externalCachePath, `module.exports = ${globals[externalKey]};`); | ||
| }) | ||
| ); | ||
|
|
||
| return { | ||
| resolve: { | ||
| alias: newAlias, | ||
| }, | ||
| }; | ||
| }, | ||
| // Replace imports with variables destructured from global scope | ||
| async transform(code: string, id: string) { | ||
| const globalsList = Object.keys(globals) as SingleGlobalName[]; | ||
| if (globalsList.every((glob) => !code.includes(glob))) return undefined; | ||
|
|
||
| const [imports] = parse(code); | ||
| const src = new MagicString(code); | ||
| imports.forEach(({ n: path, ss: startPosition, se: endPosition }) => { | ||
| const packageName = path as SingleGlobalName | undefined; | ||
| if (packageName && globalsList.includes(packageName)) { | ||
| const importStatement = src.slice(startPosition, endPosition); | ||
| const transformedImport = rewriteImport(importStatement, globals, packageName); | ||
| src.update(startPosition, endPosition, transformedImport); | ||
| } | ||
| }); | ||
|
|
||
| return { | ||
| code: src.toString(), | ||
| map: src.generateMap({ | ||
| source: id, | ||
| includeContent: true, | ||
| hires: true, | ||
| }), | ||
| }; | ||
| }, | ||
| } satisfies Plugin; | ||
| } | ||
|
|
||
| export function rewriteImport<T = true>( | ||
| importStatement: string, | ||
| globs: T extends true ? Globals : Record<string, string>, | ||
| packageName: keyof typeof globs & string | ||
| ): string { | ||
| const lookup = [ | ||
| ...replacementMap.keys(), | ||
| `.${packageName}.`, | ||
| `await import\\(.${packageName}.\\)`, | ||
| ]; | ||
| const search = new RegExp(`(${lookup.join('|')})`, 'g'); | ||
| return importStatement.replace( | ||
| search, | ||
| (match) => replacementMap.get(match) ?? globs[packageName] | ||
| ); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters