1
0
Fork 0
mirror of https://github.com/terribleplan/next.js.git synced 2024-01-19 02:48:18 +00:00
next.js/server/render.js

261 lines
6.8 KiB
JavaScript
Raw Normal View History

2016-10-19 12:58:08 +00:00
import { join } from 'path'
import { existsSync } from 'fs'
2016-10-05 23:52:50 +00:00
import { createElement } from 'react'
import { renderToString, renderToStaticMarkup } from 'react-dom/server'
import send from 'send'
import generateETag from 'etag'
import fresh from 'fresh'
2016-10-15 19:49:42 +00:00
import requireModule from './require'
import getConfig from './config'
import resolvePath from './resolve'
import { Router } from '../lib/router'
import { loadGetInitialProps } from '../lib/utils'
2016-12-16 18:42:40 +00:00
import Head, { defaultHead } from '../lib/head'
2016-10-05 23:52:50 +00:00
import App from '../lib/app'
import ErrorDebug from '../lib/error-debug'
import { flushChunks } from '../lib/dynamic'
import xssFilters from 'xss-filters'
2016-10-05 23:52:50 +00:00
export async function render (req, res, pathname, query, opts) {
const html = await renderToHTML(req, res, pathname, query, opts)
sendHTML(req, res, html, req.method, opts)
}
export function renderToHTML (req, res, pathname, query, opts) {
return doRender(req, res, pathname, query, opts)
}
export async function renderError (err, req, res, pathname, query, opts) {
const html = await renderErrorToHTML(err, req, res, query, opts)
sendHTML(req, res, html, req.method, opts)
}
export function renderErrorToHTML (err, req, res, pathname, query, opts = {}) {
return doRender(req, res, pathname, query, { ...opts, err, page: '_error' })
}
async function doRender (req, res, pathname, query, {
err,
page,
buildId,
buildStats,
hotReloader,
assetPrefix,
availableChunks,
2016-10-10 04:24:30 +00:00
dir = process.cwd(),
dev = false,
staticMarkup = false,
nextExport = false
2016-10-10 04:24:30 +00:00
} = {}) {
page = page || pathname
await ensurePage(page, { dir, hotReloader })
const dist = getConfig(dir).distDir
2016-12-16 18:42:40 +00:00
let [Component, Document] = await Promise.all([
requireModule(join(dir, dist, 'dist', 'pages', page)),
requireModule(join(dir, dist, 'dist', 'pages', '_document'))
2016-12-16 18:42:40 +00:00
])
Component = Component.default || Component
Document = Document.default || Document
const asPath = req.url
const ctx = { err, req, res, pathname, query, asPath }
2017-04-06 05:31:44 +00:00
const props = await loadGetInitialProps(Component, ctx)
2016-10-05 23:52:50 +00:00
// the response might be finshed on the getinitialprops call
if (res.finished) return
const renderPage = (enhancer = Page => Page) => {
2016-10-06 02:42:55 +00:00
const app = createElement(App, {
Component: enhancer(Component),
2016-10-06 02:42:55 +00:00
props,
router: new Router(pathname, query, asPath)
2016-10-06 02:42:55 +00:00
})
const render = staticMarkup ? renderToStaticMarkup : renderToString
let html
let head
let errorHtml = ''
2017-10-21 23:47:55 +00:00
try {
2017-10-21 23:47:55 +00:00
if (err && dev) {
errorHtml = render(createElement(ErrorDebug, { error: err }))
} else if (err) {
errorHtml = render(app)
} else {
html = render(app)
}
} finally {
head = Head.rewind() || defaultHead()
}
const chunks = loadChunks({ dev, dir, dist, availableChunks })
return { html, head, errorHtml, chunks }
2016-12-16 18:42:40 +00:00
}
2016-10-06 02:42:55 +00:00
const docProps = await loadGetInitialProps(Document, { ...ctx, renderPage })
2016-12-16 18:42:40 +00:00
if (res.finished) return
if (!Document.prototype || !Document.prototype.isReactComponent) throw new Error('_document.js is not exporting a React element')
2016-10-05 23:52:50 +00:00
const doc = createElement(Document, {
2016-12-16 18:42:40 +00:00
__NEXT_DATA__: {
2016-10-09 09:50:41 +00:00
props,
pathname,
query,
buildId,
buildStats,
assetPrefix,
nextExport,
err: (err) ? serializeError(dev, err) : null
2016-10-09 09:50:41 +00:00
},
2016-10-10 04:24:30 +00:00
dev,
dir,
2016-10-28 14:38:24 +00:00
staticMarkup,
2016-12-16 18:42:40 +00:00
...docProps
2016-10-05 23:52:50 +00:00
})
return '<!DOCTYPE html>' + renderToStaticMarkup(doc)
}
export async function renderScript (req, res, page, opts) {
try {
const dist = getConfig(opts.dir).distDir
const path = join(opts.dir, dist, 'bundles', 'pages', page)
const realPath = await resolvePath(path)
await serveStatic(req, res, realPath)
} catch (err) {
if (err.code === 'ENOENT') {
2017-04-04 20:35:25 +00:00
renderScriptError(req, res, page, err, {}, opts)
return
}
throw err
}
}
export async function renderScriptError (req, res, page, error, customFields, { dev }) {
// Asks CDNs and others to not to cache the errored page
res.setHeader('Cache-Control', 'no-store, must-revalidate')
// prevent XSS attacks by filtering the page before printing it.
page = xssFilters.uriInSingleQuotedAttr(page)
res.setHeader('Content-Type', 'text/javascript')
2017-04-04 20:35:25 +00:00
if (error.code === 'ENOENT') {
res.end(`
window.__NEXT_REGISTER_PAGE('${page}', function() {
2017-04-12 16:12:04 +00:00
var error = new Error('Page does not exist: ${page}')
2017-04-05 11:43:34 +00:00
error.statusCode = 404
return { error: error }
})
2017-04-04 20:35:25 +00:00
`)
return
}
const errorJson = {
...serializeError(dev, error),
...customFields
}
res.end(`
window.__NEXT_REGISTER_PAGE('${page}', function() {
2017-04-05 11:43:34 +00:00
var error = ${JSON.stringify(errorJson)}
return { error: error }
})
`)
}
export function sendHTML (req, res, html, method, { dev }) {
if (res.finished) return
const etag = generateETag(html)
if (fresh(req.headers, { etag })) {
res.statusCode = 304
res.end()
return
}
if (dev) {
// In dev, we should not cache pages for any reason.
// That's why we do this.
res.setHeader('Cache-Control', 'no-store, must-revalidate')
}
res.setHeader('ETag', etag)
if (!res.getHeader('Content-Type')) {
res.setHeader('Content-Type', 'text/html; charset=utf-8')
}
res.setHeader('Content-Length', Buffer.byteLength(html))
res.end(method === 'HEAD' ? null : html)
}
export function sendJSON (res, obj, method) {
if (res.finished) return
const json = JSON.stringify(obj)
res.setHeader('Content-Type', 'application/json')
res.setHeader('Content-Length', Buffer.byteLength(json))
res.end(method === 'HEAD' ? null : json)
}
function errorToJSON (err) {
2016-10-19 12:41:45 +00:00
const { name, message, stack } = err
const json = { name, message, stack }
if (err.module) {
// rawRequest contains the filename of the module which has the error.
const { rawRequest } = err.module
2016-10-19 12:41:45 +00:00
json.module = { rawRequest }
}
return json
}
function serializeError (dev, err) {
if (dev) {
return errorToJSON(err)
}
return { message: '500 - Internal Server Error.' }
}
export function serveStatic (req, res, path) {
return new Promise((resolve, reject) => {
send(req, path)
.on('directory', () => {
// We don't allow directories to be read.
const err = new Error('No directory access')
err.code = 'ENOENT'
reject(err)
})
.on('error', reject)
.pipe(res)
.on('finish', resolve)
})
}
async function ensurePage (page, { dir, hotReloader }) {
if (!hotReloader) return
if (page === '_error' || page === '_document') return
await hotReloader.ensurePage(page)
}
function loadChunks ({ dev, dir, dist, availableChunks }) {
const flushedChunks = flushChunks()
const validChunks = []
for (var chunk of flushedChunks) {
const filename = join(dir, dist, 'chunks', chunk)
const exists = dev ? existsSync(filename) : availableChunks[chunk]
if (exists) {
validChunks.push(chunk)
}
}
return validChunks
}