From 3bcb5a8e539ae50b8f8a08775ee854921e31e7d7 Mon Sep 17 00:00:00 2001
From: Mhykol <mchaeldonald62@pm.me>
Date: Fri, 18 Oct 2024 07:54:34 -0400
Subject: Added message for rate limiter

---
 app.js | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'app.js')

diff --git a/app.js b/app.js
index 9a466d8..a834f26 100644
--- a/app.js
+++ b/app.js
@@ -1,6 +1,6 @@
 const express = require('express')
 const app = express()
-const limit = require('express-limit').limit
+const rateLimit = require('express-rate-limit')
 const helmet = require('helmet')
 const bp = require('body-parser')
 const cookieParser = require('cookie-parser')
@@ -32,6 +32,11 @@ const assetDir = [
     'img',
     'vid'
 ]
+const limiter = rateLimit({
+    windowMs: 30 * 1000,
+    max: 100,
+    message: '<h1>Rate Limit Exceeded</h1><p>You have exceeded the allowed number of requests. Please try again later.</p>',
+})
 require('dotenv').config()
 const rss = require('./source/rss')({baseUrl: process.env.baseUrl})
 
@@ -52,10 +57,7 @@ class App {
         app.use(cookieParser())
         if (process.env.NODE_ENV === 'prod') {
             app.use(helmet())
-            app.use(limit({
-                max: 100, // 100 requests
-                period: 300 * 1000 // per minute (five minutes)
-            }))
+            app.use(limiter)
         }
 
         app.route('*')
-- 
cgit v1.2.3-70-g09d2