feat: Add hook option to control when plugin is ran

src/build.ts

@@ -5,7 +5,8 @@ import { copyAll, outputCopyLog } from './utils'
 export const buildPlugin = ({
   targets,
   structured,
-  silent
+  silent,
+  hook
 }: ResolvedViteStaticCopyOptions): Plugin => {
   let config: ResolvedConfig
   let output = false
@@ -20,7 +21,7 @@ export const buildPlugin = ({
       // reset for watch mode
       output = false
     },
-    async writeBundle() {
+    async [hook]() {
       // run copy only once even if multiple bundles are generated
       if (output) return
       output = true

src/options.ts

@@ -105,6 +105,11 @@ export type ViteStaticCopyOptions = {
      */
     reloadPageOnChange?: boolean
   }
+ /**
+  * Rollup hook the plugin should use during build.
+  * @default 'writeBundle'
+  */
+  hook?: string
 }
 
 export type ResolvedViteStaticCopyOptions = {
@@ -115,6 +120,7 @@ export type ResolvedViteStaticCopyOptions = {
     options: WatchOptions
     reloadPageOnChange: boolean
   }
+  hook: string
 }
 
 export const resolveOptions = (
@@ -126,5 +132,6 @@ export const resolveOptions = (
   watch: {
     options: options.watch?.options ?? {},
     reloadPageOnChange: options.watch?.reloadPageOnChange ?? false
-  }
+  },
+  hook: options.hook ?? 'writeBundle'
 })

test/fixtures/vite.hook.config.ts

@@ -0,0 +1,32 @@
+import { defineConfig, normalizePath } from 'vite'
+import { viteStaticCopy } from 'vite-plugin-static-copy'
+import { promises as fs } from 'node:fs'
+import path from 'node:path'
+import url from 'node:url'
+
+const _dirname = path.dirname(url.fileURLToPath(import.meta.url))
+
+export default defineConfig({
+  plugins: [
+    viteStaticCopy({
+      targets: [
+        {
+          src: 'foo.txt',
+          dest: 'hook1'
+        },
+      ],
+      hook: 'generateBundle',
+    }),
+    testHookPlugin(),
+  ]
+})
+
+function testHookPlugin() {
+  return {
+    name: 'test-hook-plugin',
+    async writeBundle() {
+      const filePath = normalizePath(path.resolve(_dirname, 'dist', 'hook1', 'foo.txt'));
+      await fs.access(filePath);
+    }
+  }
+}

test/tests.test.ts

@@ -137,6 +137,15 @@ describe('build', () => {
       }
     })
   }
+
+  describe('hook', () => {
+    test('should support hook option', async () => {
+      expect(
+        await build(getConfig('vite.hook.config.ts'))
+      ).to.not.throw
+    })
+  })
+
   describe('on error', () => {
     test('should throw error when it does not find the file on given src', async () => {
       let result = ''