UNPKG

next

Version:

The React Framework

156 lines (155 loc) 5.51 kB
import { HeadersAdapter } from '../web/spec-extension/adapters/headers'; import { PRERENDER_REVALIDATE_HEADER, PRERENDER_REVALIDATE_ONLY_GENERATED_HEADER } from '../../lib/constants'; import { getTracer } from '../lib/trace/tracer'; import { NodeSpan } from '../lib/trace/constants'; export function wrapApiHandler(page, handler) { return (...args)=>{ getTracer().setRootSpanAttribute('next.route', page); // Call API route method return getTracer().trace(NodeSpan.runHandler, { spanName: `executing api route (pages) ${page}` }, ()=>handler(...args)); }; } /** * * @param res response object * @param statusCode `HTTP` status code of response */ export function sendStatusCode(res, statusCode) { res.statusCode = statusCode; return res; } /** * * @param res response object * @param [statusOrUrl] `HTTP` status code of redirect * @param url URL of redirect */ export function redirect(res, statusOrUrl, url) { if (typeof statusOrUrl === 'string') { url = statusOrUrl; statusOrUrl = 307; } if (typeof statusOrUrl !== 'number' || typeof url !== 'string') { throw Object.defineProperty(new Error(`Invalid redirect arguments. Please use a single argument URL, e.g. res.redirect('/destination') or use a status code and URL, e.g. res.redirect(307, '/destination').`), "__NEXT_ERROR_CODE", { value: "E389", enumerable: false, configurable: true }); } res.writeHead(statusOrUrl, { Location: url }); res.write(url); res.end(); return res; } export function checkIsOnDemandRevalidate(req, previewProps) { const headers = HeadersAdapter.from(req.headers); const previewModeId = headers.get(PRERENDER_REVALIDATE_HEADER); const isOnDemandRevalidate = previewModeId === previewProps.previewModeId; const revalidateOnlyGenerated = headers.has(PRERENDER_REVALIDATE_ONLY_GENERATED_HEADER); return { isOnDemandRevalidate, revalidateOnlyGenerated }; } export const COOKIE_NAME_PRERENDER_BYPASS = `__prerender_bypass`; export const COOKIE_NAME_PRERENDER_DATA = `__next_preview_data`; export const RESPONSE_LIMIT_DEFAULT = 4 * 1024 * 1024; export const SYMBOL_PREVIEW_DATA = Symbol(COOKIE_NAME_PRERENDER_DATA); export const SYMBOL_CLEARED_COOKIES = Symbol(COOKIE_NAME_PRERENDER_BYPASS); export function clearPreviewData(res, options = {}) { if (SYMBOL_CLEARED_COOKIES in res) { return res; } const { serialize } = require('next/dist/compiled/cookie'); const previous = res.getHeader('Set-Cookie'); res.setHeader(`Set-Cookie`, [ ...typeof previous === 'string' ? [ previous ] : Array.isArray(previous) ? previous : [], serialize(COOKIE_NAME_PRERENDER_BYPASS, '', { // To delete a cookie, set `expires` to a date in the past: // https://tools.ietf.org/html/rfc6265#section-4.1.1 // `Max-Age: 0` is not valid, thus ignored, and the cookie is persisted. expires: new Date(0), httpOnly: true, sameSite: process.env.NODE_ENV !== 'development' ? 'none' : 'lax', secure: process.env.NODE_ENV !== 'development', path: '/', ...options.path !== undefined ? { path: options.path } : undefined }), serialize(COOKIE_NAME_PRERENDER_DATA, '', { // To delete a cookie, set `expires` to a date in the past: // https://tools.ietf.org/html/rfc6265#section-4.1.1 // `Max-Age: 0` is not valid, thus ignored, and the cookie is persisted. expires: new Date(0), httpOnly: true, sameSite: process.env.NODE_ENV !== 'development' ? 'none' : 'lax', secure: process.env.NODE_ENV !== 'development', path: '/', ...options.path !== undefined ? { path: options.path } : undefined }) ]); Object.defineProperty(res, SYMBOL_CLEARED_COOKIES, { value: true, enumerable: false }); return res; } /** * Custom error class */ export class ApiError extends Error { constructor(statusCode, message){ super(message); this.statusCode = statusCode; } } /** * Sends error in `response` * @param res response object * @param statusCode of response * @param message of response */ export function sendError(res, statusCode, message) { res.statusCode = statusCode; res.statusMessage = message; res.end(message); } /** * Execute getter function only if its needed * @param LazyProps `req` and `params` for lazyProp * @param prop name of property * @param getter function to get data */ export function setLazyProp({ req }, prop, getter) { const opts = { configurable: true, enumerable: true }; const optsReset = { ...opts, writable: true }; Object.defineProperty(req, prop, { ...opts, get: ()=>{ const value = getter(); // we set the property on the object to avoid recalculating it Object.defineProperty(req, prop, { ...optsReset, value }); return value; }, set: (value)=>{ Object.defineProperty(req, prop, { ...optsReset, value }); } }); } //# sourceMappingURL=index.js.map