130 lines
4 KiB
JavaScript
130 lines
4 KiB
JavaScript
import mkdirpModule from 'mkdirp'
|
|
import { promisify } from 'util'
|
|
import { extname, join, dirname, sep } from 'path'
|
|
import { renderToHTML } from 'next-server/dist/server/render'
|
|
import { writeFile, access } from 'fs'
|
|
import Sema from 'async-sema'
|
|
import AmpHtmlValidator from 'amphtml-validator'
|
|
import { loadComponents } from 'next-server/dist/server/load-components'
|
|
|
|
const envConfig = require('next-server/config')
|
|
const mkdirp = promisify(mkdirpModule)
|
|
const writeFileP = promisify(writeFile)
|
|
const accessP = promisify(access)
|
|
|
|
global.__NEXT_DATA__ = {
|
|
nextExport: true
|
|
}
|
|
|
|
process.on(
|
|
'message',
|
|
async ({
|
|
distDir,
|
|
buildId,
|
|
exportPaths,
|
|
exportPathMap,
|
|
outDir,
|
|
renderOpts,
|
|
serverRuntimeConfig,
|
|
concurrency
|
|
}) => {
|
|
const sema = new Sema(concurrency, { capacity: exportPaths.length })
|
|
try {
|
|
const work = async path => {
|
|
await sema.acquire()
|
|
const ampPath = `${path === '/' ? '/index' : path}.amp`
|
|
const { page, query = {} } = exportPathMap[path]
|
|
delete query.ampOnly
|
|
delete query.hasAmp
|
|
delete query.ampPath
|
|
delete query.amphtml
|
|
|
|
const req = { url: path }
|
|
const res = {}
|
|
envConfig.setConfig({
|
|
serverRuntimeConfig,
|
|
publicRuntimeConfig: renderOpts.runtimeConfig
|
|
})
|
|
|
|
let htmlFilename = `${path}${sep}index.html`
|
|
const pageExt = extname(page)
|
|
const pathExt = extname(path)
|
|
// Make sure page isn't a folder with a dot in the name e.g. `v1.2`
|
|
if (pageExt !== pathExt && pathExt !== '') {
|
|
// If the path has an extension, use that as the filename instead
|
|
htmlFilename = path
|
|
} else if (path === '/') {
|
|
// If the path is the root, just use index.html
|
|
htmlFilename = 'index.html'
|
|
}
|
|
const baseDir = join(outDir, dirname(htmlFilename))
|
|
const htmlFilepath = join(outDir, htmlFilename)
|
|
|
|
await mkdirp(baseDir)
|
|
const components = await loadComponents(distDir, buildId, page)
|
|
const curRenderOpts = { ...components, ...renderOpts, ampPath }
|
|
const html = await renderToHTML(req, res, page, query, curRenderOpts)
|
|
|
|
const validateAmp = async (html, page) => {
|
|
const validator = await AmpHtmlValidator.getInstance()
|
|
const result = validator.validateString(html)
|
|
const errors = result.errors.filter(e => e.severity === 'ERROR')
|
|
const warnings = result.errors.filter(e => e.severity !== 'ERROR')
|
|
|
|
if (warnings.length || errors.length) {
|
|
process.send({
|
|
type: 'amp-validation',
|
|
payload: {
|
|
page,
|
|
result: {
|
|
errors,
|
|
warnings
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
if (curRenderOpts.amphtml && query.amp) {
|
|
await validateAmp(html, path)
|
|
}
|
|
if (
|
|
(curRenderOpts.amphtml && !query.amp) || curRenderOpts.hasAmp
|
|
) {
|
|
// we need to render a clean AMP version
|
|
const ampHtmlFilename = `${ampPath}${sep}index.html`
|
|
const ampBaseDir = join(outDir, dirname(ampHtmlFilename))
|
|
const ampHtmlFilepath = join(outDir, ampHtmlFilename)
|
|
|
|
try {
|
|
await accessP(ampHtmlFilepath)
|
|
} catch (_) {
|
|
// make sure it doesn't exist from manual mapping
|
|
const ampHtml = await renderToHTML(req, res, page, { ...query, amp: 1 }, curRenderOpts)
|
|
|
|
await validateAmp(ampHtml, page + '?amp=1')
|
|
await mkdirp(ampBaseDir)
|
|
await writeFileP(
|
|
ampHtmlFilepath,
|
|
ampHtml,
|
|
'utf8'
|
|
)
|
|
}
|
|
}
|
|
|
|
await writeFileP(
|
|
htmlFilepath,
|
|
html,
|
|
'utf8'
|
|
)
|
|
process.send({ type: 'progress' })
|
|
sema.release()
|
|
}
|
|
await Promise.all(exportPaths.map(work))
|
|
process.send({ type: 'done' })
|
|
} catch (err) {
|
|
console.error(err)
|
|
process.send({ type: 'error', payload: err })
|
|
}
|
|
}
|
|
)
|