update

Author: li-jia-nan

src/getBabelCommonConfig.ts

@@ -2,18 +2,16 @@ import fs from 'fs-extra';
 import type { PluginItem, TransformOptions } from '@babel/core';
 import { resolve, isThereHaveBrowserslistConfig } from './utils/projectHelper';
 
+interface BabelConfigOptions {
+  enabledReactCompiler?: boolean;
+}
+
 interface BabelConfig extends TransformOptions {
   cacheDirectory?: boolean;
 }
 
-export default function getBabelCommonConfig(modules?: boolean): BabelConfig {
+function getBabelCommonConfig(modules?: boolean, options: BabelConfigOptions = {}): BabelConfig {
   const plugins: PluginItem[] = [
-    [
-      resolve('babel-plugin-react-compiler'),
-      {
-        target: '18', // 最低支持的版本是 React 18
-      },
-    ],
     [
       resolve('@babel/plugin-transform-typescript'),
       {
@@ -35,6 +33,14 @@ export default function getBabelCommonConfig(modules?: boolean): BabelConfig {
     resolve('babel-plugin-transform-dev-warning'),
     resolve('@babel/plugin-transform-private-methods'),
   ];
+  if (options.enabledReactCompiler === true) {
+    plugins.unshift([
+      resolve('babel-plugin-react-compiler'),
+      {
+        target: '18', // 最低支持的版本是 React 18
+      },
+    ]);
+  }
   return {
     presets: [
       resolve('@babel/preset-react'),
@@ -53,3 +59,5 @@ export default function getBabelCommonConfig(modules?: boolean): BabelConfig {
     plugins,
   };
 }
+
+export default getBabelCommonConfig;

src/getWebpackConfig.ts

@@ -35,9 +35,14 @@ interface GetWebpackConfigFunction {
   imageOptions: typeof imageOptions;
 }
 
+const libDir = process.env.LIB_DIR || 'components';
+
 const getWebpackConfig: GetWebpackConfigFunction = (modules?: boolean): Configuration[] => {
   const pkg: PackageJson = readJsonSync(getProjectPath('package.json'));
-  const babelConfig = getBabelCommonConfig(modules || false);
+
+  const babelConfig = getBabelCommonConfig(modules || false, {
+    enabledReactCompiler: libDir === 'dist',
+  });
 
   babelConfig.plugins.push([
     resolve('babel-plugin-import'),

src/gulpfile.ts

@@ -18,6 +18,7 @@ import replaceLib from './replaceLib';
 import checkDiff from './lint/checkDiff';
 import apiCollection from './apiCollection';
 import sortApiTable from './sortApiTable';
+import type { ICompileStream } from 'gulp-typescript/release/project';
 
 const argv = minimist(process.argv.slice(2));
 
@@ -143,8 +144,10 @@ gulp.task(
   })
 );
 
-function babelify(js, modules) {
-  const babelConfig = getBabelCommonConfig(modules);
+function babelify(js: ICompileStream['js'], modules: boolean) {
+  const babelConfig = getBabelCommonConfig(modules, {
+    enabledReactCompiler: libDir === 'dist',
+  });
   delete babelConfig.cacheDirectory;
   if (modules === false) {
     babelConfig.plugins.push(replaceLib);

src/jest/codePreprocessor.ts

@@ -36,7 +36,9 @@ const preprocessor: Preprocessor = {
   canInstrument: true,
   process(src, filePath, config, transformOptions) {
     global.__clearBabelAntdPlugin && global.__clearBabelAntdPlugin(); // eslint-disable-line
-    const babelConfig = getBabelCommonConfig();
+    const babelConfig = getBabelCommonConfig(undefined, {
+      enabledReactCompiler: libDir === 'dist',
+    });
     babelConfig.plugins = [...(babelConfig.plugins || [])];
 
     if (/\/demo\//.test(filePath)) {

src/jest/demoPreprocessor.ts

@@ -120,7 +120,10 @@ function transform(src: string, pathFilename: string): ProcessResult {
 
   global.__clearBabelAntdPlugin && global.__clearBabelAntdPlugin(); // eslint-disable-line
 
-  const babelConfig = getBabelCommonConfig();
+  const babelConfig = getBabelCommonConfig(undefined, {
+    enabledReactCompiler: libDir === 'dist',
+  });
+
   babelConfig.plugins = [...babelConfig.plugins];
   babelConfig.plugins.push(createDemo);
 

src/jest/rewriteSource.ts

@@ -1,9 +1,8 @@
-import type { NodePath } from '@babel/traverse';
-import type { ImportDeclaration } from '@babel/types';
+import * as babel from '@babel/core';
 
 function rewriteSource(
-  t: typeof import('@babel/types'),
-  path: NodePath<ImportDeclaration>,
+  t: typeof babel.types,
+  path: babel.NodePath<babel.types.ImportDeclaration>,
   libDir: string
 ): void {
   if (libDir === 'dist') {