feat(exports): add exclude option (#340)

TheAlexLichter · sxzz · web-flow · commit 9e6341a12162 · 2025-12-12T11:21:11.000+08:00
Co-authored-by: Kevin Deng &lt;sxzz@sxzz.moe&gt;
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/features/pkg/exports.test.ts b/src/features/pkg/exports.test.ts
@@ -28,7 +28,7 @@ describe.concurrent('generateExports', () => {
   test('only one entry', async ({ expect }) => {
     const results = generateExports(
       FAKE_PACKAGE_JSON,
-      { es: [genChunk('main.js')] },
+      { es: [genChunk('main.js'), genChunk('chunk.js', false)] },
       {},
     )
     await expect(results).resolves.toMatchInlineSnapshot(`
@@ -276,6 +276,72 @@ describe.concurrent('generateExports', () => {
     `)
   })
 
+  test('exclude via regex', async ({ expect }) => {
+    const results = generateExports(
+      FAKE_PACKAGE_JSON,
+      { es: [genChunk('index.js'), genChunk('foo.js'), genChunk('bar.js')] },
+      { exclude: [/bar/] },
+    )
+
+    await expect(results).resolves.toMatchInlineSnapshot(`
+      {
+        "exports": {
+          ".": "./index.js",
+          "./foo": "./foo.js",
+          "./package.json": "./package.json",
+        },
+        "main": "./index.js",
+        "module": "./index.js",
+        "publishExports": undefined,
+        "types": undefined,
+      }
+    `)
+  })
+
+  test('exclude via filename', async ({ expect }) => {
+    const results = generateExports(
+      FAKE_PACKAGE_JSON,
+      {
+        es: [genChunk('index.js'), genChunk('foo.js'), genChunk('abc/bar.js')],
+      },
+      { exclude: ['abc/bar.js'] },
+    )
+
+    await expect(results).resolves.toMatchInlineSnapshot(`
+      {
+        "exports": {
+          ".": "./index.js",
+          "./foo": "./foo.js",
+          "./package.json": "./package.json",
+        },
+        "main": "./index.js",
+        "module": "./index.js",
+        "publishExports": undefined,
+        "types": undefined,
+      }
+    `)
+  })
+
+  test('multiple excludes', async ({ expect }) => {
+    const results = generateExports(
+      FAKE_PACKAGE_JSON,
+      { es: [genChunk('foo.js'), genChunk('abc/bar.js')] },
+      { exclude: ['abc/bar.js', /foo/] },
+    )
+
+    await expect(results).resolves.toMatchInlineSnapshot(`
+      {
+        "exports": {
+          "./package.json": "./package.json",
+        },
+        "main": undefined,
+        "module": undefined,
+        "publishExports": undefined,
+        "types": undefined,
+      }
+    `)
+  })
+
   test('export all', async ({ expect }) => {
     const results = generateExports(
       FAKE_PACKAGE_JSON,
@@ -381,12 +447,12 @@ describe.concurrent('generateExports', () => {
   })
 })
 
-function genChunk(fileName: string) {
+function genChunk(fileName: string, isEntry = true) {
   // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
   return {
     type: 'chunk',
     fileName,
-    isEntry: true,
+    isEntry,
     facadeModuleId: `./SRC/${fileName}`,
     outDir: cwd,
   } as RolldownChunk
diff --git a/src/features/pkg/exports.ts b/src/features/pkg/exports.ts
@@ -2,9 +2,13 @@ import { readFile, writeFile } from 'node:fs/promises'
 import path from 'node:path'
 import { RE_DTS } from 'rolldown-plugin-dts/filename'
 import { detectIndentation } from '../../utils/format.ts'
-import { slash } from '../../utils/general.ts'
+import { matchPattern, slash } from '../../utils/general.ts'
 import type { NormalizedFormat, ResolvedConfig } from '../../config/types.ts'
-import type { ChunksByFormat, RolldownChunk } from '../../utils/chunks.ts'
+import type {
+  ChunksByFormat,
+  RolldownChunk,
+  RolldownCodeChunk,
+} from '../../utils/chunks.ts'
 import type { Awaitable } from '../../utils/types.ts'
 import type { PackageJson } from 'pkg-types'
 
@@ -21,6 +25,18 @@ export interface ExportsOptions {
    */
   all?: boolean
 
+  /**
+   * Define filenames or RegExp patterns to exclude files from exports.
+   * This is useful for excluding files that should not be part of the package exports,
+   * such as bin files or internal utilities.
+   *
+   * @example
+   * ```js
+   * exclude: ['foo.ts', /\.spec\.ts$/, /internal/]
+   * ```
+   */
+  exclude?: (RegExp | string)[]
+
   customExports?: (
     exports: Record<string, any>,
     context: {
@@ -65,10 +81,18 @@ export async function writeExports(
 
 type SubExport = Partial<Record<'cjs' | 'es' | 'src', string>>
 
+function shouldExclude(
+  fileName: string,
+  exclude?: (RegExp | string)[],
+): boolean {
+  if (!exclude?.length) return false
+  return matchPattern(fileName, exclude)
+}
+
 export async function generateExports(
   pkg: PackageJson,
   chunks: ChunksByFormat,
-  { devExports, all, customExports }: ExportsOptions,
+  { devExports, all, exclude, customExports }: ExportsOptions,
 ): Promise<{
   main: string | undefined
   module: string | undefined
@@ -90,16 +114,19 @@ export async function generateExports(
   ][]) {
     if (format !== 'es' && format !== 'cjs') continue
 
+    // Filter out non-entry chunks and excluded files
+    const filteredChunks = chunksByFormat.filter(
+      (chunk): chunk is RolldownCodeChunk =>
+        chunk.type === 'chunk' &&
+        chunk.isEntry &&
+        !shouldExclude(chunk.fileName, exclude),
+    )
+
     const onlyOneEntry =
-      chunksByFormat.filter(
-        (chunk) =>
-          chunk.type === 'chunk' &&
-          chunk.isEntry &&
-          !RE_DTS.test(chunk.fileName),
-      ).length === 1
-    for (const chunk of chunksByFormat) {
-      if (chunk.type !== 'chunk' || !chunk.isEntry) continue
+      filteredChunks.filter((chunk) => !RE_DTS.test(chunk.fileName)).length ===
+      1
 
+    for (const chunk of filteredChunks) {
       const normalizedName = slash(chunk.fileName)
       const ext = path.extname(chunk.fileName)
       let name = normalizedName.slice(0, -ext.length)
diff --git a/src/utils/chunks.ts b/src/utils/chunks.ts
@@ -2,6 +2,7 @@ import type { NormalizedFormat, ResolvedConfig } from '../config/types.ts'
 import type { OutputAsset, OutputChunk } from 'rolldown'
 
 export type RolldownChunk = (OutputChunk | OutputAsset) & { outDir: string }
+export type RolldownCodeChunk = RolldownChunk & { type: 'chunk' }
 export type ChunksByFormat = Partial<Record<NormalizedFormat, RolldownChunk[]>>
 export interface TsdownBundle extends AsyncDisposable {
   chunks: RolldownChunk[]
