diff --git a/src/node/markdownToVue.ts b/src/node/markdownToVue.ts index c944e379e12b..ccb0872f35ec 100644 --- a/src/node/markdownToVue.ts +++ b/src/node/markdownToVue.ts @@ -29,8 +29,14 @@ export interface MarkdownCompileResult { includes: string[] } -export function clearCache() { - cache.clear() +export function clearCache(file?: string) { + if (!file) { + cache.clear() + return + } + + file = JSON.stringify({ file }).slice(1) + cache.find((_, key) => key.endsWith(file!) && cache.delete(key)) } export async function createMarkdownToVueRenderFn( diff --git a/src/node/plugin.ts b/src/node/plugin.ts index 8b58f238d232..a569a04fd6e3 100644 --- a/src/node/plugin.ts +++ b/src/node/plugin.ts @@ -3,6 +3,7 @@ import c from 'picocolors' import { mergeConfig, searchForWorkspaceRoot, + type ModuleNode, type Plugin, type ResolvedConfig, type Rollup, @@ -121,6 +122,7 @@ export async function createVitePressPlugin( let siteData = site let allDeadLinks: MarkdownCompileResult['deadLinks'] = [] let config: ResolvedConfig + let importerMap: Record | undefined> = {} const vitePressPlugin: Plugin = { name: 'vitepress', @@ -212,6 +214,7 @@ export async function createVitePressPlugin( allDeadLinks.push(...deadLinks) if (includes.length) { includes.forEach((i) => { + ;(importerMap[slash(i)] ??= new Set()).add(id) this.addWatchFile(i) }) } @@ -245,12 +248,13 @@ export async function createVitePressPlugin( configDeps.forEach((file) => server.watcher.add(file)) } - const onFileAddDelete = async (added: boolean, file: string) => { + const onFileAddDelete = async (added: boolean, _file: string) => { + const file = slash(_file) // restart server on theme file creation / deletion - if (themeRE.test(slash(file))) { + if (themeRE.test(file)) { siteConfig.logger.info( c.green( - `${path.relative(process.cwd(), file)} ${ + `${path.relative(process.cwd(), _file)} ${ added ? 'created' : 'deleted' }, restarting server...\n` ), @@ -267,6 +271,10 @@ export async function createVitePressPlugin( await resolvePages(siteConfig.srcDir, siteConfig.userConfig) ) } + + if (!added && importerMap[file]) { + delete importerMap[file] + } } server.watcher .on('add', onFileAddDelete.bind(null, true)) @@ -403,10 +411,27 @@ export async function createVitePressPlugin( } } + const hmrFix: Plugin = { + name: 'vitepress:hmr-fix', + async handleHotUpdate({ file, server, modules }) { + const importers = [...(importerMap[slash(file)] || [])] + if (importers.length > 0) { + return [ + ...modules, + ...importers.map((id) => { + clearCache(id) + return server.moduleGraph.getModuleById(id) + }) + ].filter(Boolean) as ModuleNode[] + } + } + } + return [ vitePressPlugin, rewritesPlugin(siteConfig), vuePlugin, + hmrFix, webFontsPlugin(siteConfig.useWebFonts), ...(userViteConfig?.plugins || []), await localSearchPlugin(siteConfig),