feat: add afterBootstrap hook

fix #326

Author: qmhc

README.md

@@ -530,6 +530,13 @@ export interface PluginOptions {
    */
   declarationOnly?: boolean,
 
+  /**
+   * Hook called after the runtime is created.
+   *
+   * @default () => {}
+   */
+  afterBootstrap?: (runtime: Runtime) => MaybePromise<void>,
+
   /**
    * Hook called after diagnostic is emitted.
    *

README.zh-CN.md

@@ -547,6 +547,13 @@ export interface PluginOptions {
    */
   declarationOnly?: boolean,
 
+  /**
+   * 运行时创建完毕后的钩子
+   *
+   * @default () => {}
+   */
+  afterBootstrap?: (runtime: Runtime) => MaybePromise<void>,
+
   /**
    * 获取诊断信息后的钩子
    *

packages/unplugin-dts/src/plugin.ts

@@ -40,6 +40,7 @@ export const pluginFactory: UnpluginFactory<PluginOptions | undefined> = /* #__P
     copyDtsFiles = meta.framework !== 'vite',
     declarationOnly = false,
     strictOutput = true,
+    afterBootstrap,
     afterDiagnostic,
     beforeWriteFile,
     afterRollup,
@@ -214,6 +215,10 @@ export const pluginFactory: UnpluginFactory<PluginOptions | undefined> = /* #__P
         }
       }
 
+      if (typeof afterBootstrap === 'function') {
+        await unwrapPromise(afterBootstrap(runtime))
+      }
+
       handleDebug('create ts program')
       timeRecord += Date.now() - startTime
     },

packages/unplugin-dts/src/types.ts

@@ -6,9 +6,10 @@ import type {
   MaybePromise,
   Resolver,
   RollupConfig,
+  Runtime,
 } from './core'
 
-export type { Resolver, RollupConfig }
+export type { Resolver, RollupConfig, Runtime }
 
 export interface PluginOptions
   extends Omit<
@@ -24,6 +25,12 @@ export interface PluginOptions
    * @default false
    */
   declarationOnly?: boolean,
+  /**
+   * Hook called after the runtime is created.
+   *
+   * @default () => {}
+   */
+  afterBootstrap?: (runtime: Runtime) => MaybePromise<void>,
   /**
    * Hook called after diagnostic is emitted.
    *