src/decorators/Action.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 { RouteMetadata } from "../types/RouteMetadata";

export function Action(method: "GET" | "POST" | "PUT" | "PATCH" | "DELETE", uri: string, middleware: Function[] = []) {
    return (
        originalMethodOrTarget: any,
        contextOrMethodName: string | ClassMethodDecoratorContext,
        descriptor?: PropertyDescriptor
    ) => {
        if (typeof contextOrMethodName === "string") {
            const metadata: Record<string, RouteMetadata> = Reflect.getMetadata("action_methods", originalMethodOrTarget) ?? {
                [contextOrMethodName]: {
                    GET: null,
                    DELETE: null,
                    PATCH: null,
                    POST: null,
                    PUT: null
                } as RouteMetadata
            };

            metadata[contextOrMethodName] ??= {
                GET: null,
                DELETE: null,
                PATCH: null,
                POST: null,
                PUT: null
            } as RouteMetadata;

            const data = { handler: descriptor!.value, method, path: uri, middleware };

            metadata[contextOrMethodName]![method] ??= data;
            metadata[contextOrMethodName]![method]!.handler ??= data.handler;
            metadata[contextOrMethodName]![method]!.method ??= data.method;

            if (metadata[contextOrMethodName]![method]!.middleware?.length) {
                metadata[contextOrMethodName]![method]!.middleware.push(...data.middleware);
            } else {
                metadata[contextOrMethodName]![method]!.middleware = data.middleware;
            }

            Reflect.defineMetadata("action_methods", metadata, originalMethodOrTarget);
        } else {
            const metadata = (contextOrMethodName.metadata?.actionMethods ?? {
                [contextOrMethodName.name]: {
                    GET: null,
                    DELETE: null,
                    PATCH: null,
                    POST: null,
                    PUT: null
                } as RouteMetadata
            }) as Record<string, RouteMetadata>;

            const key = contextOrMethodName.name as keyof typeof metadata;

            metadata[key] ??= {
                GET: null,
                DELETE: null,
                PATCH: null,
                POST: null,
                PUT: null
            } as RouteMetadata;

            const data = { handler: originalMethodOrTarget, method, path: uri, middleware };

            metadata[key]![method] ??= data;
            metadata[key]![method]!.handler ??= data.handler;
            metadata[key]![method]!.method ??= data.method;

            if (metadata[key]![method]!.middleware?.length) {
                metadata[key]![method]!.middleware.push(...data.middleware);
            } else {
                metadata[key]![method]!.middleware = data.middleware;
            }

            (contextOrMethodName.metadata as unknown) ??= {};
            contextOrMethodName.metadata.actionMethods = metadata;
            return originalMethodOrTarget;
        }
    };
}
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 { RouteMetadata } from "../types/RouteMetadata";
21
22			export function Action(method: "GET" \| "POST" \| "PUT" \| "PATCH" \| "DELETE", uri: string, middleware: Function[] = []) {
23			return (
24			originalMethodOrTarget: any,
25			contextOrMethodName: string \| ClassMethodDecoratorContext,
26			descriptor?: PropertyDescriptor
27			) => {
28			if (typeof contextOrMethodName === "string") {
29			const metadata: Record<string, RouteMetadata> = Reflect.getMetadata("action_methods", originalMethodOrTarget) ?? {
30			[contextOrMethodName]: {
31			GET: null,
32			DELETE: null,
33			PATCH: null,
34			POST: null,
35			PUT: null
36			} as RouteMetadata
37			};
38
39			metadata[contextOrMethodName] ??= {
40			GET: null,
41			DELETE: null,
42			PATCH: null,
43			POST: null,
44			PUT: null
45			} as RouteMetadata;
46
47			const data = { handler: descriptor!.value, method, path: uri, middleware };
48
49			metadata[contextOrMethodName]![method] ??= data;
50			metadata[contextOrMethodName]![method]!.handler ??= data.handler;
51			metadata[contextOrMethodName]![method]!.method ??= data.method;
52
53			if (metadata[contextOrMethodName]![method]!.middleware?.length) {
54			metadata[contextOrMethodName]![method]!.middleware.push(...data.middleware);
55			} else {
56			metadata[contextOrMethodName]![method]!.middleware = data.middleware;
57			}
58
59			Reflect.defineMetadata("action_methods", metadata, originalMethodOrTarget);
60			} else {
61			const metadata = (contextOrMethodName.metadata?.actionMethods ?? {
62			[contextOrMethodName.name]: {
63			GET: null,
64			DELETE: null,
65			PATCH: null,
66			POST: null,
67			PUT: null
68			} as RouteMetadata
69			}) as Record<string, RouteMetadata>;
70
71			const key = contextOrMethodName.name as keyof typeof metadata;
72
73			metadata[key] ??= {
74			GET: null,
75			DELETE: null,
76			PATCH: null,
77			POST: null,
78			PUT: null
79			} as RouteMetadata;
80
81			const data = { handler: originalMethodOrTarget, method, path: uri, middleware };
82
83			metadata[key]![method] ??= data;
84			metadata[key]![method]!.handler ??= data.handler;
85			metadata[key]![method]!.method ??= data.method;
86
87			if (metadata[key]![method]!.middleware?.length) {
88			metadata[key]![method]!.middleware.push(...data.middleware);
89			} else {
90			metadata[key]![method]!.middleware = data.middleware;
91			}
92
93			(contextOrMethodName.metadata as unknown) ??= {};
94			contextOrMethodName.metadata.actionMethods = metadata;
95			return originalMethodOrTarget;
96			}
97			};
98			}