feat: respect entryFileNames

Author: sxzz

rolldown.config.ts

@@ -3,13 +3,23 @@ import { dependencies } from './package.json'
 import { dts } from './src/index'
 
 const config = defineConfig({
-  input: './src/index.ts',
+  input: ['./src/index.ts', './src/generate.ts'],
   plugins: [dts()],
   external: Object.keys(dependencies),
   platform: 'node',
   output: [
-    { dir: 'temp/esm', format: 'es' },
-    // { dir: 'temp/cjs', format: 'cjs' },
+    {
+      dir: 'temp/esm',
+      format: 'es',
+      entryFileNames: '[name].mjs',
+      chunkFileNames: '[name]-[hash].mjs',
+    },
+    // {
+    //   dir: 'temp/cjs',
+    //   format: 'cjs',
+    //   entryFileNames: '[name].cjs',
+    //   chunkFileNames: '[name]-[hash].cjs',
+    // },
   ],
 })
 

src/generate.ts

@@ -1,11 +1,12 @@
-import { basename } from 'node:path'
+import { basename, extname } from 'node:path'
 import { createResolver } from 'dts-resolver'
 import { isolatedDeclaration as oxcIsolatedDeclaration } from 'oxc-transform'
 import {
   filename_dts_to,
   filename_ts_to_dts,
   isRelative,
   RE_DTS,
+  RE_JS,
   RE_NODE_MODULES,
   RE_TS,
 } from './utils/filename'
@@ -16,11 +17,19 @@ const meta = { dtsFile: true } as const
 
 export function createGeneratePlugin({
   isolatedDeclaration,
-  inputAlias = {},
+  inputAlias,
   resolve = false,
 }: Pick<Options, 'isolatedDeclaration' | 'inputAlias' | 'resolve'>): Plugin {
-  const dtsMap = new Map<string, string>()
-  const inputAliasMap = new Map<string, string>(Object.entries(inputAlias))
+  const dtsMap = new Map<
+    string,
+    {
+      code: string
+      src: string
+    }
+  >()
+  const inputAliasMap = new Map<string, string>(
+    inputAlias && Object.entries(inputAlias),
+  )
   const resolver = createResolver()
 
   let inputOption: Record<string, string> | undefined
@@ -33,6 +42,22 @@ export function createGeneratePlugin({
       }
     },
 
+    outputOptions(options) {
+      return {
+        ...options,
+        entryFileNames(chunk) {
+          const original =
+            (typeof options.entryFileNames === 'function'
+              ? options.entryFileNames(chunk)
+              : options.entryFileNames) || '[name].js'
+          if (chunk.name.endsWith('.d')) {
+            return original.replace(RE_JS, '.$1ts')
+          }
+          return original
+        },
+      }
+    },
+
     transform: {
       order: 'pre',
       filter: {
@@ -47,27 +72,26 @@ export function createGeneratePlugin({
           code,
           isolatedDeclaration,
         )
-        if (errors.length) {
-          return this.error(errors[0])
-        }
+        if (errors.length) return this.error(errors[0])
 
         const dtsId = filename_ts_to_dts(id)
-        dtsMap.set(dtsId, dtsCode)
+        dtsMap.set(dtsId, {
+          code: dtsCode,
+          src: id,
+        })
 
         const mod = this.getModuleInfo(id)
         if (mod?.isEntry) {
-          let fileName = basename(dtsId)
-
-          if (inputAliasMap.has(fileName)) {
-            fileName = inputAliasMap.get(fileName)!
+          let name: string | undefined = basename(dtsId, extname(dtsId))
+          if (inputAliasMap.has(name)) {
+            name = inputAliasMap.get(name)!
           } else if (inputAliasMap.has(dtsId)) {
-            fileName = inputAliasMap.get(dtsId)!
+            name = inputAliasMap.get(dtsId)!
           }
-
           this.emitFile({
             type: 'chunk',
             id: dtsId,
-            fileName,
+            name,
           })
         }
       },
@@ -146,7 +170,7 @@ export function createGeneratePlugin({
       handler(id) {
         if (dtsMap.has(id)) {
           return {
-            code: dtsMap.get(id)!,
+            code: dtsMap.get(id)!.code,
             moduleSideEffects: false,
           }
         }

src/types.d.ts

@@ -0,0 +1,7 @@
+declare module 'rolldown' {
+  export interface CustomPluginOptions {
+    dtsFile?: boolean
+  }
+}
+
+export {}