fix: inserted entry not through beforeWriteFile

fix #48
qmhc · Dec 29, 2021 · e16b9b1 · e16b9b1
1 parent ab451ae
commit e16b9b1
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/plugin.ts b/src/plugin.ts
@@ -350,10 +350,11 @@ export function dtsPlugin(options: PluginOptions = {}): Plugin {
       if (insertTypesEntry) {
         const pkgPath = resolve(root, 'package.json')
         const pkg = fs.existsSync(pkgPath) ? JSON.parse(await fs.readFile(pkgPath, 'utf-8')) : {}
-        const typesPath = pkg.types ? resolve(root, pkg.types) : resolve(outputDir, 'index.d.ts')
+
+        let typesPath = pkg.types ? resolve(root, pkg.types) : resolve(outputDir, 'index.d.ts')
 
         if (!fs.existsSync(typesPath)) {
-          const content =
+          let content =
             entries
               .map(entry => {
                 let filePath = normalizePath(
@@ -367,6 +368,15 @@ export function dtsPlugin(options: PluginOptions = {}): Plugin {
               })
               .join('\n') + '\n'
 
+          if (typeof beforeWriteFile === 'function') {
+            const result = beforeWriteFile(typesPath, content)
+
+            if (result && isNativeObj(result)) {
+              typesPath = result.filePath ?? typesPath
+              content = result.content ?? content
+            }
+          }
+
           await fs.writeFile(typesPath, content, 'utf-8')
         }