2020-06-23 17:47:50 +02:00
|
|
|
import {
|
|
|
|
Compiler,
|
|
|
|
compilation as CompilationType,
|
|
|
|
Plugin,
|
|
|
|
version,
|
|
|
|
} from 'webpack'
|
2020-06-14 14:49:46 +02:00
|
|
|
import { STRING_LITERAL_DROP_BUNDLE } from '../../../next-server/lib/constants'
|
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
const isWebpack5 = parseInt(version!) === 5
|
|
|
|
|
2020-06-14 14:49:46 +02:00
|
|
|
export const ampFirstEntryNamesMap: WeakMap<
|
|
|
|
CompilationType.Compilation,
|
|
|
|
string[]
|
|
|
|
> = new WeakMap()
|
|
|
|
|
|
|
|
const PLUGIN_NAME = 'DropAmpFirstPagesPlugin'
|
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
// Prevents outputting client pages when they are not needed
|
|
|
|
export class DropClientPage implements Plugin {
|
|
|
|
ampPages = new Set()
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
apply(compiler: Compiler) {
|
|
|
|
compiler.hooks.compilation.tap(
|
|
|
|
PLUGIN_NAME,
|
2020-07-15 19:56:27 +02:00
|
|
|
(compilation: any, { normalModuleFactory }: any) => {
|
2020-06-23 17:47:50 +02:00
|
|
|
// Recursively look up the issuer till it ends up at the root
|
|
|
|
function findEntryModule(mod: any): CompilationType.Module | null {
|
|
|
|
const queue = new Set([mod])
|
|
|
|
for (const module of queue) {
|
|
|
|
if (isWebpack5) {
|
|
|
|
// @ts-ignore TODO: webpack 5 types
|
|
|
|
const incomingConnections = compilation.moduleGraph.getIncomingConnections(
|
|
|
|
module
|
|
|
|
)
|
|
|
|
|
|
|
|
for (const incomingConnection of incomingConnections) {
|
|
|
|
if (!incomingConnection.originModule) return module
|
|
|
|
queue.add(incomingConnection.originModule)
|
|
|
|
}
|
|
|
|
continue
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
for (const reason of module.reasons) {
|
|
|
|
if (!reason.module) return module
|
|
|
|
queue.add(reason.module)
|
|
|
|
}
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
return null
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
function handler(parser: any) {
|
|
|
|
function markAsAmpFirst() {
|
|
|
|
const entryModule = findEntryModule(parser.state.module)
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
if (!entryModule) {
|
|
|
|
return
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
// @ts-ignore buildInfo exists on Module
|
|
|
|
entryModule.buildInfo.NEXT_ampFirst = true
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
if (isWebpack5) {
|
|
|
|
parser.hooks.preDeclarator.tap(PLUGIN_NAME, (declarator: any) => {
|
|
|
|
if (declarator?.id?.name === STRING_LITERAL_DROP_BUNDLE) {
|
|
|
|
markAsAmpFirst()
|
|
|
|
}
|
|
|
|
})
|
|
|
|
return
|
|
|
|
}
|
2019-04-21 23:25:17 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
parser.hooks.varDeclaration
|
|
|
|
.for(STRING_LITERAL_DROP_BUNDLE)
|
|
|
|
.tap(PLUGIN_NAME, markAsAmpFirst)
|
|
|
|
}
|
2019-04-24 10:48:43 +02:00
|
|
|
|
2020-06-14 14:49:46 +02:00
|
|
|
normalModuleFactory.hooks.parser
|
|
|
|
.for('javascript/auto')
|
|
|
|
.tap(PLUGIN_NAME, handler)
|
2019-04-21 23:25:17 +02:00
|
|
|
|
2020-06-23 17:47:50 +02:00
|
|
|
normalModuleFactory.hooks.parser
|
|
|
|
.for('javascript/esm')
|
|
|
|
.tap(PLUGIN_NAME, handler)
|
|
|
|
|
|
|
|
normalModuleFactory.hooks.parser
|
|
|
|
.for('javascript/dynamic')
|
|
|
|
.tap(PLUGIN_NAME, handler)
|
|
|
|
|
2020-06-14 14:49:46 +02:00
|
|
|
if (!ampFirstEntryNamesMap.has(compilation)) {
|
|
|
|
ampFirstEntryNamesMap.set(compilation, [])
|
|
|
|
}
|
2019-04-24 10:48:43 +02:00
|
|
|
|
2020-06-14 14:49:46 +02:00
|
|
|
const ampFirstEntryNamesItem = ampFirstEntryNamesMap.get(
|
|
|
|
compilation
|
|
|
|
) as string[]
|
2019-07-25 04:16:32 +02:00
|
|
|
|
2020-06-14 14:49:46 +02:00
|
|
|
compilation.hooks.seal.tap(PLUGIN_NAME, () => {
|
2020-06-23 17:47:50 +02:00
|
|
|
if (isWebpack5) {
|
|
|
|
for (const [name, entryData] of compilation.entries) {
|
|
|
|
for (const dependency of entryData.dependencies) {
|
|
|
|
// @ts-ignore TODO: webpack 5 types
|
|
|
|
const module = compilation.moduleGraph.getModule(dependency)
|
|
|
|
if (module?.buildInfo?.NEXT_ampFirst) {
|
2020-07-20 17:14:02 +02:00
|
|
|
ampFirstEntryNamesItem.push(name)
|
2020-06-23 17:47:50 +02:00
|
|
|
// @ts-ignore @types/webpack has outdated types for webpack 5
|
|
|
|
compilation.entries.delete(name)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
// Remove preparedEntrypoint that has bundle drop marker
|
|
|
|
// This will ensure webpack does not create chunks/bundles for this particular entrypoint
|
|
|
|
for (
|
|
|
|
let i = compilation._preparedEntrypoints.length - 1;
|
|
|
|
i >= 0;
|
|
|
|
i--
|
|
|
|
) {
|
|
|
|
const entrypoint = compilation._preparedEntrypoints[i]
|
|
|
|
if (entrypoint?.module?.buildInfo?.NEXT_ampFirst) {
|
|
|
|
ampFirstEntryNamesItem.push(entrypoint.name)
|
|
|
|
compilation._preparedEntrypoints.splice(i, 1)
|
|
|
|
}
|
2019-07-25 04:16:32 +02:00
|
|
|
}
|
2020-06-14 14:49:46 +02:00
|
|
|
|
|
|
|
for (let i = compilation.entries.length - 1; i >= 0; i--) {
|
|
|
|
const entryModule = compilation.entries[i]
|
|
|
|
if (entryModule?.buildInfo?.NEXT_ampFirst) {
|
|
|
|
compilation.entries.splice(i, 1)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
)
|
2019-04-21 23:25:17 +02:00
|
|
|
}
|
|
|
|
}
|