src/decorators/RequireAuth.ts

/*
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2023 OSN Developers.
 *
 * SudoBot is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * SudoBot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
 */

import { NextFunction, Request, Response } from "express";
import RequireAuthMiddleware from "../api/middleware/RequireAuthMiddleware";
import type Client from "../core/Client";
import { Middleware } from "./Action";

export function RequireAuth(fetchUser = true) {
    return (
        originalMethodOrTarget: unknown,
        contextOrMethodName: string | ClassMethodDecoratorContext,
        _descriptor?: PropertyDescriptor
    ) => {
        if (typeof contextOrMethodName === "string") {
            const metadata = Reflect.getMetadata("auth_middleware", originalMethodOrTarget as object) ?? {};
            const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
                RequireAuthMiddleware(client, fetchUser, req, res, next);

            metadata[contextOrMethodName] ??= middleware;

            Reflect.defineMetadata("auth_middleware", metadata, originalMethodOrTarget as object);
        } else {
            const metadata = (contextOrMethodName.metadata?.authMiddleware ?? {}) as Record<string | symbol, Middleware>;
            const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
                RequireAuthMiddleware(client, fetchUser, req, res, next);

            metadata[contextOrMethodName.name] ??= middleware;
            (contextOrMethodName.metadata as unknown) ??= {};
            contextOrMethodName.metadata.authMiddleware = metadata;
            return originalMethodOrTarget as void;
        }
    };
}
1	rakinar2	577	/*
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2023 OSN Developers.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import { NextFunction, Request, Response } from "express";
21			import RequireAuthMiddleware from "../api/middleware/RequireAuthMiddleware";
22			import type Client from "../core/Client";
23			import { Middleware } from "./Action";
24
25			export function RequireAuth(fetchUser = true) {
26			return (
27			originalMethodOrTarget: unknown,
28			contextOrMethodName: string \| ClassMethodDecoratorContext,
29			_descriptor?: PropertyDescriptor
30			) => {
31			if (typeof contextOrMethodName === "string") {
32			const metadata = Reflect.getMetadata("auth_middleware", originalMethodOrTarget as object) ?? {};
33			const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
34			RequireAuthMiddleware(client, fetchUser, req, res, next);
35
36			metadata[contextOrMethodName] ??= middleware;
37
38			Reflect.defineMetadata("auth_middleware", metadata, originalMethodOrTarget as object);
39			} else {
40			const metadata = (contextOrMethodName.metadata?.authMiddleware ?? {}) as Record<string \| symbol, Middleware>;
41			const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
42			RequireAuthMiddleware(client, fetchUser, req, res, next);
43
44			metadata[contextOrMethodName.name] ??= middleware;
45			(contextOrMethodName.metadata as unknown) ??= {};
46			contextOrMethodName.metadata.authMiddleware = metadata;
47			return originalMethodOrTarget as void;
48			}
49			};
50			}