feat: support watch mode

Author: antfu

package.json

@@ -61,6 +61,7 @@
   },
   "dependencies": {
     "@rollup/pluginutils": "^4.1.1",
+    "chokidar": "^3.5.2",
     "debug": "^4.3.2",
     "fast-glob": "^3.2.7",
     "has-pkg": "^0.0.1",

pnpm-lock.yaml

@@ -1,4 +1,5 @@
 import { relative } from 'path'
+import fs from 'fs'
 import Debug from 'debug'
 import { UpdatePayload, ViteDevServer } from 'vite'
 import { throttle, toArray, slash } from '@antfu/utils'
@@ -60,19 +61,25 @@ export class Context {
     return this.transformer(code, id, path, query)
   }
 
-  setViteServer(server: ViteDevServer) {
+  setupViteServer(server: ViteDevServer) {
+    if (this._server === server)
+      return
+
     this._server = server
+    this.setupWather(server.watcher)
+  }
 
+  setupWather(watcher: fs.FSWatcher) {
     const { globs } = this.options
 
-    server.watcher
+    watcher
       .on('unlink', (path) => {
         if (!matchGlobs(path, globs))
           return
         this.removeComponents(path)
         this.onUpdate(path)
       })
-    server.watcher
+    watcher
       .on('add', (path) => {
         if (!matchGlobs(path, globs))
           return
@@ -125,6 +132,8 @@ export class Context {
   }
 
   onUpdate(path: string) {
+    this.generateDeclaration()
+
     if (!this._server)
       return
 
@@ -150,8 +159,6 @@ export class Context {
 
     if (payload.updates.length)
       this._server.ws.send(payload)
-
-    this.generateDeclaration()
   }
 
   private updateComponentNameMap() {

src/core/context.ts

@@ -1,5 +1,6 @@
 import { createUnplugin } from 'unplugin'
 import { createFilter } from '@rollup/pluginutils'
+import chokidar from 'chokidar'
 import { Options } from '../types'
 import { Context } from './context'
 import { shouldTransform } from './utils'
@@ -44,9 +45,12 @@ export default createUnplugin<Options>((options = {}) => {
           ctx.searchGlob()
           ctx.generateDeclaration()
         }
+
+        if (config.build.watch && config.command === 'build')
+          ctx.setupWather(chokidar.watch(ctx.options.globs))
       },
       configureServer(server) {
-        ctx.setViteServer(server)
+        ctx.setupViteServer(server)
       },
     },
   }