chore: update

Author: Timeless0911

packages/plugin-dts/src/index.ts

@@ -1,6 +1,10 @@
 import { fork } from 'node:child_process';
 import { basename, dirname, join, relative } from 'node:path';
-import { type RsbuildPlugin, logger } from '@rsbuild/core';
+import {
+  type EnvironmentContext,
+  type RsbuildPlugin,
+  logger,
+} from '@rsbuild/core';
 import { bundleDts } from './apiExtractor';
 import { emitDts } from './tsc';
 import { ensureTempDeclarationDir, loadTsconfig } from './utils';
@@ -12,6 +16,11 @@ export type pluginDtsOptions = {
   isWatch?: boolean;
 };
 
+export type DATA = {
+  options: pluginDtsOptions;
+  environments: Record<string, EnvironmentContext>;
+};
+
 export const PLUGIN_DTS_NAME = 'rsbuild:dts';
 
 // use ts compiler API to generate bundleless dts
@@ -36,17 +45,18 @@ export const pluginDts = (
 });
 
 if (process.send) {
-  process.on('message', async (data) => {
+  process.on('message', async (data: DATA) => {
     logger.info('Generating DTS...');
-    const { options: pluginOptions, environments } = data as any;
+    const { options: pluginOptions, environments } = data;
+    const { tsconfigPath, distPath, bundle, isWatch } = pluginOptions;
     const cwd = process.cwd();
-    const configPath = pluginOptions.tsconfigPath
-      ? join(cwd, pluginOptions.tsconfigPath)
+    const configPath = tsconfigPath
+      ? join(cwd, tsconfigPath)
       : join(cwd, 'tsconfig.json');
     const { options: rawCompilerOptions } = loadTsconfig(configPath);
     const rootDir = rawCompilerOptions.rootDir ?? 'src';
-    const outDir = pluginOptions.distPath
-      ? pluginOptions.distPath
+    const outDir = distPath
+      ? distPath
       : rawCompilerOptions.declarationDir || './dist';
 
     const getDeclarationDir = (bundle: boolean, distPath?: string) => {
@@ -56,12 +66,10 @@ if (process.send) {
       return distPath ? distPath : rawCompilerOptions.declarationDir;
     };
 
-    const declarationDir =
-      getDeclarationDir(pluginOptions.bundle, pluginOptions.distPath) ||
-      './dist';
+    const declarationDir = getDeclarationDir(bundle, distPath) || './dist';
     let entry = '';
 
-    if (pluginOptions.bundle === true) {
+    if (bundle === true) {
       const entrySourcePath = join(
         cwd,
         environments.esm?.config.source.entry?.main as string,
@@ -75,12 +83,12 @@ if (process.send) {
     }
 
     const onComplete = (isSuccess: boolean) => {
-      if (isSuccess && pluginOptions.bundle === true) {
+      if (isSuccess && bundle === true) {
         bundleDts({
           cwd,
           outDir,
           entry,
-          tsconfigPath: pluginOptions.tsconfigPath,
+          tsconfigPath,
         });
       }
     };
@@ -93,18 +101,19 @@ if (process.send) {
         declarationDir,
       },
       onComplete,
-      pluginOptions.isWatch,
+      isWatch,
     );
 
-    if (pluginOptions.bundle === true && !pluginOptions.isWatch) {
+    if (bundle === true && !isWatch) {
       bundleDts({
         cwd,
         outDir,
         entry,
-        tsconfigPath: pluginOptions.tsconfigPath,
+        tsconfigPath,
       });
     }
-
-    // process.exit();
+    if (!isWatch) {
+      process.exit();
+    }
   });
 }