Skip to content

Commit

Permalink
Use assetEmitted hook instead of afterEmit (vercel#14626)
Browse files Browse the repository at this point in the history
Makes sure this works correctly with webpack 5.
  • Loading branch information
timneutkens committed Jun 27, 2020
1 parent bd24b70 commit b4b68c0
Showing 1 changed file with 45 additions and 17 deletions.
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import { Compiler, Plugin } from 'webpack'
import { Compiler, Plugin, version } from 'webpack'
import { realpathSync } from 'fs'

const isWebpack5 = parseInt(version!) === 5

function deleteCache(path: string) {
try {
delete require.cache[realpathSync(path)]
Expand All @@ -11,30 +13,56 @@ function deleteCache(path: string) {
}
}

const PLUGIN_NAME = 'NextJsRequireCacheHotReloader'

// This plugin flushes require.cache after emitting the files. Providing 'hot reloading' of server files.
export class NextJsRequireCacheHotReloader implements Plugin {
prevAssets: any = null
previousOutputPathsWebpack5: Set<string> = new Set()
currentOutputPathsWebpack5: Set<string> = new Set()

apply(compiler: Compiler) {
compiler.hooks.afterEmit.tapAsync(
'NextJsRequireCacheHotReloader',
(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)
}
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, () => {
for (const path of this.previousOutputPathsWebpack5) {
if (!this.currentOutputPathsWebpack5.has(path)) {
deleteCache(path)
}
}
this.prevAssets = assets

callback()
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()
})
}
}

0 comments on commit b4b68c0

Please sign in to comment.