supertokens-node
Version:
NodeJS driver for SuperTokens core
181 lines (180 loc) • 7.2 kB
JavaScript
"use strict";
/* Copyright (c) 2021, VRAI Labs and/or its affiliates. All rights reserved.
*
* This software is licensed under the Apache License, Version 2.0 (the
* "License") as published by the Apache Software Foundation.
*
* You may not use this file except in compliance with the License. You may
* obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
var __importDefault =
(this && this.__importDefault) ||
function (mod) {
return mod && mod.__esModule ? mod : { default: mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.FastifyWrapper = exports.errorHandler = exports.FastifyResponse = exports.FastifyRequest = void 0;
const utils_1 = require("../../utils");
const request_1 = require("../request");
const response_1 = require("../response");
const utils_2 = require("../utils");
const supertokens_1 = __importDefault(require("../../supertokens"));
const constants_1 = require("../constants");
class FastifyRequest extends request_1.BaseRequest {
constructor(request) {
super();
this.getFormDataFromRequestBody = async () => {
return this.request.body; // NOTE: ask user to add require('fastify-formbody')
};
this.getJSONFromRequestBody = async () => {
return this.request.body;
};
this.getKeyValueFromQuery = (key) => {
if (this.request.query === undefined) {
return undefined;
}
let value = this.request.query[key];
if (value === undefined || typeof value !== "string") {
return undefined;
}
return value;
};
this.getMethod = () => {
return (0, utils_1.normaliseHttpMethod)(this.request.method);
};
this.getCookieValue = (key) => {
return (0, utils_2.getCookieValueFromHeaders)(this.request.headers, key);
};
this.getHeaderValue = (key) => {
return (0, utils_2.normalizeHeaderValue)(
(0, utils_1.getFromObjectCaseInsensitive)(key, this.request.headers)
);
};
this.getOriginalURL = () => {
return this.request.url;
};
this.original = request;
this.request = request;
}
}
exports.FastifyRequest = FastifyRequest;
class FastifyResponse extends response_1.BaseResponse {
constructor(response) {
super();
this.sendHTMLResponse = (html) => {
if (!this.response.sent) {
this.response.type("text/html");
this.response.send(html);
}
};
this.setHeader = (key, value, allowDuplicateKey) => {
try {
let existingHeaders = this.response.getHeaders();
let existingValue = existingHeaders[key.toLowerCase()];
// we have the this.response.header for compatibility with nextJS
if (existingValue === undefined) {
this.response.header(key, value);
} else if (allowDuplicateKey) {
/**
We only want to append if it does not already exist
For example if the caller is trying to add front token to the access control exposed headers property
we do not want to append if something else had already added it
*/
if (typeof existingValue !== "string" || !existingValue.includes(value)) {
this.response.header(key, existingValue + ", " + value);
}
} else {
// we overwrite the current one with the new one
this.response.header(key, value);
}
} catch (err) {
throw new Error("Error while setting header with key: " + key + " and value: " + value);
}
};
this.removeHeader = (key) => {
this.response.removeHeader(key);
};
this.setCookie = (key, value, domain, secure, httpOnly, expires, path, sameSite) => {
let serialisedCookie = (0, utils_2.serializeCookieValue)(
key,
value,
domain,
secure,
httpOnly,
expires,
path,
sameSite
);
let oldHeaders = this.response.getHeader(constants_1.COOKIE_HEADER);
if (oldHeaders === undefined) oldHeaders = [];
else if (!(oldHeaders instanceof Array)) oldHeaders = [oldHeaders.toString()];
this.response.removeHeader(constants_1.COOKIE_HEADER);
this.response.header(constants_1.COOKIE_HEADER, [
...oldHeaders.filter((h) => !h.startsWith(key + "=")),
serialisedCookie,
]);
};
/**
* @param {number} statusCode
*/
this.setStatusCode = (statusCode) => {
if (!this.response.sent) {
this.statusCode = statusCode;
}
};
/**
* @param {any} content
*/
this.sendJSONResponse = (content) => {
if (!this.response.sent) {
this.response.statusCode = this.statusCode;
this.response.send(content);
}
};
this.original = response;
this.response = response;
this.statusCode = 200;
}
}
exports.FastifyResponse = FastifyResponse;
function plugin(fastify, _, done) {
fastify.addHook("preHandler", async (req, reply) => {
let supertokens = supertokens_1.default.getInstanceOrThrowError();
let request = new FastifyRequest(req);
let response = new FastifyResponse(reply);
const userContext = (0, utils_1.makeDefaultUserContextFromAPI)(request);
try {
await supertokens.middleware(request, response, userContext);
} catch (err) {
await supertokens.errorHandler(err, request, response, userContext);
}
});
done();
}
plugin[Symbol.for("skip-override")] = true;
const errorHandler = () => {
return async (err, req, res) => {
let supertokens = supertokens_1.default.getInstanceOrThrowError();
let request = new FastifyRequest(req);
let response = new FastifyResponse(res);
let userContext = (0, utils_1.makeDefaultUserContextFromAPI)(request);
await supertokens.errorHandler(err, request, response, userContext);
};
};
exports.errorHandler = errorHandler;
exports.FastifyWrapper = {
plugin,
errorHandler: exports.errorHandler,
wrapRequest: (unwrapped) => {
return new FastifyRequest(unwrapped);
},
wrapResponse: (unwrapped) => {
return new FastifyResponse(unwrapped);
},
};