141 lines
4.8 KiB
JavaScript
141 lines
4.8 KiB
JavaScript
import { HeadersAdapter } from "../web/spec-extension/adapters/headers";
|
|
import { PRERENDER_REVALIDATE_HEADER, PRERENDER_REVALIDATE_ONLY_GENERATED_HEADER } from "../../lib/constants";
|
|
/**
|
|
*
|
|
* @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 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').`);
|
|
}
|
|
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
|