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	/*
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	}