fix(dts-plugin): ensure kill fork child process when the task is finished (#2639)

Author: 2heal1

.changeset/green-pants-sip.md

@@ -0,0 +1,5 @@
+---
+'@module-federation/dts-plugin': patch
+---
+
+fix(dts-plugin): ensure kill fork child process when the task is finished

packages/dts-plugin/src/core/lib/DtsWorker.ts

@@ -5,6 +5,7 @@ import { type RpcWorker, createRpcWorker } from '../rpc/index';
 import type { RpcMethod } from '../rpc/types';
 import type { DTSManagerOptions } from '../interfaces/DTSManagerOptions';
 import type { DTSManager } from './DTSManager';
+import { isDebugMode } from './utils';
 
 export type DtsWorkerOptions = DTSManagerOptions;
 
@@ -41,9 +42,27 @@ export class DtsWorker {
   }
 
   get controlledPromise(): ReturnType<DTSManager['generateTypes']> {
-    return Promise.resolve(this._res).then(() => {
-      this.exit();
-    });
+    const ensureChildProcessExit = () => {
+      try {
+        const pid = this.rpcWorker.process?.pid;
+        process.kill(pid, 0);
+      } catch (error) {
+        if (isDebugMode()) {
+          console.error(error);
+        }
+      }
+    };
+    return Promise.resolve(this._res)
+      .then(() => {
+        this.exit();
+        ensureChildProcessExit();
+      })
+      .catch((err) => {
+        if (isDebugMode()) {
+          console.error(err);
+        }
+        ensureChildProcessExit();
+      });
   }
 
   exit(): void {

packages/dts-plugin/src/core/lib/utils.ts

@@ -69,7 +69,10 @@ export function retrieveTypesAssetsInfo(options: RemoteOptions) {
 }
 
 export function isDebugMode() {
-  return Boolean(process.env['FEDERATION_DEBUG']);
+  return (
+    Boolean(process.env['FEDERATION_DEBUG']) ||
+    process.env['NODE_ENV'] === 'test'
+  );
 }
 
 export const isTSProject = (