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 { NextFunction, Request, Response } from "express";
import Client from "../core/Client";
import { RouteMetadata } from "../types/RouteMetadata";
import { AnyFunction } from "../types/Utils";

export type Middleware = (client: Client, request: Request, response: Response, next: NextFunction) => unknown;

export function Action(method: "GET" | "POST" | "PUT" | "PATCH" | "DELETE", uri: string, middleware: Middleware[] = []) {
    return (
        originalMethodOrTarget: unknown,
        contextOrMethodName: string | ClassMethodDecoratorContext,
        descriptor?: PropertyDescriptor
    ) => {
        if (typeof contextOrMethodName === "string") {
            const metadata: Record<string, RouteMetadata> = Reflect.getMetadata(
                "action_methods",
                originalMethodOrTarget as object
            ) ?? {
                [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: middleware as AnyFunction[] };

            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 as object);
        } 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 as AnyFunction,
                method,
                path: uri,
                middleware: middleware as AnyFunction[]
            };

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