add: rate limiter

Author: SysSn13

.gitignore

@@ -1,2 +1,3 @@
 .env
-node_modules
+node_modules
+.snyk

background.js

@@ -6,7 +6,7 @@ const { ExpressAdapter } = require("@bull-board/express");
 
 const serverAdapter = new ExpressAdapter();
 
-const { addQueue, removeQueue, setQueues, replaceQueues } = createBullBoard({
+const bullBoard = createBullBoard({
     queues: [new BullAdapter(predictQueue), new BullAdapter(jobScheduler)],
     serverAdapter: serverAdapter,
 });
@@ -34,6 +34,7 @@ const initScheduler = async () => {
         { repeat: { cron: "0 */4 * * *" } }
     );
 };
+
 initScheduler();
 
 module.exports.bullBoardServerAdapter = serverAdapter;

main.js

@@ -2,6 +2,7 @@ if (process.env.NODE_ENV !== "production") {
     require("dotenv").config();
 }
 const express = require("express");
+const rateLimit = require("express-rate-limit");
 const expressLayouts = require("express-ejs-layouts");
 const bodyParser = require("body-parser");
 
@@ -20,6 +21,17 @@ db.once("open", () => {
 
 const app = express();
 
+const limiter = rateLimit({
+    max: process.env.RATE_LIMIT || 50,
+    windowMs: process.env.RATE_LIMIT_WINDOW || 10 * 1000,
+    message: "Too many requests, please try again later.",
+});
+app.enable("trust proxy");
+app.use(limiter);
+
+// body limit
+app.use(express.json({ limit: "10kb" }));
+
 // background
 if (process.env.BACKGROUND == true) {
     const { bullBoardServerAdapter } = require("./background");
@@ -43,6 +55,15 @@ if (process.env.WEB == true) {
 
 // api
 if (!process.env.API_DISABLED) {
+    const apiLimiter = rateLimit({
+        max: process.env.API_RATE_LIMIT || 20,
+        windowMs: process.env.API_RATE_LIMIT_WINDOW || 10 * 1000,
+        message: "Too many requests, please try again later.",
+        // keyGenerator: function (req) {
+        //     return req.ip;
+        // },
+    });
+    app.use("/api/", apiLimiter);
     const apiRoutes = require("./routes/api");
     app.use("/api/v1/", apiRoutes);
     console.info("API is up.");