mirror of https://github.com/vuejs/vitepress
Daniel Kreiseder
4 years ago
committed by
GitHub
commit
8a8bcdfacb
@ -0,0 +1,4 @@
|
|||||||
|
// re-export vite client types
|
||||||
|
// with strict installers like pnpm, user won't be able to reference vite/client
|
||||||
|
// in project root
|
||||||
|
/// <reference types="vite/client" />
|
||||||
@ -0,0 +1,126 @@
|
|||||||
|
// TODO figure out why it causes full page reload
|
||||||
|
|
||||||
|
import { Plugin, ViteDevServer, loadConfigFromFile, normalizePath } from 'vite'
|
||||||
|
import { dirname, relative } from 'path'
|
||||||
|
import { isMatch } from 'micromatch'
|
||||||
|
|
||||||
|
const loaderMatch = /\.data\.(j|t)s$/
|
||||||
|
|
||||||
|
let server: ViteDevServer
|
||||||
|
|
||||||
|
interface LoaderModule {
|
||||||
|
watch: string[] | string | undefined
|
||||||
|
load: () => any
|
||||||
|
}
|
||||||
|
|
||||||
|
interface CachedLoaderModule {
|
||||||
|
base: string
|
||||||
|
pattern: string[] | undefined
|
||||||
|
loader: () => any
|
||||||
|
}
|
||||||
|
|
||||||
|
const idToLoaderModulesMap: Record<string, CachedLoaderModule | undefined> =
|
||||||
|
Object.create(null)
|
||||||
|
|
||||||
|
// During build, the load hook will be called on the same file twice
|
||||||
|
// once for client and once for server build. Not only is this wasteful, it
|
||||||
|
// also leads to a race condition in loadConfigFromFile() that results in an
|
||||||
|
// fs unlink error. So we reuse the same Promise during build to avoid double
|
||||||
|
// loading.
|
||||||
|
let idToPendingPromiseMap: Record<string, Promise<string> | undefined> =
|
||||||
|
Object.create(null)
|
||||||
|
let isBuild = false
|
||||||
|
|
||||||
|
export const staticDataPlugin: Plugin = {
|
||||||
|
name: 'vitepress:data',
|
||||||
|
|
||||||
|
configResolved(config) {
|
||||||
|
isBuild = config.command === 'build'
|
||||||
|
},
|
||||||
|
|
||||||
|
configureServer(_server) {
|
||||||
|
server = _server
|
||||||
|
},
|
||||||
|
|
||||||
|
async load(id) {
|
||||||
|
if (loaderMatch.test(id)) {
|
||||||
|
let _resolve: ((res: any) => void) | undefined
|
||||||
|
if (isBuild) {
|
||||||
|
if (idToPendingPromiseMap[id]) {
|
||||||
|
return idToPendingPromiseMap[id]
|
||||||
|
}
|
||||||
|
idToPendingPromiseMap[id] = new Promise((r) => {
|
||||||
|
_resolve = r
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const base = dirname(id)
|
||||||
|
let pattern: string[] | undefined
|
||||||
|
let loader: () => any
|
||||||
|
|
||||||
|
const existing = idToLoaderModulesMap[id]
|
||||||
|
if (existing) {
|
||||||
|
;({ pattern, loader } = existing)
|
||||||
|
} else {
|
||||||
|
// use vite's load config util as a away to load Node.js file with
|
||||||
|
// TS & native ESM support
|
||||||
|
const loaderModule = (await loadConfigFromFile({} as any, id))
|
||||||
|
?.config as LoaderModule
|
||||||
|
pattern =
|
||||||
|
typeof loaderModule.watch === 'string'
|
||||||
|
? [loaderModule.watch]
|
||||||
|
: loaderModule.watch
|
||||||
|
if (pattern) {
|
||||||
|
pattern = pattern.map((p) => {
|
||||||
|
return p.startsWith('./') ? p.slice(2) : p
|
||||||
|
})
|
||||||
|
}
|
||||||
|
loader = loaderModule.load
|
||||||
|
}
|
||||||
|
|
||||||
|
// load the data
|
||||||
|
const data = await loader()
|
||||||
|
|
||||||
|
// record loader module for HMR
|
||||||
|
if (server) {
|
||||||
|
idToLoaderModulesMap[id] = { base, pattern, loader }
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = `export const data = JSON.parse(${JSON.stringify(
|
||||||
|
JSON.stringify(data)
|
||||||
|
)})`
|
||||||
|
|
||||||
|
if (_resolve) _resolve(result)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
transform(_code, id) {
|
||||||
|
if (server && loaderMatch.test(id)) {
|
||||||
|
// register this module as a glob importer
|
||||||
|
const { base, pattern } = idToLoaderModulesMap[id]!
|
||||||
|
;(server as any)._globImporters[id] = {
|
||||||
|
module: server.moduleGraph.getModuleById(id),
|
||||||
|
importGlobs: pattern?.map((pattern) => ({ base, pattern }))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
},
|
||||||
|
|
||||||
|
handleHotUpdate(ctx) {
|
||||||
|
for (const id in idToLoaderModulesMap) {
|
||||||
|
const { base, pattern } = idToLoaderModulesMap[id]!
|
||||||
|
const isLoaderFile = normalizePath(ctx.file) === id
|
||||||
|
if (isLoaderFile) {
|
||||||
|
// invalidate loader file
|
||||||
|
delete idToLoaderModulesMap[id]
|
||||||
|
}
|
||||||
|
if (
|
||||||
|
isLoaderFile ||
|
||||||
|
(pattern && isMatch(relative(base, ctx.file), pattern))
|
||||||
|
) {
|
||||||
|
ctx.modules.push(server.moduleGraph.getModuleById(id)!)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue