UNPKG

@vitejs/plugin-react-swc

Version:
238 lines (228 loc) 8.72 kB
import { createRequire } from "node:module"; import { readFileSync } from "node:fs"; import { dirname, join } from "node:path"; import { fileURLToPath } from "node:url"; import { transform } from "@swc/core"; import * as vite from "vite"; import { exactRegex } from "@rolldown/pluginutils"; //#region ../common/refresh-utils.ts const runtimePublicPath = "/@react-refresh"; const reactCompRE = /extends\s+(?:React\.)?(?:Pure)?Component/; const refreshContentRE = /\$RefreshReg\$\(/; const preambleCode = `import { injectIntoGlobalHook } from "__BASE__${runtimePublicPath.slice(1)}"; injectIntoGlobalHook(window); window.$RefreshReg$ = () => {}; window.$RefreshSig$ = () => (type) => type;`; const getPreambleCode = (base) => preambleCode.replace("__BASE__", base); const avoidSourceMapOption = Symbol(); function addRefreshWrapper(code, map, pluginName, id, reactRefreshHost = "") { const hasRefresh = refreshContentRE.test(code); const onlyReactComp = !hasRefresh && reactCompRE.test(code); const normalizedMap = map === avoidSourceMapOption ? null : map; if (!hasRefresh && !onlyReactComp) return { code, map: normalizedMap }; const avoidSourceMap = map === avoidSourceMapOption; const newMap = typeof normalizedMap === "string" ? JSON.parse(normalizedMap) : normalizedMap; let newCode = code; if (hasRefresh) { const refreshHead = removeLineBreaksIfNeeded(`let prevRefreshReg; let prevRefreshSig; if (import.meta.hot && !inWebWorker) { if (!window.$RefreshReg$) { throw new Error( "${pluginName} can't detect preamble. Something is wrong." ); } prevRefreshReg = window.$RefreshReg$; prevRefreshSig = window.$RefreshSig$; window.$RefreshReg$ = RefreshRuntime.getRefreshReg(${JSON.stringify(id)}); window.$RefreshSig$ = RefreshRuntime.createSignatureFunctionForTransform; } `, avoidSourceMap); newCode = `${refreshHead}${newCode} if (import.meta.hot && !inWebWorker) { window.$RefreshReg$ = prevRefreshReg; window.$RefreshSig$ = prevRefreshSig; } `; if (newMap) newMap.mappings = ";".repeat(16) + newMap.mappings; } const sharedHead = removeLineBreaksIfNeeded(`import * as RefreshRuntime from "${reactRefreshHost}${runtimePublicPath}"; const inWebWorker = typeof WorkerGlobalScope !== 'undefined' && self instanceof WorkerGlobalScope; `, avoidSourceMap); newCode = `${sharedHead}${newCode} if (import.meta.hot && !inWebWorker) { RefreshRuntime.__hmr_import(import.meta.url).then((currentExports) => { RefreshRuntime.registerExportsForReactRefresh(${JSON.stringify(id)}, currentExports); import.meta.hot.accept((nextExports) => { if (!nextExports) return; const invalidateMessage = RefreshRuntime.validateRefreshBoundaryAndEnqueueUpdate(${JSON.stringify(id)}, currentExports, nextExports); if (invalidateMessage) import.meta.hot.invalidate(invalidateMessage); }); }); } `; if (newMap) newMap.mappings = ";;;" + newMap.mappings; return { code: newCode, map: newMap }; } function removeLineBreaksIfNeeded(code, enabled) { return enabled ? code.replace(/\n/g, "") : code; } //#endregion //#region ../common/warning.ts const silenceUseClientWarning = (userConfig) => ({ rollupOptions: { onwarn(warning, defaultHandler) { if (warning.code === "MODULE_LEVEL_DIRECTIVE" && (warning.message.includes("use client") || warning.message.includes("use server"))) return; if (warning.code === "SOURCEMAP_ERROR" && warning.message.includes("resolve original location") && warning.pos === 0) return; if (userConfig.build?.rollupOptions?.onwarn) userConfig.build.rollupOptions.onwarn(warning, defaultHandler); else defaultHandler(warning); } } }); //#endregion //#region src/index.ts const _dirname = typeof __dirname !== "undefined" ? __dirname : dirname(fileURLToPath(import.meta.url)); const resolve = createRequire(typeof __filename !== "undefined" ? __filename : import.meta.url).resolve; const react = (_options) => { let hmrDisabled = false; const options = { jsxImportSource: _options?.jsxImportSource ?? "react", tsDecorators: _options?.tsDecorators, plugins: _options?.plugins ? _options?.plugins.map((el) => [resolve(el[0]), el[1]]) : void 0, devTarget: _options?.devTarget ?? "es2020", parserConfig: _options?.parserConfig, reactRefreshHost: _options?.reactRefreshHost, useAtYourOwnRisk_mutateSwcOptions: _options?.useAtYourOwnRisk_mutateSwcOptions, disableOxcRecommendation: _options?.disableOxcRecommendation }; return [ { name: "vite:react-swc:resolve-runtime", apply: "serve", enforce: "pre", resolveId: { filter: { id: exactRegex(runtimePublicPath) }, handler: (id) => id === runtimePublicPath ? id : void 0 }, load: { filter: { id: exactRegex(runtimePublicPath) }, handler: (id) => id === runtimePublicPath ? readFileSync(join(_dirname, "refresh-runtime.js"), "utf-8").replace(/__README_URL__/g, "https://github.com/vitejs/vite-plugin-react/tree/main/packages/plugin-react-swc") : void 0 } }, { name: "vite:react-swc", apply: "serve", config: () => ({ esbuild: false, oxc: false, optimizeDeps: { include: [`${options.jsxImportSource}/jsx-dev-runtime`], ..."rolldownVersion" in vite ? { rollupOptions: { jsx: { mode: "automatic" } } } : { esbuildOptions: { jsx: "automatic" } } } }), configResolved(config) { if (config.server.hmr === false) hmrDisabled = true; const mdxIndex = config.plugins.findIndex((p) => p.name === "@mdx-js/rollup"); if (mdxIndex !== -1 && mdxIndex > config.plugins.findIndex((p) => p.name === "vite:react-swc")) throw new Error("[vite:react-swc] The MDX plugin should be placed before this plugin"); if ("rolldownVersion" in vite && !options.plugins && !options.useAtYourOwnRisk_mutateSwcOptions && !options.disableOxcRecommendation) config.logger.warn("[vite:react-swc] We recommend switching to `@vitejs/plugin-react-oxc` for improved performance as no swc plugins are used. More information at https://vite.dev/rolldown"); }, transformIndexHtml: (_, config) => { if (!hmrDisabled) return [{ tag: "script", attrs: { type: "module" }, children: getPreambleCode(config.server.config.base) }]; }, async transform(code, _id, transformOptions) { const id = _id.split("?")[0]; const refresh = !transformOptions?.ssr && !hmrDisabled; const result = await transformWithOptions(id, code, options.devTarget, options, { refresh, development: true, runtime: "automatic", importSource: options.jsxImportSource }); if (!result) return; if (!refresh) return result; return addRefreshWrapper(result.code, result.map, "@vitejs/plugin-react-swc", id, options.reactRefreshHost); } }, options.plugins ? { name: "vite:react-swc", apply: "build", enforce: "pre", config: (userConfig) => ({ build: silenceUseClientWarning(userConfig) }), transform: (code, _id) => transformWithOptions(_id.split("?")[0], code, "esnext", options, { runtime: "automatic", importSource: options.jsxImportSource }) } : { name: "vite:react-swc", apply: "build", config: (userConfig) => ({ build: silenceUseClientWarning(userConfig), esbuild: { jsx: "automatic", jsxImportSource: options.jsxImportSource, tsconfigRaw: { compilerOptions: { useDefineForClassFields: true } } } }) } ]; }; const transformWithOptions = async (id, code, target, options, reactConfig) => { const decorators = options?.tsDecorators ?? false; const parser = options.parserConfig ? options.parserConfig(id) : id.endsWith(".tsx") ? { syntax: "typescript", tsx: true, decorators } : id.endsWith(".ts") || id.endsWith(".mts") ? { syntax: "typescript", tsx: false, decorators } : id.endsWith(".jsx") ? { syntax: "ecmascript", jsx: true } : id.endsWith(".mdx") ? { syntax: "ecmascript", jsx: true } : void 0; if (!parser) return; let result; try { const swcOptions = { filename: id, swcrc: false, configFile: false, sourceMaps: true, jsc: { target, parser, experimental: { plugins: options.plugins }, transform: { useDefineForClassFields: true, react: reactConfig } } }; if (options.useAtYourOwnRisk_mutateSwcOptions) options.useAtYourOwnRisk_mutateSwcOptions(swcOptions); result = await transform(code, swcOptions); } catch (e) { const message = e.message; const fileStartIndex = message.indexOf("╭─["); if (fileStartIndex !== -1) { const match = message.slice(fileStartIndex).match(/:(\d+):(\d+)\]/); if (match) { e.line = match[1]; e.column = match[2]; } } throw e; } return result; }; var src_default = react; //#endregion export { src_default as default };