src/decorators/Validate.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 { ZodSchema } from "zod";
import ValidateMiddleware from "../api/middleware/ValidateMiddleware";
import type Client from "../core/Client";

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

            metadata[contextOrMethodName] ??= middleware;
            Reflect.defineMetadata("validation_middleware", metadata, originalMethodOrTarget as object);
        } else {
            const metadata = (contextOrMethodName.metadata?.validationMiddleware ?? {}) as Record<
                string | symbol,
                (client: Client, req: Request, res: Response, next: NextFunction) => ReturnType<typeof ValidateMiddleware>
            >;

            const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
                ValidateMiddleware(schema, req, res, next);

            metadata[contextOrMethodName.name] ??= middleware;
            (contextOrMethodName.metadata as unknown) ??= {};
            contextOrMethodName.metadata.validationMiddleware = metadata;
            return originalMethodOrTarget as void;
        }
    };
}
1	/*
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 { ZodSchema } from "zod";
22	import ValidateMiddleware from "../api/middleware/ValidateMiddleware";
23	import type Client from "../core/Client";
24
25	export function Validate(schema: ZodSchema) {
26	return (
27	originalMethodOrTarget: unknown,
28	contextOrMethodName: string \| ClassMethodDecoratorContext,
29	_descriptor?: PropertyDescriptor
30	) => {
31	if (typeof contextOrMethodName === "string") {
32	const metadata = Reflect.getMetadata("validation_middleware", originalMethodOrTarget as object) ?? {};
33	const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
34	ValidateMiddleware(schema, req, res, next);
35
36	metadata[contextOrMethodName] ??= middleware;
37	Reflect.defineMetadata("validation_middleware", metadata, originalMethodOrTarget as object);
38	} else {
39	const metadata = (contextOrMethodName.metadata?.validationMiddleware ?? {}) as Record<
40	string \| symbol,
41	(client: Client, req: Request, res: Response, next: NextFunction) => ReturnType<typeof ValidateMiddleware>
42	>;
43
44	const middleware = (client: Client, req: Request, res: Response, next: NextFunction) =>
45	ValidateMiddleware(schema, req, res, next);
46
47	metadata[contextOrMethodName.name] ??= middleware;
48	(contextOrMethodName.metadata as unknown) ??= {};
49	contextOrMethodName.metadata.validationMiddleware = metadata;
50	return originalMethodOrTarget as void;
51	}
52	};
53	}