mirror of
https://github.com/TriliumNext/Notes.git
synced 2025-08-12 20:02:28 +08:00
50 lines
1.5 KiB
TypeScript
50 lines
1.5 KiB
TypeScript
import type { Application, NextFunction, Request, Response } from "express";
|
|
import log from "../services/log.js";
|
|
import NotFoundError from "../errors/not_found_error.js";
|
|
import ForbiddenError from "../errors/forbidden_error.js";
|
|
import HttpError from "../errors/http_error.js";
|
|
|
|
function register(app: Application) {
|
|
|
|
app.use((err: unknown | Error, req: Request, res: Response, next: NextFunction) => {
|
|
|
|
const isCsrfTokenError = typeof err === "object"
|
|
&& err
|
|
&& "code" in err
|
|
&& err.code === "EBADCSRFTOKEN";
|
|
|
|
if (isCsrfTokenError) {
|
|
log.error(`Invalid CSRF token: ${req.headers["x-csrf-token"]}, secret: ${req.cookies["_csrf"]}`);
|
|
return next(new ForbiddenError("Invalid CSRF token"));
|
|
}
|
|
|
|
return next(err);
|
|
});
|
|
|
|
// catch 404 and forward to error handler
|
|
app.use((req, res, next) => {
|
|
const err = new NotFoundError(`Router not found for request ${req.method} ${req.url}`);
|
|
next(err);
|
|
});
|
|
|
|
// error handler
|
|
app.use((err: unknown | Error, req: Request, res: Response, _next: NextFunction) => {
|
|
|
|
const statusCode = (err instanceof HttpError) ? err.statusCode : 500;
|
|
const errMessage = (err instanceof Error && statusCode !== 404)
|
|
? err
|
|
: `${statusCode} ${req.method} ${req.url}`;
|
|
|
|
log.info(errMessage);
|
|
|
|
res.status(statusCode).send({
|
|
message: err instanceof Error ? err.message : "Unknown Error"
|
|
});
|
|
|
|
});
|
|
}
|
|
|
|
export default {
|
|
register
|
|
};
|