src/api/Server.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* 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 DiscordClient from "../client/Client";
import Service from "../utils/structures/Service";
import express, { Express, Router as ExpressRouter } from 'express';
import Router from "./Router";
import path from "path";
import rateLimit from 'express-rate-limit';
import cors from 'cors';

export interface ServerOptions {
    port?: number;
}

export default class Server extends Service {
    port: number;
    express: Express;
    router: Router;

    constructor(client: DiscordClient, { port = 4000 }: ServerOptions = {}) {
        super(client);
        this.port = port;
        this.express = express();
        this.router = new Router(client, this, { routesDir: path.resolve(__dirname, 'routes') });
    }

    async boot() {
        type methods = 'get' | 'post' | 'put' | 'patch' | 'delete';
        const expressRouter = ExpressRouter();

        expressRouter.use(cors());

        expressRouter.use(rateLimit({
            windowMs: 1 * 60 * 1000,
            max: 50, 
            standardHeaders: true,
            legacyHeaders: true,
            message: { code: 429, error: 'Too many requests at a time. Please try again later.' },
        }));

        expressRouter.use(express.json());
        expressRouter.use(express.urlencoded({ extended: true }));

        await this.router.loadRoutes();

        for (const route of this.router.routes) {            
            // console.log(route.callback[1], route.callback[0].middleware()[route.callback[1]]);            
            expressRouter[route.method.toLowerCase() as methods](route.path, ...(route.callback[0].middleware()[route.callback[1]] as any[] ?? []), ...route.middlewareList as any[], await route.getCallbackFunction());
        }

        this.express.use(expressRouter);
    }

    async run() {
        await this.boot();
        
        this.express.listen(this.port, () => {
            console.log(`API Server listening at port ${this.port}`);
        });
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2022 OSN Inc.
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 DiscordClient from "../client/Client";
21	import Service from "../utils/structures/Service";
22	import express, { Express, Router as ExpressRouter } from 'express';
23	import Router from "./Router";
24	import path from "path";
25	import rateLimit from 'express-rate-limit';
26	import cors from 'cors';
27
28	export interface ServerOptions {
29	port?: number;
30	}
31
32	export default class Server extends Service {
33	port: number;
34	express: Express;
35	router: Router;
36
37	constructor(client: DiscordClient, { port = 4000 }: ServerOptions = {}) {
38	super(client);
39	this.port = port;
40	this.express = express();
41	this.router = new Router(client, this, { routesDir: path.resolve(__dirname, 'routes') });
42	}
43
44	async boot() {
45	type methods = 'get' \| 'post' \| 'put' \| 'patch' \| 'delete';
46	const expressRouter = ExpressRouter();
47
48	expressRouter.use(cors());
49
50	expressRouter.use(rateLimit({
51	windowMs: 1 * 60 * 1000,
52	max: 50,
53	standardHeaders: true,
54	legacyHeaders: true,
55	message: { code: 429, error: 'Too many requests at a time. Please try again later.' },
56	}));
57
58	expressRouter.use(express.json());
59	expressRouter.use(express.urlencoded({ extended: true }));
60
61	await this.router.loadRoutes();
62
63	for (const route of this.router.routes) {
64	// console.log(route.callback[1], route.callback[0].middleware()[route.callback[1]]);
65	expressRouter[route.method.toLowerCase() as methods](route.path, ...(route.callback[0].middleware()[route.callback[1]] as any[] ?? []), ...route.middlewareList as any[], await route.getCallbackFunction());
66	}
67
68	this.express.use(expressRouter);
69	}
70
71	async run() {
72	await this.boot();
73
74	this.express.listen(this.port, () => {
75	console.log(`API Server listening at port ${this.port}`);
76	});
77	}
78	}