src/api/Route.ts

import { Request, Response as ExpressResponse } from "express";
import Response from "./Response";

export default class Route {
    constructor(public readonly method: string, public readonly path: string, public readonly callback: [Object, string], public middlewareList: Array<any> = []) {

    }

    middleware(...middleware: Object[]) {
        this.middlewareList = [...this.middlewareList, ...middleware];
    }

    async getCallbackFunction(...args: any[]) {
        const [controller, method] = this.callback;
        return async (req: Request, res: ExpressResponse) => {
            let output = (controller as { [key: string]: Function })[method].call(controller, req, res, ...args);

            if (output instanceof Promise) {
                output = await output;
            }

            if (output instanceof Response) {
                res.status(output.status);

                for (const header in output.headers) {
                    res.setHeader(header, output.headers[header]);
                }

                if (typeof output.body === 'object') {
                    res.json(output.body);
                }
                else {
                    res.send(output.body);
                }
            }
            else {
                if (typeof output === 'object') {
                    return res.json(output);
                }
    
                return res.send(output);
            }
        };
    }
}
1	rakin	325	import { Request, Response as ExpressResponse } from "express";
2			import Response from "./Response";
3	rakin	324
4	rakin	323	export default class Route {
5	rakin	324	constructor(public readonly method: string, public readonly path: string, public readonly callback: [Object, string], public middlewareList: Array<any> = []) {
6	rakin	323
7			}
8
9			middleware(...middleware: Object[]) {
10			this.middlewareList = [...this.middlewareList, ...middleware];
11			}
12
13	rakin	325	async getCallbackFunction(...args: any[]) {
14	rakin	323	const [controller, method] = this.callback;
15	rakin	325	return async (req: Request, res: ExpressResponse) => {
16			let output = (controller as { [key: string]: Function })[method].call(controller, req, res, ...args);
17	rakin	324
18	rakin	325	if (output instanceof Promise) {
19			output = await output;
20	rakin	324	}
21
22	rakin	325	if (output instanceof Response) {
23			res.status(output.status);
24
25			for (const header in output.headers) {
26			res.setHeader(header, output.headers[header]);
27			}
28
29			if (typeof output.body === 'object') {
30			res.json(output.body);
31			}
32			else {
33			res.send(output.body);
34			}
35			}
36			else {
37			if (typeof output === 'object') {
38			return res.json(output);
39			}
40
41			return res.send(output);
42			}
43	rakin	324	};
44	rakin	323	}
45			}