rsnext/packages/next/build/webpack/plugins/nextjs-require-cache-hot-reloader.ts

83 lines
2.4 KiB
TypeScript

import { webpack } from 'next/dist/compiled/webpack/webpack'
import { isWebpack5 } from 'next/dist/compiled/webpack/webpack'
import { realpathSync } from 'fs'
import path from 'path'
function deleteCache(filePath: string) {
try {
delete require.cache[realpathSync(filePath)]
} catch (e) {
if (e.code !== 'ENOENT') throw e
} finally {
delete require.cache[filePath]
}
}
const PLUGIN_NAME = 'NextJsRequireCacheHotReloader'
// This plugin flushes require.cache after emitting the files. Providing 'hot reloading' of server files.
export class NextJsRequireCacheHotReloader implements webpack.Plugin {
prevAssets: any = null
previousOutputPathsWebpack5: Set<string> = new Set()
currentOutputPathsWebpack5: Set<string> = new Set()
apply(compiler: webpack.Compiler) {
if (isWebpack5) {
// @ts-ignored Webpack has this hooks
compiler.hooks.assetEmitted.tap(
PLUGIN_NAME,
(_file: any, { targetPath }: any) => {
this.currentOutputPathsWebpack5.add(targetPath)
deleteCache(targetPath)
}
)
compiler.hooks.afterEmit.tap(PLUGIN_NAME, (compilation) => {
const runtimeChunkPath = path.join(
compilation.outputOptions.path,
'webpack-runtime.js'
)
deleteCache(runtimeChunkPath)
// we need to make sure to clear all server entries from cache
// since they can have a stale webpack-runtime cache
// which needs to always be in-sync
const entries = [...compilation.entries.keys()].filter((entry) =>
entry.toString().startsWith('pages/')
)
entries.forEach((page) => {
const outputPath = path.join(
compilation.outputOptions.path,
page + '.js'
)
deleteCache(outputPath)
})
})
this.previousOutputPathsWebpack5 = new Set(
this.currentOutputPathsWebpack5
)
this.currentOutputPathsWebpack5.clear()
return
}
compiler.hooks.afterEmit.tapAsync(PLUGIN_NAME, (compilation, callback) => {
const { assets } = compilation
if (this.prevAssets) {
for (const f of Object.keys(assets)) {
deleteCache(assets[f].existsAt)
}
for (const f of Object.keys(this.prevAssets)) {
if (!assets[f]) {
deleteCache(this.prevAssets[f].existsAt)
}
}
}
this.prevAssets = assets
callback()
})
}
}