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/index.js

306 lines
8.1 KiB
JavaScript
Raw Normal View History

2016-10-19 12:41:45 +00:00
import { resolve, join } from 'path'
import { parse } from 'url'
import fs from 'mz/fs'
import http, { STATUS_CODES } from 'http'
import {
renderToHTML,
renderErrorToHTML,
renderJSON,
renderErrorJSON,
sendHTML,
serveStatic,
serveStaticWithGzip
} from './render'
2016-10-05 23:52:50 +00:00
import Router from './router'
2016-10-17 07:05:46 +00:00
import HotReloader from './hot-reloader'
2016-10-19 12:41:45 +00:00
import { resolveFromList } from './resolve'
import getConfig from './config'
// We need to go up one more level since we are in the `dist` directory
import pkg from '../../package'
2016-10-05 23:52:50 +00:00
export default class Server {
constructor ({ dir = '.', dev = false, staticMarkup = false, quiet = false } = {}) {
2016-10-06 11:05:52 +00:00
this.dir = resolve(dir)
this.dev = dev
this.quiet = quiet
this.renderOpts = { dir: this.dir, dev, staticMarkup }
2016-10-05 23:52:50 +00:00
this.router = new Router()
2016-12-21 14:39:08 +00:00
this.hotReloader = dev ? new HotReloader(this.dir, { quiet }) : null
this.http = null
2016-12-27 23:27:51 +00:00
this.config = getConfig(this.dir)
this.defineRoutes()
}
2016-10-05 23:52:50 +00:00
getRequestHandler () {
return (req, res) => {
2016-10-09 09:25:38 +00:00
this.run(req, res)
.catch((err) => {
if (!this.quiet) console.error(err)
res.statusCode = 500
res.end(STATUS_CODES[500])
2016-10-05 23:52:50 +00:00
})
}
2016-10-05 23:52:50 +00:00
}
async prepare () {
2016-10-17 07:07:41 +00:00
if (this.hotReloader) {
await this.hotReloader.start()
}
this.renderOpts.buildId = await this.readBuildId()
2016-10-17 07:07:41 +00:00
}
async close () {
if (this.hotReloader) {
await this.hotReloader.stop()
}
New test setup (#640) * Use jest-cli instead of gulp plugin. * Use jest-cli instead of gulp plugin. * Move fixtures into the examples dir. * Move test code of example app to the basic example. * Add isolated tests for server/resolve * Allow tests to use cheerio. * Use portfinder to get a unique port. * Move back integration tests into the example dir. * Introduce next-test-utils. * Remove gulp-jest * Add coveralls support. * Use transpiled version of code in dist. This is to make sure same file gets covered by both unit/isolated tests and integration tests. * Add support for source maps. * Use code from dist always. * Use nyc to stop instrument. * Add integration test suite for production usage. * Use jest-cli. * Add support for running e2e tests. * Check gzipPath with fs.stat before serving Otherwise, serve package might throw issues other than ENOENT * Install chromedriver with npm install. * Install chrome on travis-ci. * Add --forceExit to Jest. * Run tests only on Node v6. That's because selenium-webdriver only supports Node 6 LTS. * Use chromedriver NPM module to install chromedriver. * Use wd as the webdriver client. * Run chromedriver before tests. * Run travis for both node 4 and 6 * Remove unwanted npm install script. * Move some common text utilities to next-test-utils * Add lint checks and testing in npm prepublish hook. * Use npm on travis-ci. We are having some caching issues with yarn and chromedriver. * Make tests work on windows.\n But chromedriver doesn't work. * Clean up dependencies. * Run chromedriver in background without any tools. * Fix a typo in the code. * Use ES6 features used in node4 inside the gulpfile. * Add some comments. * Add support for running in windows. * Stop chromedriver properly on windows. * Fix typos.
2017-01-12 04:14:49 +00:00
if (this.http) {
await new Promise((resolve, reject) => {
this.http.close((err) => {
if (err) return reject(err)
return resolve()
})
})
}
}
2016-10-17 07:07:41 +00:00
defineRoutes () {
const routes = {
'/_next-prefetcher.js': async (req, res, params) => {
const p = join(__dirname, '../client/next-prefetcher-bundle.js')
await this.serveStatic(req, res, p)
},
'/_next/:buildId/main.js': async (req, res, params) => {
this.handleBuildId(params.buildId, res)
const p = join(this.dir, '.next/main.js')
await this.serveStaticWithGzip(req, res, p)
},
'/_next/:buildId/commons.js': async (req, res, params) => {
this.handleBuildId(params.buildId, res)
const p = join(this.dir, '.next/commons.js')
await this.serveStaticWithGzip(req, res, p)
},
'/_next/:buildId/pages/:path*': async (req, res, params) => {
this.handleBuildId(params.buildId, res)
const paths = params.path || ['index']
const pathname = `/${paths.join('/')}`
await this.renderJSON(req, res, pathname)
},
'/_next/:path+': async (req, res, params) => {
const p = join(__dirname, '..', 'client', ...(params.path || []))
await this.serveStatic(req, res, p)
},
'/static/:path+': async (req, res, params) => {
const p = join(this.dir, 'static', ...(params.path || []))
await this.serveStatic(req, res, p)
},
'/:path*': async (req, res) => {
const { pathname, query } = parse(req.url, true)
await this.render(req, res, pathname, query)
}
}
2016-10-05 23:52:50 +00:00
for (const method of ['GET', 'HEAD']) {
for (const p of Object.keys(routes)) {
this.router.add(method, p, routes[p])
}
}
}
2016-10-05 23:52:50 +00:00
async start (port) {
await this.prepare()
this.http = http.createServer(this.getRequestHandler())
await new Promise((resolve, reject) => {
this.http.listen(port, (err) => {
if (err) return reject(err)
resolve()
})
2016-10-05 23:52:50 +00:00
})
}
async run (req, res) {
if (this.hotReloader) {
await this.hotReloader.run(req, res)
}
2016-10-05 23:52:50 +00:00
const fn = this.router.match(req, res)
if (fn) {
await fn()
return
}
if (req.method === 'GET' || req.method === 'HEAD') {
2016-10-05 23:52:50 +00:00
await this.render404(req, res)
} else {
res.statusCode = 501
res.end(STATUS_CODES[501])
2016-10-05 23:52:50 +00:00
}
}
async render (req, res, pathname, query) {
if (this.config.poweredByHeader) {
res.setHeader('X-Powered-By', `Next.js ${pkg.version}`)
}
const html = await this.renderToHTML(req, res, pathname, query)
sendHTML(res, html, req.method)
}
2016-10-19 12:41:45 +00:00
async renderToHTML (req, res, pathname, query) {
if (this.dev) {
const compilationErr = this.getCompilationError(pathname)
if (compilationErr) {
res.statusCode = 500
return this.renderErrorToHTML(compilationErr, req, res, pathname, query)
}
}
try {
return await renderToHTML(req, res, pathname, query, this.renderOpts)
} catch (err) {
if (err.code === 'ENOENT') {
res.statusCode = 404
return this.renderErrorToHTML(null, req, res, pathname, query)
} else {
if (!this.quiet) console.error(err)
res.statusCode = 500
return this.renderErrorToHTML(err, req, res, pathname, query)
2016-10-05 23:52:50 +00:00
}
}
}
async renderError (err, req, res, pathname, query) {
const html = await this.renderErrorToHTML(err, req, res, pathname, query)
sendHTML(res, html, req.method)
2016-10-05 23:52:50 +00:00
}
async renderErrorToHTML (err, req, res, pathname, query) {
if (this.dev) {
const compilationErr = this.getCompilationError('/_error')
if (compilationErr) {
res.statusCode = 500
return renderErrorToHTML(compilationErr, req, res, pathname, query, this.renderOpts)
}
}
2016-10-19 12:41:45 +00:00
try {
return await renderErrorToHTML(err, req, res, pathname, query, this.renderOpts)
} catch (err2) {
if (this.dev) {
if (!this.quiet) console.error(err2)
res.statusCode = 500
return renderErrorToHTML(err2, req, res, pathname, query, this.renderOpts)
} else {
throw err2
2016-10-05 23:52:50 +00:00
}
}
}
2016-10-10 04:24:30 +00:00
async render404 (req, res) {
const { pathname, query } = parse(req.url, true)
res.statusCode = 404
this.renderError(null, req, res, pathname, query)
}
async renderJSON (req, res, page) {
if (this.dev) {
const compilationErr = this.getCompilationError(page)
if (compilationErr) {
return this.renderErrorJSON(compilationErr, req, res)
}
}
2016-10-10 04:24:30 +00:00
try {
await renderJSON(req, res, page, this.renderOpts)
} catch (err) {
if (err.code === 'ENOENT') {
res.statusCode = 404
return this.renderErrorJSON(null, req, res)
} else {
if (!this.quiet) console.error(err)
res.statusCode = 500
return this.renderErrorJSON(err, req, res)
}
}
2016-10-10 04:24:30 +00:00
}
async renderErrorJSON (err, req, res) {
if (this.dev) {
const compilationErr = this.getCompilationError('/_error')
if (compilationErr) {
res.statusCode = 500
return renderErrorJSON(compilationErr, req, res, this.renderOpts)
}
}
return renderErrorJSON(err, req, res, this.renderOpts)
2016-10-05 23:52:50 +00:00
}
2016-10-19 12:41:45 +00:00
async serveStaticWithGzip (req, res, path) {
this._serveStatic(req, res, () => {
return serveStaticWithGzip(req, res, path)
})
}
serveStatic (req, res, path) {
this._serveStatic(req, res, () => {
return serveStatic(req, res, path)
})
}
async _serveStatic (req, res, fn) {
try {
await fn()
} catch (err) {
if (err.code === 'ENOENT') {
this.render404(req, res)
} else {
throw err
}
}
}
async readBuildId () {
const buildIdPath = join(this.dir, '.next', 'BUILD_ID')
try {
const buildId = await fs.readFile(buildIdPath, 'utf8')
return buildId.trim()
} catch (err) {
if (err.code === 'ENOENT') {
return '-'
} else {
throw err
}
}
}
handleBuildId (buildId, res) {
if (this.dev) return
if (buildId !== this.renderOpts.buildId) {
const errorMessage = 'Build id mismatch!' +
'Seems like the server and the client version of files are not the same.'
throw new Error(errorMessage)
}
res.setHeader('Cache-Control', 'max-age=365000000, immutable')
}
getCompilationError (page) {
2016-10-19 12:41:45 +00:00
if (!this.hotReloader) return
const errors = this.hotReloader.getCompilationErrors()
if (!errors.size) return
const id = join(this.dir, '.next', 'bundles', 'pages', page)
const p = resolveFromList(id, errors.keys())
if (p) return errors.get(p)[0]
2016-10-19 12:41:45 +00:00
}
2016-10-05 23:52:50 +00:00
}