80 lines
1.8 KiB
TypeScript
80 lines
1.8 KiB
TypeScript
|
import type { IncomingMessage, ServerResponse } from "http";
|
||
|
|
import type { CookieSerializeOptions } from "cookie";
|
||
|
|
import nookies from "nookies";
|
||
|
|
|
||
|
|
const defaultOptions: CookieSerializeOptions = {
|
||
|
|
httpOnly: true,
|
||
|
|
sameSite: "lax",
|
||
|
|
path: "/",
|
||
|
|
};
|
||
|
|
|
||
|
|
export function getCookies(req?: BaseParams["req"]) {
|
||
|
|
const context = buildContext({ req });
|
||
|
|
|
||
|
|
return nookies.get(context);
|
||
|
|
}
|
||
|
|
|
||
|
|
type SetCookieParams = BaseParams & {
|
||
|
|
value: string;
|
||
|
|
options?: CookieSerializeOptions;
|
||
|
|
};
|
||
|
|
|
||
|
|
export function setCookie(params: SetCookieParams) {
|
||
|
|
const { req, res, name, value } = params;
|
||
|
|
const context = buildContext({ res });
|
||
|
|
const options: CookieSerializeOptions = {
|
||
|
|
...defaultOptions,
|
||
|
|
...params.options,
|
||
|
|
secure: isSecureEnvironment(req),
|
||
|
|
};
|
||
|
|
|
||
|
|
return nookies.set(context, name, value, options);
|
||
|
|
}
|
||
|
|
|
||
|
|
type DestroyCookieParams = BaseParams & {
|
||
|
|
options?: CookieSerializeOptions;
|
||
|
|
};
|
||
|
|
|
||
|
|
export function destroyCookie(params: DestroyCookieParams) {
|
||
|
|
const { res, name } = params;
|
||
|
|
const context = buildContext({ res });
|
||
|
|
const options = Object.assign({}, defaultOptions, params.options);
|
||
|
|
|
||
|
|
return nookies.destroy(context, name, options);
|
||
|
|
}
|
||
|
|
|
||
|
|
function isSecureEnvironment(req: IncomingMessage | null | undefined): boolean {
|
||
|
|
if (process.env.NODE_ENV !== "production") {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!req || !req.headers || !req.headers.host) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
const host =
|
||
|
|
(req.headers.host.indexOf(":") > -1 &&
|
||
|
|
req.headers.host.split(":")[0]) ||
|
||
|
|
req.headers.host;
|
||
|
|
|
||
|
|
return !["localhost", "127.0.0.1"].includes(host);
|
||
|
|
}
|
||
|
|
|
||
|
|
type BaseParams = {
|
||
|
|
req?: IncomingMessage | null;
|
||
|
|
res?: ServerResponse | null;
|
||
|
|
name: string;
|
||
|
|
};
|
||
|
|
|
||
|
|
function buildContext({ req, res }: Pick<BaseParams, "req" | "res">) {
|
||
|
|
if (req !== null && typeof req !== "undefined") {
|
||
|
|
return { req };
|
||
|
|
}
|
||
|
|
|
||
|
|
if (res !== null && typeof res !== "undefined") {
|
||
|
|
return { res };
|
||
|
|
}
|
||
|
|
|
||
|
|
return null;
|
||
|
|
}
|