feat: add EntryChunkPlugin to handle shebang and shims

packages/core/src/config.ts

@@ -600,10 +600,14 @@ const composeFormatConfig = ({
   }
 };
 
+type DeepRequired<T> = Required<{
+  [K in keyof T]: T[K] extends Required<T[K]> ? T[K] : DeepRequired<T[K]>;
+}>;
+
 const composeShimsConfig = (
   format: Format,
   shims?: Shims,
-): { rsbuildConfig: RsbuildConfig; resolvedShims: Shims } => {
+): { rsbuildConfig: RsbuildConfig; enabledShims: DeepRequired<Shims> } => {
   const resolvedShims = {
     cjs: {
       'import.meta.url': shims?.cjs?.['import.meta.url'] ?? true,
@@ -615,6 +619,23 @@ const composeShimsConfig = (
     },
   };
 
+  const enabledShims = {
+    cjs:
+      format === 'cjs'
+        ? resolvedShims.cjs
+        : {
+            'import.meta.url': false,
+          },
+    esm:
+      format === 'esm'
+        ? resolvedShims.esm
+        : {
+            __filename: false,
+            __dirname: false,
+            require: false,
+          },
+  };
+
   let rsbuildConfig: RsbuildConfig = {};
   switch (format) {
     case 'esm': {
@@ -648,7 +669,7 @@ const composeShimsConfig = (
       throw new Error(`Unsupported format: ${format}`);
   }
 
-  return { rsbuildConfig, resolvedShims };
+  return { rsbuildConfig, enabledShims };
 };
 
 export const composeModuleImportWarn = (request: string): string => {
@@ -1020,7 +1041,7 @@ const composeTargetConfig = (
 const composeExternalHelpersConfig = (
   externalHelpers: boolean,
   pkgJson?: PkgJson,
-): RsbuildConfig => {
+) => {
   let defaultConfig = {
     tools: {
       swc: {
@@ -1077,7 +1098,7 @@ async function composeLibRsbuildConfig(config: LibConfig, configPath: string) {
     redirect = {},
     umdName,
   } = config;
-  const { rsbuildConfig: shimsConfig, resolvedShims } = composeShimsConfig(
+  const { rsbuildConfig: shimsConfig, enabledShims } = composeShimsConfig(
     format!,
     shims,
   );
@@ -1124,8 +1145,7 @@ async function composeLibRsbuildConfig(config: LibConfig, configPath: string) {
   );
   const cssConfig = composeCssConfig(lcp, config.bundle);
   const entryChunkConfig = composeEntryChunkConfig({
-    enabledImportMetaUrlShim:
-      format === 'cjs' && !!resolvedShims?.cjs?.['import.meta.url'],
+    enabledImportMetaUrlShim: enabledShims.cjs['import.meta.url'],
   });
   const dtsConfig = await composeDtsConfig(config, dtsExtension);
   const externalsWarnConfig = composeExternalsWarnConfig(

packages/plugin-dts/src/dts.ts

@@ -190,6 +190,7 @@ export async function generateDts(data: DtsGenOptions): Promise<void> {
       : join(cwd, entryPath);
     const relativePath = relative(rootDir, dirname(entrySourcePath));
     entry = join(declarationDir!, relativePath, basename(entrySourcePath))
+      // Remove query in file path, such as RSLIB_ENTRY_QUERY.
       .replace(/\?.*$/, '')
       .replace(
         /\.(js|mjs|jsx|ts|mts|tsx|cjs|cts|cjsx|ctsx|mjsx|mtsx)$/,

tests/scripts/shared.ts

@@ -1,6 +1,6 @@
 import assert from 'node:assert';
 import fs from 'node:fs';
-import path, { dirname, join, normalize } from 'node:path';
+import { basename, dirname, join, normalize } from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { pluginModuleFederation } from '@module-federation/rsbuild-plugin';
 import {
@@ -327,9 +327,9 @@ export function queryContent(
     basename?: boolean;
   } = {},
 ): string | null {
-  const basename = options?.basename ?? false;
+  const useBasename = options?.basename ?? false;
   const matched = Object.entries(contents).find(([key]) => {
-    const toQueried = basename ? path.basename(key) : key;
+    const toQueried = useBasename ? basename(key) : key;
     return typeof query === 'string'
       ? toQueried === query
       : query.test(toQueried);