2020-07-01 16:59:18 +02:00
|
|
|
import { IncomingMessage, ServerResponse } from 'http'
|
2020-03-13 10:40:10 +01:00
|
|
|
import { isResSent } from '../lib/utils'
|
2020-11-06 03:33:14 +01:00
|
|
|
import generateETag from 'etag'
|
2020-07-01 16:59:18 +02:00
|
|
|
import fresh from 'next/dist/compiled/fresh'
|
2020-03-13 10:40:10 +01:00
|
|
|
|
2020-11-14 08:12:47 +01:00
|
|
|
type PayloadOptions =
|
|
|
|
| { private: true }
|
|
|
|
| { private: boolean; stateful: true }
|
|
|
|
| { private: boolean; stateful: false; revalidate: number | false }
|
|
|
|
|
|
|
|
export function setRevalidateHeaders(
|
|
|
|
res: ServerResponse,
|
|
|
|
options: PayloadOptions
|
|
|
|
) {
|
|
|
|
if (options.private || options.stateful) {
|
|
|
|
if (options.private || !res.hasHeader('Cache-Control')) {
|
|
|
|
res.setHeader(
|
|
|
|
'Cache-Control',
|
|
|
|
`private, no-cache, no-store, max-age=0, must-revalidate`
|
|
|
|
)
|
|
|
|
}
|
|
|
|
} else if (typeof options.revalidate === 'number') {
|
|
|
|
if (options.revalidate < 1) {
|
|
|
|
throw new Error(
|
|
|
|
`invariant: invalid Cache-Control duration provided: ${options.revalidate} < 1`
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
res.setHeader(
|
|
|
|
'Cache-Control',
|
|
|
|
`s-maxage=${options.revalidate}, stale-while-revalidate`
|
|
|
|
)
|
|
|
|
} else if (options.revalidate === false) {
|
|
|
|
res.setHeader('Cache-Control', `s-maxage=31536000, stale-while-revalidate`)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-13 10:40:10 +01:00
|
|
|
export function sendPayload(
|
2020-07-01 16:59:18 +02:00
|
|
|
req: IncomingMessage,
|
2020-03-13 10:40:10 +01:00
|
|
|
res: ServerResponse,
|
|
|
|
payload: any,
|
|
|
|
type: 'html' | 'json',
|
2020-07-27 22:19:30 +02:00
|
|
|
{
|
|
|
|
generateEtags,
|
|
|
|
poweredByHeader,
|
|
|
|
}: { generateEtags: boolean; poweredByHeader: boolean },
|
2020-11-14 08:12:47 +01:00
|
|
|
options?: PayloadOptions
|
2020-03-13 10:40:10 +01:00
|
|
|
): void {
|
|
|
|
if (isResSent(res)) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-07-27 22:19:30 +02:00
|
|
|
if (poweredByHeader && type === 'html') {
|
|
|
|
res.setHeader('X-Powered-By', 'Next.js')
|
|
|
|
}
|
|
|
|
|
2020-07-01 16:59:18 +02:00
|
|
|
const etag = generateEtags ? generateETag(payload) : undefined
|
2020-11-10 05:40:26 +01:00
|
|
|
if (sendEtagResponse(req, res, etag)) {
|
2020-07-01 16:59:18 +02:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-07-27 22:19:30 +02:00
|
|
|
if (!res.getHeader('Content-Type')) {
|
|
|
|
res.setHeader(
|
|
|
|
'Content-Type',
|
|
|
|
type === 'json' ? 'application/json' : 'text/html; charset=utf-8'
|
|
|
|
)
|
|
|
|
}
|
2020-03-13 10:40:10 +01:00
|
|
|
res.setHeader('Content-Length', Buffer.byteLength(payload))
|
|
|
|
if (options != null) {
|
2020-11-14 08:12:47 +01:00
|
|
|
setRevalidateHeaders(res, options)
|
2020-03-13 10:40:10 +01:00
|
|
|
}
|
2020-07-27 22:19:30 +02:00
|
|
|
res.end(req.method === 'HEAD' ? null : payload)
|
2020-03-13 10:40:10 +01:00
|
|
|
}
|
2020-11-10 05:40:26 +01:00
|
|
|
|
|
|
|
export function sendEtagResponse(
|
|
|
|
req: IncomingMessage,
|
|
|
|
res: ServerResponse,
|
|
|
|
etag: string | undefined
|
|
|
|
): boolean {
|
|
|
|
if (etag) {
|
|
|
|
/**
|
|
|
|
* The server generating a 304 response MUST generate any of the
|
|
|
|
* following header fields that would have been sent in a 200 (OK)
|
|
|
|
* response to the same request: Cache-Control, Content-Location, Date,
|
|
|
|
* ETag, Expires, and Vary. https://tools.ietf.org/html/rfc7232#section-4.1
|
|
|
|
*/
|
|
|
|
res.setHeader('ETag', etag)
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fresh(req.headers, { etag })) {
|
|
|
|
res.statusCode = 304
|
|
|
|
res.end()
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
return false
|
|
|
|
}
|