feat: support frok process

Author: Timeless0911

packages/plugin-dts/modern.config.ts

@@ -15,6 +15,7 @@ export default defineConfig({
       autoExtension: true,
       externals,
       dts: false,
+      shims: true,
       define,
     },
     {
@@ -24,6 +25,7 @@ export default defineConfig({
       autoExtension: true,
       externals,
       dts: false,
+      shims: true,
       define,
     },
     {

packages/plugin-dts/src/index.ts

@@ -1,5 +1,6 @@
+import { fork } from 'node:child_process';
 import { basename, dirname, join, relative } from 'node:path';
-import type { RsbuildPlugin } from '@rsbuild/core';
+import { type RsbuildPlugin, logger } from '@rsbuild/core';
 import { bundleDts } from './apiExtractor';
 import { emitDts } from './tsc';
 import { ensureTempDeclarationDir, loadTsconfig } from './utils';
@@ -24,78 +25,86 @@ export const pluginDts = (
   name: PLUGIN_DTS_NAME,
 
   setup(api) {
-    const { tsconfigPath, distPath, bundle, isWatch } = options;
-    let isFirstCompile = true;
-
     api.onBeforeBuild(async ({ environments }) => {
-      const cwd = process.cwd();
-      const configPath = tsconfigPath
-        ? join(cwd, tsconfigPath)
-        : join(cwd, 'tsconfig.json');
-      const { options: rawCompilerOptions } = loadTsconfig(configPath);
-      const rootDir = rawCompilerOptions.rootDir ?? 'src';
-      const outDir = distPath
-        ? distPath
-        : rawCompilerOptions.declarationDir || './dist';
+      const childProcess = fork(__filename, [], {
+        stdio: 'inherit',
+      });
 
-      const getDeclarationDir = (bundle: boolean, distPath?: string) => {
-        if (bundle) {
-          return ensureTempDeclarationDir();
-        }
-        return distPath ? distPath : rawCompilerOptions.declarationDir;
-      };
+      childProcess.send({ options, environments });
+    });
+  },
+});
 
-      const declarationDir = getDeclarationDir(bundle, distPath) || './dist';
-      let entry = '';
+if (process.send) {
+  process.on('message', async (data) => {
+    logger.info('Generating DTS...');
+    const { options: pluginOptions, environments } = data as any;
+    const cwd = process.cwd();
+    const configPath = pluginOptions.tsconfigPath
+      ? join(cwd, pluginOptions.tsconfigPath)
+      : join(cwd, 'tsconfig.json');
+    const { options: rawCompilerOptions } = loadTsconfig(configPath);
+    const rootDir = rawCompilerOptions.rootDir ?? 'src';
+    const outDir = pluginOptions.distPath
+      ? pluginOptions.distPath
+      : rawCompilerOptions.declarationDir || './dist';
 
-      if (options.bundle === true) {
-        const entrySourcePath = join(
-          cwd,
-          environments.esm?.config.source.entry?.main as string,
-        );
-        const relativePath = relative(rootDir, dirname(entrySourcePath));
-        entry = join(
-          declarationDir!,
-          relativePath,
-          basename(entrySourcePath),
-        ).replace(/\.(m?js|jsx?|m?ts|tsx?|c?js)$/, '.d.ts');
+    const getDeclarationDir = (bundle: boolean, distPath?: string) => {
+      if (bundle) {
+        return ensureTempDeclarationDir();
       }
+      return distPath ? distPath : rawCompilerOptions.declarationDir;
+    };
 
-      const onComplete = (isSuccess: boolean) => {
-        if (isSuccess && options.bundle === true) {
-          bundleDts({
-            cwd,
-            outDir,
-            entry,
-            tsconfigPath,
-          });
-        }
-      };
+    const declarationDir =
+      getDeclarationDir(pluginOptions.bundle, pluginOptions.distPath) ||
+      './dist';
+    let entry = '';
 
-      emitDts(
-        {
-          cwd,
-          configPath,
-          rootDir,
-          declarationDir,
-        },
-        isFirstCompile,
-        onComplete,
-        isWatch,
+    if (pluginOptions.bundle === true) {
+      const entrySourcePath = join(
+        cwd,
+        environments.esm?.config.source.entry?.main as string,
       );
+      const relativePath = relative(rootDir, dirname(entrySourcePath));
+      entry = join(
+        declarationDir!,
+        relativePath,
+        basename(entrySourcePath),
+      ).replace(/\.(m?js|jsx?|m?ts|tsx?|c?js)$/, '.d.ts');
+    }
 
-      if (options.bundle === true && !isWatch) {
+    const onComplete = (isSuccess: boolean) => {
+      if (isSuccess && pluginOptions.bundle === true) {
         bundleDts({
           cwd,
           outDir,
           entry,
-          tsconfigPath,
+          tsconfigPath: pluginOptions.tsconfigPath,
         });
       }
+    };
 
-      if (isFirstCompile) {
-        isFirstCompile = false;
-      }
-    });
-  },
-});
+    emitDts(
+      {
+        cwd,
+        configPath,
+        rootDir,
+        declarationDir,
+      },
+      onComplete,
+      pluginOptions.isWatch,
+    );
+
+    if (pluginOptions.bundle === true && !pluginOptions.isWatch) {
+      bundleDts({
+        cwd,
+        outDir,
+        entry,
+        tsconfigPath: pluginOptions.tsconfigPath,
+      });
+    }
+
+    // process.exit();
+  });
+}

packages/plugin-dts/src/tsc.ts

@@ -11,7 +11,6 @@ export type emitDtsOptions = {
 
 export function emitDts(
   options: emitDtsOptions,
-  isFirstCompile: boolean,
   onComplete: (isSuccess: boolean) => void,
   isWatch = false,
 ) {
@@ -73,7 +72,7 @@ export function emitDts(
     }
 
     logger.info('TypeScript compilation succeeded\n');
-  } else if (isFirstCompile) {
+  } else {
     const createProgram = ts.createSemanticDiagnosticsBuilderProgram;
     const formatHost: ts.FormatDiagnosticsHost = {
       getCanonicalFileName: (path) => path,