feat(better-define): support vite hmr

sxzz · sxzz · commit c7a60bfd35f3 · 2022-11-20T14:00:55.000+08:00
closes #153
diff --git a/.changeset/few-pigs-switch.md b/.changeset/few-pigs-switch.md
@@ -0,0 +1,5 @@
+---
+'@vue-macros/better-define': minor
+---
+
+support vite hmr
diff --git a/packages/better-define/src/index.ts b/packages/better-define/src/index.ts
@@ -2,8 +2,9 @@ import { existsSync } from 'node:fs'
 import { createUnplugin } from 'unplugin'
 import { createFilter } from '@rollup/pluginutils'
 import { REGEX_SETUP_SFC, REGEX_VUE_SFC } from '@vue-macros/common'
-import { setResolveTSFileIdImpl } from '@vue-macros/api'
+import { setResolveTSFileIdImpl, tsFileCache } from '@vue-macros/api'
 import { transformBetterDefine } from './core'
+import type { ModuleNode } from 'vite'
 import type { ResolveTSFileIdImpl } from '@vue-macros/api'
 import type { PluginContext } from 'rollup'
 import type { FilterPattern } from '@rollup/pluginutils'
@@ -32,6 +33,20 @@ export default createUnplugin<Options | undefined>((userOptions = {}, meta) => {
   const options = resolveOptions(userOptions)
   const filter = createFilter(options.include, options.exclude)
 
+  const referencedFiles = new Map<
+    string /* file */,
+    Set<string /* importer */>
+  >()
+
+  function collectReferencedFile(importer: string, file: string) {
+    if (!importer) return
+    if (!referencedFiles.has(file)) {
+      referencedFiles.set(file, new Set([importer]))
+    } else {
+      referencedFiles.get(file)!.add(importer)
+    }
+  }
+
   return {
     name,
     enforce: 'pre',
@@ -42,10 +57,16 @@ export default createUnplugin<Options | undefined>((userOptions = {}, meta) => {
         const resolveFn: ResolveTSFileIdImpl = async (id, importer) => {
           let resolved = (await ctx.resolve(id, importer))?.id
           if (!resolved) return
-          if (existsSync(resolved)) return resolved
+          if (existsSync(resolved)) {
+            collectReferencedFile(importer, resolved)
+            return resolved
+          }
 
           resolved = (await ctx.resolve(resolved))?.id
-          if (resolved && existsSync(resolved)) return resolved
+          if (resolved && existsSync(resolved)) {
+            collectReferencedFile(importer, resolved)
+            return resolved
+          }
         }
         setResolveTSFileIdImpl(resolveFn)
       }
@@ -68,6 +89,25 @@ export default createUnplugin<Options | undefined>((userOptions = {}, meta) => {
       configResolved(config) {
         options.isProduction = config.isProduction
       },
+
+      handleHotUpdate({ file, server }) {
+        function getAffectedModules(file: string): Set<ModuleNode> {
+          if (!referencedFiles.has(file)) return new Set([])
+          const modules = new Set<ModuleNode>([])
+          for (const importer of referencedFiles.get(file)!) {
+            const mods = server.moduleGraph.getModulesByFile(importer)
+            if (mods) mods.forEach((m) => modules.add(m))
+
+            getAffectedModules(importer).forEach((m) => modules.add(m))
+          }
+          return modules
+        }
+
+        if (tsFileCache[file]) delete tsFileCache[file]
+
+        const modules = getAffectedModules(file)
+        return Array.from(modules)
+      },
     },
   }
 })
diff --git a/playground/vue3/src/examples/better-define/another-types.ts b/playground/vue3/src/examples/better-define/another-types.ts
@@ -0,0 +1,3 @@
+export type AnotherType = {
+  baz: string
+}
diff --git a/playground/vue3/src/examples/better-define/types.ts b/playground/vue3/src/examples/better-define/types.ts
@@ -1,5 +1,7 @@
+import type { AnotherType } from './another-types'
+
 export type Str = string
-export interface BaseProps {
+export interface BaseProps extends AnotherType {
   name: Str
   age: number
 }

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +'@vue-macros/better-define': minor
 +---
++
 +support vite hmr
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+export type AnotherType = {`
	`2`	`+ baz: string`
	`3`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,7 @@`
	`1`	`+import type { AnotherType } from './another-types'`
	`2`	`+`
`1`	`3`	`export type Str = string`
`2`		`-export interface BaseProps {`
	`4`	`+export interface BaseProps extends AnotherType {`
`3`	`5`	`name: Str`
`4`	`6`	`age: number`
`5`	`7`	`}`